CIndex.cpp revision 162e1c1b487352434552147967c3dd296ebee2f7
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"
33b846debc1b22a37228efe4aa87b34482d15b6a3cBenjamin Kramer#include "clang/Lex/PreprocessingRecord.h"
3433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor#include "clang/Lex/Preprocessor.h"
35a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor#include "llvm/ADT/STLExtras.h"
36d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek#include "llvm/ADT/Optional.h"
37f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor#include "llvm/ADT/StringSwitch.h"
38d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek#include "clang/Analysis/Support/SaveAndRestore.h"
39c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar#include "llvm/Support/CrashRecoveryContext.h"
4048615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar#include "llvm/Support/PrettyStackTrace.h"
4102465750c8c3fa96b1e7e596b02297e24361dc4fDouglas Gregor#include "llvm/Support/MemoryBuffer.h"
42358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor#include "llvm/Support/raw_ostream.h"
437a07fcb8f10fe45ea65a0a41798eb1c40777bde4Douglas Gregor#include "llvm/Support/Timer.h"
4403013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Mutex.h"
4503013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Program.h"
4603013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Signals.h"
4703013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Threading.h"
4837f1ea0eb08a00fa90edbecb427cfbb50ca0f4d0Ted Kremenek#include "llvm/Support/Compiler.h"
49fc0622155fa61349698a8fd0053773c37d9f7ac4Ted Kremenek
5050398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroffusing namespace clang;
5116c440a377b7ec8b722a2e2c7c864f75c95bd305Ted Kremenekusing namespace clang::cxcursor;
52ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenekusing namespace clang::cxstring;
5350398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff
54a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenekstatic CXTranslationUnit MakeCXTranslationUnit(ASTUnit *TU) {
55a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  if (!TU)
56a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return 0;
57a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit D = new CXTranslationUnitImpl();
58a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  D->TUData = TU;
59a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  D->StringPool = createCXStringPool();
60a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return D;
61a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek}
62a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek
6333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// \brief The result of comparing two source ranges.
6433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregorenum RangeComparisonResult {
6533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief Either the ranges overlap or one of the ranges is invalid.
6633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  RangeOverlap,
67f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
6833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The first range ends before the second range starts.
6933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  RangeBefore,
70f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
7133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The first range starts after the second range ends.
7233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  RangeAfter
7333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor};
7433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
75f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek/// \brief Compare two source ranges to determine their relative position in
7633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// the translation unit.
77f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekstatic RangeComparisonResult RangeCompare(SourceManager &SM,
78f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                          SourceRange R1,
7933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor                                          SourceRange R2) {
8033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  assert(R1.isValid() && "First range is invalid?");
8133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  assert(R2.isValid() && "Second range is invalid?");
82a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (R1.getEnd() != R2.getBegin() &&
83d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar      SM.isBeforeInTranslationUnit(R1.getEnd(), R2.getBegin()))
8433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    return RangeBefore;
85a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (R2.getEnd() != R1.getBegin() &&
86d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar      SM.isBeforeInTranslationUnit(R2.getEnd(), R1.getBegin()))
8733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    return RangeAfter;
8833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  return RangeOverlap;
8933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
9033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
91fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek/// \brief Determine if a source location falls within, before, or after a
92fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek///   a given source range.
93fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekstatic RangeComparisonResult LocationCompare(SourceManager &SM,
94fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                             SourceLocation L, SourceRange R) {
95fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  assert(R.isValid() && "First range is invalid?");
96fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  assert(L.isValid() && "Second range is invalid?");
97a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (L == R.getBegin() || L == R.getEnd())
98fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeOverlap;
99fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (SM.isBeforeInTranslationUnit(L, R.getBegin()))
100fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeBefore;
101fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (SM.isBeforeInTranslationUnit(R.getEnd(), L))
102fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeAfter;
103fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  return RangeOverlap;
104fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek}
105fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
10676dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// \brief Translate a Clang source range into a CIndex source range.
10776dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar///
10876dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// Clang internally represents ranges where the end location points to the
10976dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// start of the token at the end. However, for external clients it is more
11076dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// useful to have a CXSourceRange be a proper half-open interval. This routine
11176dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// does the appropriate translation.
112f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed KremenekCXSourceRange cxloc::translateSourceRange(const SourceManager &SM,
11376dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                                          const LangOptions &LangOpts,
1140a76aae8c03cb7dd7bdbe683485560afaf695959Chris Lattner                                          const CharSourceRange &R) {
11576dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  // We want the last character in this location, so we will adjust the
1166a5a23f8e7fb65e028c8092bc1d1a1d9dfe2e9bcDouglas Gregor  // location accordingly.
11776dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  SourceLocation EndLoc = R.getEnd();
118a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (EndLoc.isValid() && EndLoc.isMacroID())
119a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    EndLoc = SM.getSpellingLoc(EndLoc);
1200a76aae8c03cb7dd7bdbe683485560afaf695959Chris Lattner  if (R.isTokenRange() && !EndLoc.isInvalid() && EndLoc.isFileID()) {
1216a5a23f8e7fb65e028c8092bc1d1a1d9dfe2e9bcDouglas Gregor    unsigned Length = Lexer::MeasureTokenLength(EndLoc, SM, LangOpts);
12276dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar    EndLoc = EndLoc.getFileLocWithOffset(Length);
12376dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  }
12476dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar
12576dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  CXSourceRange Result = { { (void *)&SM, (void *)&LangOpts },
12676dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                           R.getBegin().getRawEncoding(),
12776dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                           EndLoc.getRawEncoding() };
12876dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  return Result;
12976dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar}
1301db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
1318a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek//===----------------------------------------------------------------------===//
13233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor// Cursor visitor.
1338a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek//===----------------------------------------------------------------------===//
1348a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek
13589922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroffnamespace {
136c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
137c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekclass VisitorJob {
138c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekpublic:
139cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek  enum Kind { DeclVisitKind, StmtVisitKind, MemberExprPartsKind,
140e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek              TypeLocVisitKind, OverloadExprPartsKind,
14160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek              DeclRefExprPartsKind, LabelRefVisitKind,
142f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek              ExplicitTemplateArgsVisitKind,
143f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek              NestedNameSpecifierVisitKind,
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
163c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenektypedef llvm::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
1867d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  // MaxPCHLevel - the maximum PCH level of declarations that we will pass on
1877d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  // to the visitor. Declarations with a PCH level greater than this value will
1887d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  // be suppressed.
1897d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  unsigned MaxPCHLevel;
19033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
19104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor  /// \brief Whether we should visit the preprocessing record entries last,
19204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor  /// after visiting other declarations.
19304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor  bool VisitPreprocessorLast;
19404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
19533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief When valid, a source range to which the cursor should restrict
19633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// its search.
19733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  SourceRange RegionOfInterest;
198f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
199d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually remove.  This part of a hack to support proper
200d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // iteration over all Decls contained lexically within an ObjC container.
201d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  DeclContext::decl_iterator *DI_current;
202d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  DeclContext::decl_iterator DE_current;
203d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
204d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  // Cache of pre-allocated worklists for data-recursion walk of Stmts.
205d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  llvm::SmallVector<VisitorWorkList*, 5> WorkListFreeList;
206d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  llvm::SmallVector<VisitorWorkList*, 5> WorkListCache;
207d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek
208b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  using DeclVisitor<CursorVisitor, bool>::Visit;
2097d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  using TypeLocVisitor<CursorVisitor, bool>::Visit;
210f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
211f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  /// \brief Determine whether this particular source range comes before, comes
212f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  /// after, or overlaps the region of interest.
21333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  ///
214d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar  /// \param R a half-open source range retrieved from the abstract syntax tree.
215f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  RangeComparisonResult CompareRegionOfInterest(SourceRange R);
216f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2170f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek  class SetParentRAII {
2180f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    CXCursor &Parent;
2190f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    Decl *&StmtParent;
2200f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    CXCursor OldParent;
2210f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek
2220f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek  public:
2230f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    SetParentRAII(CXCursor &Parent, Decl *&StmtParent, CXCursor NewParent)
2240f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek      : Parent(Parent), StmtParent(StmtParent), OldParent(Parent)
2250f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    {
2260f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek      Parent = NewParent;
2270f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek      if (clang_isDeclaration(Parent.kind))
2280f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek        StmtParent = getCursorDecl(Parent);
2290f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    }
2300f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek
2310f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    ~SetParentRAII() {
2320f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek      Parent = OldParent;
2330f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek      if (clang_isDeclaration(Parent.kind))
2340f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek        StmtParent = getCursorDecl(Parent);
2350f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    }
2360f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek  };
2370f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek
238b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorpublic:
239a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CursorVisitor(CXTranslationUnit TU, CXCursorVisitor Visitor,
240a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                CXClientData ClientData,
241f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                unsigned MaxPCHLevel,
24204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                bool VisitPreprocessorLast,
24333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor                SourceRange RegionOfInterest = SourceRange())
244a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    : TU(TU), AU(static_cast<ASTUnit*>(TU->TUData)),
245a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      Visitor(Visitor), ClientData(ClientData),
24604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      MaxPCHLevel(MaxPCHLevel), VisitPreprocessorLast(VisitPreprocessorLast),
24704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      RegionOfInterest(RegionOfInterest), DI_current(0)
248b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  {
249b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Parent.kind = CXCursor_NoDeclFound;
250b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Parent.data[0] = 0;
251b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Parent.data[1] = 0;
252b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Parent.data[2] = 0;
253f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    StmtParent = 0;
254b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
255f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
256d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  ~CursorVisitor() {
257d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    // Free the pre-allocated worklists for data-recursion.
258d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    for (llvm::SmallVectorImpl<VisitorWorkList*>::iterator
259d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek          I = WorkListCache.begin(), E = WorkListCache.end(); I != E; ++I) {
260d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek      delete *I;
261d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    }
262d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
263d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek
264a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *getASTUnit() const { return static_cast<ASTUnit*>(TU->TUData); }
265a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit getTU() const { return TU; }
266ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek
26733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  bool Visit(CXCursor Cursor, bool CheckedRegionOfInterest = false);
268788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
269788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  std::pair<PreprocessingRecord::iterator, PreprocessingRecord::iterator>
270788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    getPreprocessedEntities();
271788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
272b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  bool VisitChildren(CXCursor Parent);
273f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2747d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  // Declaration visitors
275162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  bool VisitTypeAliasDecl(TypeAliasDecl *D);
27609dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek  bool VisitAttributes(Decl *D);
2771ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  bool VisitBlockDecl(BlockDecl *B);
2783064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  bool VisitCXXRecordDecl(CXXRecordDecl *D);
279d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  llvm::Optional<bool> shouldVisitCursor(CXCursor C);
280b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  bool VisitDeclContext(DeclContext *DC);
28179758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitTranslationUnitDecl(TranslationUnitDecl *D);
28279758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitTypedefDecl(TypedefDecl *D);
28379758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitTagDecl(TagDecl *D);
2840ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  bool VisitClassTemplateSpecializationDecl(ClassTemplateSpecializationDecl *D);
28574dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  bool VisitClassTemplatePartialSpecializationDecl(
28674dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor                                     ClassTemplatePartialSpecializationDecl *D);
287fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitTemplateTypeParmDecl(TemplateTypeParmDecl *D);
2884540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitEnumConstantDecl(EnumConstantDecl *D);
28979758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitDeclaratorDecl(DeclaratorDecl *DD);
2904540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitFunctionDecl(FunctionDecl *ND);
29179758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitFieldDecl(FieldDecl *D);
29279758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitVarDecl(VarDecl *);
29384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  bool VisitNonTypeTemplateParmDecl(NonTypeTemplateParmDecl *D);
294fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitFunctionTemplateDecl(FunctionTemplateDecl *D);
29539d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  bool VisitClassTemplateDecl(ClassTemplateDecl *D);
29684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  bool VisitTemplateTemplateParmDecl(TemplateTemplateParmDecl *D);
29779758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCMethodDecl(ObjCMethodDecl *ND);
2984540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitObjCContainerDecl(ObjCContainerDecl *D);
29979758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCCategoryDecl(ObjCCategoryDecl *ND);
30079758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCProtocolDecl(ObjCProtocolDecl *PID);
30123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  bool VisitObjCPropertyDecl(ObjCPropertyDecl *PD);
30279758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCInterfaceDecl(ObjCInterfaceDecl *D);
3034540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitObjCImplDecl(ObjCImplDecl *D);
30479758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D);
3051ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  bool VisitObjCImplementationDecl(ObjCImplementationDecl *D);
30679758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  // FIXME: ObjCCompatibleAliasDecl requires aliased-class locations.
30779758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCForwardProtocolDecl(ObjCForwardProtocolDecl *D);
30879758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCClassDecl(ObjCClassDecl *D);
309a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  bool VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *PD);
310a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  bool VisitLinkageSpecDecl(LinkageSpecDecl *D);
3118f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  bool VisitNamespaceDecl(NamespaceDecl *D);
3126931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  bool VisitNamespaceAliasDecl(NamespaceAliasDecl *D);
3130a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  bool VisitUsingDirectiveDecl(UsingDirectiveDecl *D);
3147e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  bool VisitUsingDecl(UsingDecl *D);
3157e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  bool VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D);
3167e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  bool VisitUnresolvedUsingTypenameDecl(UnresolvedUsingTypenameDecl *D);
3170a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
31801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  // Name visitor
31901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  bool VisitDeclarationNameInfo(DeclarationNameInfo Name);
320c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  bool VisitNestedNameSpecifier(NestedNameSpecifier *NNS, SourceRange Range);
321dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  bool VisitNestedNameSpecifierLoc(NestedNameSpecifierLoc NNS);
32201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
323fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // Template visitors
324fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitTemplateParameters(const TemplateParameterList *Params);
3250b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  bool VisitTemplateName(TemplateName Name, SourceLocation Loc);
326fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitTemplateArgumentLoc(const TemplateArgumentLoc &TAL);
327fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
3287d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  // Type visitors
32901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  bool VisitQualifiedTypeLoc(QualifiedTypeLoc TL);
330f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitBuiltinTypeLoc(BuiltinTypeLoc TL);
3317d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  bool VisitTypedefTypeLoc(TypedefTypeLoc TL);
332f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitUnresolvedUsingTypeLoc(UnresolvedUsingTypeLoc TL);
333f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitTagTypeLoc(TagTypeLoc TL);
334fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitTemplateTypeParmTypeLoc(TemplateTypeParmTypeLoc TL);
335f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL);
336c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  bool VisitObjCObjectTypeLoc(ObjCObjectTypeLoc TL);
337f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitObjCObjectPointerTypeLoc(ObjCObjectPointerTypeLoc TL);
338075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara  bool VisitParenTypeLoc(ParenTypeLoc TL);
339f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitPointerTypeLoc(PointerTypeLoc TL);
340f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitBlockPointerTypeLoc(BlockPointerTypeLoc TL);
341f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitMemberPointerTypeLoc(MemberPointerTypeLoc TL);
342f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitLValueReferenceTypeLoc(LValueReferenceTypeLoc TL);
343f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitRValueReferenceTypeLoc(RValueReferenceTypeLoc TL);
34401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  bool VisitFunctionTypeLoc(FunctionTypeLoc TL, bool SkipResultType = false);
345f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitArrayTypeLoc(ArrayTypeLoc TL);
346fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitTemplateSpecializationTypeLoc(TemplateSpecializationTypeLoc TL);
3472332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  // FIXME: Implement visitors here when the unimplemented TypeLocs get
3482332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  // implemented
3492332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  bool VisitTypeOfExprTypeLoc(TypeOfExprTypeLoc TL);
3507536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor  bool VisitPackExpansionTypeLoc(PackExpansionTypeLoc TL);
3512332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  bool VisitTypeOfTypeLoc(TypeOfTypeLoc TL);
3522494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor  bool VisitDependentNameTypeLoc(DependentNameTypeLoc TL);
35394fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  bool VisitDependentTemplateSpecializationTypeLoc(
35494fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor                                    DependentTemplateSpecializationTypeLoc TL);
3559e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor  bool VisitElaboratedTypeLoc(ElaboratedTypeLoc TL);
3562494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
357c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  // Data-recursive visitor functions.
358c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  bool IsInRegionOfInterest(CXCursor C);
359c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  bool RunVisitorWorkList(VisitorWorkList &WL);
360c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  void EnqueueWorkList(VisitorWorkList &WL, Stmt *S);
361cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  LLVM_ATTRIBUTE_NOINLINE bool Visit(Stmt *S);
362b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor};
363f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
364b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor} // end anonymous namespace
3650d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
366a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorstatic SourceRange getRawCursorExtent(CXCursor C);
3676653798ff5ce6deb58112777e21307ccc453133dDouglas Gregorstatic SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr);
3686653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
369a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
37033e9abd21083a0191a7676a04b497006d2da184dDouglas GregorRangeComparisonResult CursorVisitor::CompareRegionOfInterest(SourceRange R) {
371a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return RangeCompare(AU->getSourceManager(), R, RegionOfInterest);
37233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
37333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
374b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \brief Visit the given cursor and, if requested by the visitor,
375b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// its children.
376b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor///
37733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// \param Cursor the cursor to visit.
37833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor///
37933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// \param CheckRegionOfInterest if true, then the caller already checked that
38033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// this cursor is within the region of interest.
38133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor///
382b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \returns true if the visitation should be aborted, false if it
383b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// should continue.
38433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregorbool CursorVisitor::Visit(CXCursor Cursor, bool CheckedRegionOfInterest) {
385b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isInvalid(Cursor.kind))
386b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return false;
387f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
388b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isDeclaration(Cursor.kind)) {
389b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Decl *D = getCursorDecl(Cursor);
390b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    assert(D && "Invalid declaration cursor");
391b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    if (D->getPCHLevel() > MaxPCHLevel)
392b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return false;
393b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor
394b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    if (D->isImplicit())
395b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return false;
396b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
3970d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
39833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  // If we have a range of interest, and this cursor doesn't intersect with it,
39933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  // we're done.
40033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (RegionOfInterest.isValid() && !CheckedRegionOfInterest) {
401a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    SourceRange Range = getRawCursorExtent(Cursor);
402f408f32aa9ae3d97bc656267dc5d78fa7d03499bDaniel Dunbar    if (Range.isInvalid() || CompareRegionOfInterest(Range))
40333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor      return false;
40433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  }
405f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
406b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  switch (Visitor(Cursor, Parent, ClientData)) {
407b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Break:
408b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
4090d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
410b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Continue:
411b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return false;
4122e331b938b38057e333fab0ba841130ea8467794Douglas Gregor
413b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Recurse:
414b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return VisitChildren(Cursor);
415b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
4160d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
417fd64377225a6a140bddb3f997d52a036486f9360Douglas Gregor  return false;
418b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
4190d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
420788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregorstd::pair<PreprocessingRecord::iterator, PreprocessingRecord::iterator>
421788f5a1242c04762f91eaa7565c07b9865846d88Douglas GregorCursorVisitor::getPreprocessedEntities() {
422788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  PreprocessingRecord &PPRec
423a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    = *AU->getPreprocessor().getPreprocessingRecord();
424788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
425788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  bool OnlyLocalDecls
42632038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    = !AU->isMainFileAST() && AU->getOnlyLocalDecls();
42732038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor
42832038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor  if (OnlyLocalDecls && RegionOfInterest.isValid()) {
42932038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    // If we would only look at local declarations but we have a region of
43032038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    // interest, check whether that region of interest is in the main file.
43132038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    // If not, we should traverse all declarations.
43232038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    // FIXME: My kingdom for a proper binary search approach to finding
43332038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    // cursors!
43432038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    std::pair<FileID, unsigned> Location
43532038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor      = AU->getSourceManager().getDecomposedInstantiationLoc(
43632038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor                                                   RegionOfInterest.getBegin());
43732038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    if (Location.first != AU->getSourceManager().getMainFileID())
43832038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor      OnlyLocalDecls = false;
43932038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor  }
440788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
44189d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  PreprocessingRecord::iterator StartEntity, EndEntity;
44289d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  if (OnlyLocalDecls) {
44389d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor    StartEntity = AU->pp_entity_begin();
44489d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor    EndEntity = AU->pp_entity_end();
44589d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  } else {
44689d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor    StartEntity = PPRec.begin();
44789d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor    EndEntity = PPRec.end();
44889d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  }
44989d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor
450788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  // There is no region of interest; we have to walk everything.
451788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  if (RegionOfInterest.isInvalid())
45289d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor    return std::make_pair(StartEntity, EndEntity);
453788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
454788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  // Find the file in which the region of interest lands.
455a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  SourceManager &SM = AU->getSourceManager();
456788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  std::pair<FileID, unsigned> Begin
457788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    = SM.getDecomposedInstantiationLoc(RegionOfInterest.getBegin());
458788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  std::pair<FileID, unsigned> End
459788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    = SM.getDecomposedInstantiationLoc(RegionOfInterest.getEnd());
460788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
461788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  // The region of interest spans files; we have to walk everything.
462788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  if (Begin.first != End.first)
46389d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor    return std::make_pair(StartEntity, EndEntity);
464788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
465788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  ASTUnit::PreprocessedEntitiesByFileMap &ByFileMap
466a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    = AU->getPreprocessedEntitiesByFile();
467788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  if (ByFileMap.empty()) {
468788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    // Build the mapping from files to sets of preprocessed entities.
46989d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor    for (PreprocessingRecord::iterator E = StartEntity; E != EndEntity; ++E) {
470788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor      std::pair<FileID, unsigned> P
471788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor        = SM.getDecomposedInstantiationLoc((*E)->getSourceRange().getBegin());
47289d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor
473788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor      ByFileMap[P.first].push_back(*E);
474788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    }
475788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  }
476788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
477788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  return std::make_pair(ByFileMap[Begin.first].begin(),
478788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor                        ByFileMap[Begin.first].end());
479788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor}
480788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
481b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \brief Visit the children of the given cursor.
482a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek///
483b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \returns true if the visitation should be aborted, false if it
484b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// should continue.
485f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekbool CursorVisitor::VisitChildren(CXCursor Cursor) {
486c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  if (clang_isReference(Cursor.kind) &&
487c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      Cursor.kind != CXCursor_CXXBaseSpecifier) {
488a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    // By definition, references have no children.
489a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return false;
490a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  }
491f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
492f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // Set the Parent field to Cursor, then back to its old value once we're
493b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  // done.
4940f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek  SetParentRAII SetParent(Parent, StmtParent, Cursor);
495f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
496b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isDeclaration(Cursor.kind)) {
497b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Decl *D = getCursorDecl(Cursor);
49806d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (!D)
49906d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return false;
50006d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
501539311e0221df256c70c1c3080c8af847cd29dffTed Kremenek    return VisitAttributes(D) || Visit(D);
502b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
503f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
50406d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  if (clang_isStatement(Cursor.kind)) {
50506d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (Stmt *S = getCursorStmt(Cursor))
50606d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return Visit(S);
50706d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
50806d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    return false;
50906d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  }
51006d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
51106d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  if (clang_isExpression(Cursor.kind)) {
51206d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (Expr *E = getCursorExpr(Cursor))
51306d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return Visit(E);
51406d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
51506d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    return false;
51606d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  }
517f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
518b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isTranslationUnit(Cursor.kind)) {
519a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    CXTranslationUnit tu = getCursorTU(Cursor);
520a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    ASTUnit *CXXUnit = static_cast<ASTUnit*>(tu->TUData);
52104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
52204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor    int VisitOrder[2] = { VisitPreprocessorLast, !VisitPreprocessorLast };
52304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor    for (unsigned I = 0; I != 2; ++I) {
52404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      if (VisitOrder[I]) {
52504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        if (!CXXUnit->isMainFileAST() && CXXUnit->getOnlyLocalDecls() &&
52604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            RegionOfInterest.isInvalid()) {
52704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          for (ASTUnit::top_level_iterator TL = CXXUnit->top_level_begin(),
52804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                                        TLEnd = CXXUnit->top_level_end();
52904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor               TL != TLEnd; ++TL) {
53004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            if (Visit(MakeCXCursor(*TL, tu), true))
53104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor              return true;
53204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          }
53304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        } else if (VisitDeclContext(
53404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                                CXXUnit->getASTContext().getTranslationUnitDecl()))
5357b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor          return true;
53604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        continue;
5377b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor      }
5383178cb674ac8c3b59e1791e14d38d48619a1b621Bob Wilson
53904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      // Walk the preprocessing record.
54004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      if (CXXUnit->getPreprocessor().getPreprocessingRecord()) {
54104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        // FIXME: Once we have the ability to deserialize a preprocessing record,
54204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        // do so.
54304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        PreprocessingRecord::iterator E, EEnd;
54404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        for (llvm::tie(E, EEnd) = getPreprocessedEntities(); E != EEnd; ++E) {
54504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          if (MacroInstantiation *MI = dyn_cast<MacroInstantiation>(*E)) {
54604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            if (Visit(MakeMacroInstantiationCursor(MI, tu)))
54704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor              return true;
54804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
54904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            continue;
55004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          }
5510396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor
55204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          if (MacroDefinition *MD = dyn_cast<MacroDefinition>(*E)) {
55304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            if (Visit(MakeMacroDefinitionCursor(MD, tu)))
55404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor              return true;
55504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
55604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            continue;
55704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          }
558ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
55904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          if (InclusionDirective *ID = dyn_cast<InclusionDirective>(*E)) {
56004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            if (Visit(MakeInclusionDirectiveCursor(ID, tu)))
56104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor              return true;
56204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
56304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            continue;
56404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          }
565ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor        }
5660396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor      }
5670396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    }
56804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
5697b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor    return false;
570b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
571f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
572c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  if (Cursor.kind == CXCursor_CXXBaseSpecifier) {
573c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor    if (CXXBaseSpecifier *Base = getCursorCXXBaseSpecifier(Cursor)) {
574c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      if (TypeSourceInfo *BaseTSInfo = Base->getTypeSourceInfo()) {
575c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor        return Visit(BaseTSInfo->getTypeLoc());
576c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      }
577c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor    }
578c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  }
579c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor
580b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  // Nothing to visit at the moment.
581b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
582dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
583dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
5841ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenekbool CursorVisitor::VisitBlockDecl(BlockDecl *B) {
585fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall  if (Visit(B->getSignatureAsWritten()->getTypeLoc()))
586fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall    return true;
5871ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
588664cffd330611d78fc0286f539589920a37ca328Ted Kremenek  if (Stmt *Body = B->getBody())
589664cffd330611d78fc0286f539589920a37ca328Ted Kremenek    return Visit(MakeCXCursor(Body, StmtParent, TU));
590664cffd330611d78fc0286f539589920a37ca328Ted Kremenek
591664cffd330611d78fc0286f539589920a37ca328Ted Kremenek  return false;
5921ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek}
5931ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
594d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenekllvm::Optional<bool> CursorVisitor::shouldVisitCursor(CXCursor Cursor) {
595d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (RegionOfInterest.isValid()) {
5966653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    SourceRange Range = getFullCursorExtent(Cursor, AU->getSourceManager());
597d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (Range.isInvalid())
598d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return llvm::Optional<bool>();
5996653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
600d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    switch (CompareRegionOfInterest(Range)) {
601d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeBefore:
602d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration comes before the region of interest; skip it.
603d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return llvm::Optional<bool>();
60423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
605d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeAfter:
606d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration comes after the region of interest; we're done.
607d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
608d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar
609d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeOverlap:
610d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration overlaps the region of interest; visit it.
611d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      break;
612d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
613d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
614d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  return true;
615d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek}
616f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
617d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenekbool CursorVisitor::VisitDeclContext(DeclContext *DC) {
618d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  DeclContext::decl_iterator I = DC->decls_begin(), E = DC->decls_end();
619f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
620d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually remove.  This part of a hack to support proper
621d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // iteration over all Decls contained lexically within an ObjC container.
622d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SaveAndRestore<DeclContext::decl_iterator*> DI_saved(DI_current, &I);
623d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SaveAndRestore<DeclContext::decl_iterator> DE_saved(DE_current, E);
624f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
625d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for ( ; I != E; ++I) {
626d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    Decl *D = *I;
627d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (D->getLexicalDeclContext() != DC)
628d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
629d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    CXCursor Cursor = MakeCXCursor(D, TU);
630d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    const llvm::Optional<bool> &V = shouldVisitCursor(Cursor);
631d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.hasValue())
632d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
633d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.getValue())
634d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
635d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar    if (Visit(Cursor, true))
636b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
637b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
638b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
639dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
640dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
6411ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTranslationUnitDecl(TranslationUnitDecl *D) {
6421ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  llvm_unreachable("Translation units are visited directly by Visit()");
6431ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
6441ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6451ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
646162e1c1b487352434552147967c3dd296ebee2f7Richard Smithbool CursorVisitor::VisitTypeAliasDecl(TypeAliasDecl *D) {
647162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
648162e1c1b487352434552147967c3dd296ebee2f7Richard Smith    return Visit(TSInfo->getTypeLoc());
649162e1c1b487352434552147967c3dd296ebee2f7Richard Smith
650162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  return false;
651162e1c1b487352434552147967c3dd296ebee2f7Richard Smith}
652162e1c1b487352434552147967c3dd296ebee2f7Richard Smith
6531ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTypedefDecl(TypedefDecl *D) {
6541ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
6551ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(TSInfo->getTypeLoc());
656f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
6571ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
6581ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6591ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6601ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTagDecl(TagDecl *D) {
6611ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitDeclContext(D);
6621ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6631ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6640ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregorbool CursorVisitor::VisitClassTemplateSpecializationDecl(
6650ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor                                          ClassTemplateSpecializationDecl *D) {
6660ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  bool ShouldVisitBody = false;
6670ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  switch (D->getSpecializationKind()) {
6680ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_Undeclared:
6690ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ImplicitInstantiation:
6700ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    // Nothing to visit
6710ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    return false;
6720ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6730ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitInstantiationDeclaration:
6740ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitInstantiationDefinition:
6750ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    break;
6760ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6770ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitSpecialization:
6780ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    ShouldVisitBody = true;
6790ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    break;
6800ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  }
6810ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6820ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  // Visit the template arguments used in the specialization.
6830ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  if (TypeSourceInfo *SpecType = D->getTypeAsWritten()) {
6840ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    TypeLoc TL = SpecType->getTypeLoc();
6850ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    if (TemplateSpecializationTypeLoc *TSTLoc
6860ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor          = dyn_cast<TemplateSpecializationTypeLoc>(&TL)) {
6870ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor      for (unsigned I = 0, N = TSTLoc->getNumArgs(); I != N; ++I)
6880ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor        if (VisitTemplateArgumentLoc(TSTLoc->getArgLoc(I)))
6890ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor          return true;
6900ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    }
6910ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  }
6920ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6930ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  if (ShouldVisitBody && VisitCXXRecordDecl(D))
6940ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    return true;
6950ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6960ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  return false;
6970ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor}
6980ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
69974dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregorbool CursorVisitor::VisitClassTemplatePartialSpecializationDecl(
70074dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor                                   ClassTemplatePartialSpecializationDecl *D) {
70174dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // FIXME: Visit the "outer" template parameter lists on the TagDecl
70274dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // before visiting these template parameters.
70374dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
70474dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    return true;
70574dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
70674dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // Visit the partial specialization arguments.
70774dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  const TemplateArgumentLoc *TemplateArgs = D->getTemplateArgsAsWritten();
70874dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  for (unsigned I = 0, N = D->getNumTemplateArgsAsWritten(); I != N; ++I)
70974dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    if (VisitTemplateArgumentLoc(TemplateArgs[I]))
71074dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor      return true;
71174dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
71274dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  return VisitCXXRecordDecl(D);
71374dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor}
71474dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
715fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateTypeParmDecl(TemplateTypeParmDecl *D) {
71684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  // Visit the default argument.
71784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
71884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    if (TypeSourceInfo *DefArg = D->getDefaultArgumentInfo())
71984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      if (Visit(DefArg->getTypeLoc()))
72084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor        return true;
72184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
722fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
723fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
724fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
7251ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitEnumConstantDecl(EnumConstantDecl *D) {
7261ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *Init = D->getInitExpr())
7271ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(MakeCXCursor(Init, StmtParent, TU));
7281ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
7291ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
7301ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
7317d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorbool CursorVisitor::VisitDeclaratorDecl(DeclaratorDecl *DD) {
7327d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  if (TypeSourceInfo *TSInfo = DD->getTypeSourceInfo())
7337d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    if (Visit(TSInfo->getTypeLoc()))
7347d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      return true;
7357d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
736c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  // Visit the nested-name-specifier, if present.
737c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = DD->getQualifierLoc())
738c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
739c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      return true;
740c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor
7417d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  return false;
7427d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor}
7437d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
744a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor/// \brief Compare two base or member initializers based on their source order.
745cbb67480094b3bcb5b715acd827cbad55e2a204cSean Huntstatic int CompareCXXCtorInitializers(const void* Xp, const void *Yp) {
746cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt  CXXCtorInitializer const * const *X
747cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt    = static_cast<CXXCtorInitializer const * const *>(Xp);
748cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt  CXXCtorInitializer const * const *Y
749cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt    = static_cast<CXXCtorInitializer const * const *>(Yp);
750a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
751a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  if ((*X)->getSourceOrder() < (*Y)->getSourceOrder())
752a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return -1;
753a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  else if ((*X)->getSourceOrder() > (*Y)->getSourceOrder())
754a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return 1;
755a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  else
756a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return 0;
757a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor}
758a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
759b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitFunctionDecl(FunctionDecl *ND) {
76001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  if (TypeSourceInfo *TSInfo = ND->getTypeSourceInfo()) {
76101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the function declaration's syntactic components in the order
76201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // written. This requires a bit of work.
763723df245307a530da5433dfb43accf187dc3e243Abramo Bagnara    TypeLoc TL = TSInfo->getTypeLoc().IgnoreParens();
76401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    FunctionTypeLoc *FTL = dyn_cast<FunctionTypeLoc>(&TL);
76501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
76601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // If we have a function declared directly (without the use of a typedef),
76701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // visit just the return type. Otherwise, just visit the function's type
76801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // now.
76901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if ((FTL && !isa<CXXConversionDecl>(ND) && Visit(FTL->getResultLoc())) ||
77001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor        (!FTL && Visit(TL)))
77101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
77201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
773c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // Visit the nested-name-specifier, if present.
774c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (NestedNameSpecifierLoc QualifierLoc = ND->getQualifierLoc())
775c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      if (VisitNestedNameSpecifierLoc(QualifierLoc))
776c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor        return true;
77701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
77801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the declaration name.
77901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (VisitDeclarationNameInfo(ND->getNameInfo()))
78001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
78101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
78201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Visit explicitly-specified template arguments!
78301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
78401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the function parameters, if we have a function type.
78501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (FTL && VisitFunctionTypeLoc(*FTL, true))
78601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
78701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
78801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Attributes?
78901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  }
79001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
791a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  if (ND->isThisDeclarationADefinition()) {
792a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    if (CXXConstructorDecl *Constructor = dyn_cast<CXXConstructorDecl>(ND)) {
793a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Find the initializers that were written in the source.
794cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt      llvm::SmallVector<CXXCtorInitializer *, 4> WrittenInits;
795a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      for (CXXConstructorDecl::init_iterator I = Constructor->init_begin(),
796a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor                                          IEnd = Constructor->init_end();
797a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor           I != IEnd; ++I) {
798a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        if (!(*I)->isWritten())
799a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor          continue;
800a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
801a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        WrittenInits.push_back(*I);
802a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      }
803a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
804a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Sort the initializers in source order
805a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      llvm::array_pod_sort(WrittenInits.begin(), WrittenInits.end(),
806cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt                           &CompareCXXCtorInitializers);
807a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
808a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Visit the initializers in source order
809a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      for (unsigned I = 0, N = WrittenInits.size(); I != N; ++I) {
810cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt        CXXCtorInitializer *Init = WrittenInits[I];
81100eb3f9c5b33e3d99aee1f8b75dd9c9678fdd66bFrancois Pichet        if (Init->isAnyMemberInitializer()) {
81200eb3f9c5b33e3d99aee1f8b75dd9c9678fdd66bFrancois Pichet          if (Visit(MakeCursorMemberRef(Init->getAnyMember(),
813a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor                                        Init->getMemberLocation(), TU)))
814a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
815a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        } else if (TypeSourceInfo *BaseInfo = Init->getBaseClassInfo()) {
816a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor          if (Visit(BaseInfo->getTypeLoc()))
817a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
818a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        }
819a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
820a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        // Visit the initializer value.
821a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        if (Expr *Initializer = Init->getInit())
822a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor          if (Visit(MakeCXCursor(Initializer, ND, TU)))
823a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
824a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      }
825a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
826a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
827a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    if (Visit(MakeCXCursor(ND->getBody(), StmtParent, TU)))
828a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return true;
829a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  }
830f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
831b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
832b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
833dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
8341ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitFieldDecl(FieldDecl *D) {
8351ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (VisitDeclaratorDecl(D))
8361ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
837f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8381ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *BitWidth = D->getBitWidth())
8391ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(MakeCXCursor(BitWidth, StmtParent, TU));
840f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8411ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
8421ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8431ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
8441ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitVarDecl(VarDecl *D) {
8451ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (VisitDeclaratorDecl(D))
8461ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
847f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8481ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *Init = D->getInit())
8491ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(MakeCXCursor(Init, StmtParent, TU));
850f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8511ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
8521ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8531ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
85484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregorbool CursorVisitor::VisitNonTypeTemplateParmDecl(NonTypeTemplateParmDecl *D) {
85584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (VisitDeclaratorDecl(D))
85684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
85784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
85884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
85984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    if (Expr *DefArg = D->getDefaultArgument())
86084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      return Visit(MakeCXCursor(DefArg, StmtParent, TU));
86184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
86284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  return false;
86384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor}
86484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
865fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitFunctionTemplateDecl(FunctionTemplateDecl *D) {
866fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // FIXME: Visit the "outer" template parameter lists on the FunctionDecl
867fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // before visiting these template parameters.
868fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
869fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return true;
870fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
871fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return VisitFunctionDecl(D->getTemplatedDecl());
872fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
873fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
87439d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregorbool CursorVisitor::VisitClassTemplateDecl(ClassTemplateDecl *D) {
87539d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  // FIXME: Visit the "outer" template parameter lists on the TagDecl
87639d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  // before visiting these template parameters.
87739d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
87839d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor    return true;
87939d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor
88039d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  return VisitCXXRecordDecl(D->getTemplatedDecl());
88139d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor}
88239d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor
88384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregorbool CursorVisitor::VisitTemplateTemplateParmDecl(TemplateTemplateParmDecl *D) {
88484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
88584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
88684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
88784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited() &&
88884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      VisitTemplateArgumentLoc(D->getDefaultArgument()))
88984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
89084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
89184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  return false;
89284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor}
89384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
8941ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCMethodDecl(ObjCMethodDecl *ND) {
8954bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor  if (TypeSourceInfo *TSInfo = ND->getResultTypeSourceInfo())
8964bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor    if (Visit(TSInfo->getTypeLoc()))
8974bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor      return true;
8984bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor
899f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  for (ObjCMethodDecl::param_iterator P = ND->param_begin(),
9001ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       PEnd = ND->param_end();
9011ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       P != PEnd; ++P) {
9021ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    if (Visit(MakeCXCursor(*P, TU)))
9031ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
9041ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  }
905f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
9061ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (ND->isThisDeclarationADefinition() &&
9071ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      Visit(MakeCXCursor(ND->getBody(), StmtParent, TU)))
9081ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
909f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
9101ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
9111ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
9121ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
913d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremeneknamespace {
914d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  struct ContainerDeclsSort {
915d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    SourceManager &SM;
916d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    ContainerDeclsSort(SourceManager &sm) : SM(sm) {}
917d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    bool operator()(Decl *A, Decl *B) {
918d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L_A = A->getLocStart();
919d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L_B = B->getLocStart();
920d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      assert(L_A.isValid() && L_B.isValid());
921d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return SM.isBeforeInTranslationUnit(L_A, L_B);
922d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
923d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  };
924d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek}
925d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
926a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregorbool CursorVisitor::VisitObjCContainerDecl(ObjCContainerDecl *D) {
927d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually convert back to just 'VisitDeclContext()'.  Essentially
928d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // an @implementation can lexically contain Decls that are not properly
929d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // nested in the AST.  When we identify such cases, we need to retrofit
930d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // this nesting here.
931d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (!DI_current)
932d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    return VisitDeclContext(D);
933d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
934d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Scan the Decls that immediately come after the container
935d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // in the current DeclContext.  If any fall within the
936d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // container's lexical region, stash them into a vector
937d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // for later processing.
938d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  llvm::SmallVector<Decl *, 24> DeclsInContainer;
939d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SourceLocation EndLoc = D->getSourceRange().getEnd();
940a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  SourceManager &SM = AU->getSourceManager();
941d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (EndLoc.isValid()) {
942d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    DeclContext::decl_iterator next = *DI_current;
943d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    while (++next != DE_current) {
944d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      Decl *D_next = *next;
945d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      if (!D_next)
946d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        break;
947d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L = D_next->getLocStart();
948d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      if (!L.isValid())
949d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        break;
950d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      if (SM.isBeforeInTranslationUnit(L, EndLoc)) {
951d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        *DI_current = next;
952d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        DeclsInContainer.push_back(D_next);
953d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        continue;
954d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      }
955d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      break;
956d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
957d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
958d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
959d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // The common case.
960d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (DeclsInContainer.empty())
961d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    return VisitDeclContext(D);
962d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
963d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Get all the Decls in the DeclContext, and sort them with the
964d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // additional ones we've collected.  Then visit them.
965d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for (DeclContext::decl_iterator I = D->decls_begin(), E = D->decls_end();
966d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek       I!=E; ++I) {
967d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    Decl *subDecl = *I;
9680582c897ec7261b4c6af0fe26dc2a0b6b54d266cTed Kremenek    if (!subDecl || subDecl->getLexicalDeclContext() != D ||
9690582c897ec7261b4c6af0fe26dc2a0b6b54d266cTed Kremenek        subDecl->getLocStart().isInvalid())
970d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
971d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    DeclsInContainer.push_back(subDecl);
972d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
973d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
974d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Now sort the Decls so that they appear in lexical order.
975d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  std::sort(DeclsInContainer.begin(), DeclsInContainer.end(),
976d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek            ContainerDeclsSort(SM));
977d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
978d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Now visit the decls.
979d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for (llvm::SmallVectorImpl<Decl*>::iterator I = DeclsInContainer.begin(),
980d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek         E = DeclsInContainer.end(); I != E; ++I) {
981d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    CXCursor Cursor = MakeCXCursor(*I, TU);
982d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    const llvm::Optional<bool> &V = shouldVisitCursor(Cursor);
983d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.hasValue())
984d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
985d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.getValue())
986d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
987d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (Visit(Cursor, true))
988d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return true;
989d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
990d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  return false;
991a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor}
992a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor
993b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitObjCCategoryDecl(ObjCCategoryDecl *ND) {
994b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (Visit(MakeCursorObjCClassRef(ND->getClassInterface(), ND->getLocation(),
995b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                   TU)))
996b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
997f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
99878db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  ObjCCategoryDecl::protocol_loc_iterator PL = ND->protocol_loc_begin();
99978db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  for (ObjCCategoryDecl::protocol_iterator I = ND->protocol_begin(),
100078db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor         E = ND->protocol_end(); I != E; ++I, ++PL)
1001b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1002b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
1003f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1004a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitObjCContainerDecl(ND);
1005dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1006dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
10071ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCProtocolDecl(ObjCProtocolDecl *PID) {
10081ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  ObjCProtocolDecl::protocol_loc_iterator PL = PID->protocol_loc_begin();
10091ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCProtocolDecl::protocol_iterator I = PID->protocol_begin(),
10101ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       E = PID->protocol_end(); I != E; ++I, ++PL)
10111ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
10121ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
1013f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10141ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCContainerDecl(PID);
10151ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
10161ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
101723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenekbool CursorVisitor::VisitObjCPropertyDecl(ObjCPropertyDecl *PD) {
101883cb94269015bf2770ade71e616c5322ea7e76e1Douglas Gregor  if (PD->getTypeSourceInfo() && Visit(PD->getTypeSourceInfo()->getTypeLoc()))
1019fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall    return true;
1020fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall
102123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // FIXME: This implements a workaround with @property declarations also being
102223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // installed in the DeclContext for the @interface.  Eventually this code
102323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // should be removed.
102423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCCategoryDecl *CDecl = dyn_cast<ObjCCategoryDecl>(PD->getDeclContext());
102523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!CDecl || !CDecl->IsClassExtension())
102623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
102723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
102823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCInterfaceDecl *ID = CDecl->getClassInterface();
102923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!ID)
103023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
103123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
103223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  IdentifierInfo *PropertyId = PD->getIdentifier();
103323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCPropertyDecl *prevDecl =
103423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(ID), PropertyId);
103523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
103623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!prevDecl)
103723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
103823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
103923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // Visit synthesized methods since they will be skipped when visiting
104023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // the @interface.
104123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (ObjCMethodDecl *MD = prevDecl->getGetterMethodDecl())
1042a054fb46b1fb596d1719b89d2d9a5be3c32a4b0dTed Kremenek    if (MD->isSynthesized() && MD->getLexicalDeclContext() == CDecl)
104323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek      if (Visit(MakeCXCursor(MD, TU)))
104423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return true;
104523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
104623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (ObjCMethodDecl *MD = prevDecl->getSetterMethodDecl())
1047a054fb46b1fb596d1719b89d2d9a5be3c32a4b0dTed Kremenek    if (MD->isSynthesized() && MD->getLexicalDeclContext() == CDecl)
104823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek      if (Visit(MakeCXCursor(MD, TU)))
104923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return true;
105023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
105123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  return false;
105223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek}
105323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
1054b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitObjCInterfaceDecl(ObjCInterfaceDecl *D) {
1055dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek  // Issue callbacks for super class.
1056b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (D->getSuperClass() &&
1057b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1058f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        D->getSuperClassLoc(),
1059b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                        TU)))
1060b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
1061f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
106278db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  ObjCInterfaceDecl::protocol_loc_iterator PL = D->protocol_loc_begin();
106378db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  for (ObjCInterfaceDecl::protocol_iterator I = D->protocol_begin(),
106478db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor         E = D->protocol_end(); I != E; ++I, ++PL)
1065b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1066b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
1067f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1068a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitObjCContainerDecl(D);
1069dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1070dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
10711ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCImplDecl(ObjCImplDecl *D) {
10721ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCContainerDecl(D);
10731ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
10741ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
10751ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D) {
1076ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  // 'ID' could be null when dealing with invalid code.
1077ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  if (ObjCInterfaceDecl *ID = D->getClassInterface())
1078ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek    if (Visit(MakeCursorObjCClassRef(ID, D->getLocation(), TU)))
1079ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek      return true;
1080f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10811ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCImplDecl(D);
10821ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
10831ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
10841ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCImplementationDecl(ObjCImplementationDecl *D) {
10851ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor#if 0
10861ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // Issue callbacks for super class.
10871ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // FIXME: No source location information!
10881ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (D->getSuperClass() &&
10891ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1090f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        D->getSuperClassLoc(),
10911ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor                                        TU)))
1092a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return true;
10931ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor#endif
1094f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10951ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCImplDecl(D);
1096dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1097dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
10981ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCForwardProtocolDecl(ObjCForwardProtocolDecl *D) {
10991ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  ObjCForwardProtocolDecl::protocol_loc_iterator PL = D->protocol_loc_begin();
11001ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCForwardProtocolDecl::protocol_iterator I = D->protocol_begin(),
11011ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor                                                  E = D->protocol_end();
11021ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       I != E; ++I, ++PL)
1103b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1104b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
1105f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1106f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return false;
1107dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1108dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
11091ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCClassDecl(ObjCClassDecl *D) {
11101ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCClassDecl::iterator C = D->begin(), CEnd = D->end(); C != CEnd; ++C)
11111ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    if (Visit(MakeCursorObjCClassRef(C->getInterface(), C->getLocation(), TU)))
11121ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
1113f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
11141ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
1115dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
11165e4bc590b0ea010e38372d0b4a0aab578a746fe6Benjamin Kramer
1117a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregorbool CursorVisitor::VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *PD) {
1118a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  if (ObjCIvarDecl *Ivar = PD->getPropertyIvarDecl())
1119a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor    return Visit(MakeCursorMemberRef(Ivar, PD->getPropertyIvarDeclLoc(), TU));
1120a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor
1121a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  return false;
1122a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor}
1123a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor
11248f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenekbool CursorVisitor::VisitNamespaceDecl(NamespaceDecl *D) {
11258f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  return VisitDeclContext(D);
11268f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek}
11278f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek
11286931900f43cea558c6974075256c07728dbfecc6Douglas Gregorbool CursorVisitor::VisitNamespaceAliasDecl(NamespaceAliasDecl *D) {
1129c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
11300cfaf6a270ecd0f5c7e541a8047c87948317548bDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
11310cfaf6a270ecd0f5c7e541a8047c87948317548bDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1132c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
11336931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
11346931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  return Visit(MakeCursorNamespaceRef(D->getAliasedNamespace(),
11356931900f43cea558c6974075256c07728dbfecc6Douglas Gregor                                      D->getTargetNameLoc(), TU));
11366931900f43cea558c6974075256c07728dbfecc6Douglas Gregor}
11376931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
11387e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUsingDecl(UsingDecl *D) {
1139c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1140dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1141dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1142c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1143dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
11447e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
11451f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (Visit(MakeCursorOverloadedDeclRef(D, D->getLocation(), TU)))
11461f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return true;
11471f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
11487e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return VisitDeclarationNameInfo(D->getNameInfo());
11497e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11507e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
11510a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregorbool CursorVisitor::VisitUsingDirectiveDecl(UsingDirectiveDecl *D) {
1152c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1153db9924191092b4d426cc066637d81698211846aaDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1154db9924191092b4d426cc066637d81698211846aaDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1155c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
11560a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
11570a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  return Visit(MakeCursorNamespaceRef(D->getNominatedNamespaceAsWritten(),
11580a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor                                      D->getIdentLocation(), TU));
11590a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor}
11600a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
11617e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D) {
1162c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1163dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1164dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1165c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1166dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
1167c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
11687e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return VisitDeclarationNameInfo(D->getNameInfo());
11697e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11707e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
11717e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUnresolvedUsingTypenameDecl(
11727e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor                                               UnresolvedUsingTypenameDecl *D) {
1173c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1174dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1175dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1176c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1177c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
11787e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return false;
11797e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11807e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
118101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitDeclarationNameInfo(DeclarationNameInfo Name) {
118201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  switch (Name.getName().getNameKind()) {
118301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::Identifier:
118401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXLiteralOperatorName:
118501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXOperatorName:
118601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXUsingDirective:
118701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
118801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
118901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXConstructorName:
119001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXDestructorName:
119101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXConversionFunctionName:
119201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (TypeSourceInfo *TSInfo = Name.getNamedTypeInfo())
119301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return Visit(TSInfo->getTypeLoc());
119401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
119501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
119601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCZeroArgSelector:
119701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCOneArgSelector:
119801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCMultiArgSelector:
119901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Per-identifier location info?
120001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
120101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  }
120201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
120301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  return false;
120401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor}
120501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
1206c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregorbool CursorVisitor::VisitNestedNameSpecifier(NestedNameSpecifier *NNS,
1207c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                             SourceRange Range) {
1208c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // FIXME: This whole routine is a hack to work around the lack of proper
1209c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // source information in nested-name-specifiers (PR5791). Since we do have
1210c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // a beginning source location, we can visit the first component of the
1211c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // nested-name-specifier, if it's a single-token component.
1212c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  if (!NNS)
1213c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    return false;
1214c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1215c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Get the first component in the nested-name-specifier.
1216c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  while (NestedNameSpecifier *Prefix = NNS->getPrefix())
1217c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    NNS = Prefix;
1218c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1219c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  switch (NNS->getKind()) {
1220c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Namespace:
1221c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    return Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(), Range.getBegin(),
1222c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                        TU));
1223c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
122414aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor  case NestedNameSpecifier::NamespaceAlias:
122514aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor    return Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
122614aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor                                        Range.getBegin(), TU));
122714aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor
1228c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::TypeSpec: {
1229c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // If the type has a form where we know that the beginning of the source
1230c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // range matches up with a reference cursor. Visit the appropriate reference
1231c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // cursor.
1232f4c7371fb1d3cebcfb40abad4537bb82515704eaJohn McCall    const Type *T = NNS->getAsType();
1233c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TypedefType *Typedef = dyn_cast<TypedefType>(T))
1234c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return Visit(MakeCursorTypeRef(Typedef->getDecl(), Range.getBegin(), TU));
1235c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TagType *Tag = dyn_cast<TagType>(T))
1236c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return Visit(MakeCursorTypeRef(Tag->getDecl(), Range.getBegin(), TU));
1237c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TemplateSpecializationType *TST
1238c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                      = dyn_cast<TemplateSpecializationType>(T))
1239c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return VisitTemplateName(TST->getTemplateName(), Range.getBegin());
1240c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    break;
1241c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  }
1242c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1243c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::TypeSpecWithTemplate:
1244c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Global:
1245c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Identifier:
1246c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    break;
1247c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  }
1248c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1249c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  return false;
1250c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor}
1251c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1252dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregorbool
1253dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas GregorCursorVisitor::VisitNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
1254dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  llvm::SmallVector<NestedNameSpecifierLoc, 4> Qualifiers;
1255dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  for (; Qualifier; Qualifier = Qualifier.getPrefix())
1256dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    Qualifiers.push_back(Qualifier);
1257dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1258dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  while (!Qualifiers.empty()) {
1259dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    NestedNameSpecifierLoc Q = Qualifiers.pop_back_val();
1260dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    NestedNameSpecifier *NNS = Q.getNestedNameSpecifier();
1261dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    switch (NNS->getKind()) {
1262dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Namespace:
1263dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(),
1264c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor                                       Q.getLocalBeginLoc(),
1265dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor                                       TU)))
1266dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1267dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1268dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1269dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1270dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::NamespaceAlias:
1271dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
1272c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor                                       Q.getLocalBeginLoc(),
1273dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor                                       TU)))
1274dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1275dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1276dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1277dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1278dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::TypeSpec:
1279dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::TypeSpecWithTemplate:
1280dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(Q.getTypeLoc()))
1281dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1282dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1283dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1284dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1285dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Global:
1286dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Identifier:
1287dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1288dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    }
1289dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
1290dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1291dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  return false;
1292dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor}
1293dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1294fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateParameters(
1295fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                          const TemplateParameterList *Params) {
1296fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  if (!Params)
1297fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1298fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1299fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  for (TemplateParameterList::const_iterator P = Params->begin(),
1300fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                          PEnd = Params->end();
1301fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor       P != PEnd; ++P) {
1302fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Visit(MakeCXCursor(*P, TU)))
1303fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return true;
1304fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  }
1305fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1306fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1307fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1308fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
13090b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregorbool CursorVisitor::VisitTemplateName(TemplateName Name, SourceLocation Loc) {
13100b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  switch (Name.getKind()) {
13110b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::Template:
13120b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return Visit(MakeCursorTemplateRef(Name.getAsTemplateDecl(), Loc, TU));
13130b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
13140b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::OverloadedTemplate:
13151f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    // Visit the overloaded template set.
13161f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (Visit(MakeCursorOverloadedDeclRef(Name, Loc, TU)))
13171f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return true;
13181f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
13190b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return false;
13200b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
13210b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::DependentTemplate:
13220b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    // FIXME: Visit nested-name-specifier.
13230b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return false;
13240b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
13250b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::QualifiedTemplate:
13260b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    // FIXME: Visit nested-name-specifier.
13270b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return Visit(MakeCursorTemplateRef(
13280b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                                  Name.getAsQualifiedTemplateName()->getDecl(),
13290b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                                       Loc, TU));
13301aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor
13311aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor  case TemplateName::SubstTemplateTemplateParmPack:
13321aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor    return Visit(MakeCursorTemplateRef(
13331aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor                  Name.getAsSubstTemplateTemplateParmPack()->getParameterPack(),
13341aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor                                       Loc, TU));
13350b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  }
13360b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
13370b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  return false;
13380b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor}
13390b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
1340fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateArgumentLoc(const TemplateArgumentLoc &TAL) {
1341fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  switch (TAL.getArgument().getKind()) {
1342fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Null:
1343fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Integral:
1344fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Pack:
1345fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
134687dd697dcc8ecb64df73ae64d61b8c80ff0c157cDouglas Gregor
1347fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Type:
1348fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (TypeSourceInfo *TSInfo = TAL.getTypeSourceInfo())
1349fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return Visit(TSInfo->getTypeLoc());
1350fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1351fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1352fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Declaration:
1353fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Expr *E = TAL.getSourceDeclExpression())
1354fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return Visit(MakeCXCursor(E, StmtParent, TU));
1355fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1356fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1357fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Expression:
1358fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Expr *E = TAL.getSourceExpression())
1359fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return Visit(MakeCXCursor(E, StmtParent, TU));
1360fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1361fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1362fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Template:
1363a7fc901a2e39bfe55bfcff5934b2d9fdf9656491Douglas Gregor  case TemplateArgument::TemplateExpansion:
1364b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor    if (VisitNestedNameSpecifierLoc(TAL.getTemplateQualifierLoc()))
1365b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor      return true;
1366b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor
1367a7fc901a2e39bfe55bfcff5934b2d9fdf9656491Douglas Gregor    return VisitTemplateName(TAL.getArgument().getAsTemplateOrTemplatePattern(),
13680b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                             TAL.getTemplateNameLoc());
1369fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  }
1370fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1371fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1372fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1373fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1374a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenekbool CursorVisitor::VisitLinkageSpecDecl(LinkageSpecDecl *D) {
1375a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  return VisitDeclContext(D);
1376a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek}
1377a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek
137801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitQualifiedTypeLoc(QualifiedTypeLoc TL) {
137901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  return Visit(TL.getUnqualifiedLoc());
138001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor}
138101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
1382f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitBuiltinTypeLoc(BuiltinTypeLoc TL) {
1383a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTContext &Context = AU->getASTContext();
1384f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1385f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // Some builtin types (such as Objective-C's "id", "sel", and
1386f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // "Class") have associated declarations. Create cursors for those.
1387f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  QualType VisitType;
1388f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  switch (TL.getType()->getAs<BuiltinType>()->getKind()) {
13896b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Void:
1390f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Bool:
13916b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Char_U:
13926b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UChar:
1393f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Char16:
1394f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Char32:
13956b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UShort:
13966b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UInt:
13976b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ULong:
13986b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ULongLong:
13996b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UInt128:
1400f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Char_S:
14016b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::SChar:
14023f59c975aa5d047f7edd1b900b5e885c38af0ef7Chris Lattner  case BuiltinType::WChar_U:
14033f59c975aa5d047f7edd1b900b5e885c38af0ef7Chris Lattner  case BuiltinType::WChar_S:
14046b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Short:
1405f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Int:
1406f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Long:
1407c4174cc4b9b657abb77d0825de473ea29cf48297Ted Kremenek  case BuiltinType::LongLong:
14086b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Int128:
14096b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Float:
14106b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Double:
14116b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::LongDouble:
1412f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::NullPtr:
1413f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Overload:
14146b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Dependent:
14151de4d4e8cb2e9c88809fea8092bc6e835a5473d2John McCall  case BuiltinType::UnknownAny:
1416f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
14176b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
1418f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::ObjCId:
1419f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    VisitType = Context.getObjCIdType();
1420f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
14216b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
14226b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ObjCClass:
14236b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek    VisitType = Context.getObjCClassType();
14246b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek    break;
14256b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
1426f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::ObjCSel:
1427f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    VisitType = Context.getObjCSelType();
1428f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
1429f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1430f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1431f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (!VisitType.isNull()) {
1432f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    if (const TypedefType *Typedef = VisitType->getAs<TypedefType>())
1433f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek      return Visit(MakeCursorTypeRef(Typedef->getDecl(), TL.getBuiltinLoc(),
1434f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                     TU));
1435f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1436f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1437f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1438f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1439f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
14407d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorbool CursorVisitor::VisitTypedefTypeLoc(TypedefTypeLoc TL) {
1441162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  return Visit(MakeCursorTypeRef(TL.getTypedefNameDecl(), TL.getNameLoc(), TU));
14427d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor}
14437d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
1444f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitUnresolvedUsingTypeLoc(UnresolvedUsingTypeLoc TL) {
1445f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1446f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1447f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1448f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitTagTypeLoc(TagTypeLoc TL) {
1449f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1450f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1451f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1452fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateTypeParmTypeLoc(TemplateTypeParmTypeLoc TL) {
14531f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  // FIXME: We can't visit the template type parameter, because there's
1454fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // no context information with which we can match up the depth/index in the
1455fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // type to the appropriate
1456fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1457fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1458fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1459f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL) {
1460f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(MakeCursorObjCClassRef(TL.getIFaceDecl(), TL.getNameLoc(), TU)))
1461f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1462f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1463c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  return false;
1464c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall}
1465c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall
1466c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCallbool CursorVisitor::VisitObjCObjectTypeLoc(ObjCObjectTypeLoc TL) {
1467c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  if (TL.hasBaseTypeAsWritten() && Visit(TL.getBaseLoc()))
1468c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall    return true;
1469c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall
1470f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  for (unsigned I = 0, N = TL.getNumProtocols(); I != N; ++I) {
1471f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    if (Visit(MakeCursorObjCProtocolRef(TL.getProtocol(I), TL.getProtocolLoc(I),
1472f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                        TU)))
1473f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor      return true;
1474f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1475f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1476f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1477f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1478f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1479f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitObjCObjectPointerTypeLoc(ObjCObjectPointerTypeLoc TL) {
1480c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  return Visit(TL.getPointeeLoc());
1481f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1482f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1483075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnarabool CursorVisitor::VisitParenTypeLoc(ParenTypeLoc TL) {
1484075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara  return Visit(TL.getInnerLoc());
1485075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara}
1486075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara
1487f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitPointerTypeLoc(PointerTypeLoc TL) {
1488f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1489f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1490f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1491f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitBlockPointerTypeLoc(BlockPointerTypeLoc TL) {
1492f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1493f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1494f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1495f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitMemberPointerTypeLoc(MemberPointerTypeLoc TL) {
1496f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1497f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1498f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1499f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitLValueReferenceTypeLoc(LValueReferenceTypeLoc TL) {
1500f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return Visit(TL.getPointeeLoc());
1501f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1502f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1503f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitRValueReferenceTypeLoc(RValueReferenceTypeLoc TL) {
1504f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return Visit(TL.getPointeeLoc());
1505f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1506f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
150701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitFunctionTypeLoc(FunctionTypeLoc TL,
150801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor                                         bool SkipResultType) {
150901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  if (!SkipResultType && Visit(TL.getResultLoc()))
1510f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1511f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1512f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
15135dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek    if (Decl *D = TL.getArg(I))
15145dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek      if (Visit(MakeCXCursor(D, TU)))
15155dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek        return true;
1516f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1517f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1518f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1519f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1520f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitArrayTypeLoc(ArrayTypeLoc TL) {
1521f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(TL.getElementLoc()))
1522f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1523f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1524f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Expr *Size = TL.getSizeExpr())
1525f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return Visit(MakeCXCursor(Size, StmtParent, TU));
1526f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1527f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1528f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1529f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1530fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateSpecializationTypeLoc(
1531fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                             TemplateSpecializationTypeLoc TL) {
15320b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  // Visit the template name.
15330b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  if (VisitTemplateName(TL.getTypePtr()->getTemplateName(),
15340b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                        TL.getTemplateNameLoc()))
15350b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return true;
1536fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1537fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // Visit the template arguments.
1538fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
1539fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
1540fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return true;
1541fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1542fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1543fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1544fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
15452332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregorbool CursorVisitor::VisitTypeOfExprTypeLoc(TypeOfExprTypeLoc TL) {
15462332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  return Visit(MakeCXCursor(TL.getUnderlyingExpr(), StmtParent, TU));
15472332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor}
15482332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15492332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregorbool CursorVisitor::VisitTypeOfTypeLoc(TypeOfTypeLoc TL) {
15502332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
15512332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor    return Visit(TSInfo->getTypeLoc());
15522332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15532332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  return false;
15542332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor}
15552332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15562494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregorbool CursorVisitor::VisitDependentNameTypeLoc(DependentNameTypeLoc TL) {
15572494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor  if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
15582494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    return true;
15592494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
15602494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor  return false;
15612494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor}
15622494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
156394fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregorbool CursorVisitor::VisitDependentTemplateSpecializationTypeLoc(
156494fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor                                    DependentTemplateSpecializationTypeLoc TL) {
156594fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  // Visit the nested-name-specifier, if there is one.
156694fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  if (TL.getQualifierLoc() &&
156794fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor      VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
156894fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor    return true;
156994fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
157094fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  // Visit the template arguments.
157194fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
157294fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor    if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
157394fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor      return true;
157494fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
157594fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  return false;
157694fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor}
157794fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
15789e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregorbool CursorVisitor::VisitElaboratedTypeLoc(ElaboratedTypeLoc TL) {
15799e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor  if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
15809e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor    return true;
15819e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor
15829e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor  return Visit(TL.getNamedTypeLoc());
15839e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor}
15849e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor
15857536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregorbool CursorVisitor::VisitPackExpansionTypeLoc(PackExpansionTypeLoc TL) {
15867536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor  return Visit(TL.getPatternLoc());
15877536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor}
15887536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor
15893064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenekbool CursorVisitor::VisitCXXRecordDecl(CXXRecordDecl *D) {
1590c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  // Visit the nested-name-specifier, if present.
1591c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1592c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1593c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      return true;
1594c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor
15953064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  if (D->isDefinition()) {
15963064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    for (CXXRecordDecl::base_class_iterator I = D->bases_begin(),
15973064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek         E = D->bases_end(); I != E; ++I) {
15983064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      if (Visit(cxcursor::MakeCursorCXXBaseSpecifier(I, TU)))
15993064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek        return true;
16003064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
16013064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  }
16023064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek
16033064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  return VisitTagDecl(D);
16043064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek}
16053064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek
160609dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenekbool CursorVisitor::VisitAttributes(Decl *D) {
1607cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt  for (AttrVec::const_iterator i = D->attr_begin(), e = D->attr_end();
1608cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt       i != e; ++i)
1609cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt    if (Visit(MakeCXCursor(*i, D, TU)))
161009dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek        return true;
161109dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
161209dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek  return false;
161309dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek}
161409dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
1615c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
1616c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek// Data-recursive visitor methods.
1617c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
1618c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
161928a719433411ef782b582946823bc648ddcc4533Ted Kremeneknamespace {
1620035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek#define DEF_JOB(NAME, DATA, KIND)\
1621035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass NAME : public VisitorJob {\
1622035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:\
1623035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  NAME(DATA *d, CXCursor parent) : VisitorJob(parent, VisitorJob::KIND, d) {} \
1624035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) { return VJ->getKind() == KIND; }\
1625f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DATA *get() const { return static_cast<DATA*>(data[0]); }\
1626035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1627035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1628035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(StmtVisit, Stmt, StmtVisitKind)
1629035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(MemberExprParts, MemberExpr, MemberExprPartsKind)
1630e4979ccb5960608edce73f3b274eb7c2de15dac5Ted KremenekDEF_JOB(DeclRefExprParts, DeclRefExpr, DeclRefExprPartsKind)
1631035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(OverloadExprParts, OverloadExpr, OverloadExprPartsKind)
163260608ec12d17168a3d1f415409a6a6eaf6d94508Ted KremenekDEF_JOB(ExplicitTemplateArgsVisit, ExplicitTemplateArgumentList,
163360608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        ExplicitTemplateArgsVisitKind)
163494d96291cd041adc5731a2294828a9c20e450b74Douglas GregorDEF_JOB(SizeOfPackExprParts, SizeOfPackExpr, SizeOfPackExprPartsKind)
1635035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek#undef DEF_JOB
1636035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1637035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass DeclVisit : public VisitorJob {
1638035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:
1639035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  DeclVisit(Decl *d, CXCursor parent, bool isFirst) :
1640035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    VisitorJob(parent, VisitorJob::DeclVisitKind,
1641035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek               d, isFirst ? (void*) 1 : (void*) 0) {}
1642035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) {
164382f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    return VJ->getKind() == DeclVisitKind;
1644035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1645f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  Decl *get() const { return static_cast<Decl*>(data[0]); }
1646f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  bool isFirst() const { return data[1] ? true : false; }
1647035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1648035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass TypeLocVisit : public VisitorJob {
1649035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:
1650035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  TypeLocVisit(TypeLoc tl, CXCursor parent) :
1651035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    VisitorJob(parent, VisitorJob::TypeLocVisitKind,
1652035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek               tl.getType().getAsOpaquePtr(), tl.getOpaqueData()) {}
1653035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1654035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1655035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    return VJ->getKind() == TypeLocVisitKind;
1656035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1657035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
165882f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek  TypeLoc get() const {
1659f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    QualType T = QualType::getFromOpaquePtr(data[0]);
1660f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return TypeLoc(T, data[1]);
1661035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1662035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1663035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1664ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekclass LabelRefVisit : public VisitorJob {
1665ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekpublic:
1666ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  LabelRefVisit(LabelDecl *LD, SourceLocation labelLoc, CXCursor parent)
1667ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner    : VisitorJob(parent, VisitorJob::LabelRefVisitKind, LD,
1668dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 labelLoc.getPtrEncoding()) {}
1669ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek
1670ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1671ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek    return VJ->getKind() == VisitorJob::LabelRefVisitKind;
1672ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  }
1673ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  LabelDecl *get() const { return static_cast<LabelDecl*>(data[0]); }
1674ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  SourceLocation getLoc() const {
1675dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin    return SourceLocation::getFromPtrEncoding(data[1]); }
1676f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek};
1677f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekclass NestedNameSpecifierVisit : public VisitorJob {
1678f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekpublic:
1679f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  NestedNameSpecifierVisit(NestedNameSpecifier *NS, SourceRange R,
1680f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek                           CXCursor parent)
1681f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    : VisitorJob(parent, VisitorJob::NestedNameSpecifierVisitKind,
1682dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 NS, R.getBegin().getPtrEncoding(),
1683dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 R.getEnd().getPtrEncoding()) {}
1684f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1685f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return VJ->getKind() == VisitorJob::NestedNameSpecifierVisitKind;
1686f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1687f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  NestedNameSpecifier *get() const {
1688f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return static_cast<NestedNameSpecifier*>(data[0]);
1689f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1690f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  SourceRange getSourceRange() const {
1691f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    SourceLocation A =
1692f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      SourceLocation::getFromRawEncoding((unsigned)(uintptr_t) data[1]);
1693f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    SourceLocation B =
1694f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      SourceLocation::getFromRawEncoding((unsigned)(uintptr_t) data[2]);
1695f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return SourceRange(A, B);
1696f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1697f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek};
1698f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1699f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorclass NestedNameSpecifierLocVisit : public VisitorJob {
1700f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorpublic:
1701f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  NestedNameSpecifierLocVisit(NestedNameSpecifierLoc Qualifier, CXCursor parent)
1702f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    : VisitorJob(parent, VisitorJob::NestedNameSpecifierLocVisitKind,
1703f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                 Qualifier.getNestedNameSpecifier(),
1704f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                 Qualifier.getOpaqueData()) { }
1705f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1706f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  static bool classof(const VisitorJob *VJ) {
1707f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    return VJ->getKind() == VisitorJob::NestedNameSpecifierLocVisitKind;
1708f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  }
1709f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1710f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  NestedNameSpecifierLoc get() const {
1711f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    return NestedNameSpecifierLoc(static_cast<NestedNameSpecifier*>(data[0]),
1712f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                                  data[1]);
1713f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  }
1714f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor};
1715f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1716f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekclass DeclarationNameInfoVisit : public VisitorJob {
1717f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekpublic:
1718f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DeclarationNameInfoVisit(Stmt *S, CXCursor parent)
1719f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    : VisitorJob(parent, VisitorJob::DeclarationNameInfoVisitKind, S) {}
1720f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1721f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return VJ->getKind() == VisitorJob::DeclarationNameInfoVisitKind;
1722f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1723f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DeclarationNameInfo get() const {
1724f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    Stmt *S = static_cast<Stmt*>(data[0]);
1725f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    switch (S->getStmtClass()) {
1726f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    default:
1727f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      llvm_unreachable("Unhandled Stmt");
1728f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    case Stmt::CXXDependentScopeMemberExprClass:
1729f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      return cast<CXXDependentScopeMemberExpr>(S)->getMemberNameInfo();
1730f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    case Stmt::DependentScopeDeclRefExprClass:
1731f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      return cast<DependentScopeDeclRefExpr>(S)->getNameInfo();
1732f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    }
1733f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1734ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek};
1735cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekclass MemberRefVisit : public VisitorJob {
1736cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekpublic:
1737cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  MemberRefVisit(FieldDecl *D, SourceLocation L, CXCursor parent)
1738cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    : VisitorJob(parent, VisitorJob::MemberRefVisitKind, D,
1739dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 L.getPtrEncoding()) {}
1740cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  static bool classof(const VisitorJob *VJ) {
1741cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return VJ->getKind() == VisitorJob::MemberRefVisitKind;
1742cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1743cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  FieldDecl *get() const {
1744cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return static_cast<FieldDecl*>(data[0]);
1745cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1746cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  SourceLocation getLoc() const {
1747cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return SourceLocation::getFromRawEncoding((unsigned)(uintptr_t) data[1]);
1748cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1749cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek};
175028a719433411ef782b582946823bc648ddcc4533Ted Kremenekclass EnqueueVisitor : public StmtVisitor<EnqueueVisitor, void> {
175128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  VisitorWorkList &WL;
175228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  CXCursor Parent;
175328a719433411ef782b582946823bc648ddcc4533Ted Kremenekpublic:
175428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueVisitor(VisitorWorkList &wl, CXCursor parent)
175528a719433411ef782b582946823bc648ddcc4533Ted Kremenek    : WL(wl), Parent(parent) {}
175628a719433411ef782b582946823bc648ddcc4533Ted Kremenek
1757ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  void VisitAddrLabelExpr(AddrLabelExpr *E);
175873d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitBlockExpr(BlockExpr *B);
175928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
1760083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  void VisitCompoundStmt(CompoundStmt *S);
176111b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  void VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) { /* Do nothing. */ }
1762f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E);
176311b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  void VisitCXXNewExpr(CXXNewExpr *E);
17646d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  void VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E);
176528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E);
1766cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E);
176773d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E);
1768b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  void VisitCXXTypeidExpr(CXXTypeidExpr *E);
176955b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  void VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E);
17701e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  void VisitCXXUuidofExpr(CXXUuidofExpr *E);
1771e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek  void VisitDeclRefExpr(DeclRefExpr *D);
1772035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  void VisitDeclStmt(DeclStmt *S);
1773f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E);
1774cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitDesignatedInitExpr(DesignatedInitExpr *E);
177528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitExplicitCastExpr(ExplicitCastExpr *E);
177628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitForStmt(ForStmt *FS);
1777ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  void VisitGotoStmt(GotoStmt *GS);
177828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitIfStmt(IfStmt *If);
177928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitInitListExpr(InitListExpr *IE);
178028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitMemberExpr(MemberExpr *M);
1781cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitOffsetOfExpr(OffsetOfExpr *E);
178273d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitObjCEncodeExpr(ObjCEncodeExpr *E);
178328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitObjCMessageExpr(ObjCMessageExpr *M);
178428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitOverloadExpr(OverloadExpr *E);
1785f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbourne  void VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E);
178628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitStmt(Stmt *S);
178728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitSwitchStmt(SwitchStmt *S);
178828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitWhileStmt(WhileStmt *W);
17892939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek  void VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E);
17906ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  void VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E);
179128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitUnresolvedMemberExpr(UnresolvedMemberExpr *U);
17929d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  void VisitVAArgExpr(VAArgExpr *E);
179394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  void VisitSizeOfPackExpr(SizeOfPackExpr *E);
1794ee8aff06f6a96214731de17b2cb6df407c6c1820Douglas Gregor
179528a719433411ef782b582946823bc648ddcc4533Ted Kremenekprivate:
1796f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void AddDeclarationNameInfo(Stmt *S);
1797f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void AddNestedNameSpecifier(NestedNameSpecifier *NS, SourceRange R);
1798f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  void AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier);
179960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  void AddExplicitTemplateArgs(const ExplicitTemplateArgumentList *A);
1800cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void AddMemberRef(FieldDecl *D, SourceLocation L);
180128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void AddStmt(Stmt *S);
1802035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  void AddDecl(Decl *D, bool isFirst = true);
180328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void AddTypeLoc(TypeSourceInfo *TI);
180428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void EnqueueChildren(Stmt *S);
180528a719433411ef782b582946823bc648ddcc4533Ted Kremenek};
180628a719433411ef782b582946823bc648ddcc4533Ted Kremenek} // end anonyous namespace
180728a719433411ef782b582946823bc648ddcc4533Ted Kremenek
1808f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::AddDeclarationNameInfo(Stmt *S) {
1809f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  // 'S' should always be non-null, since it comes from the
1810f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  // statement we are visiting.
1811f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  WL.push_back(DeclarationNameInfoVisit(S, Parent));
1812f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
1813f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::AddNestedNameSpecifier(NestedNameSpecifier *N,
1814f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek                                            SourceRange R) {
1815f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  if (N)
1816f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    WL.push_back(NestedNameSpecifierVisit(N, R, Parent));
1817f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
1818f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1819f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorvoid
1820f3db29fff6a583ecda823cf909ab7737d8d30129Douglas GregorEnqueueVisitor::AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
1821f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  if (Qualifier)
1822f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    WL.push_back(NestedNameSpecifierLocVisit(Qualifier, Parent));
1823f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor}
1824f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
182528a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::AddStmt(Stmt *S) {
182628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (S)
182728a719433411ef782b582946823bc648ddcc4533Ted Kremenek    WL.push_back(StmtVisit(S, Parent));
182828a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1829035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekvoid EnqueueVisitor::AddDecl(Decl *D, bool isFirst) {
183028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (D)
1831035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    WL.push_back(DeclVisit(D, Parent, isFirst));
183228a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
183360608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenekvoid EnqueueVisitor::
183460608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  AddExplicitTemplateArgs(const ExplicitTemplateArgumentList *A) {
183560608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  if (A)
183660608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek    WL.push_back(ExplicitTemplateArgsVisit(
183760608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek                        const_cast<ExplicitTemplateArgumentList*>(A), Parent));
183860608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek}
1839cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::AddMemberRef(FieldDecl *D, SourceLocation L) {
1840cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  if (D)
1841cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    WL.push_back(MemberRefVisit(D, L, Parent));
1842cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
184328a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::AddTypeLoc(TypeSourceInfo *TI) {
184428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (TI)
184528a719433411ef782b582946823bc648ddcc4533Ted Kremenek    WL.push_back(TypeLocVisit(TI->getTypeLoc(), Parent));
184628a719433411ef782b582946823bc648ddcc4533Ted Kremenek }
184728a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::EnqueueChildren(Stmt *S) {
1848a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  unsigned size = WL.size();
18497502c1d3ce8bb97bcc4f7bebef507040bd93b26fJohn McCall  for (Stmt::child_range Child = S->children(); Child; ++Child) {
185028a719433411ef782b582946823bc648ddcc4533Ted Kremenek    AddStmt(*Child);
1851a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  }
1852a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  if (size == WL.size())
1853a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek    return;
1854a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  // Now reverse the entries we just added.  This will match the DFS
1855a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  // ordering performed by the worklist.
1856a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
1857a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  std::reverse(I, E);
1858a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek}
1859ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekvoid EnqueueVisitor::VisitAddrLabelExpr(AddrLabelExpr *E) {
1860ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  WL.push_back(LabelRefVisit(E->getLabel(), E->getLabelLoc(), Parent));
1861ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek}
186273d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitBlockExpr(BlockExpr *B) {
186373d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddDecl(B->getBlockDecl());
186473d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
186528a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
186628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(E);
186728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(E->getTypeSourceInfo());
186828a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1869083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenekvoid EnqueueVisitor::VisitCompoundStmt(CompoundStmt *S) {
1870083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  for (CompoundStmt::reverse_body_iterator I = S->body_rbegin(),
1871083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek        E = S->body_rend(); I != E; ++I) {
1872083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek    AddStmt(*I);
1873083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  }
187411b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek}
1875f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::
1876f64d80306144f978148ba92f36f7cea7b671dd34Ted KremenekVisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E) {
1877f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
1878f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddDeclarationNameInfo(E);
18797c3179cf463c3b3b8c21dbb955f933ba50b74f28Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
18807c3179cf463c3b3b8c21dbb955f933ba50b74f28Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
1881f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  if (!E->isImplicitAccess())
1882f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    AddStmt(E->getBase());
1883f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
188411b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenekvoid EnqueueVisitor::VisitCXXNewExpr(CXXNewExpr *E) {
188511b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the initializer or constructor arguments.
188611b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  for (unsigned I = E->getNumConstructorArgs(); I > 0; --I)
188711b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek    AddStmt(E->getConstructorArg(I-1));
188811b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the array size, if any.
188911b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  AddStmt(E->getArraySize());
189011b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the allocated type.
189111b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  AddTypeLoc(E->getAllocatedTypeSourceInfo());
189211b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the placement arguments.
189311b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  for (unsigned I = E->getNumPlacementArgs(); I > 0; --I)
189411b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek    AddStmt(E->getPlacementArg(I-1));
189511b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek}
189628a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *CE) {
18978b8d8c90f2d8ac651d14b57f116d20b3c911ac7fTed Kremenek  for (unsigned I = CE->getNumArgs(); I > 1 /* Yes, this is 1 */; --I)
18988b8d8c90f2d8ac651d14b57f116d20b3c911ac7fTed Kremenek    AddStmt(CE->getArg(I-1));
189928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(CE->getCallee());
190028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(CE->getArg(0));
190128a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1902cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
1903cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the name of the type being destroyed.
1904cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getDestroyedTypeInfo());
1905cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the scope type that looks disturbingly like the nested-name-specifier
1906cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // but isn't.
1907cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getScopeTypeInfo());
1908cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the nested-name-specifier.
1909f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
1910f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
1911cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit base expression.
1912cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddStmt(E->getBase());
1913cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
19146d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenekvoid EnqueueVisitor::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
19156d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
19166d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek}
191773d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
191873d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  EnqueueChildren(E);
191973d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
192073d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
1921b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenekvoid EnqueueVisitor::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
1922b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  EnqueueChildren(E);
1923b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  if (E->isTypeOperand())
1924b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek    AddTypeLoc(E->getTypeOperandSourceInfo());
1925b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek}
192655b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek
192755b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenekvoid EnqueueVisitor::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr
192855b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek                                                     *E) {
192955b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  EnqueueChildren(E);
193055b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
193155b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek}
19321e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenekvoid EnqueueVisitor::VisitCXXUuidofExpr(CXXUuidofExpr *E) {
19331e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  EnqueueChildren(E);
19341e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  if (E->isTypeOperand())
19351e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek    AddTypeLoc(E->getTypeOperandSourceInfo());
19361e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek}
1937e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenekvoid EnqueueVisitor::VisitDeclRefExpr(DeclRefExpr *DR) {
193860608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  if (DR->hasExplicitTemplateArgs()) {
193960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek    AddExplicitTemplateArgs(&DR->getExplicitTemplateArgs());
194060608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  }
1941e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek  WL.push_back(DeclRefExprParts(DR, Parent));
1942e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek}
1943f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
1944f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
1945f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddDeclarationNameInfo(E);
194600cf3cc2718671aa48e8da264a523b0058a8591eDouglas Gregor  AddNestedNameSpecifierLoc(E->getQualifierLoc());
1947f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
1948035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekvoid EnqueueVisitor::VisitDeclStmt(DeclStmt *S) {
1949035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  unsigned size = WL.size();
1950035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  bool isFirst = true;
1951035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  for (DeclStmt::decl_iterator D = S->decl_begin(), DEnd = S->decl_end();
1952035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek       D != DEnd; ++D) {
1953035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    AddDecl(*D, isFirst);
1954035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    isFirst = false;
1955035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1956035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  if (size == WL.size())
1957035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    return;
1958035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  // Now reverse the entries we just added.  This will match the DFS
1959035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  // ordering performed by the worklist.
1960035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
1961035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  std::reverse(I, E);
1962035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek}
1963cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
1964cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddStmt(E->getInit());
1965cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  typedef DesignatedInitExpr::Designator Designator;
1966cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  for (DesignatedInitExpr::reverse_designators_iterator
1967cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek         D = E->designators_rbegin(), DEnd = E->designators_rend();
1968cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek         D != DEnd; ++D) {
1969cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    if (D->isFieldDesignator()) {
1970cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      if (FieldDecl *Field = D->getField())
1971cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        AddMemberRef(Field, D->getFieldLoc());
1972cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
1973cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
1974cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    if (D->isArrayDesignator()) {
1975cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      AddStmt(E->getArrayIndex(*D));
1976cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
1977cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
1978cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    assert(D->isArrayRangeDesignator() && "Unknown designator kind");
1979cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    AddStmt(E->getArrayRangeEnd(*D));
1980cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    AddStmt(E->getArrayRangeStart(*D));
1981cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1982cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
198328a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitExplicitCastExpr(ExplicitCastExpr *E) {
198428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(E);
198528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(E->getTypeInfoAsWritten());
198628a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
198728a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitForStmt(ForStmt *FS) {
198828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getBody());
198928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getInc());
199028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getCond());
199128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(FS->getConditionVariable());
199228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getInit());
199328a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1994ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekvoid EnqueueVisitor::VisitGotoStmt(GotoStmt *GS) {
1995ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  WL.push_back(LabelRefVisit(GS->getLabel(), GS->getLabelLoc(), Parent));
1996ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek}
199728a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitIfStmt(IfStmt *If) {
199828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getElse());
199928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getThen());
200028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getCond());
200128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(If->getConditionVariable());
200228a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
200328a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitInitListExpr(InitListExpr *IE) {
200428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  // We care about the syntactic form of the initializer list, only.
200528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (InitListExpr *Syntactic = IE->getSyntacticForm())
200628a719433411ef782b582946823bc648ddcc4533Ted Kremenek    IE = Syntactic;
200728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(IE);
200828a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
200928a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitMemberExpr(MemberExpr *M) {
201089629a746019a42797495b091711a1d68467e88aDouglas Gregor  WL.push_back(MemberExprParts(M, Parent));
201189629a746019a42797495b091711a1d68467e88aDouglas Gregor
201289629a746019a42797495b091711a1d68467e88aDouglas Gregor  // If the base of the member access expression is an implicit 'this', don't
201389629a746019a42797495b091711a1d68467e88aDouglas Gregor  // visit it.
201489629a746019a42797495b091711a1d68467e88aDouglas Gregor  // FIXME: If we ever want to show these implicit accesses, this will be
201589629a746019a42797495b091711a1d68467e88aDouglas Gregor  // unfortunate. However, clang_getCursor() relies on this behavior.
201675e85048e73fcde2ce9d8a48dfdb1220e132eb59Douglas Gregor  if (!M->isImplicitAccess())
201775e85048e73fcde2ce9d8a48dfdb1220e132eb59Douglas Gregor    AddStmt(M->getBase());
201828a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
201973d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
202073d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddTypeLoc(E->getEncodedTypeSourceInfo());
202173d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
202228a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitObjCMessageExpr(ObjCMessageExpr *M) {
202328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(M);
202428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(M->getClassReceiverTypeInfo());
202528a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
2026cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitOffsetOfExpr(OffsetOfExpr *E) {
2027cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the components of the offsetof expression.
2028cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  for (unsigned N = E->getNumComponents(), I = N; I > 0; --I) {
2029cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    typedef OffsetOfExpr::OffsetOfNode OffsetOfNode;
2030cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    const OffsetOfNode &Node = E->getComponent(I-1);
2031cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    switch (Node.getKind()) {
2032cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Array:
2033cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      AddStmt(E->getIndexExpr(Node.getArrayExprIndex()));
2034cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      break;
2035cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Field:
203606dec892b5300b43263d25c5476b506c9d6cfbadAbramo Bagnara      AddMemberRef(Node.getField(), Node.getSourceRange().getEnd());
2037cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      break;
2038cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Identifier:
2039cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Base:
2040cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
2041cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
2042cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
2043cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the type into which we're computing the offset.
2044cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
2045cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
204628a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitOverloadExpr(OverloadExpr *E) {
204760608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
20486045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek  WL.push_back(OverloadExprParts(E, Parent));
20496045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek}
2050f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbournevoid EnqueueVisitor::VisitUnaryExprOrTypeTraitExpr(
2051f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbourne                                              UnaryExprOrTypeTraitExpr *E) {
20526d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  EnqueueChildren(E);
20536d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  if (E->isArgumentType())
20546d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek    AddTypeLoc(E->getArgumentTypeInfo());
20556d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek}
205628a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitStmt(Stmt *S) {
205728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(S);
205828a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
205928a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitSwitchStmt(SwitchStmt *S) {
206028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(S->getBody());
206128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(S->getCond());
206228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(S->getConditionVariable());
206328a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
2064fafa75aebadef8d6b44a920e3f40529f150a5574Ted Kremenek
206528a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitWhileStmt(WhileStmt *W) {
206628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(W->getBody());
206728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(W->getCond());
206828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(W->getConditionVariable());
206928a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
20702939b6f356161f572712d4d6310b65f9599e3675Ted Kremenekvoid EnqueueVisitor::VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E) {
20712939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek  AddTypeLoc(E->getQueriedTypeSourceInfo());
20722939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek}
20736ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet
20746ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichetvoid EnqueueVisitor::VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E) {
20756ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  AddTypeLoc(E->getRhsTypeSourceInfo());
20760a03a3f98b14006a54bcac9e8908a7c9f50e519fFrancois Pichet  AddTypeLoc(E->getLhsTypeSourceInfo());
20776ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet}
20786ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet
207928a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *U) {
208028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  VisitOverloadExpr(U);
208128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (!U->isImplicitAccess())
208228a719433411ef782b582946823bc648ddcc4533Ted Kremenek    AddStmt(U->getBase());
208328a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
20849d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenekvoid EnqueueVisitor::VisitVAArgExpr(VAArgExpr *E) {
20859d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  AddStmt(E->getSubExpr());
20869d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  AddTypeLoc(E->getWrittenTypeInfo());
20879d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek}
208894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregorvoid EnqueueVisitor::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
208994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  WL.push_back(SizeOfPackExprParts(E, Parent));
209094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor}
20916045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek
2092c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekvoid CursorVisitor::EnqueueWorkList(VisitorWorkList &WL, Stmt *S) {
209328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueVisitor(WL, MakeCXCursor(S, StmtParent, TU)).Visit(S);
2094c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2095c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2096c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekbool CursorVisitor::IsInRegionOfInterest(CXCursor C) {
2097c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  if (RegionOfInterest.isValid()) {
2098c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    SourceRange Range = getRawCursorExtent(C);
2099c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    if (Range.isInvalid() || CompareRegionOfInterest(Range))
2100c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      return false;
2101c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  }
2102c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  return true;
2103c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2104c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2105c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekbool CursorVisitor::RunVisitorWorkList(VisitorWorkList &WL) {
2106c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  while (!WL.empty()) {
2107c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    // Dequeue the worklist item.
210882f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    VisitorJob LI = WL.back();
210982f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    WL.pop_back();
211082f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek
2111c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    // Set the Parent field, then back to its old value once we're done.
2112c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    SetParentRAII SetParent(Parent, StmtParent, LI.getParent());
2113c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2114c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    switch (LI.getKind()) {
2115f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek      case VisitorJob::DeclVisitKind: {
211682f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        Decl *D = cast<DeclVisit>(&LI)->get();
2117f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        if (!D)
2118f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek          continue;
2119f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek
2120f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        // For now, perform default visitation for Decls.
212182f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        if (Visit(MakeCXCursor(D, TU, cast<DeclVisit>(&LI)->isFirst())))
2122f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek            return true;
2123f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek
2124f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        continue;
2125f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek      }
212660608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek      case VisitorJob::ExplicitTemplateArgsVisitKind: {
212760608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        const ExplicitTemplateArgumentList *ArgList =
212860608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek          cast<ExplicitTemplateArgsVisit>(&LI)->get();
212960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        for (const TemplateArgumentLoc *Arg = ArgList->getTemplateArgs(),
213060608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek               *ArgEnd = Arg + ArgList->NumTemplateArgs;
213160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek               Arg != ArgEnd; ++Arg) {
213260608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek          if (VisitTemplateArgumentLoc(*Arg))
213360608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek            return true;
213460608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        }
213560608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        continue;
213660608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek      }
2137cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek      case VisitorJob::TypeLocVisitKind: {
2138cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek        // Perform default visitation for TypeLocs.
213982f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        if (Visit(cast<TypeLocVisit>(&LI)->get()))
2140cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek          return true;
2141cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek        continue;
2142cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek      }
2143ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek      case VisitorJob::LabelRefVisitKind: {
2144ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner        LabelDecl *LS = cast<LabelRefVisit>(&LI)->get();
2145e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek        if (LabelStmt *stmt = LS->getStmt()) {
2146e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek          if (Visit(MakeCursorLabelRef(stmt, cast<LabelRefVisit>(&LI)->getLoc(),
2147e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek                                       TU))) {
2148e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek            return true;
2149e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek          }
2150e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek        }
2151ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek        continue;
2152ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek      }
2153f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
2154f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      case VisitorJob::NestedNameSpecifierVisitKind: {
2155f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        NestedNameSpecifierVisit *V = cast<NestedNameSpecifierVisit>(&LI);
2156f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        if (VisitNestedNameSpecifier(V->get(), V->getSourceRange()))
2157f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek          return true;
2158f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        continue;
2159f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      }
2160f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
2161f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor      case VisitorJob::NestedNameSpecifierLocVisitKind: {
2162f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        NestedNameSpecifierLocVisit *V = cast<NestedNameSpecifierLocVisit>(&LI);
2163f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        if (VisitNestedNameSpecifierLoc(V->get()))
2164f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor          return true;
2165f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        continue;
2166f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor      }
2167f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
2168f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      case VisitorJob::DeclarationNameInfoVisitKind: {
2169f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        if (VisitDeclarationNameInfo(cast<DeclarationNameInfoVisit>(&LI)
2170f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek                                     ->get()))
2171f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek          return true;
2172f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        continue;
2173f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      }
2174cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      case VisitorJob::MemberRefVisitKind: {
2175cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        MemberRefVisit *V = cast<MemberRefVisit>(&LI);
2176cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        if (Visit(MakeCursorMemberRef(V->get(), V->getLoc(), TU)))
2177cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          return true;
2178cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        continue;
2179cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      }
2180c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      case VisitorJob::StmtVisitKind: {
218182f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        Stmt *S = cast<StmtVisit>(&LI)->get();
21828c269ac75569454a049385b1246140db5f2b6faaTed Kremenek        if (!S)
21838c269ac75569454a049385b1246140db5f2b6faaTed Kremenek          continue;
21848c269ac75569454a049385b1246140db5f2b6faaTed Kremenek
2185f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        // Update the current cursor.
2186c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        CXCursor Cursor = MakeCXCursor(S, StmtParent, TU);
2187cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        if (!IsInRegionOfInterest(Cursor))
2188cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          continue;
2189cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        switch (Visitor(Cursor, Parent, ClientData)) {
2190cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Break: return true;
2191cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Continue: break;
2192cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Recurse:
2193cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek            EnqueueWorkList(WL, S);
219482f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek            break;
2195c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        }
219682f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        continue;
2197c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      }
2198c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      case VisitorJob::MemberExprPartsKind: {
2199c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Handle the other pieces in the MemberExpr besides the base.
220082f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        MemberExpr *M = cast<MemberExprParts>(&LI)->get();
2201c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2202c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the nested-name-specifier
220340d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = M->getQualifierLoc())
220440d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
2205c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek            return true;
2206c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2207c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the declaration name.
2208c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        if (VisitDeclarationNameInfo(M->getMemberNameInfo()))
2209c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          return true;
2210c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2211c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the explicitly-specified template arguments, if any.
2212c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        if (M->hasExplicitTemplateArgs()) {
2213c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          for (const TemplateArgumentLoc *Arg = M->getTemplateArgs(),
2214c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek               *ArgEnd = Arg + M->getNumTemplateArgs();
2215c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek               Arg != ArgEnd; ++Arg) {
2216c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek            if (VisitTemplateArgumentLoc(*Arg))
2217c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek              return true;
2218c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          }
2219c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        }
2220c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        continue;
2221c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      }
2222e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek      case VisitorJob::DeclRefExprPartsKind: {
222382f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        DeclRefExpr *DR = cast<DeclRefExprParts>(&LI)->get();
2224e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        // Visit nested-name-specifier, if present.
222540d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = DR->getQualifierLoc())
222640d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
2227e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek            return true;
2228e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        // Visit declaration name.
2229e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        if (VisitDeclarationNameInfo(DR->getNameInfo()))
2230e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek          return true;
2231e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        continue;
2232e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek      }
22336045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek      case VisitorJob::OverloadExprPartsKind: {
223482f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        OverloadExpr *O = cast<OverloadExprParts>(&LI)->get();
22356045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the nested-name-specifier.
22364c9be89bb615ec07eb3ed507c8fa9d0baa8a5ad7Douglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = O->getQualifierLoc())
22374c9be89bb615ec07eb3ed507c8fa9d0baa8a5ad7Douglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
22386045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek            return true;
22396045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the declaration name.
22406045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        if (VisitDeclarationNameInfo(O->getNameInfo()))
22416045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek          return true;
22426045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the overloaded declaration reference.
22436045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        if (Visit(MakeCursorOverloadedDeclRef(O, TU)))
22446045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek          return true;
22456045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        continue;
22466045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek      }
224794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      case VisitorJob::SizeOfPackExprPartsKind: {
224894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        SizeOfPackExpr *E = cast<SizeOfPackExprParts>(&LI)->get();
224994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        NamedDecl *Pack = E->getPack();
225094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        if (isa<TemplateTypeParmDecl>(Pack)) {
225194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          if (Visit(MakeCursorTypeRef(cast<TemplateTypeParmDecl>(Pack),
225294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor                                      E->getPackLoc(), TU)))
225394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor            return true;
225494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
225594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          continue;
225694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        }
225794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
225894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        if (isa<TemplateTemplateParmDecl>(Pack)) {
225994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          if (Visit(MakeCursorTemplateRef(cast<TemplateTemplateParmDecl>(Pack),
226094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor                                          E->getPackLoc(), TU)))
226194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor            return true;
226294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
226394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          continue;
226494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        }
226594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
226694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        // Non-type template parameter packs and function parameter packs are
226794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        // treated like DeclRefExpr cursors.
226894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        continue;
226994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      }
2270c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    }
2271c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  }
2272c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  return false;
2273c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2274c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2275cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekbool CursorVisitor::Visit(Stmt *S) {
2276d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  VisitorWorkList *WL = 0;
2277d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  if (!WorkListFreeList.empty()) {
2278d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL = WorkListFreeList.back();
2279d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL->clear();
2280d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WorkListFreeList.pop_back();
2281d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
2282d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  else {
2283d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL = new VisitorWorkList();
2284d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WorkListCache.push_back(WL);
2285d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
2286d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  EnqueueWorkList(*WL, S);
2287d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  bool result = RunVisitorWorkList(*WL);
2288d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  WorkListFreeList.push_back(WL);
2289d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  return result;
2290c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2291c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2292c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
2293c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek// Misc. API hooks.
2294c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
2295c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
22968c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregorstatic llvm::sys::Mutex EnableMultithreadingMutex;
22978c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregorstatic bool EnabledMultithreading;
22988c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor
22995e4bc590b0ea010e38372d0b4a0aab578a746fe6Benjamin Kramerextern "C" {
23000a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas GregorCXIndex clang_createIndex(int excludeDeclarationsFromPCH,
23010a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor                          int displayDiagnostics) {
230248615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  // Disable pretty stack trace functionality, which will otherwise be a very
230348615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  // poor citizen of the world and set up all sorts of signal handlers.
230448615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  llvm::DisablePrettyStackTrace = true;
230548615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar
2306c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  // We use crash recovery to make some of our APIs more reliable, implicitly
2307c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  // enable it.
2308c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  llvm::CrashRecoveryContext::Enable();
2309c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar
23108c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  // Enable support for multithreading in LLVM.
23118c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  {
23128c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    llvm::sys::ScopedLock L(EnableMultithreadingMutex);
23138c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    if (!EnabledMultithreading) {
23148c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor      llvm::llvm_start_multithreaded();
23158c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor      EnabledMultithreading = true;
23168c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    }
23178c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  }
23188c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor
2319a030b7cf5e6aad5889b1b662b6979840bc75f87fDouglas Gregor  CIndexer *CIdxr = new CIndexer();
2320e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  if (excludeDeclarationsFromPCH)
2321e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff    CIdxr->setOnlyLocalDecls();
23220a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  if (displayDiagnostics)
23230a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    CIdxr->setDisplayDiagnostics();
2324e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  return CIdxr;
2325600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
2326600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
23279ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarvoid clang_disposeIndex(CXIndex CIdx) {
23282b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (CIdx)
23292b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    delete static_cast<CIndexer *>(CIdx);
23302bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff}
23312bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff
2332d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenekvoid clang_toggleCrashRecovery(unsigned isEnabled) {
2333d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek  if (isEnabled)
2334d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek    llvm::CrashRecoveryContext::Enable();
2335d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek  else
2336d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek    llvm::CrashRecoveryContext::Disable();
2337d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek}
2338d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek
23399ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXTranslationUnit clang_createTranslationUnit(CXIndex CIdx,
2340a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                              const char *ast_filename) {
23412b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CIdx)
23422b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    return 0;
2343f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
23447d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
2345389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  FileSystemOptions FileSystemOpts;
2346389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  FileSystemOpts.WorkingDir = CXXIdx->getWorkingDirectory();
23470d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
234828019772db70d4547be05a042eb950bc910f134fDouglas Gregor  llvm::IntrusiveRefCntPtr<Diagnostic> Diags;
2349a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *TU = ASTUnit::LoadFromASTFile(ast_filename, Diags, FileSystemOpts,
2350a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                  CXXIdx->getOnlyLocalDecls(),
2351a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                  0, 0, true);
2352a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return MakeCXTranslationUnit(TU);
2353600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
2354600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
2355b1c031be513705d924038f497279b9b599868ba1Douglas Gregorunsigned clang_defaultEditingTranslationUnitOptions() {
23562a2c50b330e7754499f42173616a36865b5f313bDouglas Gregor  return CXTranslationUnit_PrecompiledPreamble |
235799ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor         CXTranslationUnit_CacheCompletionResults |
235899ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor         CXTranslationUnit_CXXPrecompiledPreamble;
2359b1c031be513705d924038f497279b9b599868ba1Douglas Gregor}
2360b1c031be513705d924038f497279b9b599868ba1Douglas Gregor
23619ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXTranslationUnit
23629ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarclang_createTranslationUnitFromSourceFile(CXIndex CIdx,
23639ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          const char *source_filename,
23649ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          int num_command_line_args,
23652ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor                                          const char * const *command_line_args,
23664db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor                                          unsigned num_unsaved_files,
2367a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                          struct CXUnsavedFile *unsaved_files) {
23685a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor  return clang_parseTranslationUnit(CIdx, source_filename,
23695a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                    command_line_args, num_command_line_args,
23705a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                    unsaved_files, num_unsaved_files,
23715a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                 CXTranslationUnit_DetailedPreprocessingRecord);
23725a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor}
237319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
237419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbarstruct ParseTranslationUnitInfo {
237519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXIndex CIdx;
237619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  const char *source_filename;
23772ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor  const char *const *command_line_args;
237819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  int num_command_line_args;
237919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  struct CXUnsavedFile *unsaved_files;
238019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned num_unsaved_files;
238119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned options;
238219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXTranslationUnit result;
238319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar};
2384b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbarstatic void clang_parseTranslationUnit_Impl(void *UserData) {
238519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  ParseTranslationUnitInfo *PTUI =
238619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    static_cast<ParseTranslationUnitInfo*>(UserData);
238719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXIndex CIdx = PTUI->CIdx;
238819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  const char *source_filename = PTUI->source_filename;
23892ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor  const char * const *command_line_args = PTUI->command_line_args;
239019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  int num_command_line_args = PTUI->num_command_line_args;
239119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  struct CXUnsavedFile *unsaved_files = PTUI->unsaved_files;
239219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned num_unsaved_files = PTUI->num_unsaved_files;
239319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned options = PTUI->options;
239419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  PTUI->result = 0;
23955a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor
23962b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CIdx)
239719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    return;
2398f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2399e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
2400e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff
240144c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  bool PrecompilePreamble = options & CXTranslationUnit_PrecompiledPreamble;
2402df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor  bool CompleteTranslationUnit
2403df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor    = ((options & CXTranslationUnit_Incomplete) == 0);
240487c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  bool CacheCodeCompetionResults
240587c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    = options & CXTranslationUnit_CacheCompletionResults;
240699ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor  bool CXXPrecompilePreamble
240799ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor    = options & CXTranslationUnit_CXXPrecompiledPreamble;
240899ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor  bool CXXChainedPCH
240999ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor    = options & CXTranslationUnit_CXXChainedPCH;
241087c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor
24115352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  // Configure the diagnostics.
24125352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  DiagnosticOptions DiagOpts;
241325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::IntrusiveRefCntPtr<Diagnostic>
241425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Diags(CompilerInstance::createDiagnostics(DiagOpts, num_command_line_args,
241525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                                command_line_args));
241625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
241725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
241825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<Diagnostic,
241925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    llvm::CrashRecoveryContextReleaseRefCleanup<Diagnostic> >
242025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    DiagCleanup(Diags.getPtr());
242125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
242225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::OwningPtr<std::vector<ASTUnit::RemappedFile> >
242325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles(new std::vector<ASTUnit::RemappedFile>());
242425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
242525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
242625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<
242725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
2428f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
24294db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  for (unsigned I = 0; I != num_unsaved_files; ++I) {
2430a0a270c0f1c0a4e3482438bdc5f4a7bd3d25f0a6Chris Lattner    llvm::StringRef Data(unsaved_files[I].Contents, unsaved_files[I].Length);
2431f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    const llvm::MemoryBuffer *Buffer
2432a0a270c0f1c0a4e3482438bdc5f4a7bd3d25f0a6Chris Lattner      = llvm::MemoryBuffer::getMemBufferCopy(Data, unsaved_files[I].Filename);
243325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles->push_back(std::make_pair(unsaved_files[I].Filename,
243425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                            Buffer));
24354db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  }
2436f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
243725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::OwningPtr<std::vector<const char *> >
243825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args(new std::vector<const char*>());
243925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
244025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this method.
244125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<std::vector<const char*> >
244225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    ArgsCleanup(Args.get());
244325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
244452ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // Since the Clang C library is primarily used by batch tools dealing with
244552ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // (often very broken) source code, where spell-checking can have a
244652ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // significant negative impact on performance (particularly when
244752ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // precompiled headers are involved), we disable it by default.
2448b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  // Only do this if we haven't found a spell-checking-related argument.
2449b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  bool FoundSpellCheckingArgument = false;
2450b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  for (int I = 0; I != num_command_line_args; ++I) {
2451b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    if (strcmp(command_line_args[I], "-fno-spell-checking") == 0 ||
2452b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        strcmp(command_line_args[I], "-fspell-checking") == 0) {
2453b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      FoundSpellCheckingArgument = true;
2454b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      break;
2455e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff    }
2456b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  }
2457b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  if (!FoundSpellCheckingArgument)
245825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-fno-spell-checking");
2459b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor
246025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  Args->insert(Args->end(), command_line_args,
246125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek               command_line_args + num_command_line_args);
2462d93256e55673a17d18543397ec462416acb13792Douglas Gregor
2463c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // The 'source_filename' argument is optional.  If the caller does not
2464c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // specify it then it is assumed that the source file is specified
2465c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // in the actual argument list.
2466c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // Put the source file after command_line_args otherwise if '-x' flag is
2467c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // present it will be unused.
2468c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  if (source_filename)
246925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back(source_filename);
2470c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis
247144c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  // Do we need the detailed preprocessing record?
247244c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  if (options & CXTranslationUnit_DetailedPreprocessingRecord) {
247325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-Xclang");
247425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-detailed-preprocessing-record");
247544c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  }
247644c181aec37789f25f6c15543c164416f72e562aDouglas Gregor
2477026f6911bb985c800a54446de9f6da8745ae025aArgyrios Kyrtzidis  unsigned NumErrors = Diags->getClient()->getNumErrors();
2478b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  llvm::OwningPtr<ASTUnit> Unit(
24794ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek    ASTUnit::LoadFromCommandLine(Args->size() ? &(*Args)[0] : 0
24804ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 /* vector::data() not portable */,
24814ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 Args->size() ? (&(*Args)[0] + Args->size()) :0,
2482b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 Diags,
2483b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CXXIdx->getClangResourcesPath(),
2484b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CXXIdx->getOnlyLocalDecls(),
2485e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor                                 /*CaptureDiagnostics=*/true,
24864ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 RemappedFiles->size() ? &(*RemappedFiles)[0]:0,
248725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                 RemappedFiles->size(),
2488299a4a967b02c9f0d0d94ad8560e3ced893f9116Argyrios Kyrtzidis                                 /*RemappedFilesKeepOriginalName=*/true,
2489b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 PrecompilePreamble,
2490b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CompleteTranslationUnit,
249199ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor                                 CacheCodeCompetionResults,
249299ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor                                 CXXPrecompilePreamble,
249399ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor                                 CXXChainedPCH));
2494b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor
2495026f6911bb985c800a54446de9f6da8745ae025aArgyrios Kyrtzidis  if (NumErrors != Diags->getClient()->getNumErrors()) {
2496b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    // Make sure to check that 'Unit' is non-NULL.
2497b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    if (CXXIdx->getDisplayDiagnostics() && Unit.get()) {
2498b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      for (ASTUnit::stored_diag_iterator D = Unit->stored_diag_begin(),
2499b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                      DEnd = Unit->stored_diag_end();
2500b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor           D != DEnd; ++D) {
2501b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        CXStoredDiagnostic Diag(*D, Unit->getASTContext().getLangOptions());
2502b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        CXString Msg = clang_formatDiagnostic(&Diag,
2503b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                    clang_defaultDiagnosticDisplayOptions());
2504b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        fprintf(stderr, "%s\n", clang_getCString(Msg));
2505b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        clang_disposeString(Msg);
2506b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      }
2507274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor#ifdef LLVM_ON_WIN32
2508b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      // On Windows, force a flush, since there may be multiple copies of
2509b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      // stderr and stdout in the file system, all with different buffers
2510b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      // but writing to the same device.
2511b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      fflush(stderr);
2512b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor#endif
2513b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    }
2514a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  }
2515d93256e55673a17d18543397ec462416acb13792Douglas Gregor
2516a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  PTUI->result = MakeCXTranslationUnit(Unit.take());
251719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar}
251819ffd492a31a25fb691098bf79f317e5f3edf177Daniel DunbarCXTranslationUnit clang_parseTranslationUnit(CXIndex CIdx,
251919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             const char *source_filename,
25202ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor                                         const char * const *command_line_args,
252119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             int num_command_line_args,
25229e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                            struct CXUnsavedFile *unsaved_files,
252319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             unsigned num_unsaved_files,
252419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             unsigned options) {
252519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  ParseTranslationUnitInfo PTUI = { CIdx, source_filename, command_line_args,
25269e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                    num_command_line_args, unsaved_files,
25279e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                    num_unsaved_files, options, 0 };
252819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  llvm::CrashRecoveryContext CRC;
252919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
2530bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  if (!RunSafely(CRC, clang_parseTranslationUnit_Impl, &PTUI)) {
253160a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "libclang: crash detected during parsing: {\n");
253260a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'source_filename' : '%s'\n", source_filename);
253360a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'command_line_args' : [");
253460a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    for (int i = 0; i != num_command_line_args; ++i) {
253560a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      if (i)
253660a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar        fprintf(stderr, ", ");
253760a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      fprintf(stderr, "'%s'", command_line_args[i]);
253860a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    }
253960a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "],\n");
254060a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'unsaved_files' : [");
254160a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    for (unsigned i = 0; i != num_unsaved_files; ++i) {
254260a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      if (i)
254360a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar        fprintf(stderr, ", ");
254460a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      fprintf(stderr, "('%s', '...', %ld)", unsaved_files[i].Filename,
254560a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar              unsaved_files[i].Length);
254660a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    }
254760a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "],\n");
254860a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'options' : %d,\n", options);
254960a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "}\n");
255060a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar
255119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    return 0;
255219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  }
255319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
255419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  return PTUI.result;
25555b7d8e254f6c2855b37b5521c0aee0a560dab237Steve Naroff}
25565b7d8e254f6c2855b37b5521c0aee0a560dab237Steve Naroff
25571999844e7a18786e61e619e1dc6c789827541863Douglas Gregorunsigned clang_defaultSaveOptions(CXTranslationUnit TU) {
25581999844e7a18786e61e619e1dc6c789827541863Douglas Gregor  return CXSaveTranslationUnit_None;
25591999844e7a18786e61e619e1dc6c789827541863Douglas Gregor}
25601999844e7a18786e61e619e1dc6c789827541863Douglas Gregor
25611999844e7a18786e61e619e1dc6c789827541863Douglas Gregorint clang_saveTranslationUnit(CXTranslationUnit TU, const char *FileName,
25621999844e7a18786e61e619e1dc6c789827541863Douglas Gregor                              unsigned options) {
25637ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor  if (!TU)
25647ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor    return 1;
25657ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor
2566a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return static_cast<ASTUnit *>(TU->TUData)->Save(FileName);
25677ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor}
256819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
25699ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarvoid clang_disposeTranslationUnit(CXTranslationUnit CTUnit) {
2570ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  if (CTUnit) {
2571ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    // If the translation unit has been marked as unsafe to free, just discard
2572ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    // it.
2573a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    if (static_cast<ASTUnit *>(CTUnit->TUData)->isUnsafeToFree())
2574ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar      return;
2575ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2576a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    delete static_cast<ASTUnit *>(CTUnit->TUData);
2577a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    disposeCXStringPool(CTUnit->StringPool);
2578a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    delete CTUnit;
2579ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  }
25802bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff}
25810d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
2582e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregorunsigned clang_defaultReparseOptions(CXTranslationUnit TU) {
2583e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor  return CXReparse_None;
2584e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor}
2585e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor
2586ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbarstruct ReparseTranslationUnitInfo {
2587ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  CXTranslationUnit TU;
2588ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned num_unsaved_files;
2589ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  struct CXUnsavedFile *unsaved_files;
2590ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned options;
2591ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  int result;
2592ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar};
2593593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2594b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbarstatic void clang_reparseTranslationUnit_Impl(void *UserData) {
2595ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  ReparseTranslationUnitInfo *RTUI =
2596ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    static_cast<ReparseTranslationUnitInfo*>(UserData);
2597ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  CXTranslationUnit TU = RTUI->TU;
2598ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned num_unsaved_files = RTUI->num_unsaved_files;
2599ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  struct CXUnsavedFile *unsaved_files = RTUI->unsaved_files;
2600ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned options = RTUI->options;
2601ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  (void) options;
2602ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  RTUI->result = 1;
2603ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2604abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  if (!TU)
2605ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    return;
2606593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2607a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
2608593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
2609abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor
261025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::OwningPtr<std::vector<ASTUnit::RemappedFile> >
261125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles(new std::vector<ASTUnit::RemappedFile>());
261225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
261325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
261425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<
261525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
261625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
2617abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  for (unsigned I = 0; I != num_unsaved_files; ++I) {
2618abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor    llvm::StringRef Data(unsaved_files[I].Contents, unsaved_files[I].Length);
2619abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor    const llvm::MemoryBuffer *Buffer
26201abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor      = llvm::MemoryBuffer::getMemBufferCopy(Data, unsaved_files[I].Filename);
262125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles->push_back(std::make_pair(unsaved_files[I].Filename,
262225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                            Buffer));
2623abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  }
2624abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor
26254ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek  if (!CXXUnit->Reparse(RemappedFiles->size() ? &(*RemappedFiles)[0] : 0,
26264ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                        RemappedFiles->size()))
2627593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor    RTUI->result = 0;
2628abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor}
2629593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2630ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbarint clang_reparseTranslationUnit(CXTranslationUnit TU,
2631ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 unsigned num_unsaved_files,
2632ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 struct CXUnsavedFile *unsaved_files,
2633ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 unsigned options) {
2634ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  ReparseTranslationUnitInfo RTUI = { TU, num_unsaved_files, unsaved_files,
2635ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                      options, 0 };
2636ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  llvm::CrashRecoveryContext CRC;
2637ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2638bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  if (!RunSafely(CRC, clang_reparseTranslationUnit_Impl, &RTUI)) {
2639b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbar    fprintf(stderr, "libclang: crash detected during reparsing\n");
2640a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    static_cast<ASTUnit *>(TU->TUData)->setUnsafeToFree(true);
2641ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    return 1;
2642ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  }
2643ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
26441dfb26af4d6aa4f7818e256659a79f1ec2cba784Ted Kremenek
2645ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  return RTUI.result;
2646ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar}
2647ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2648df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor
26499ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXString clang_getTranslationUnitSpelling(CXTranslationUnit CTUnit) {
26502b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CTUnit)
2651ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
2652f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2653a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit->TUData);
2654ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString(CXXUnit->getOriginalSourceFileName(), true);
2655af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff}
26561eb79b58e56b99cf557d5d353586a10c5360364dDaniel Dunbar
26577eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas GregorCXCursor clang_getTranslationUnitCursor(CXTranslationUnit TU) {
2658b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  CXCursor Result = { CXCursor_TranslationUnit, { 0, 0, TU } };
26597eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  return Result;
26607eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor}
26617eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
2662fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
2663600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
2664fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
26651db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor// CXSourceLocation and CXSourceRange Operations.
26661db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor//===----------------------------------------------------------------------===//
26671db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
2668b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregorextern "C" {
2669b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXSourceLocation clang_getNullLocation() {
26705352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  CXSourceLocation Result = { { 0, 0 }, 0 };
2671b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  return Result;
2672b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2673b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
2674b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregorunsigned clang_equalLocations(CXSourceLocation loc1, CXSourceLocation loc2) {
267590a6b9e1e4d4d1995ada044e319d6e722b07a6b4Daniel Dunbar  return (loc1.ptr_data[0] == loc2.ptr_data[0] &&
267690a6b9e1e4d4d1995ada044e319d6e722b07a6b4Daniel Dunbar          loc1.ptr_data[1] == loc2.ptr_data[1] &&
267790a6b9e1e4d4d1995ada044e319d6e722b07a6b4Daniel Dunbar          loc1.int_data == loc2.int_data);
2678b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2679b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
2680b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXSourceLocation clang_getLocation(CXTranslationUnit tu,
2681b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor                                   CXFile file,
2682b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor                                   unsigned line,
2683b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor                                   unsigned column) {
268442748ec5cb2d75fe0dbb3a6db5aee6c11b5dc190Douglas Gregor  if (!tu || !file)
2685b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor    return clang_getNullLocation();
268642748ec5cb2d75fe0dbb3a6db5aee6c11b5dc190Douglas Gregor
268786a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  bool Logging = ::getenv("LIBCLANG_LOGGING");
2688a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
268986a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  const FileEntry *File = static_cast<const FileEntry *>(file);
2690b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  SourceLocation SLoc
269186a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor    = CXXUnit->getSourceManager().getLocation(File, line, column);
269286a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  if (SLoc.isInvalid()) {
269386a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor    if (Logging)
269486a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor      llvm::errs() << "clang_getLocation(\"" << File->getName()
269586a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor                   << "\", " << line << ", " << column << ") = invalid\n";
269686a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor    return clang_getNullLocation();
269786a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  }
269886a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor
269986a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  if (Logging)
270086a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor    llvm::errs() << "clang_getLocation(\"" << File->getName()
270186a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor                 << "\", " << line << ", " << column << ") = "
270286a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor                 << SLoc.getRawEncoding() << "\n";
270383889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall
270483889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  return cxloc::translateSourceLocation(CXXUnit->getASTContext(), SLoc);
270583889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall}
270683889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall
270783889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid ChisnallCXSourceLocation clang_getLocationForOffset(CXTranslationUnit tu,
270883889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall                                            CXFile file,
270983889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall                                            unsigned offset) {
271083889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  if (!tu || !file)
271183889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall    return clang_getNullLocation();
271283889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall
2713a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
271483889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  SourceLocation Start
271583889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall    = CXXUnit->getSourceManager().getLocation(
271683889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall                                        static_cast<const FileEntry *>(file),
271783889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall                                              1, 1);
271883889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  if (Start.isInvalid()) return clang_getNullLocation();
271983889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall
272083889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  SourceLocation SLoc = Start.getFileLocWithOffset(offset);
272183889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall
272283889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  if (SLoc.isInvalid()) return clang_getNullLocation();
2723f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
27241a9a0bc472ee4fec72ee8be8b575fb66ca600d1bTed Kremenek  return cxloc::translateSourceLocation(CXXUnit->getASTContext(), SLoc);
2725b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2726b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
27275352ac06d8f6194825bb2a99ffa009b61bafb503Douglas GregorCXSourceRange clang_getNullRange() {
27285352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  CXSourceRange Result = { { 0, 0 }, 0, 0 };
27295352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  return Result;
27305352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor}
2731d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar
2732b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXSourceRange clang_getRange(CXSourceLocation begin, CXSourceLocation end) {
27335352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  if (begin.ptr_data[0] != end.ptr_data[0] ||
27345352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor      begin.ptr_data[1] != end.ptr_data[1])
27355352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
2736f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2737f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  CXSourceRange Result = { { begin.ptr_data[0], begin.ptr_data[1] },
27385352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor                           begin.int_data, end.int_data };
2739b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  return Result;
2740b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
27419d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek} // end: extern "C"
2742b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
27439d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenekstatic void createNullLocation(CXFile *file, unsigned *line,
27449d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek                               unsigned *column, unsigned *offset) {
27459d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  if (file)
27469d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek   *file = 0;
27479d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  if (line)
27489d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek   *line = 0;
27499d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  if (column)
27509d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek   *column = 0;
27519d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  if (offset)
27529d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek   *offset = 0;
27539d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  return;
27549d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek}
27559d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek
27569d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenekextern "C" {
275746766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregorvoid clang_getInstantiationLocation(CXSourceLocation location,
275846766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor                                    CXFile *file,
275946766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor                                    unsigned *line,
276046766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor                                    unsigned *column,
276146766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor                                    unsigned *offset) {
27621db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  SourceLocation Loc = SourceLocation::getFromRawEncoding(location.int_data);
27631db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
2764bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar  if (!location.ptr_data[0] || Loc.isInvalid()) {
27659d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek    createNullLocation(file, line, column, offset);
276646766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor    return;
276746766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor  }
276846766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor
2769bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar  const SourceManager &SM =
2770bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar    *static_cast<const SourceManager*>(location.ptr_data[0]);
27711db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  SourceLocation InstLoc = SM.getInstantiationLoc(Loc);
27721db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
27739d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  // Check that the FileID is invalid on the instantiation location.
27749d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  // This can manifest in invalid code.
27759d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  FileID fileID = SM.getFileID(InstLoc);
27769d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  const SrcMgr::SLocEntry &sloc = SM.getSLocEntry(fileID);
27779d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  if (!sloc.isFile()) {
27789d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek    createNullLocation(file, line, column, offset);
27799d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek    return;
27809d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  }
27819d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek
27821db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  if (file)
27839d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek    *file = (void *)SM.getFileEntryForSLocEntry(sloc);
27841db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  if (line)
27851db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor    *line = SM.getInstantiationLineNumber(InstLoc);
27861db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  if (column)
27871db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor    *column = SM.getInstantiationColumnNumber(InstLoc);
2788e69517ce61638f12c9abe4605753a45275ac4e37Douglas Gregor  if (offset)
278946766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor    *offset = SM.getDecomposedLoc(InstLoc).second;
2790e69517ce61638f12c9abe4605753a45275ac4e37Douglas Gregor}
2791e69517ce61638f12c9abe4605753a45275ac4e37Douglas Gregor
2792a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregorvoid clang_getSpellingLocation(CXSourceLocation location,
2793a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor                               CXFile *file,
2794a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor                               unsigned *line,
2795a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor                               unsigned *column,
2796a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor                               unsigned *offset) {
2797a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  SourceLocation Loc = SourceLocation::getFromRawEncoding(location.int_data);
2798a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
2799a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (!location.ptr_data[0] || Loc.isInvalid()) {
2800a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    if (file)
2801a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      *file = 0;
2802a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    if (line)
2803a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      *line = 0;
2804a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    if (column)
2805a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      *column = 0;
2806a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    if (offset)
2807a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      *offset = 0;
2808a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    return;
2809a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  }
2810a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
2811a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  const SourceManager &SM =
2812a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *static_cast<const SourceManager*>(location.ptr_data[0]);
2813a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  SourceLocation SpellLoc = Loc;
2814a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (SpellLoc.isMacroID()) {
2815a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    SourceLocation SimpleSpellingLoc = SM.getImmediateSpellingLoc(SpellLoc);
2816a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    if (SimpleSpellingLoc.isFileID() &&
2817a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor        SM.getFileEntryForID(SM.getDecomposedLoc(SimpleSpellingLoc).first))
2818a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      SpellLoc = SimpleSpellingLoc;
2819a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    else
2820a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      SpellLoc = SM.getInstantiationLoc(SpellLoc);
2821a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  }
2822a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
2823a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  std::pair<FileID, unsigned> LocInfo = SM.getDecomposedLoc(SpellLoc);
2824a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  FileID FID = LocInfo.first;
2825a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  unsigned FileOffset = LocInfo.second;
2826a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
2827a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (file)
2828a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *file = (void *)SM.getFileEntryForID(FID);
2829a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (line)
2830a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *line = SM.getLineNumber(FID, FileOffset);
2831a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (column)
2832a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *column = SM.getColumnNumber(FID, FileOffset);
2833a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (offset)
2834a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *offset = FileOffset;
2835a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor}
2836a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
28371db19dea8d221f27be46332d668d1e2decb7f1abDouglas GregorCXSourceLocation clang_getRangeStart(CXSourceRange range) {
2838f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  CXSourceLocation Result = { { range.ptr_data[0], range.ptr_data[1] },
28395352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor                              range.begin_int_data };
28401db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  return Result;
28411db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor}
28421db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
28431db19dea8d221f27be46332d668d1e2decb7f1abDouglas GregorCXSourceLocation clang_getRangeEnd(CXSourceRange range) {
2844bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar  CXSourceLocation Result = { { range.ptr_data[0], range.ptr_data[1] },
28455352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor                              range.end_int_data };
28461db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  return Result;
28471db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor}
28481db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
2849b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor} // end: extern "C"
2850b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
28511db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor//===----------------------------------------------------------------------===//
2852fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXFile Operations.
2853fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2854fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2855fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
285674844072411bae91d5dbb89955d200cbe1e0a1c8Ted KremenekCXString clang_getFileName(CXFile SFile) {
285798258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (!SFile)
2858a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return createCXString((const char*)NULL);
2859f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
286088145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  FileEntry *FEnt = static_cast<FileEntry *>(SFile);
286174844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek  return createCXString(FEnt->getName());
286288145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff}
286388145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff
286488145034694ed5267fa6fa5febc54fadc02bd479Steve Narofftime_t clang_getFileTime(CXFile SFile) {
286598258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (!SFile)
286698258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor    return 0;
2867f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
286888145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  FileEntry *FEnt = static_cast<FileEntry *>(SFile);
286988145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  return FEnt->getModificationTime();
2870ee9405e807d7c447c0143c2bd865b759192e97b3Steve Naroff}
2871f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2872b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXFile clang_getFile(CXTranslationUnit tu, const char *file_name) {
2873b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  if (!tu)
2874b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor    return 0;
2875f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2876a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
2877f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2878b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  FileManager &FMgr = CXXUnit->getFileManager();
287939b49bcaaddb1049234fca9500c0ac02c088e23dChris Lattner  return const_cast<FileEntry *>(FMgr.getFile(file_name));
2880b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2881f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2882fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
2883fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2884fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2885fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXCursor Operations.
2886fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2887fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2888fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekstatic Decl *getDeclFromExpr(Stmt *E) {
2889db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor  if (CastExpr *CE = dyn_cast<CastExpr>(E))
2890db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor    return getDeclFromExpr(CE->getSubExpr());
2891db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2892fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (DeclRefExpr *RefExpr = dyn_cast<DeclRefExpr>(E))
2893fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return RefExpr->getDecl();
289438f28c1189142429384996409fffbc57f67b5c60Douglas Gregor  if (BlockDeclRefExpr *RefExpr = dyn_cast<BlockDeclRefExpr>(E))
289538f28c1189142429384996409fffbc57f67b5c60Douglas Gregor    return RefExpr->getDecl();
2896fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (MemberExpr *ME = dyn_cast<MemberExpr>(E))
2897fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return ME->getMemberDecl();
2898fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (ObjCIvarRefExpr *RE = dyn_cast<ObjCIvarRefExpr>(E))
2899fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return RE->getDecl();
2900db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor  if (ObjCPropertyRefExpr *PRE = dyn_cast<ObjCPropertyRefExpr>(E))
290112f78a6741a4cb3d904340f8d3d2714568b50e7aJohn McCall    return PRE->isExplicitProperty() ? PRE->getExplicitProperty() : 0;
2902db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2903fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (CallExpr *CE = dyn_cast<CallExpr>(E))
2904fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return getDeclFromExpr(CE->getCallee());
290593798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  if (CXXConstructExpr *CE = llvm::dyn_cast<CXXConstructExpr>(E))
290693798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    if (!CE->isElidable())
290793798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    return CE->getConstructor();
2908fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (ObjCMessageExpr *OME = dyn_cast<ObjCMessageExpr>(E))
2909fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return OME->getMethodDecl();
2910f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2911db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor  if (ObjCProtocolExpr *PE = dyn_cast<ObjCProtocolExpr>(E))
2912db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor    return PE->getProtocol();
2913c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor  if (SubstNonTypeTemplateParmPackExpr *NTTP
2914c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor                              = dyn_cast<SubstNonTypeTemplateParmPackExpr>(E))
2915c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor    return NTTP->getParameterPack();
291694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  if (SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
291794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor    if (isa<NonTypeTemplateParmDecl>(SizeOfPack->getPack()) ||
291894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        isa<ParmVarDecl>(SizeOfPack->getPack()))
291994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      return SizeOfPack->getPack();
2920db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2921fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  return 0;
2922fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek}
2923ee9405e807d7c447c0143c2bd865b759192e97b3Steve Naroff
2924c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbarstatic SourceLocation getLocationFromExpr(Expr *E) {
2925c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (ObjCMessageExpr *Msg = dyn_cast<ObjCMessageExpr>(E))
2926c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return /*FIXME:*/Msg->getLeftLoc();
2927c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
2928c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return DRE->getLocation();
292938f28c1189142429384996409fffbc57f67b5c60Douglas Gregor  if (BlockDeclRefExpr *RefExpr = dyn_cast<BlockDeclRefExpr>(E))
293038f28c1189142429384996409fffbc57f67b5c60Douglas Gregor    return RefExpr->getLocation();
2931c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (MemberExpr *Member = dyn_cast<MemberExpr>(E))
2932c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return Member->getMemberLoc();
2933c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (ObjCIvarRefExpr *Ivar = dyn_cast<ObjCIvarRefExpr>(E))
2934c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return Ivar->getLocation();
293594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  if (SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
293694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor    return SizeOfPack->getPackLoc();
293794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
2938c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  return E->getLocStart();
2939c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar}
2940c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar
2941fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
2942f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2943f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekunsigned clang_visitChildren(CXCursor parent,
2944b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor                             CXCursorVisitor visitor,
2945b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor                             CXClientData client_data) {
2946a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CursorVisitor CursorVis(getCursorTU(parent), visitor, client_data,
294704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                          getCursorASTUnit(parent)->getMaxPCHLevel(),
294804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                          false);
2949b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return CursorVis.VisitChildren(parent);
2950b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
2951b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor
29523387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#ifndef __has_feature
29533387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#define __has_feature(x) 0
29543387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#endif
29553387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#if __has_feature(blocks)
29563387c65a094a02b2a94c05111d035a97d3d5c794David Chisnalltypedef enum CXChildVisitResult
29573387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall     (^CXCursorVisitorBlock)(CXCursor cursor, CXCursor parent);
29583387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
29593387c65a094a02b2a94c05111d035a97d3d5c794David Chisnallstatic enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
29603387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall    CXClientData client_data) {
29613387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
29623387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return block(cursor, parent);
29633387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
29643387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#else
29653387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall// If we are compiled with a compiler that doesn't have native blocks support,
29663387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall// define and call the block manually, so the
29673387c65a094a02b2a94c05111d035a97d3d5c794David Chisnalltypedef struct _CXChildVisitResult
29683387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall{
29693387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	void *isa;
29703387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	int flags;
29713387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	int reserved;
29729e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar	enum CXChildVisitResult(*invoke)(struct _CXChildVisitResult*, CXCursor,
29739e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                         CXCursor);
29743387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall} *CXCursorVisitorBlock;
29753387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
29763387c65a094a02b2a94c05111d035a97d3d5c794David Chisnallstatic enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
29773387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall    CXClientData client_data) {
29783387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
29793387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return block->invoke(block, cursor, parent);
29803387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
29813387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#endif
29823387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
29833387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
29849e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbarunsigned clang_visitChildrenWithBlock(CXCursor parent,
29859e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                      CXCursorVisitorBlock block) {
29863387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return clang_visitChildren(parent, visitWithBlock, block);
29873387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
29883387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
298978205d4bada39d95097e766af9eb30cdd0159461Douglas Gregorstatic CXString getDeclSpelling(Decl *D) {
299078205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D);
2991e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor  if (!ND) {
2992e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor    if (ObjCPropertyImplDecl *PropImpl =llvm::dyn_cast<ObjCPropertyImplDecl>(D))
2993e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor      if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
2994e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor        return createCXString(Property->getIdentifier()->getName());
2995e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
2996ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
2997e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor  }
2998e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
299978205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
3000ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(OMD->getSelector().getAsString());
3001f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
300278205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ObjCCategoryImplDecl *CIMP = dyn_cast<ObjCCategoryImplDecl>(ND))
300378205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // No, this isn't the same as the code below. getIdentifier() is non-virtual
300478205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // and returns different names. NamedDecl returns the class name and
300578205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // ObjCCategoryImplDecl returns the category name.
3006ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(CIMP->getIdentifier()->getNameStart());
3007f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
30080a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  if (isa<UsingDirectiveDecl>(D))
30090a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor    return createCXString("");
30100a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
301150aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  llvm::SmallString<1024> S;
301250aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  llvm::raw_svector_ostream os(S);
301350aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  ND->printName(os);
301450aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek
301550aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  return createCXString(os.str());
301678205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor}
3017f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
30189ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXString clang_getCursorSpelling(CXCursor C) {
30197eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  if (clang_isTranslationUnit(C.kind))
3020a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return clang_getTranslationUnitSpelling(
3021a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                            static_cast<CXTranslationUnit>(C.data[2]));
30227eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
3023f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  if (clang_isReference(C.kind)) {
3024f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff    switch (C.kind) {
3025acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCSuperClassRef: {
30262e331b938b38057e333fab0ba841130ea8467794Douglas Gregor      ObjCInterfaceDecl *Super = getCursorObjCSuperClassRef(C).first;
3027ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(Super->getIdentifier()->getNameStart());
3028acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
3029acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCClassRef: {
30301adb082a709f7b588f03672999294e061234b2cfDouglas Gregor      ObjCInterfaceDecl *Class = getCursorObjCClassRef(C).first;
3031ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(Class->getIdentifier()->getNameStart());
3032acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
3033acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCProtocolRef: {
303478db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor      ObjCProtocolDecl *OID = getCursorObjCProtocolRef(C).first;
3035f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      assert(OID && "getCursorSpelling(): Missing protocol decl");
3036ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(OID->getIdentifier()->getNameStart());
3037acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
30383064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
30393064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      CXXBaseSpecifier *B = getCursorCXXBaseSpecifier(C);
30403064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      return createCXString(B->getType().getAsString());
30413064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
30427d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    case CXCursor_TypeRef: {
30437d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      TypeDecl *Type = getCursorTypeRef(C).first;
30447d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      assert(Type && "Missing type decl");
30457d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3046ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(getCursorContext(C).getTypeDeclType(Type).
3047ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek                              getAsString());
30487d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    }
30490b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef: {
30500b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      TemplateDecl *Template = getCursorTemplateRef(C).first;
30516931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      assert(Template && "Missing template decl");
30520b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
30530b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return createCXString(Template->getNameAsString());
30540b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    }
30556931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
30566931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef: {
30576931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      NamedDecl *NS = getCursorNamespaceRef(C).first;
30586931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      assert(NS && "Missing namespace decl");
30596931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
30606931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return createCXString(NS->getNameAsString());
30616931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    }
30627d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3063a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef: {
3064a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      FieldDecl *Field = getCursorMemberRef(C).first;
3065a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      assert(Field && "Missing member decl");
3066a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
3067a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return createCXString(Field->getNameAsString());
3068a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
3069a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
307036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef: {
307136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      LabelStmt *Label = getCursorLabelRef(C).first;
307236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      assert(Label && "Missing label");
307336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
3074ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner      return createCXString(Label->getName());
307536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    }
307636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
30771f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef: {
30781f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
30791f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (Decl *D = Storage.dyn_cast<Decl *>()) {
30801f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        if (NamedDecl *ND = dyn_cast<NamedDecl>(D))
30811f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor          return createCXString(ND->getNameAsString());
30821f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString("");
30831f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      }
30841f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
30851f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString(E->getName().getAsString());
30861f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      OverloadedTemplateStorage *Ovl
30871f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        = Storage.get<OverloadedTemplateStorage*>();
30881f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (Ovl->size() == 0)
30891f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString("");
30901f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return createCXString((*Ovl->begin())->getNameAsString());
30911f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    }
30921f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3093acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    default:
3094ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString("<not implemented>");
3095f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff    }
3096f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  }
309797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
309897b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind)) {
309997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    Decl *D = getDeclFromExpr(getCursorExpr(C));
310097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    if (D)
310178205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor      return getDeclSpelling(D);
3102ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
310397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  }
310497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
310536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind)) {
310636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    Stmt *S = getCursorStmt(C);
310736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    if (LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S))
3108ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner      return createCXString(Label->getName());
310936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
311036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return createCXString("");
311136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  }
311236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
31134ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor  if (C.kind == CXCursor_MacroInstantiation)
31144ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor    return createCXString(getCursorMacroInstantiation(C)->getName()
31154ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor                                                           ->getNameStart());
31164ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor
3117572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition)
3118572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return createCXString(getCursorMacroDefinition(C)->getName()
3119572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor                                                           ->getNameStart());
3120572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3121ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective)
3122ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return createCXString(getCursorInclusionDirective(C)->getFileName());
3123ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
312460cbfacd947590f83257a4191566dda92fbbde69Douglas Gregor  if (clang_isDeclaration(C.kind))
312560cbfacd947590f83257a4191566dda92fbbde69Douglas Gregor    return getDeclSpelling(getCursorDecl(C));
3126e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3127ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString("");
3128f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff}
3129f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff
3130358559d8d7b458c5f64941842383a16e61f0828dDouglas GregorCXString clang_getCursorDisplayName(CXCursor C) {
3131358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (!clang_isDeclaration(C.kind))
3132358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return clang_getCursorSpelling(C);
3133358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3134358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  Decl *D = getCursorDecl(C);
3135358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (!D)
3136358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString("");
3137358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3138358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  PrintingPolicy &Policy = getCursorContext(C).PrintingPolicy;
3139358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(D))
3140358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    D = FunTmpl->getTemplatedDecl();
3141358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3142358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (FunctionDecl *Function = dyn_cast<FunctionDecl>(D)) {
3143358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::SmallString<64> Str;
3144358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3145358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << Function->getNameAsString();
3146358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (Function->getPrimaryTemplate())
3147358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << "<>";
3148358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << "(";
3149358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    for (unsigned I = 0, N = Function->getNumParams(); I != N; ++I) {
3150358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (I)
3151358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3152358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << Function->getParamDecl(I)->getType().getAsString(Policy);
3153358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3154358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3155358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (Function->isVariadic()) {
3156358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (Function->getNumParams())
3157358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3158358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << "...";
3159358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3160358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ")";
3161358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3162358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3163358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3164358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D)) {
3165358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::SmallString<64> Str;
3166358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3167358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ClassTemplate->getNameAsString();
3168358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << "<";
3169358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    TemplateParameterList *Params = ClassTemplate->getTemplateParameters();
3170358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    for (unsigned I = 0, N = Params->size(); I != N; ++I) {
3171358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (I)
3172358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3173358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3174358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      NamedDecl *Param = Params->getParam(I);
3175358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (Param->getIdentifier()) {
3176358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << Param->getIdentifier()->getName();
3177358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        continue;
3178358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      }
3179358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3180358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      // There is no parameter name, which makes this tricky. Try to come up
3181358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      // with something useful that isn't too long.
3182358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
3183358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << (TTP->wasDeclaredWithTypename()? "typename" : "class");
3184358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      else if (NonTypeTemplateParmDecl *NTTP
3185358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                                    = dyn_cast<NonTypeTemplateParmDecl>(Param))
3186358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << NTTP->getType().getAsString(Policy);
3187358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      else
3188358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << "template<...> class";
3189358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3190358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3191358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ">";
3192358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3193358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3194358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3195358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (ClassTemplateSpecializationDecl *ClassSpec
3196358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                              = dyn_cast<ClassTemplateSpecializationDecl>(D)) {
3197358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    // If the type was explicitly written, use that.
3198358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (TypeSourceInfo *TSInfo = ClassSpec->getTypeAsWritten())
3199358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      return createCXString(TSInfo->getType().getAsString(Policy));
3200358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3201358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::SmallString<64> Str;
3202358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3203358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ClassSpec->getNameAsString();
3204358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << TemplateSpecializationType::PrintTemplateArgumentList(
3205910f8008fea79120489a53593fe971b0b8a4a740Douglas Gregor                                      ClassSpec->getTemplateArgs().data(),
3206910f8008fea79120489a53593fe971b0b8a4a740Douglas Gregor                                      ClassSpec->getTemplateArgs().size(),
3207358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                                                                Policy);
3208358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3209358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3210358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3211358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  return clang_getCursorSpelling(C);
3212358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor}
3213358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3214e68fff6fc083c6270d835216a3de0b82c6ef0310Ted KremenekCXString clang_getCursorKindSpelling(enum CXCursorKind Kind) {
321589922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  switch (Kind) {
3216e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_FunctionDecl:
3217e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("FunctionDecl");
3218e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TypedefDecl:
3219e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TypedefDecl");
3220e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_EnumDecl:
3221e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("EnumDecl");
3222e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_EnumConstantDecl:
3223e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("EnumConstantDecl");
3224e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_StructDecl:
3225e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("StructDecl");
3226e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnionDecl:
3227e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnionDecl");
3228e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ClassDecl:
3229e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ClassDecl");
3230e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_FieldDecl:
3231e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("FieldDecl");
3232e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_VarDecl:
3233e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("VarDecl");
3234e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ParmDecl:
3235e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ParmDecl");
3236e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCInterfaceDecl:
3237e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCInterfaceDecl");
3238e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCCategoryDecl:
3239e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCCategoryDecl");
3240e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCProtocolDecl:
3241e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCProtocolDecl");
3242e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCPropertyDecl:
3243e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCPropertyDecl");
3244e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCIvarDecl:
3245e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCIvarDecl");
3246e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCInstanceMethodDecl:
3247e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCInstanceMethodDecl");
3248e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCClassMethodDecl:
3249e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCClassMethodDecl");
3250e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCImplementationDecl:
3251e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCImplementationDecl");
3252e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCCategoryImplDecl:
3253e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCCategoryImplDecl");
32548bd5a69999cfd06b6b5a58fdd04e4f802b2df5a4Ted Kremenek  case CXCursor_CXXMethod:
32558bd5a69999cfd06b6b5a58fdd04e4f802b2df5a4Ted Kremenek      return createCXString("CXXMethod");
3256e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedDecl:
3257e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedDecl");
3258e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCSuperClassRef:
3259e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCSuperClassRef");
3260e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCProtocolRef:
3261e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCProtocolRef");
3262e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCClassRef:
3263e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCClassRef");
3264e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TypeRef:
3265e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TypeRef");
32660b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case CXCursor_TemplateRef:
32670b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return createCXString("TemplateRef");
32686931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  case CXCursor_NamespaceRef:
32696931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    return createCXString("NamespaceRef");
3270a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  case CXCursor_MemberRef:
3271a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return createCXString("MemberRef");
327236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  case CXCursor_LabelRef:
327336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return createCXString("LabelRef");
32741f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case CXCursor_OverloadedDeclRef:
32751f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return createCXString("OverloadedDeclRef");
3276e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedExpr:
3277e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedExpr");
32781ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  case CXCursor_BlockExpr:
32791ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek      return createCXString("BlockExpr");
3280e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_DeclRefExpr:
3281e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("DeclRefExpr");
3282e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_MemberRefExpr:
3283e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("MemberRefExpr");
3284e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_CallExpr:
3285e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("CallExpr");
3286e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCMessageExpr:
3287e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCMessageExpr");
3288e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedStmt:
3289e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedStmt");
329036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  case CXCursor_LabelStmt:
329136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return createCXString("LabelStmt");
3292e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_InvalidFile:
3293e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("InvalidFile");
3294292db6401f040795db3ea4e00fc02622d6c3ba1dTed Kremenek  case CXCursor_InvalidCode:
3295292db6401f040795db3ea4e00fc02622d6c3ba1dTed Kremenek    return createCXString("InvalidCode");
3296e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_NoDeclFound:
3297e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("NoDeclFound");
3298e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_NotImplemented:
3299e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("NotImplemented");
3300e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TranslationUnit:
3301e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TranslationUnit");
3302e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek  case CXCursor_UnexposedAttr:
3303e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek      return createCXString("UnexposedAttr");
3304e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek  case CXCursor_IBActionAttr:
3305e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek      return createCXString("attribute(ibaction)");
33069f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  case CXCursor_IBOutletAttr:
33079f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor     return createCXString("attribute(iboutlet)");
3308857e918a8a40deb128840308a318bf623d68295fTed Kremenek  case CXCursor_IBOutletCollectionAttr:
3309857e918a8a40deb128840308a318bf623d68295fTed Kremenek      return createCXString("attribute(iboutletcollection)");
33109f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  case CXCursor_PreprocessingDirective:
33119f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    return createCXString("preprocessing directive");
3312572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  case CXCursor_MacroDefinition:
3313572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return createCXString("macro definition");
33144807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  case CXCursor_MacroInstantiation:
33154807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor    return createCXString("macro instantiation");
3316ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  case CXCursor_InclusionDirective:
3317ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return createCXString("inclusion directive");
33188f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  case CXCursor_Namespace:
33198f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek    return createCXString("Namespace");
3320a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  case CXCursor_LinkageSpec:
3321a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek    return createCXString("LinkageSpec");
33223064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  case CXCursor_CXXBaseSpecifier:
33233064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    return createCXString("C++ base class specifier");
332401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_Constructor:
332501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXConstructor");
332601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_Destructor:
332701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXDestructor");
332801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_ConversionFunction:
332901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXConversion");
3330fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_TemplateTypeParameter:
3331fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("TemplateTypeParameter");
3332fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_NonTypeTemplateParameter:
3333fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("NonTypeTemplateParameter");
3334fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_TemplateTemplateParameter:
3335fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("TemplateTemplateParameter");
3336fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_FunctionTemplate:
3337fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("FunctionTemplate");
333839d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  case CXCursor_ClassTemplate:
333939d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor    return createCXString("ClassTemplate");
334074dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  case CXCursor_ClassTemplatePartialSpecialization:
334174dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    return createCXString("ClassTemplatePartialSpecialization");
33426931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  case CXCursor_NamespaceAlias:
33436931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    return createCXString("NamespaceAlias");
33440a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  case CXCursor_UsingDirective:
33450a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor    return createCXString("UsingDirective");
33467e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  case CXCursor_UsingDeclaration:
33477e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor    return createCXString("UsingDeclaration");
3348162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  case CXCursor_TypeAliasDecl:
3349162e1c1b487352434552147967c3dd296ebee2f7Richard Smith      return createCXString("TypeAliasDecl");
335089922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  }
3351e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3352deb06bd3566e18f677e76bc435d478b033fe328bTed Kremenek  llvm_unreachable("Unhandled CXCursorKind");
3353a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return createCXString((const char*) 0);
3354600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
335589922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff
3356e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenekenum CXChildVisitResult GetCursorVisitor(CXCursor cursor,
3357e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek                                         CXCursor parent,
335833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor                                         CXClientData client_data) {
335933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  CXCursor *BestCursor = static_cast<CXCursor *>(client_data);
336093798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor
336193798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // If our current best cursor is the construction of a temporary object,
336293798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // don't replace that cursor with a type reference, because we want
336393798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // clang_getCursor() to point at the constructor.
336493798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  if (clang_isExpression(BestCursor->kind) &&
336593798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor      isa<CXXTemporaryObjectExpr>(getCursorExpr(*BestCursor)) &&
336693798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor      cursor.kind == CXCursor_TypeRef)
336793798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    return CXChildVisit_Recurse;
336893798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor
336985fe1560b061b5f93a52dbd07cddd6e808854710Douglas Gregor  // Don't override a preprocessing cursor with another preprocessing
337085fe1560b061b5f93a52dbd07cddd6e808854710Douglas Gregor  // cursor; we want the outermost preprocessing cursor.
337185fe1560b061b5f93a52dbd07cddd6e808854710Douglas Gregor  if (clang_isPreprocessing(cursor.kind) &&
337285fe1560b061b5f93a52dbd07cddd6e808854710Douglas Gregor      clang_isPreprocessing(BestCursor->kind))
337385fe1560b061b5f93a52dbd07cddd6e808854710Douglas Gregor    return CXChildVisit_Recurse;
337485fe1560b061b5f93a52dbd07cddd6e808854710Douglas Gregor
337533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  *BestCursor = cursor;
337633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  return CXChildVisit_Recurse;
337733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
3378e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3379b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXCursor clang_getCursor(CXTranslationUnit TU, CXSourceLocation Loc) {
3380b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  if (!TU)
3381f462989fe8d6f59ab2d7d0fe2b4b96292ce706eaTed Kremenek    return clang_getNullCursor();
3382e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3383a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
3384bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
3385bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor
3386a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  // Translate the given source location to make it point at the beginning of
3387a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  // the token under the cursor.
3388a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek  SourceLocation SLoc = cxloc::translateSourceLocation(Loc);
3389a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek
3390a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek  // Guard against an invalid SourceLocation, or we may assert in one
3391a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek  // of the following calls.
3392a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek  if (SLoc.isInvalid())
3393a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek    return clang_getNullCursor();
3394a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek
339540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  bool Logging = getenv("LIBCLANG_LOGGING");
3396a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  SLoc = Lexer::GetBeginningOfToken(SLoc, CXXUnit->getSourceManager(),
3397a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor                                    CXXUnit->getASTContext().getLangOptions());
3398a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
339933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  CXCursor Result = MakeCXCursorInvalid(CXCursor_NoDeclFound);
340033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (SLoc.isValid()) {
340133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    // FIXME: Would be great to have a "hint" cursor, then walk from that
340233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    // hint cursor upward until we find a cursor whose source range encloses
340333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    // the region of interest, rather than starting from the translation unit.
3404a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    CXCursor Parent = clang_getTranslationUnitCursor(TU);
3405a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    CursorVisitor CursorVis(TU, GetCursorVisitor, &Result,
340604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                            Decl::MaxPCHLevel, true, SourceLocation(SLoc));
340733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    CursorVis.VisitChildren(Parent);
340877128ddd3077fc045751a55bb3226802b15d5510Steve Naroff  }
340940749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
341040749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  if (Logging) {
341140749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXFile SearchFile;
341240749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    unsigned SearchLine, SearchColumn;
341340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXFile ResultFile;
341440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    unsigned ResultLine, ResultColumn;
34156653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    CXString SearchFileName, ResultFileName, KindSpelling, USR;
34166653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    const char *IsDef = clang_isCursorDefinition(Result)? " (Definition)" : "";
341740749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXSourceLocation ResultLoc = clang_getCursorLocation(Result);
341840749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
341940749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_getInstantiationLocation(Loc, &SearchFile, &SearchLine, &SearchColumn,
342040749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor                                   0);
342140749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_getInstantiationLocation(ResultLoc, &ResultFile, &ResultLine,
342240749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor                                   &ResultColumn, 0);
342340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    SearchFileName = clang_getFileName(SearchFile);
342440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    ResultFileName = clang_getFileName(ResultFile);
342540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    KindSpelling = clang_getCursorKindSpelling(Result.kind);
34266653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    USR = clang_getCursorUSR(Result);
34276653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    fprintf(stderr, "clang_getCursor(%s:%d:%d) = %s(%s:%d:%d):%s%s\n",
342840749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor            clang_getCString(SearchFileName), SearchLine, SearchColumn,
342940749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor            clang_getCString(KindSpelling),
34306653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor            clang_getCString(ResultFileName), ResultLine, ResultColumn,
34316653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor            clang_getCString(USR), IsDef);
343240749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(SearchFileName);
343340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(ResultFileName);
343440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(KindSpelling);
34356653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    clang_disposeString(USR);
34360aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor
34370aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    CXCursor Definition = clang_getCursorDefinition(Result);
34380aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    if (!clang_equalCursors(Definition, clang_getNullCursor())) {
34390aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXSourceLocation DefinitionLoc = clang_getCursorLocation(Definition);
34400aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXString DefinitionKindSpelling
34410aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor                                = clang_getCursorKindSpelling(Definition.kind);
34420aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXFile DefinitionFile;
34430aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      unsigned DefinitionLine, DefinitionColumn;
34440aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_getInstantiationLocation(DefinitionLoc, &DefinitionFile,
34450aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor                                     &DefinitionLine, &DefinitionColumn, 0);
34460aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXString DefinitionFileName = clang_getFileName(DefinitionFile);
34470aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      fprintf(stderr, "  -> %s(%s:%d:%d)\n",
34480aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              clang_getCString(DefinitionKindSpelling),
34490aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              clang_getCString(DefinitionFileName),
34500aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              DefinitionLine, DefinitionColumn);
34510aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_disposeString(DefinitionFileName);
34520aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_disposeString(DefinitionKindSpelling);
34530aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    }
345440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  }
345540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
3456e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  return Result;
345777128ddd3077fc045751a55bb3226802b15d5510Steve Naroff}
345877128ddd3077fc045751a55bb3226802b15d5510Steve Naroff
3459738855554394a6afcf39cc8345fd22c3756b8dd0Ted KremenekCXCursor clang_getNullCursor(void) {
34605bfb8c128c2ac8eb4032afc180cdc400a0f953caDouglas Gregor  return MakeCXCursorInvalid(CXCursor_InvalidFile);
3461738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek}
3462738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek
3463738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenekunsigned clang_equalCursors(CXCursor X, CXCursor Y) {
3464283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  return X == Y;
3465738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek}
34660d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
34679ce5584553054d0cb934940586aca0186e87fa57Douglas Gregorunsigned clang_hashCursor(CXCursor C) {
34689ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  unsigned Index = 0;
34699ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  if (clang_isExpression(C.kind) || clang_isStatement(C.kind))
34709ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor    Index = 1;
34719ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor
34729ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  return llvm::DenseMapInfo<std::pair<unsigned, void*> >::getHashValue(
34739ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor                                        std::make_pair(C.kind, C.data[Index]));
34749ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor}
34759ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor
34769ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isInvalid(enum CXCursorKind K) {
347777128ddd3077fc045751a55bb3226802b15d5510Steve Naroff  return K >= CXCursor_FirstInvalid && K <= CXCursor_LastInvalid;
347877128ddd3077fc045751a55bb3226802b15d5510Steve Naroff}
347977128ddd3077fc045751a55bb3226802b15d5510Steve Naroff
34809ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isDeclaration(enum CXCursorKind K) {
348189922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  return K >= CXCursor_FirstDecl && K <= CXCursor_LastDecl;
348289922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff}
34832d4d629d8a0de5112c7ae9d05c03ddbf6dcd956aSteve Naroff
34849ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isReference(enum CXCursorKind K) {
3485f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  return K >= CXCursor_FirstRef && K <= CXCursor_LastRef;
3486f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff}
3487f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff
348897b9872d5775446cb8aca1380e437649fe848d91Douglas Gregorunsigned clang_isExpression(enum CXCursorKind K) {
348997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  return K >= CXCursor_FirstExpr && K <= CXCursor_LastExpr;
349097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor}
349197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
349297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregorunsigned clang_isStatement(enum CXCursorKind K) {
349397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  return K >= CXCursor_FirstStmt && K <= CXCursor_LastStmt;
349497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor}
349597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
34967eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregorunsigned clang_isTranslationUnit(enum CXCursorKind K) {
34977eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  return K == CXCursor_TranslationUnit;
34987eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor}
34997eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
35009f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregorunsigned clang_isPreprocessing(enum CXCursorKind K) {
35019f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  return K >= CXCursor_FirstPreprocessing && K <= CXCursor_LastPreprocessing;
35029f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor}
35039f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor
3504ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenekunsigned clang_isUnexposed(enum CXCursorKind K) {
3505ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek  switch (K) {
3506ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedDecl:
3507ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedExpr:
3508ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedStmt:
3509ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedAttr:
3510ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek      return true;
3511ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    default:
3512ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek      return false;
3513ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek  }
3514ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek}
3515ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek
35169ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXCursorKind clang_getCursorKind(CXCursor C) {
35179efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff  return C.kind;
35189efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff}
35199efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff
352098258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas GregorCXSourceLocation clang_getCursorLocation(CXCursor C) {
352198258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (clang_isReference(C.kind)) {
3522f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    switch (C.kind) {
3523f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCSuperClassRef: {
3524f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCInterfaceDecl *, SourceLocation> P
3525f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCSuperClassRef(C);
3526a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3527f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
3528f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor
3529f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCProtocolRef: {
3530f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCProtocolDecl *, SourceLocation> P
3531f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCProtocolRef(C);
3532a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3533f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
3534f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor
3535f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCClassRef: {
3536f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCInterfaceDecl *, SourceLocation> P
3537f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCClassRef(C);
3538a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3539f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
35407d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3541f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_TypeRef: {
35427d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      std::pair<TypeDecl *, SourceLocation> P = getCursorTypeRef(C);
3543a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
35447d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    }
35450b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
35460b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef: {
35470b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      std::pair<TemplateDecl *, SourceLocation> P = getCursorTemplateRef(C);
35480b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
35490b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    }
35500b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
35516931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef: {
35526931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      std::pair<NamedDecl *, SourceLocation> P = getCursorNamespaceRef(C);
35536931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
35546931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    }
35556931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
3556a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef: {
3557a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      std::pair<FieldDecl *, SourceLocation> P = getCursorMemberRef(C);
3558a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3559a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
3560a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
35613064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
35621b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      CXXBaseSpecifier *BaseSpec = getCursorCXXBaseSpecifier(C);
35631b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      if (!BaseSpec)
35641b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor        return clang_getNullLocation();
35651b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor
35661b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      if (TypeSourceInfo *TSInfo = BaseSpec->getTypeSourceInfo())
35671b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor        return cxloc::translateSourceLocation(getCursorContext(C),
35681b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor                                            TSInfo->getTypeLoc().getBeginLoc());
35691b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor
35701b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C),
35711b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor                                        BaseSpec->getSourceRange().getBegin());
35723064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
3573f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
357436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef: {
357536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      std::pair<LabelStmt *, SourceLocation> P = getCursorLabelRef(C);
357636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C), P.second);
357736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    }
357836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
35791f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
35801f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C),
35811f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                                          getCursorOverloadedDeclRef(C).second);
35821f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3583f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    default:
3584f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      // FIXME: Need a way to enumerate all non-reference cases.
3585f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      llvm_unreachable("Missed a reference kind");
3586f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
358798258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  }
358897b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
358997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind))
3590f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    return cxloc::translateSourceLocation(getCursorContext(C),
359197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor                                   getLocationFromExpr(getCursorExpr(C)));
359297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
359336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind))
359436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C),
359536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor                                          getCursorStmt(C)->getLocStart());
359636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
35979f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  if (C.kind == CXCursor_PreprocessingDirective) {
35989f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    SourceLocation L = cxcursor::getCursorPreprocessingDirective(C).getBegin();
35999f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
36009f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  }
36014807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor
36024807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  if (C.kind == CXCursor_MacroInstantiation) {
36034ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor    SourceLocation L
36044ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor      = cxcursor::getCursorMacroInstantiation(C)->getSourceRange().getBegin();
36054807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
36064807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  }
3607572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3608572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition) {
3609572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    SourceLocation L = cxcursor::getCursorMacroDefinition(C)->getLocation();
3610572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
3611572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  }
3612ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3613ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective) {
3614ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    SourceLocation L
3615ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor      = cxcursor::getCursorInclusionDirective(C)->getSourceRange().getBegin();
3616ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
3617ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  }
3618ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
36199a700d277c38d9afaa7cb3fe93a714bfe9b62eecTed Kremenek  if (C.kind < CXCursor_FirstDecl || C.kind > CXCursor_LastDecl)
36205352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullLocation();
362198258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor
3622f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  Decl *D = getCursorDecl(C);
3623f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  SourceLocation Loc = D->getLocation();
3624f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(D))
3625f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    Loc = Class->getClassLoc();
3626007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // FIXME: Multiple variables declared in a single declaration
3627007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // currently lack the information needed to correctly determine their
3628007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // ranges when accounting for the type-specifier.  We use context
3629007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
3630007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // and if so, whether it is the first decl.
3631007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
3632007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    if (!cxcursor::isFirstInDeclGroup(C))
3633007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek      Loc = VD->getLocation();
3634007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  }
3635007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek
36362ca54feee89d7277fb967e3247a64f40ef155a82Douglas Gregor  return cxloc::translateSourceLocation(getCursorContext(C), Loc);
363788145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff}
3638a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor
3639a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor} // end extern "C"
3640a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
3641a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorstatic SourceRange getRawCursorExtent(CXCursor C) {
3642a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  if (clang_isReference(C.kind)) {
3643a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor    switch (C.kind) {
3644a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCSuperClassRef:
3645a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return  getCursorObjCSuperClassRef(C).second;
3646f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3647a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCProtocolRef:
3648a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorObjCProtocolRef(C).second;
3649f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3650a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCClassRef:
3651a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorObjCClassRef(C).second;
36527d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3653a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_TypeRef:
3654a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorTypeRef(C).second;
36550b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
36560b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef:
36570b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return getCursorTemplateRef(C).second;
36580b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
36596931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef:
36606931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return getCursorNamespaceRef(C).second;
3661a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
3662a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef:
3663a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return getCursorMemberRef(C).second;
3664a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
36653064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier:
36661b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      return getCursorCXXBaseSpecifier(C)->getSourceRange();
3667f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
366836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef:
366936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return getCursorLabelRef(C).second;
367036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
36711f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
36721f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return getCursorOverloadedDeclRef(C).second;
36731f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3674a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    default:
3675a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      // FIXME: Need a way to enumerate all non-reference cases.
3676a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      llvm_unreachable("Missed a reference kind");
3677a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor    }
3678a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  }
367997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
368097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind))
3681a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return getCursorExpr(C)->getSourceRange();
368233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
368333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (clang_isStatement(C.kind))
3684a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return getCursorStmt(C)->getSourceRange();
3685f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3686a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (C.kind == CXCursor_PreprocessingDirective)
3687a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return cxcursor::getCursorPreprocessingDirective(C);
36884807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor
3689a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (C.kind == CXCursor_MacroInstantiation)
3690a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return cxcursor::getCursorMacroInstantiation(C)->getSourceRange();
3691572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3692a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (C.kind == CXCursor_MacroDefinition)
3693a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return cxcursor::getCursorMacroDefinition(C)->getSourceRange();
3694ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3695ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective)
3696ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return cxcursor::getCursorInclusionDirective(C)->getSourceRange();
3697ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3698007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  if (C.kind >= CXCursor_FirstDecl && C.kind <= CXCursor_LastDecl) {
3699007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    Decl *D = cxcursor::getCursorDecl(C);
3700007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    SourceRange R = D->getSourceRange();
3701007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // FIXME: Multiple variables declared in a single declaration
3702007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // currently lack the information needed to correctly determine their
3703007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // ranges when accounting for the type-specifier.  We use context
3704007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
3705007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // and if so, whether it is the first decl.
3706007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
3707007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek      if (!cxcursor::isFirstInDeclGroup(C))
3708007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek        R.setBegin(VD->getLocation());
3709007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    }
3710007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    return R;
3711007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  }
37126653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  return SourceRange();
37136653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor}
37146653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
37156653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor/// \brief Retrieves the "raw" cursor extent, which is then extended to include
37166653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor/// the decl-specifier-seq for declarations.
37176653798ff5ce6deb58112777e21307ccc453133dDouglas Gregorstatic SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr) {
37186653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  if (C.kind >= CXCursor_FirstDecl && C.kind <= CXCursor_LastDecl) {
37196653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    Decl *D = cxcursor::getCursorDecl(C);
37206653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    SourceRange R = D->getSourceRange();
37212494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
37222494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // Adjust the start of the location for declarations preceded by
37232494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // declaration specifiers.
37242494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    SourceLocation StartLoc;
37256653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    if (const DeclaratorDecl *DD = dyn_cast<DeclaratorDecl>(D)) {
37262494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
37272494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
37282494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    } else if (TypedefDecl *Typedef = dyn_cast<TypedefDecl>(D)) {
37292494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
37302494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
37312494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    }
37326653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
37332494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (StartLoc.isValid() && R.getBegin().isValid() &&
37342494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        SrcMgr.isBeforeInTranslationUnit(StartLoc, R.getBegin()))
37352494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      R.setBegin(StartLoc);
37362494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
37372494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // FIXME: Multiple variables declared in a single declaration
37382494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // currently lack the information needed to correctly determine their
37392494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // ranges when accounting for the type-specifier.  We use context
37402494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
37412494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // and if so, whether it is the first decl.
37422494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
37432494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (!cxcursor::isFirstInDeclGroup(C))
37442494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        R.setBegin(VD->getLocation());
37456653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    }
37466653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
37476653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    return R;
37486653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  }
37496653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
37506653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  return getRawCursorExtent(C);
37516653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor}
3752a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
3753a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorextern "C" {
3754a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
3755a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas GregorCXSourceRange clang_getCursorExtent(CXCursor C) {
3756a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  SourceRange R = getRawCursorExtent(C);
3757a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (R.isInvalid())
37585352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
3759f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3760a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  return cxloc::translateSourceRange(getCursorContext(C), R);
3761a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor}
3762c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor
3763c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas GregorCXCursor clang_getCursorReferenced(CXCursor C) {
3764b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (clang_isInvalid(C.kind))
3765b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getNullCursor();
3766f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3767a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit tu = getCursorTU(C);
37681f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (clang_isDeclaration(C.kind)) {
37691f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Decl *D = getCursorDecl(C);
37701f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (UsingDecl *Using = dyn_cast<UsingDecl>(D))
3771a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Using, D->getLocation(), tu);
37721f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (ObjCClassDecl *Classes = dyn_cast<ObjCClassDecl>(D))
3773a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Classes, D->getLocation(), tu);
37741f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (ObjCForwardProtocolDecl *Protocols
37751f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                                        = dyn_cast<ObjCForwardProtocolDecl>(D))
3776a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Protocols, D->getLocation(), tu);
3777e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor    if (ObjCPropertyImplDecl *PropImpl =llvm::dyn_cast<ObjCPropertyImplDecl>(D))
3778e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor      if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
3779e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor        return MakeCXCursor(Property, tu);
3780e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
3781c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    return C;
37821f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  }
37831f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
378497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind)) {
37851f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Expr *E = getCursorExpr(C);
37861f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Decl *D = getDeclFromExpr(E);
378797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    if (D)
3788a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(D, tu);
37891f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
37901f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (OverloadExpr *Ovl = dyn_cast_or_null<OverloadExpr>(E))
3791a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Ovl, tu);
37921f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
379397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    return clang_getNullCursor();
379497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  }
379597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
379636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind)) {
379736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    Stmt *S = getCursorStmt(C);
379836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    if (GotoStmt *Goto = dyn_cast_or_null<GotoStmt>(S))
379937c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek      if (LabelDecl *label = Goto->getLabel())
380037c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek        if (LabelStmt *labelS = label->getStmt())
380137c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek        return MakeCXCursor(labelS, getCursorDecl(C), tu);
380236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
380336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return clang_getNullCursor();
380436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  }
380536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
3806bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor  if (C.kind == CXCursor_MacroInstantiation) {
3807bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor    if (MacroDefinition *Def = getCursorMacroInstantiation(C)->getDefinition())
3808a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeMacroDefinitionCursor(Def, tu);
3809bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor  }
3810bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor
3811c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  if (!clang_isReference(C.kind))
3812c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    return clang_getNullCursor();
3813f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3814c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  switch (C.kind) {
3815c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    case CXCursor_ObjCSuperClassRef:
3816a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorObjCSuperClassRef(C).first, tu);
3817f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3818f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCProtocolRef: {
3819a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorObjCProtocolRef(C).first, tu);
3820f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3821f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCClassRef:
3822a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorObjCClassRef(C).first, tu );
38237d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3824f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_TypeRef:
3825a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorTypeRef(C).first, tu );
38260b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
38270b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef:
3828a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorTemplateRef(C).first, tu );
38290b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
38306931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef:
3831a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorNamespaceRef(C).first, tu );
38326931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
3833a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef:
3834a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorMemberRef(C).first, tu );
3835a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
38363064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
38373064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      CXXBaseSpecifier *B = cxcursor::getCursorCXXBaseSpecifier(C);
38383064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      return clang_getTypeDeclaration(cxtype::MakeCXType(B->getType(),
3839a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                                         tu ));
38403064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
3841f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
384236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef:
384336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      // FIXME: We end up faking the "parent" declaration here because we
384436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      // don't want to make CXCursor larger.
384536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return MakeCXCursor(getCursorLabelRef(C).first,
3846a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek               static_cast<ASTUnit*>(tu->TUData)->getASTContext()
3847a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          .getTranslationUnitDecl(),
3848a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          tu);
384936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
38501f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
38511f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return C;
38521f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3853c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    default:
3854c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      // We would prefer to enumerate all non-reference cursor kinds here.
3855c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      llvm_unreachable("Unhandled reference cursor kind");
3856c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      break;
3857c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    }
3858c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  }
3859f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3860c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  return clang_getNullCursor();
3861c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor}
3862c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor
3863b699866820102a69d83d6ac6941985c5ef4e8c40Douglas GregorCXCursor clang_getCursorDefinition(CXCursor C) {
3864b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (clang_isInvalid(C.kind))
3865b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getNullCursor();
3866f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3867a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(C);
3868f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3869b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  bool WasReference = false;
387097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isReference(C.kind) || clang_isExpression(C.kind)) {
3871b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    C = clang_getCursorReferenced(C);
3872b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    WasReference = true;
3873b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
3874b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3875bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor  if (C.kind == CXCursor_MacroInstantiation)
3876bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor    return clang_getCursorReferenced(C);
3877bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor
3878b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!clang_isDeclaration(C.kind))
3879b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3880b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3881b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  Decl *D = getCursorDecl(C);
3882b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!D)
3883b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3884f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3885b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  switch (D->getKind()) {
3886b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds that don't really separate the notions of
3887b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // declaration and definition.
3888b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Namespace:
3889b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Typedef:
3890162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  case Decl::TypeAlias:
3891b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TemplateTypeParm:
3892b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::EnumConstant:
3893b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Field:
3894d98114647e16796a976b04af79975b4f0eacf22bBenjamin Kramer  case Decl::IndirectField:
3895b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCIvar:
3896b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCAtDefsField:
3897b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ImplicitParam:
3898b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ParmVar:
3899b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::NonTypeTemplateParm:
3900b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TemplateTemplateParm:
3901b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCategoryImpl:
3902b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCImplementation:
39036206d53f67613958ae1b023aba337ebb46f11a8bAbramo Bagnara  case Decl::AccessSpec:
3904b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::LinkageSpec:
3905b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCPropertyImpl:
3906b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FileScopeAsm:
3907b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::StaticAssert:
3908b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Block:
3909ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  case Decl::Label:  // FIXME: Is this right??
3910b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return C;
3911b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3912b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds that don't make any sense here, but are
3913b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // nonetheless harmless.
3914b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TranslationUnit:
3915b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    break;
3916b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3917b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds for which the definition is not resolvable.
3918b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UnresolvedUsingTypename:
3919b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UnresolvedUsingValue:
3920b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    break;
3921b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3922b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UsingDirective:
3923b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return MakeCXCursor(cast<UsingDirectiveDecl>(D)->getNominatedNamespace(),
3924a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                        TU);
3925b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3926b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::NamespaceAlias:
3927a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(cast<NamespaceAliasDecl>(D)->getNamespace(), TU);
3928b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3929b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Enum:
3930b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Record:
3931b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXRecord:
3932b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplateSpecialization:
3933b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplatePartialSpecialization:
3934952b017601f9c82b51119c3a1600f1312a833db9Douglas Gregor    if (TagDecl *Def = cast<TagDecl>(D)->getDefinition())
3935a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def, TU);
3936b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3937b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3938b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Function:
3939b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXMethod:
3940b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXConstructor:
3941b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXDestructor:
3942b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXConversion: {
3943b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    const FunctionDecl *Def = 0;
3944b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (cast<FunctionDecl>(D)->getBody(Def))
3945a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(const_cast<FunctionDecl *>(Def), TU);
3946b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3947b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
3948b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3949b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Var: {
395031310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    // Ask the variable if it has a definition.
395131310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    if (VarDecl *Def = cast<VarDecl>(D)->getDefinition())
3952a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def, TU);
395331310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    return clang_getNullCursor();
3954b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
3955f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3956b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FunctionTemplate: {
3957b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    const FunctionDecl *Def = 0;
3958b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (cast<FunctionTemplateDecl>(D)->getTemplatedDecl()->getBody(Def))
3959a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def->getDescribedFunctionTemplate(), TU);
3960b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3961b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
3962f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3963b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplate: {
3964b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (RecordDecl *Def = cast<ClassTemplateDecl>(D)->getTemplatedDecl()
3965952b017601f9c82b51119c3a1600f1312a833db9Douglas Gregor                                                            ->getDefinition())
39660b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return MakeCXCursor(cast<CXXRecordDecl>(Def)->getDescribedClassTemplate(),
3967a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          TU);
3968b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3969b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
3970b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
39711f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case Decl::Using:
39721f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return MakeCursorOverloadedDeclRef(cast<UsingDecl>(D),
3973a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                       D->getLocation(), TU);
3974b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3975b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UsingShadow:
3976b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getCursorDefinition(
3977f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                       MakeCXCursor(cast<UsingShadowDecl>(D)->getTargetDecl(),
3978a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                    TU));
3979b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3980b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCMethod: {
3981b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    ObjCMethodDecl *Method = cast<ObjCMethodDecl>(D);
3982b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (Method->isThisDeclarationADefinition())
3983b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      return C;
3984b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3985b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // Dig out the method definition in the associated
3986b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // @implementation, if we have it.
3987b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: The ASTs should make finding the definition easier.
3988b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCInterfaceDecl *Class
3989b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                       = dyn_cast<ObjCInterfaceDecl>(Method->getDeclContext()))
3990b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (ObjCImplementationDecl *ClassImpl = Class->getImplementation())
3991b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor        if (ObjCMethodDecl *Def = ClassImpl->getMethod(Method->getSelector(),
3992b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                                                  Method->isInstanceMethod()))
3993b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor          if (Def->isThisDeclarationADefinition())
3994a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek            return MakeCXCursor(Def, TU);
3995b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3996b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3997b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
3998b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3999b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCategory:
4000b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCCategoryImplDecl *Impl
4001b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                               = cast<ObjCCategoryDecl>(D)->getImplementation())
4002a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Impl, TU);
4003b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4004b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4005b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCProtocol:
4006b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (!cast<ObjCProtocolDecl>(D)->isForwardDecl())
4007b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      return C;
4008b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4009b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4010b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCInterface:
4011b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // There are two notions of a "definition" for an Objective-C
4012b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // class: the interface and its implementation. When we resolved a
4013b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // reference to an Objective-C class, produce the @interface as
4014b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // the definition; when we were provided with the interface,
4015b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // produce the @implementation as the definition.
4016b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (WasReference) {
4017b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (!cast<ObjCInterfaceDecl>(D)->isForwardDecl())
4018b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor        return C;
4019b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    } else if (ObjCImplementationDecl *Impl
4020b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                              = cast<ObjCInterfaceDecl>(D)->getImplementation())
4021a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Impl, TU);
4022b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4023f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4024b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCProperty:
4025b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: We don't really know where to find the
4026b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // ObjCPropertyImplDecls that implement this property.
4027b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4028b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4029b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCompatibleAlias:
4030b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCInterfaceDecl *Class
4031b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor          = cast<ObjCCompatibleAliasDecl>(D)->getClassInterface())
4032b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (!Class->isForwardDecl())
4033a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek        return MakeCXCursor(Class, TU);
4034f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4035b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4036b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
40371f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case Decl::ObjCForwardProtocol:
40381f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return MakeCursorOverloadedDeclRef(cast<ObjCForwardProtocolDecl>(D),
4039a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                       D->getLocation(), TU);
4040b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
40411f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case Decl::ObjCClass:
40429e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar    return MakeCursorOverloadedDeclRef(cast<ObjCClassDecl>(D), D->getLocation(),
4043a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                       TU);
4044b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4045b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Friend:
4046b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (NamedDecl *Friend = cast<FriendDecl>(D)->getFriendDecl())
4047a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
4048b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4049b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4050b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FriendTemplate:
4051b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (NamedDecl *Friend = cast<FriendTemplateDecl>(D)->getFriendDecl())
4052a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
4053b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4054b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4055b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4056b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  return clang_getNullCursor();
4057b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor}
4058b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4059b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregorunsigned clang_isCursorDefinition(CXCursor C) {
4060b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!clang_isDeclaration(C.kind))
4061b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return 0;
4062b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4063b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  return clang_getCursorDefinition(C) == C;
4064b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor}
4065b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
40661a9d0503b67a499797141af0fd6d315d5045f0eaDouglas GregorCXCursor clang_getCanonicalCursor(CXCursor C) {
40671a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  if (!clang_isDeclaration(C.kind))
40681a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor    return C;
40691a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
40701a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  if (Decl *D = getCursorDecl(C))
40711a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor    return MakeCXCursor(D->getCanonicalDecl(), getCursorTU(C));
40721a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
40731a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  return C;
40741a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor}
40751a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
40761f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregorunsigned clang_getNumOverloadedDecls(CXCursor C) {
40777c432dd959609a3689c2e4406450c092e6d76d6dDouglas Gregor  if (C.kind != CXCursor_OverloadedDeclRef)
40781f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return 0;
40791f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40801f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
40811f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
40821f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return E->getNumDecls();
40831f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40841f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadedTemplateStorage *S
40851f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                              = Storage.dyn_cast<OverloadedTemplateStorage*>())
40861f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return S->size();
40871f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40881f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  Decl *D = Storage.get<Decl*>();
40891f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (UsingDecl *Using = dyn_cast<UsingDecl>(D))
4090826faa22bae112e01293a58534a40711043cce65Argyrios Kyrtzidis    return Using->shadow_size();
40911f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (ObjCClassDecl *Classes = dyn_cast<ObjCClassDecl>(D))
40921f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return Classes->size();
40931f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (ObjCForwardProtocolDecl *Protocols =dyn_cast<ObjCForwardProtocolDecl>(D))
40941f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return Protocols->protocol_size();
40951f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40961f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  return 0;
40971f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor}
40981f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40991f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas GregorCXCursor clang_getOverloadedDecl(CXCursor cursor, unsigned index) {
41007c432dd959609a3689c2e4406450c092e6d76d6dDouglas Gregor  if (cursor.kind != CXCursor_OverloadedDeclRef)
41011f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return clang_getNullCursor();
41021f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
41031f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (index >= clang_getNumOverloadedDecls(cursor))
41041f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return clang_getNullCursor();
41051f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
4106a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(cursor);
41071f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(cursor).first;
41081f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
4109a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(E->decls_begin()[index], TU);
41101f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
41111f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadedTemplateStorage *S
41121f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                              = Storage.dyn_cast<OverloadedTemplateStorage*>())
4113a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(S->begin()[index], TU);
41141f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
41151f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  Decl *D = Storage.get<Decl*>();
41161f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (UsingDecl *Using = dyn_cast<UsingDecl>(D)) {
41171f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    // FIXME: This is, unfortunately, linear time.
41181f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    UsingDecl::shadow_iterator Pos = Using->shadow_begin();
41191f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    std::advance(Pos, index);
4120a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(cast<UsingShadowDecl>(*Pos)->getTargetDecl(), TU);
41211f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  }
41221f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
41231f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (ObjCClassDecl *Classes = dyn_cast<ObjCClassDecl>(D))
4124a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(Classes->begin()[index].getInterface(), TU);
41251f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
41261f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (ObjCForwardProtocolDecl *Protocols = dyn_cast<ObjCForwardProtocolDecl>(D))
4127a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(Protocols->protocol_begin()[index], TU);
41281f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
41291f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  return clang_getNullCursor();
41301f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor}
41311f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
41320d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbarvoid clang_getDefinitionSpellingAndExtent(CXCursor C,
41334ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          const char **startBuf,
41344ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          const char **endBuf,
41354ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *startLine,
41364ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *startColumn,
41374ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *endLine,
41389ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          unsigned *endColumn) {
4139283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  assert(getCursorDecl(C) && "CXCursor has null decl");
4140283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  NamedDecl *ND = static_cast<NamedDecl *>(getCursorDecl(C));
41414ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  FunctionDecl *FD = dyn_cast<FunctionDecl>(ND);
41424ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  CompoundStmt *Body = dyn_cast<CompoundStmt>(FD->getBody());
4143f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
41444ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  SourceManager &SM = FD->getASTContext().getSourceManager();
41454ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startBuf = SM.getCharacterData(Body->getLBracLoc());
41464ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endBuf = SM.getCharacterData(Body->getRBracLoc());
41474ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startLine = SM.getSpellingLineNumber(Body->getLBracLoc());
41484ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startColumn = SM.getSpellingColumnNumber(Body->getLBracLoc());
41494ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endLine = SM.getSpellingLineNumber(Body->getRBracLoc());
41504ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endColumn = SM.getSpellingColumnNumber(Body->getRBracLoc());
41514ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff}
4152f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
41530a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregorvoid clang_enableStackTraces(void) {
41540a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  llvm::sys::PrintStackTraceOnErrorSignal();
41550a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor}
41560a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor
4157995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbarvoid clang_executeOnThread(void (*fn)(void*), void *user_data,
4158995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar                           unsigned stack_size) {
4159995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar  llvm::llvm_execute_on_thread(fn, user_data, stack_size);
4160995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar}
4161995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar
4162fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
4163fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
4164fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
4165fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor// Token-based Operations.
4166fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor//===----------------------------------------------------------------------===//
4167fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4168fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor/* CXToken layout:
4169fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[0]: a CXTokenKind
4170fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[1]: starting token location
4171fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[2]: token length
4172fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[3]: reserved
4173f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek *   ptr_data: for identifiers and keywords, an IdentifierInfo*.
4174fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   otherwise unused.
4175fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor */
4176fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregorextern "C" {
4177fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4178fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXTokenKind clang_getTokenKind(CXToken CXTok) {
4179fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  return static_cast<CXTokenKind>(CXTok.int_data[0]);
4180fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4181fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4182fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXString clang_getTokenSpelling(CXTranslationUnit TU, CXToken CXTok) {
4183fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  switch (clang_getTokenKind(CXTok)) {
4184fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Identifier:
4185fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Keyword:
4186fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // We know we have an IdentifierInfo*, so use that.
4187ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(static_cast<IdentifierInfo *>(CXTok.ptr_data)
4188ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek                            ->getNameStart());
4189fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4190fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Literal: {
4191fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // We have stashed the starting pointer in the ptr_data field. Use it.
4192fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    const char *Text = static_cast<const char *>(CXTok.ptr_data);
4193ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(llvm::StringRef(Text, CXTok.int_data[2]));
4194fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  }
4195f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4196fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Punctuation:
4197fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Comment:
4198fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    break;
4199fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  }
4200f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4201f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // We have to find the starting buffer pointer the hard way, by
4202fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // deconstructing the source location.
4203a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4204fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit)
4205ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
4206f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4207fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  SourceLocation Loc = SourceLocation::getFromRawEncoding(CXTok.int_data[1]);
4208fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> LocInfo
4209fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    = CXXUnit->getSourceManager().getDecomposedLoc(Loc);
4210f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  bool Invalid = false;
4211f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  llvm::StringRef Buffer
4212f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    = CXXUnit->getSourceManager().getBufferData(LocInfo.first, &Invalid);
4213f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  if (Invalid)
4214aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor    return createCXString("");
4215fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4216f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  return createCXString(Buffer.substr(LocInfo.second, CXTok.int_data[2]));
4217fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4218f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4219fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXSourceLocation clang_getTokenLocation(CXTranslationUnit TU, CXToken CXTok) {
4220a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4221fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit)
4222fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return clang_getNullLocation();
4223f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4224fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  return cxloc::translateSourceLocation(CXXUnit->getASTContext(),
4225fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                        SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
4226fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4227fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4228fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXSourceRange clang_getTokenExtent(CXTranslationUnit TU, CXToken CXTok) {
4229a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
42305352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  if (!CXXUnit)
42315352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
4232f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4233f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return cxloc::translateSourceRange(CXXUnit->getASTContext(),
4234fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                        SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
4235fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4236f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4237fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregorvoid clang_tokenize(CXTranslationUnit TU, CXSourceRange Range,
4238fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                    CXToken **Tokens, unsigned *NumTokens) {
4239fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (Tokens)
4240fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    *Tokens = 0;
4241fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (NumTokens)
4242fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    *NumTokens = 0;
4243f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4244a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4245fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit || !Tokens || !NumTokens)
4246fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4247f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4248bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
4249bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor
425085b988fdfa6adab6d43e16efd19ad4f3f7e2b49bDaniel Dunbar  SourceRange R = cxloc::translateCXSourceRange(Range);
4251fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (R.isInvalid())
4252fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4253f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4254fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  SourceManager &SourceMgr = CXXUnit->getSourceManager();
4255fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> BeginLocInfo
4256fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    = SourceMgr.getDecomposedLoc(R.getBegin());
4257fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> EndLocInfo
4258fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    = SourceMgr.getDecomposedLoc(R.getEnd());
4259f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4260fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // Cannot tokenize across files.
4261fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (BeginLocInfo.first != EndLocInfo.first)
4262fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4263f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4264f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // Create a lexer
4265f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  bool Invalid = false;
4266f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  llvm::StringRef Buffer
4267f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
426847a3fcd4afe122b23f9e7b6148f147bfa460cfe8Douglas Gregor  if (Invalid)
426947a3fcd4afe122b23f9e7b6148f147bfa460cfe8Douglas Gregor    return;
4270aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor
4271fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
4272fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor            CXXUnit->getASTContext().getLangOptions(),
4273f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer            Buffer.begin(), Buffer.data() + BeginLocInfo.second, Buffer.end());
4274fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Lex.SetCommentRetentionState(true);
4275f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4276fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // Lex tokens until we hit the end of the range.
4277f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  const char *EffectiveBufferEnd = Buffer.data() + EndLocInfo.second;
4278fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  llvm::SmallVector<CXToken, 32> CXTokens;
4279fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Token Tok;
4280096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall  bool previousWasAt = false;
4281fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  do {
4282fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // Lex the next token
4283fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    Lex.LexFromRawLexer(Tok);
4284fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    if (Tok.is(tok::eof))
4285fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      break;
4286f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4287fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // Initialize the CXToken.
4288fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXToken CXTok;
4289f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4290fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    //   - Common fields
4291fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[1] = Tok.getLocation().getRawEncoding();
4292fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[2] = Tok.getLength();
4293fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[3] = 0;
4294f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4295fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    //   - Kind-specific fields
4296fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    if (Tok.isLiteral()) {
4297fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Literal;
4298fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = (void *)Tok.getLiteralData();
4299c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara    } else if (Tok.is(tok::raw_identifier)) {
4300aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor      // Lookup the identifier to determine whether we have a keyword.
4301fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      IdentifierInfo *II
4302c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara        = CXXUnit->getPreprocessor().LookUpIdentifierInfo(Tok);
4303aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek
4304096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall      if ((II->getObjCKeywordID() != tok::objc_not_keyword) && previousWasAt) {
4305aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek        CXTok.int_data[0] = CXToken_Keyword;
4306aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      }
4307aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      else {
4308c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara        CXTok.int_data[0] = Tok.is(tok::identifier)
4309c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara          ? CXToken_Identifier
4310c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara          : CXToken_Keyword;
4311aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      }
4312fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = II;
4313fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else if (Tok.is(tok::comment)) {
4314fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Comment;
4315fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = 0;
4316fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else {
4317fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Punctuation;
4318fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = 0;
4319fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    }
4320fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTokens.push_back(CXTok);
4321096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall    previousWasAt = Tok.is(tok::at);
4322fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  } while (Lex.getBufferLocation() <= EffectiveBufferEnd);
4323f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4324fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (CXTokens.empty())
4325fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4326f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4327fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  *Tokens = (CXToken *)malloc(sizeof(CXToken) * CXTokens.size());
4328fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  memmove(*Tokens, CXTokens.data(), sizeof(CXToken) * CXTokens.size());
4329fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  *NumTokens = CXTokens.size();
4330fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
43310045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
43326db610934bedc6896393c1e1099525b35380acd6Ted Kremenekvoid clang_disposeTokens(CXTranslationUnit TU,
43336db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                         CXToken *Tokens, unsigned NumTokens) {
43346db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  free(Tokens);
43356db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
43366db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
43376db610934bedc6896393c1e1099525b35380acd6Ted Kremenek} // end: extern "C"
43386db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
43396db610934bedc6896393c1e1099525b35380acd6Ted Kremenek//===----------------------------------------------------------------------===//
43406db610934bedc6896393c1e1099525b35380acd6Ted Kremenek// Token annotation APIs.
43416db610934bedc6896393c1e1099525b35380acd6Ted Kremenek//===----------------------------------------------------------------------===//
43426db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
43430045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregortypedef llvm::DenseMap<unsigned, CXCursor> AnnotateTokensData;
4344fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekstatic enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
4345fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                     CXCursor parent,
4346fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                     CXClientData client_data);
43476db610934bedc6896393c1e1099525b35380acd6Ted Kremeneknamespace {
43486db610934bedc6896393c1e1099525b35380acd6Ted Kremenekclass AnnotateTokensWorker {
43496db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  AnnotateTokensData &Annotated;
435011949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  CXToken *Tokens;
435111949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  CXCursor *Cursors;
435211949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  unsigned NumTokens;
4353fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  unsigned TokIdx;
43544419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  unsigned PreprocessingTokIdx;
4355fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CursorVisitor AnnotateVis;
4356fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceManager &SrcMgr;
4357f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  bool HasContextSensitiveKeywords;
4358f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4359fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  bool MoreTokens() const { return TokIdx < NumTokens; }
4360fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  unsigned NextToken() const { return TokIdx; }
4361fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void AdvanceToken() { ++TokIdx; }
4362fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceLocation GetTokenLoc(unsigned tokI) {
4363fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[1]);
4364fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4365fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
43666db610934bedc6896393c1e1099525b35380acd6Ted Kremenekpublic:
436711949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  AnnotateTokensWorker(AnnotateTokensData &annotated,
4368fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                       CXToken *tokens, CXCursor *cursors, unsigned numTokens,
4369a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                       CXTranslationUnit tu, SourceRange RegionOfInterest)
437011949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek    : Annotated(annotated), Tokens(tokens), Cursors(cursors),
43714419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      NumTokens(numTokens), TokIdx(0), PreprocessingTokIdx(0),
4372a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      AnnotateVis(tu,
4373a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                  AnnotateTokensVisitor, this,
437404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                  Decl::MaxPCHLevel, true, RegionOfInterest),
4375f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      SrcMgr(static_cast<ASTUnit*>(tu->TUData)->getSourceManager()),
4376f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      HasContextSensitiveKeywords(false) { }
437711949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4378fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void VisitChildren(CXCursor C) { AnnotateVis.VisitChildren(C); }
43796db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  enum CXChildVisitResult Visit(CXCursor cursor, CXCursor parent);
4380fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void AnnotateTokens(CXCursor parent);
4381ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek  void AnnotateTokens() {
4382a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    AnnotateTokens(clang_getTranslationUnitCursor(AnnotateVis.getTU()));
4383ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek  }
4384f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4385f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  /// \brief Determine whether the annotator saw any cursors that have
4386f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  /// context-sensitive keywords.
4387f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  bool hasContextSensitiveKeywords() const {
4388f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    return HasContextSensitiveKeywords;
4389f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
43906db610934bedc6896393c1e1099525b35380acd6Ted Kremenek};
43916db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
43920045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
4393fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekvoid AnnotateTokensWorker::AnnotateTokens(CXCursor parent) {
4394fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Walk the AST within the region of interest, annotating tokens
4395fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // along the way.
4396fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  VisitChildren(parent);
4397fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4398fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = 0 ; I < TokIdx ; ++I) {
439911949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek    AnnotateTokensData::iterator Pos = Annotated.find(Tokens[I].int_data[1]);
44004419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    if (Pos != Annotated.end() &&
44014419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        (clang_isInvalid(Cursors[I].kind) ||
44024419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor         Pos->second.kind != CXCursor_PreprocessingDirective))
4403fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      Cursors[I] = Pos->second;
4404fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4405fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4406fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Finish up annotating any tokens left.
4407fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (!MoreTokens())
4408fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return;
440911949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4410fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const CXCursor &C = clang_getNullCursor();
4411fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = TokIdx ; I < NumTokens ; ++I) {
4412fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    AnnotateTokensData::iterator Pos = Annotated.find(Tokens[I].int_data[1]);
4413fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = (Pos == Annotated.end()) ? C : Pos->second;
441411949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  }
441511949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek}
441611949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
44176db610934bedc6896393c1e1099525b35380acd6Ted Kremenekenum CXChildVisitResult
44184419b675577d7c281a659fab1fec10e1bfbe04c5Douglas GregorAnnotateTokensWorker::Visit(CXCursor cursor, CXCursor parent) {
4419fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CXSourceLocation Loc = clang_getCursorLocation(cursor);
44204419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  SourceRange cursorRange = getRawCursorExtent(cursor);
442181d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor  if (cursorRange.isInvalid())
442281d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor    return CXChildVisit_Recurse;
4423f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4424f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  if (!HasContextSensitiveKeywords) {
4425f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // Objective-C properties can have context-sensitive keywords.
4426f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    if (cursor.kind == CXCursor_ObjCPropertyDecl) {
4427f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (ObjCPropertyDecl *Property
4428f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                  = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(cursor)))
4429f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        HasContextSensitiveKeywords = Property->getPropertyAttributesAsWritten() != 0;
4430f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4431f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // Objective-C methods can have context-sensitive keywords.
4432f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_ObjCInstanceMethodDecl ||
4433f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ObjCClassMethodDecl) {
4434f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (ObjCMethodDecl *Method
4435f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(cursor))) {
4436f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (Method->getObjCDeclQualifier())
4437f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4438f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        else {
4439f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
4440f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                                           PEnd = Method->param_end();
4441f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor               P != PEnd; ++P) {
4442f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            if ((*P)->getObjCDeclQualifier()) {
4443f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              HasContextSensitiveKeywords = true;
4444f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              break;
4445f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            }
4446f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          }
4447f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
4448f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4449f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4450f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // C++ methods can have context-sensitive keywords.
4451f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_CXXMethod) {
4452f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (CXXMethodDecl *Method
4453f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                  = dyn_cast_or_null<CXXMethodDecl>(getCursorDecl(cursor))) {
4454f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (Method->hasAttr<FinalAttr>() || Method->hasAttr<OverrideAttr>())
4455f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4456f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4457f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4458f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // C++ classes can have context-sensitive keywords.
4459f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_StructDecl ||
4460f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassDecl ||
4461f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassTemplate ||
4462f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassTemplatePartialSpecialization) {
4463f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Decl *D = getCursorDecl(cursor))
4464f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (D->hasAttr<FinalAttr>())
4465f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4466f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4467f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
4468f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
44694419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  if (clang_isPreprocessing(cursor.kind)) {
44704419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // For macro instantiations, just note where the beginning of the macro
44714419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // instantiation occurs.
44724419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    if (cursor.kind == CXCursor_MacroInstantiation) {
44734419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      Annotated[Loc.int_data] = cursor;
44744419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      return CXChildVisit_Recurse;
44754419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
44764419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
44774419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Items in the preprocessing record are kept separate from items in
44784419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // declarations, so we keep a separate token index.
44794419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    unsigned SavedTokIdx = TokIdx;
44804419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    TokIdx = PreprocessingTokIdx;
44814419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
44824419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Skip tokens up until we catch up to the beginning of the preprocessing
44834419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // entry.
44844419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    while (MoreTokens()) {
44854419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      const unsigned I = NextToken();
44864419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      SourceLocation TokLoc = GetTokenLoc(I);
44874419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
44884419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeBefore:
44894419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        AdvanceToken();
44904419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        continue;
44914419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeAfter:
44924419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeOverlap:
44934419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        break;
44944419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      }
44954419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      break;
44964419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
44974419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
44984419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Look at all of the tokens within this range.
44994419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    while (MoreTokens()) {
45004419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      const unsigned I = NextToken();
45014419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      SourceLocation TokLoc = GetTokenLoc(I);
45024419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
45034419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeBefore:
45044419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        assert(0 && "Infeasible");
45054419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeAfter:
45064419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        break;
45074419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeOverlap:
45084419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        Cursors[I] = cursor;
45094419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        AdvanceToken();
45104419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        continue;
45114419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      }
45124419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      break;
45134419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
45144419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
45154419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Save the preprocessing token index; restore the non-preprocessing
45164419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // token index.
45174419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    PreprocessingTokIdx = TokIdx;
45184419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    TokIdx = SavedTokIdx;
45190045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor    return CXChildVisit_Recurse;
45200045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  }
4521fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4522fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (cursorRange.isInvalid())
4523fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return CXChildVisit_Continue;
4524a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek
4525fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceLocation L = SourceLocation::getFromRawEncoding(Loc.int_data);
4526fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4527a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  // Adjust the annotated range based specific declarations.
4528a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  const enum CXCursorKind cursorK = clang_getCursorKind(cursor);
4529a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  if (cursorK >= CXCursor_FirstDecl && cursorK <= CXCursor_LastDecl) {
453023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    Decl *D = cxcursor::getCursorDecl(cursor);
453123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    // Don't visit synthesized ObjC methods, since they have no syntatic
453223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    // representation in the source.
453323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D)) {
453423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek      if (MD->isSynthesized())
453523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return CXChildVisit_Continue;
453623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    }
45372494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
45382494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    SourceLocation StartLoc;
453923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    if (const DeclaratorDecl *DD = dyn_cast<DeclaratorDecl>(D)) {
45402494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
45412494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
45422494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    } else if (TypedefDecl *Typedef = dyn_cast<TypedefDecl>(D)) {
45432494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
45442494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
4545a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek    }
45462494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
45472494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (StartLoc.isValid() && L.isValid() &&
45482494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        SrcMgr.isBeforeInTranslationUnit(StartLoc, L))
45492494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      cursorRange.setBegin(StartLoc);
4550a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  }
455181d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor
45523f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // If the location of the cursor occurs within a macro instantiation, record
45533f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // the spelling location of the cursor in our annotation map.  We can then
45543f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // paper over the token labelings during a post-processing step to try and
45553f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // get cursor mappings for tokens that are the *arguments* of a macro
45563f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // instantiation.
45573f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  if (L.isMacroID()) {
45583f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    unsigned rawEncoding = SrcMgr.getSpellingLoc(L).getRawEncoding();
45593f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // Only invalidate the old annotation if it isn't part of a preprocessing
45603f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // directive.  Here we assume that the default construction of CXCursor
45613f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // results in CXCursor.kind being an initialized value (i.e., 0).  If
45623f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // this isn't the case, we can fix by doing lookup + insertion.
45634419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
45643f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    CXCursor &oldC = Annotated[rawEncoding];
45653f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    if (!clang_isPreprocessing(oldC.kind))
45663f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek      oldC = cursor;
45673f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  }
45683f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek
4569fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const enum CXCursorKind K = clang_getCursorKind(parent);
4570fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const CXCursor updateC =
4571d8b0a84d586bc0a08695968acf2f169c9d01da69Ted Kremenek    (clang_isInvalid(K) || K == CXCursor_TranslationUnit)
4572d8b0a84d586bc0a08695968acf2f169c9d01da69Ted Kremenek     ? clang_getNullCursor() : parent;
4573fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4574fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  while (MoreTokens()) {
4575fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    const unsigned I = NextToken();
4576fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    SourceLocation TokLoc = GetTokenLoc(I);
4577fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
4578fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeBefore:
4579fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        Cursors[I] = updateC;
4580fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        AdvanceToken();
4581fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        continue;
4582fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeAfter:
4583fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeOverlap:
4584fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        break;
4585fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    }
4586fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    break;
4587fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4588fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4589fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Visit children to get their cursor information.
4590fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned BeforeChildren = NextToken();
4591fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  VisitChildren(cursor);
4592fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned AfterChildren = NextToken();
4593fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4594fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Adjust 'Last' to the last token within the extent of the cursor.
4595fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  while (MoreTokens()) {
4596fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    const unsigned I = NextToken();
4597fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    SourceLocation TokLoc = GetTokenLoc(I);
4598fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
4599fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeBefore:
4600fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        assert(0 && "Infeasible");
4601fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeAfter:
4602fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        break;
4603fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeOverlap:
4604fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        Cursors[I] = updateC;
4605fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        AdvanceToken();
4606fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        continue;
4607fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    }
4608fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    break;
4609fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4610fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned Last = NextToken();
46116db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
4612fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Scan the tokens that are at the beginning of the cursor, but are not
4613fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // capture by the child cursors.
4614fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4615fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // For AST elements within macros, rely on a post-annotate pass to
4616fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // to correctly annotate the tokens with cursors.  Otherwise we can
4617fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // get confusing results of having tokens that map to cursors that really
4618fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // are expanded by an instantiation.
4619fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (L.isMacroID())
4620fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    cursor = clang_getNullCursor();
4621fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4622fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = BeforeChildren; I != AfterChildren; ++I) {
4623fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    if (!clang_isInvalid(clang_getCursorKind(Cursors[I])))
4624fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      break;
46254419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
4626fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = cursor;
4627fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4628fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Scan the tokens that are at the end of the cursor, but are not captured
4629fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // but the child cursors.
4630fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = AfterChildren; I != Last; ++I)
4631fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = cursor;
4632fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4633fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  TokIdx = Last;
4634fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  return CXChildVisit_Continue;
46350045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor}
46360045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
46376db610934bedc6896393c1e1099525b35380acd6Ted Kremenekstatic enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
46386db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                                     CXCursor parent,
46396db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                                     CXClientData client_data) {
46406db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  return static_cast<AnnotateTokensWorker*>(client_data)->Visit(cursor, parent);
46416db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
46426db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
46436628a614c504263ae539462f049d523dd07ac1baTed Kremeneknamespace {
46446628a614c504263ae539462f049d523dd07ac1baTed Kremenek  struct clang_annotateTokens_Data {
46456628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXTranslationUnit TU;
46466628a614c504263ae539462f049d523dd07ac1baTed Kremenek    ASTUnit *CXXUnit;
46476628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXToken *Tokens;
46486628a614c504263ae539462f049d523dd07ac1baTed Kremenek    unsigned NumTokens;
46496628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXCursor *Cursors;
46506628a614c504263ae539462f049d523dd07ac1baTed Kremenek  };
4651ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek}
4652ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek
46536628a614c504263ae539462f049d523dd07ac1baTed Kremenek// This gets run a separate thread to avoid stack blowout.
46546628a614c504263ae539462f049d523dd07ac1baTed Kremenekstatic void clang_annotateTokensImpl(void *UserData) {
46556628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXTranslationUnit TU = ((clang_annotateTokens_Data*)UserData)->TU;
46566628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit *CXXUnit = ((clang_annotateTokens_Data*)UserData)->CXXUnit;
46576628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXToken *Tokens = ((clang_annotateTokens_Data*)UserData)->Tokens;
46586628a614c504263ae539462f049d523dd07ac1baTed Kremenek  const unsigned NumTokens = ((clang_annotateTokens_Data*)UserData)->NumTokens;
46596628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXCursor *Cursors = ((clang_annotateTokens_Data*)UserData)->Cursors;
4660fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
46610396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // Determine the region of interest, which contains all of the tokens.
46620045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  SourceRange RegionOfInterest;
46636628a614c504263ae539462f049d523dd07ac1baTed Kremenek  RegionOfInterest.setBegin(
46646628a614c504263ae539462f049d523dd07ac1baTed Kremenek    cxloc::translateSourceLocation(clang_getTokenLocation(TU, Tokens[0])));
46656628a614c504263ae539462f049d523dd07ac1baTed Kremenek  RegionOfInterest.setEnd(
46666628a614c504263ae539462f049d523dd07ac1baTed Kremenek    cxloc::translateSourceLocation(clang_getTokenLocation(TU,
46676628a614c504263ae539462f049d523dd07ac1baTed Kremenek                                                         Tokens[NumTokens-1])));
4668fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
46690396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // A mapping from the source locations found when re-lexing or traversing the
46700396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // region of interest to the corresponding cursors.
46710045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  AnnotateTokensData Annotated;
46726628a614c504263ae539462f049d523dd07ac1baTed Kremenek
4673fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Relex the tokens within the source range to look for preprocessing
46740396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // directives.
46759f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  SourceManager &SourceMgr = CXXUnit->getSourceManager();
46769f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  std::pair<FileID, unsigned> BeginLocInfo
46779f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    = SourceMgr.getDecomposedLoc(RegionOfInterest.getBegin());
46789f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  std::pair<FileID, unsigned> EndLocInfo
46799f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    = SourceMgr.getDecomposedLoc(RegionOfInterest.getEnd());
46806628a614c504263ae539462f049d523dd07ac1baTed Kremenek
46819f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  llvm::StringRef Buffer;
46820396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  bool Invalid = false;
46830396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  if (BeginLocInfo.first == EndLocInfo.first &&
46840396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor      ((Buffer = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid)),true) &&
46850396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor      !Invalid) {
46869f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
46879f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor              CXXUnit->getASTContext().getLangOptions(),
4688fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek              Buffer.begin(), Buffer.data() + BeginLocInfo.second,
46894ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor              Buffer.end());
46909f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    Lex.SetCommentRetentionState(true);
46916628a614c504263ae539462f049d523dd07ac1baTed Kremenek
4692fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    // Lex tokens in raw mode until we hit the end of the range, to avoid
46939f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    // entering #includes or expanding macros.
46944807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor    while (true) {
46959f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor      Token Tok;
46969f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor      Lex.LexFromRawLexer(Tok);
46976628a614c504263ae539462f049d523dd07ac1baTed Kremenek
46989f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    reprocess:
46999f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor      if (Tok.is(tok::hash) && Tok.isAtStartOfLine()) {
47009f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        // We have found a preprocessing directive. Gobble it up so that we
47019e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar        // don't see it while preprocessing these tokens later, but keep track
47029e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar        // of all of the token locations inside this preprocessing directive so
47039e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar        // that we can annotate them appropriately.
47049f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        //
47059f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        // FIXME: Some simple tests here could identify macro definitions and
47069f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        // #undefs, to provide specific cursor kinds for those.
47076628a614c504263ae539462f049d523dd07ac1baTed Kremenek        llvm::SmallVector<SourceLocation, 32> Locations;
47089f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        do {
47099f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor          Locations.push_back(Tok.getLocation());
4710fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek          Lex.LexFromRawLexer(Tok);
47119f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        } while (!Tok.isAtStartOfLine() && !Tok.is(tok::eof));
47126628a614c504263ae539462f049d523dd07ac1baTed Kremenek
47139f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        using namespace cxcursor;
47149f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        CXCursor Cursor
47156628a614c504263ae539462f049d523dd07ac1baTed Kremenek        = MakePreprocessingDirectiveCursor(SourceRange(Locations.front(),
47166628a614c504263ae539462f049d523dd07ac1baTed Kremenek                                                       Locations.back()),
4717a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                           TU);
47189f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        for (unsigned I = 0, N = Locations.size(); I != N; ++I) {
47199f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor          Annotated[Locations[I].getRawEncoding()] = Cursor;
47209f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        }
47216628a614c504263ae539462f049d523dd07ac1baTed Kremenek
47229f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        if (Tok.isAtStartOfLine())
47239f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor          goto reprocess;
47246628a614c504263ae539462f049d523dd07ac1baTed Kremenek
47259f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        continue;
47269f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor      }
47276628a614c504263ae539462f049d523dd07ac1baTed Kremenek
47284807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor      if (Tok.is(tok::eof))
47299f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        break;
47309f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    }
47314ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor  }
47326628a614c504263ae539462f049d523dd07ac1baTed Kremenek
47330396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // Annotate all of the source locations in the region of interest that map to
4734fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // a specific cursor.
4735fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  AnnotateTokensWorker W(Annotated, Tokens, Cursors, NumTokens,
4736a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                         TU, RegionOfInterest);
47376628a614c504263ae539462f049d523dd07ac1baTed Kremenek
47386c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // FIXME: We use a ridiculous stack size here because the data-recursion
47396c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // algorithm uses a large stack frame than the non-data recursive version,
47406c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // and AnnotationTokensWorker currently transforms the data-recursion
47416c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // algorithm back into a traditional recursion by explicitly calling
47426c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // VisitChildren().  We will need to remove this explicit recursive call.
47436628a614c504263ae539462f049d523dd07ac1baTed Kremenek  W.AnnotateTokens();
47446628a614c504263ae539462f049d523dd07ac1baTed Kremenek
4745f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  // If we ran into any entities that involve context-sensitive keywords,
4746f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  // take another pass through the tokens to mark them as such.
4747f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  if (W.hasContextSensitiveKeywords()) {
4748f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    for (unsigned I = 0; I != NumTokens; ++I) {
4749f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (clang_getTokenKind(Tokens[I]) != CXToken_Identifier)
4750f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
4751f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4752f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ObjCPropertyDecl) {
4753f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
4754f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (ObjCPropertyDecl *Property
47556628a614c504263ae539462f049d523dd07ac1baTed Kremenek            = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(Cursors[I]))) {
4756f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          if (Property->getPropertyAttributesAsWritten() != 0 &&
4757f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              llvm::StringSwitch<bool>(II->getName())
47586628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("readonly", true)
47596628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("assign", true)
47606628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("readwrite", true)
47616628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("retain", true)
47626628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("copy", true)
47636628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("nonatomic", true)
47646628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("atomic", true)
47656628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("getter", true)
47666628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("setter", true)
47676628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Default(false))
4768f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            Tokens[I].int_data[0] = CXToken_Keyword;
4769f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
4770f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
4771f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4772f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4773f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ObjCInstanceMethodDecl ||
4774f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Cursors[I].kind == CXCursor_ObjCClassMethodDecl) {
4775f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
4776f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (llvm::StringSwitch<bool>(II->getName())
47776628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("in", true)
47786628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("out", true)
47796628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("inout", true)
47806628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("oneway", true)
47816628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("bycopy", true)
47826628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("byref", true)
47836628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Default(false))
4784f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Tokens[I].int_data[0] = CXToken_Keyword;
4785f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
4786f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4787f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4788f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_CXXMethod) {
4789f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
4790f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (CXXMethodDecl *Method
47916628a614c504263ae539462f049d523dd07ac1baTed Kremenek            = dyn_cast_or_null<CXXMethodDecl>(getCursorDecl(Cursors[I]))) {
4792f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          if ((Method->hasAttr<FinalAttr>() ||
4793f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor               Method->hasAttr<OverrideAttr>()) &&
4794f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              Method->getLocation().getRawEncoding() != Tokens[I].int_data[1] &&
4795f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              llvm::StringSwitch<bool>(II->getName())
47966628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("final", true)
47976628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("override", true)
47986628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Default(false))
4799f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            Tokens[I].int_data[0] = CXToken_Keyword;
4800f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
4801f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
4802f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4803f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4804f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ClassDecl ||
4805f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Cursors[I].kind == CXCursor_StructDecl ||
4806f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Cursors[I].kind == CXCursor_ClassTemplate) {
4807f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
4808f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (II->getName() == "final") {
4809f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          // We have to be careful with 'final', since it could be the name
4810f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          // of a member class rather than the context-sensitive keyword.
4811f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          // So, check whether the cursor associated with this
4812f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Decl *D = getCursorDecl(Cursors[I]);
4813f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          if (CXXRecordDecl *Record = dyn_cast_or_null<CXXRecordDecl>(D)) {
4814f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            if ((Record->hasAttr<FinalAttr>()) &&
4815f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                Record->getIdentifier() != II)
4816f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              Tokens[I].int_data[0] = CXToken_Keyword;
4817f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          } else if (ClassTemplateDecl *ClassTemplate
48186628a614c504263ae539462f049d523dd07ac1baTed Kremenek                     = dyn_cast_or_null<ClassTemplateDecl>(D)) {
4819f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            CXXRecordDecl *Record = ClassTemplate->getTemplatedDecl();
4820f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            if ((Record->hasAttr<FinalAttr>()) &&
4821f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                Record->getIdentifier() != II)
48226628a614c504263ae539462f049d523dd07ac1baTed Kremenek              Tokens[I].int_data[0] = CXToken_Keyword;
4823f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          }
4824f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
48256628a614c504263ae539462f049d523dd07ac1baTed Kremenek        continue;
4826f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4827f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4828f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
4829fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
48306628a614c504263ae539462f049d523dd07ac1baTed Kremenek
48316628a614c504263ae539462f049d523dd07ac1baTed Kremenekextern "C" {
48326628a614c504263ae539462f049d523dd07ac1baTed Kremenek
48336628a614c504263ae539462f049d523dd07ac1baTed Kremenekvoid clang_annotateTokens(CXTranslationUnit TU,
48346628a614c504263ae539462f049d523dd07ac1baTed Kremenek                          CXToken *Tokens, unsigned NumTokens,
48356628a614c504263ae539462f049d523dd07ac1baTed Kremenek                          CXCursor *Cursors) {
48366628a614c504263ae539462f049d523dd07ac1baTed Kremenek
48376628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (NumTokens == 0 || !Tokens || !Cursors)
48386628a614c504263ae539462f049d523dd07ac1baTed Kremenek    return;
48396628a614c504263ae539462f049d523dd07ac1baTed Kremenek
48406628a614c504263ae539462f049d523dd07ac1baTed Kremenek  // Any token we don't specifically annotate will have a NULL cursor.
48416628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXCursor C = clang_getNullCursor();
48426628a614c504263ae539462f049d523dd07ac1baTed Kremenek  for (unsigned I = 0; I != NumTokens; ++I)
48436628a614c504263ae539462f049d523dd07ac1baTed Kremenek    Cursors[I] = C;
48446628a614c504263ae539462f049d523dd07ac1baTed Kremenek
48456628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
48466628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (!CXXUnit)
48476628a614c504263ae539462f049d523dd07ac1baTed Kremenek    return;
48486628a614c504263ae539462f049d523dd07ac1baTed Kremenek
48496628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
48506628a614c504263ae539462f049d523dd07ac1baTed Kremenek
48516628a614c504263ae539462f049d523dd07ac1baTed Kremenek  clang_annotateTokens_Data data = { TU, CXXUnit, Tokens, NumTokens, Cursors };
48526628a614c504263ae539462f049d523dd07ac1baTed Kremenek  llvm::CrashRecoveryContext CRC;
48536628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (!RunSafely(CRC, clang_annotateTokensImpl, &data,
48546628a614c504263ae539462f049d523dd07ac1baTed Kremenek                 GetSafetyThreadStackSize() * 2)) {
48556628a614c504263ae539462f049d523dd07ac1baTed Kremenek    fprintf(stderr, "libclang: crash detected while annotating tokens\n");
48566628a614c504263ae539462f049d523dd07ac1baTed Kremenek  }
48576628a614c504263ae539462f049d523dd07ac1baTed Kremenek}
48586628a614c504263ae539462f049d523dd07ac1baTed Kremenek
4859fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor} // end: extern "C"
4860fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4861fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor//===----------------------------------------------------------------------===//
486216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek// Operations for querying linkage of a cursor.
486316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek//===----------------------------------------------------------------------===//
486416b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
486516b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenekextern "C" {
486616b4259aecaa22b642d35d36fd89965ed700c1e0Ted KremenekCXLinkageKind clang_getCursorLinkage(CXCursor cursor) {
48670396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  if (!clang_isDeclaration(cursor.kind))
48680396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    return CXLinkage_Invalid;
48690396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor
487016b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  Decl *D = cxcursor::getCursorDecl(cursor);
487116b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  if (NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D))
487216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek    switch (ND->getLinkage()) {
487316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case NoLinkage: return CXLinkage_NoLinkage;
487416b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case InternalLinkage: return CXLinkage_Internal;
487516b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case UniqueExternalLinkage: return CXLinkage_UniqueExternal;
487616b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case ExternalLinkage: return CXLinkage_External;
487716b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek    };
487816b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
487916b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  return CXLinkage_Invalid;
488016b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek}
488116b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek} // end: extern "C"
488216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
488316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek//===----------------------------------------------------------------------===//
488445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek// Operations for querying language of a cursor.
488545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek//===----------------------------------------------------------------------===//
488645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
488745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenekstatic CXLanguageKind getDeclLanguage(const Decl *D) {
488845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  switch (D->getKind()) {
488945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    default:
489045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      break;
489145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ImplicitParam:
489245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCAtDefsField:
489345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCategory:
489445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCategoryImpl:
489545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCClass:
489645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCompatibleAlias:
489745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCForwardProtocol:
489845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCImplementation:
489945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCInterface:
490045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCIvar:
490145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCMethod:
490245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCProperty:
490345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCPropertyImpl:
490445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCProtocol:
490545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      return CXLanguage_ObjC;
490645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXConstructor:
490745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXConversion:
490845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXDestructor:
490945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXMethod:
491045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXRecord:
491145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplate:
491245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplatePartialSpecialization:
491345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplateSpecialization:
491445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Friend:
491545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::FriendTemplate:
491645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::FunctionTemplate:
491745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::LinkageSpec:
491845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Namespace:
491945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::NamespaceAlias:
492045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::NonTypeTemplateParm:
492145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::StaticAssert:
492245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::TemplateTemplateParm:
492345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::TemplateTypeParm:
492445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UnresolvedUsingTypename:
492545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UnresolvedUsingValue:
492645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Using:
492745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UsingDirective:
492845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UsingShadow:
492945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      return CXLanguage_CPlusPlus;
493045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  }
493145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
493245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  return CXLanguage_C;
493345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek}
493445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
493545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenekextern "C" {
493658ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
493758ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregorenum CXAvailabilityKind clang_getCursorAvailability(CXCursor cursor) {
493858ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor  if (clang_isDeclaration(cursor.kind))
493958ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor    if (Decl *D = cxcursor::getCursorDecl(cursor)) {
49400a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      if (isa<FunctionDecl>(D) && cast<FunctionDecl>(D)->isDeleted())
494158ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor        return CXAvailability_Available;
494258ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
49430a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      switch (D->getAvailability()) {
49440a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Available:
49450a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_NotYetIntroduced:
49460a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor        return CXAvailability_Available;
49470a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
49480a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Deprecated:
494958ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor        return CXAvailability_Deprecated;
49500a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
49510a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Unavailable:
49520a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor        return CXAvailability_NotAvailable;
49530a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      }
495458ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor    }
49550a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
495658ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor  return CXAvailability_Available;
495758ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor}
495858ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
495945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted KremenekCXLanguageKind clang_getCursorLanguage(CXCursor cursor) {
496045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  if (clang_isDeclaration(cursor.kind))
496145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    return getDeclLanguage(cxcursor::getCursorDecl(cursor));
496245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
496345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  return CXLanguage_Invalid;
496445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek}
49653910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49663910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// \brief If the given cursor is the "templated" declaration
49673910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// descibing a class or function template, return the class or
49683910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// function template.
49693910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregorstatic Decl *maybeGetTemplateCursor(Decl *D) {
49703910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (!D)
49713910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    return 0;
49723910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49733910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
49743910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    if (FunctionTemplateDecl *FunTmpl = FD->getDescribedFunctionTemplate())
49753910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return FunTmpl;
49763910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49773910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(D))
49783910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    if (ClassTemplateDecl *ClassTmpl = RD->getDescribedClassTemplate())
49793910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return ClassTmpl;
49803910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49813910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  return D;
49823910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor}
49833910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49842be5bc9ad3981347a000742f81b91ab3080f1214Douglas GregorCXCursor clang_getCursorSemanticParent(CXCursor cursor) {
49852be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isDeclaration(cursor.kind)) {
49862be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor)) {
49872be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor      DeclContext *DC = D->getDeclContext();
49883910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      if (!DC)
49893910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor        return clang_getNullCursor();
49903910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49913910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
49923910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor                          getCursorTU(cursor));
49932be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    }
49942be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
49952be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
49962be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isStatement(cursor.kind) || clang_isExpression(cursor.kind)) {
49972be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor))
4998a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(D, getCursorTU(cursor));
49992be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
50002be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
50012be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  return clang_getNullCursor();
50022be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor}
50032be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
50042be5bc9ad3981347a000742f81b91ab3080f1214Douglas GregorCXCursor clang_getCursorLexicalParent(CXCursor cursor) {
50052be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isDeclaration(cursor.kind)) {
50062be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor)) {
50072be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor      DeclContext *DC = D->getLexicalDeclContext();
50083910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      if (!DC)
50093910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor        return clang_getNullCursor();
50103910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
50113910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
50123910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor                          getCursorTU(cursor));
50132be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    }
50142be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
50152be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
50162be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  // FIXME: Note that we can't easily compute the lexical context of a
50172be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  // statement or expression, so we return nothing.
50182be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  return clang_getNullCursor();
50192be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor}
50202be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
50219f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregorstatic void CollectOverriddenMethods(DeclContext *Ctx,
50229f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                     ObjCMethodDecl *Method,
50239f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                            llvm::SmallVectorImpl<ObjCMethodDecl *> &Methods) {
50249f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!Ctx)
50259f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
50269f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50279f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  // If we have a class or category implementation, jump straight to the
50289f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  // interface.
50299f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (ObjCImplDecl *Impl = dyn_cast<ObjCImplDecl>(Ctx))
50309f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return CollectOverriddenMethods(Impl->getClassInterface(), Method, Methods);
50319f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50329f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  ObjCContainerDecl *Container = dyn_cast<ObjCContainerDecl>(Ctx);
50339f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!Container)
50349f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
50359f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50369f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  // Check whether we have a matching method at this level.
50379f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (ObjCMethodDecl *Overridden = Container->getMethod(Method->getSelector(),
50389f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                                    Method->isInstanceMethod()))
50399f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    if (Method != Overridden) {
50409f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      // We found an override at this level; there is no need to look
50419f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      // into other protocols or categories.
50429f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      Methods.push_back(Overridden);
50439f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      return;
50449f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    }
50459f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50469f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
50479f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    for (ObjCProtocolDecl::protocol_iterator P = Protocol->protocol_begin(),
50489f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                          PEnd = Protocol->protocol_end();
50499f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor         P != PEnd; ++P)
50509f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      CollectOverriddenMethods(*P, Method, Methods);
50519f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  }
50529f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50539f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(Container)) {
50549f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    for (ObjCCategoryDecl::protocol_iterator P = Category->protocol_begin(),
50559f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                          PEnd = Category->protocol_end();
50569f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor         P != PEnd; ++P)
50579f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      CollectOverriddenMethods(*P, Method, Methods);
50589f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  }
50599f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50609f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (ObjCInterfaceDecl *Interface = dyn_cast<ObjCInterfaceDecl>(Container)) {
50619f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    for (ObjCInterfaceDecl::protocol_iterator P = Interface->protocol_begin(),
50629f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                           PEnd = Interface->protocol_end();
50639f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor         P != PEnd; ++P)
50649f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      CollectOverriddenMethods(*P, Method, Methods);
50659f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50669f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    for (ObjCCategoryDecl *Category = Interface->getCategoryList();
50679f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor         Category; Category = Category->getNextClassCategory())
50689f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      CollectOverriddenMethods(Category, Method, Methods);
50699f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50709f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    // We only look into the superclass if we haven't found anything yet.
50719f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    if (Methods.empty())
50729f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      if (ObjCInterfaceDecl *Super = Interface->getSuperClass())
50739f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor        return CollectOverriddenMethods(Super, Method, Methods);
50749f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  }
50759f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor}
50769f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50779f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregorvoid clang_getOverriddenCursors(CXCursor cursor,
50789f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                CXCursor **overridden,
50799f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                unsigned *num_overridden) {
50809f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (overridden)
50819f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    *overridden = 0;
50829f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (num_overridden)
50839f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    *num_overridden = 0;
50849f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!overridden || !num_overridden)
50859f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
50869f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50879f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!clang_isDeclaration(cursor.kind))
50889f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
50899f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50909f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  Decl *D = getCursorDecl(cursor);
50919f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!D)
50929f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
50939f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50949f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  // Handle C++ member functions.
5095a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(cursor);
50969f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (CXXMethodDecl *CXXMethod = dyn_cast<CXXMethodDecl>(D)) {
50979f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    *num_overridden = CXXMethod->size_overridden_methods();
50989f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    if (!*num_overridden)
50999f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      return;
51009f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
51019f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    *overridden = new CXCursor [*num_overridden];
51029f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    unsigned I = 0;
51039f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    for (CXXMethodDecl::method_iterator
51049f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor              M = CXXMethod->begin_overridden_methods(),
51059f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor           MEnd = CXXMethod->end_overridden_methods();
51069f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor         M != MEnd; (void)++M, ++I)
5107a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      (*overridden)[I] = MakeCXCursor(const_cast<CXXMethodDecl*>(*M), TU);
51089f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
51099f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  }
51109f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
51119f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(D);
51129f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!Method)
51139f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
51149f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
51159f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  // Handle Objective-C methods.
51169f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  llvm::SmallVector<ObjCMethodDecl *, 4> Methods;
51179f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  CollectOverriddenMethods(Method->getDeclContext(), Method, Methods);
51189f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
51199f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (Methods.empty())
51209f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
51219f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
51229f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  *num_overridden = Methods.size();
51239f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  *overridden = new CXCursor [Methods.size()];
51249f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  for (unsigned I = 0, N = Methods.size(); I != N; ++I)
5125a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    (*overridden)[I] = MakeCXCursor(Methods[I], TU);
51269f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor}
51279f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
51289f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregorvoid clang_disposeOverriddenCursors(CXCursor *overridden) {
51299f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  delete [] overridden;
51309f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor}
51319f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
5132ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas GregorCXFile clang_getIncludedFile(CXCursor cursor) {
5133ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (cursor.kind != CXCursor_InclusionDirective)
5134ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return 0;
5135ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
5136ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  InclusionDirective *ID = getCursorInclusionDirective(cursor);
5137ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  return (void *)ID->getFile();
5138ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor}
5139ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
514045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek} // end: extern "C"
514145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
51429ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
51439ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek//===----------------------------------------------------------------------===//
51449ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek// C++ AST instrospection.
51459ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek//===----------------------------------------------------------------------===//
51469ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
51479ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenekextern "C" {
51489ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenekunsigned clang_CXXMethod_isStatic(CXCursor C) {
51499ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek  if (!clang_isDeclaration(C.kind))
51509ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek    return 0;
515149f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor
515249f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  CXXMethodDecl *Method = 0;
515349f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  Decl *D = cxcursor::getCursorDecl(C);
515449f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast_or_null<FunctionTemplateDecl>(D))
515549f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor    Method = dyn_cast<CXXMethodDecl>(FunTmpl->getTemplatedDecl());
515649f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  else
515749f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor    Method = dyn_cast_or_null<CXXMethodDecl>(D);
515849f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  return (Method && Method->isStatic()) ? 1 : 0;
515940b492a43bac3ed0c465772aa6921d011cfc273fTed Kremenek}
5160b12903e1a4b8d1b611b8c7e4f910665d628e68cdTed Kremenek
51619ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek} // end: extern "C"
51629ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
516345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek//===----------------------------------------------------------------------===//
516495f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek// Attribute introspection.
516595f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek//===----------------------------------------------------------------------===//
516695f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
516795f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenekextern "C" {
516895f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted KremenekCXType clang_getIBOutletCollectionType(CXCursor C) {
516995f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek  if (C.kind != CXCursor_IBOutletCollectionAttr)
5170a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return cxtype::MakeCXType(QualType(), cxcursor::getCursorTU(C));
517195f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
517295f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek  IBOutletCollectionAttr *A =
517395f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek    cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(C));
517495f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
5175841b238087d6cdb21c2443b7429cb85bd1f9fce2Douglas Gregor  return cxtype::MakeCXType(A->getInterFace(), cxcursor::getCursorTU(C));
517695f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek}
517795f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek} // end: extern "C"
517895f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
517995f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek//===----------------------------------------------------------------------===//
518004bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek// Misc. utility functions.
518104bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek//===----------------------------------------------------------------------===//
5182f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
5183abdce7abc8a22dd2fe79a05c0b71864039bd8296Daniel Dunbar/// Default to using an 8 MB stack size on "safety" threads.
5184abdce7abc8a22dd2fe79a05c0b71864039bd8296Daniel Dunbarstatic unsigned SafetyStackThreadSize = 8 << 20;
5185bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5186bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarnamespace clang {
5187bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5188bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarbool RunSafely(llvm::CrashRecoveryContext &CRC,
51896c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek               void (*Fn)(void*), void *UserData,
51906c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek               unsigned Size) {
51916c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  if (!Size)
51926c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek    Size = GetSafetyThreadStackSize();
51936c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  if (Size)
5194bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar    return CRC.RunSafelyOnThread(Fn, UserData, Size);
5195bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  return CRC.RunSafely(Fn, UserData);
5196bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5197bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5198bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarunsigned GetSafetyThreadStackSize() {
5199bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  return SafetyStackThreadSize;
5200bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5201bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5202bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarvoid SetSafetyThreadStackSize(unsigned Value) {
5203bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  SafetyStackThreadSize = Value;
5204bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5205bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5206bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5207bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
520804bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenekextern "C" {
520904bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
5210a2a9d6e4e5b6001b86b7dfc5db1ea296ce29a3d3Ted KremenekCXString clang_getClangVersion() {
5211ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString(getClangFullVersion());
521204bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek}
521304bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
521404bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek} // end: extern "C"
5215