CIndex.cpp revision 4ee9926671913ea6189ef9840a244d7c4385a7d5
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
27509dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek  bool VisitAttributes(Decl *D);
2761ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  bool VisitBlockDecl(BlockDecl *B);
2773064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  bool VisitCXXRecordDecl(CXXRecordDecl *D);
278d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  llvm::Optional<bool> shouldVisitCursor(CXCursor C);
279b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  bool VisitDeclContext(DeclContext *DC);
28079758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitTranslationUnitDecl(TranslationUnitDecl *D);
28179758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitTypedefDecl(TypedefDecl *D);
28279758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitTagDecl(TagDecl *D);
2830ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  bool VisitClassTemplateSpecializationDecl(ClassTemplateSpecializationDecl *D);
28474dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  bool VisitClassTemplatePartialSpecializationDecl(
28574dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor                                     ClassTemplatePartialSpecializationDecl *D);
286fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitTemplateTypeParmDecl(TemplateTypeParmDecl *D);
2874540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitEnumConstantDecl(EnumConstantDecl *D);
28879758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitDeclaratorDecl(DeclaratorDecl *DD);
2894540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitFunctionDecl(FunctionDecl *ND);
29079758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitFieldDecl(FieldDecl *D);
29179758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitVarDecl(VarDecl *);
29284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  bool VisitNonTypeTemplateParmDecl(NonTypeTemplateParmDecl *D);
293fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitFunctionTemplateDecl(FunctionTemplateDecl *D);
29439d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  bool VisitClassTemplateDecl(ClassTemplateDecl *D);
29584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  bool VisitTemplateTemplateParmDecl(TemplateTemplateParmDecl *D);
29679758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCMethodDecl(ObjCMethodDecl *ND);
2974540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitObjCContainerDecl(ObjCContainerDecl *D);
29879758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCCategoryDecl(ObjCCategoryDecl *ND);
29979758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCProtocolDecl(ObjCProtocolDecl *PID);
30023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  bool VisitObjCPropertyDecl(ObjCPropertyDecl *PD);
30179758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCInterfaceDecl(ObjCInterfaceDecl *D);
3024540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitObjCImplDecl(ObjCImplDecl *D);
30379758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D);
3041ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  bool VisitObjCImplementationDecl(ObjCImplementationDecl *D);
30579758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  // FIXME: ObjCCompatibleAliasDecl requires aliased-class locations.
30679758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCForwardProtocolDecl(ObjCForwardProtocolDecl *D);
30779758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCClassDecl(ObjCClassDecl *D);
308a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  bool VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *PD);
309a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  bool VisitLinkageSpecDecl(LinkageSpecDecl *D);
3108f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  bool VisitNamespaceDecl(NamespaceDecl *D);
3116931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  bool VisitNamespaceAliasDecl(NamespaceAliasDecl *D);
3120a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  bool VisitUsingDirectiveDecl(UsingDirectiveDecl *D);
3137e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  bool VisitUsingDecl(UsingDecl *D);
3147e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  bool VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D);
3157e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  bool VisitUnresolvedUsingTypenameDecl(UnresolvedUsingTypenameDecl *D);
3160a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
31701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  // Name visitor
31801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  bool VisitDeclarationNameInfo(DeclarationNameInfo Name);
319c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  bool VisitNestedNameSpecifier(NestedNameSpecifier *NNS, SourceRange Range);
320dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  bool VisitNestedNameSpecifierLoc(NestedNameSpecifierLoc NNS);
32101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
322fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // Template visitors
323fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitTemplateParameters(const TemplateParameterList *Params);
3240b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  bool VisitTemplateName(TemplateName Name, SourceLocation Loc);
325fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitTemplateArgumentLoc(const TemplateArgumentLoc &TAL);
326fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
3277d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  // Type visitors
32801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  bool VisitQualifiedTypeLoc(QualifiedTypeLoc TL);
329f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitBuiltinTypeLoc(BuiltinTypeLoc TL);
3307d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  bool VisitTypedefTypeLoc(TypedefTypeLoc TL);
331f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitUnresolvedUsingTypeLoc(UnresolvedUsingTypeLoc TL);
332f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitTagTypeLoc(TagTypeLoc TL);
333fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitTemplateTypeParmTypeLoc(TemplateTypeParmTypeLoc TL);
334f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL);
335c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  bool VisitObjCObjectTypeLoc(ObjCObjectTypeLoc TL);
336f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitObjCObjectPointerTypeLoc(ObjCObjectPointerTypeLoc TL);
337075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara  bool VisitParenTypeLoc(ParenTypeLoc TL);
338f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitPointerTypeLoc(PointerTypeLoc TL);
339f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitBlockPointerTypeLoc(BlockPointerTypeLoc TL);
340f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitMemberPointerTypeLoc(MemberPointerTypeLoc TL);
341f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitLValueReferenceTypeLoc(LValueReferenceTypeLoc TL);
342f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitRValueReferenceTypeLoc(RValueReferenceTypeLoc TL);
34301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  bool VisitFunctionTypeLoc(FunctionTypeLoc TL, bool SkipResultType = false);
344f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitArrayTypeLoc(ArrayTypeLoc TL);
345fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitTemplateSpecializationTypeLoc(TemplateSpecializationTypeLoc TL);
3462332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  // FIXME: Implement visitors here when the unimplemented TypeLocs get
3472332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  // implemented
3482332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  bool VisitTypeOfExprTypeLoc(TypeOfExprTypeLoc TL);
3497536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor  bool VisitPackExpansionTypeLoc(PackExpansionTypeLoc TL);
3502332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  bool VisitTypeOfTypeLoc(TypeOfTypeLoc TL);
3512494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor  bool VisitDependentNameTypeLoc(DependentNameTypeLoc TL);
35294fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  bool VisitDependentTemplateSpecializationTypeLoc(
35394fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor                                    DependentTemplateSpecializationTypeLoc TL);
3549e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor  bool VisitElaboratedTypeLoc(ElaboratedTypeLoc TL);
3552494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
356c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  // Data-recursive visitor functions.
357c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  bool IsInRegionOfInterest(CXCursor C);
358c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  bool RunVisitorWorkList(VisitorWorkList &WL);
359c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  void EnqueueWorkList(VisitorWorkList &WL, Stmt *S);
360cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  LLVM_ATTRIBUTE_NOINLINE bool Visit(Stmt *S);
361b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor};
362f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
363b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor} // end anonymous namespace
3640d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
365a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorstatic SourceRange getRawCursorExtent(CXCursor C);
3666653798ff5ce6deb58112777e21307ccc453133dDouglas Gregorstatic SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr);
3676653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
368a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
36933e9abd21083a0191a7676a04b497006d2da184dDouglas GregorRangeComparisonResult CursorVisitor::CompareRegionOfInterest(SourceRange R) {
370a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return RangeCompare(AU->getSourceManager(), R, RegionOfInterest);
37133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
37233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
373b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \brief Visit the given cursor and, if requested by the visitor,
374b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// its children.
375b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor///
37633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// \param Cursor the cursor to visit.
37733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor///
37833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// \param CheckRegionOfInterest if true, then the caller already checked that
37933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// this cursor is within the region of interest.
38033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor///
381b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \returns true if the visitation should be aborted, false if it
382b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// should continue.
38333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregorbool CursorVisitor::Visit(CXCursor Cursor, bool CheckedRegionOfInterest) {
384b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isInvalid(Cursor.kind))
385b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return false;
386f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
387b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isDeclaration(Cursor.kind)) {
388b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Decl *D = getCursorDecl(Cursor);
389b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    assert(D && "Invalid declaration cursor");
390b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    if (D->getPCHLevel() > MaxPCHLevel)
391b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return false;
392b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor
393b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    if (D->isImplicit())
394b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return false;
395b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
3960d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
39733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  // If we have a range of interest, and this cursor doesn't intersect with it,
39833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  // we're done.
39933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (RegionOfInterest.isValid() && !CheckedRegionOfInterest) {
400a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    SourceRange Range = getRawCursorExtent(Cursor);
401f408f32aa9ae3d97bc656267dc5d78fa7d03499bDaniel Dunbar    if (Range.isInvalid() || CompareRegionOfInterest(Range))
40233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor      return false;
40333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  }
404f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
405b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  switch (Visitor(Cursor, Parent, ClientData)) {
406b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Break:
407b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
4080d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
409b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Continue:
410b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return false;
4112e331b938b38057e333fab0ba841130ea8467794Douglas Gregor
412b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Recurse:
413b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return VisitChildren(Cursor);
414b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
4150d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
416fd64377225a6a140bddb3f997d52a036486f9360Douglas Gregor  return false;
417b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
4180d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
419788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregorstd::pair<PreprocessingRecord::iterator, PreprocessingRecord::iterator>
420788f5a1242c04762f91eaa7565c07b9865846d88Douglas GregorCursorVisitor::getPreprocessedEntities() {
421788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  PreprocessingRecord &PPRec
422a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    = *AU->getPreprocessor().getPreprocessingRecord();
423788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
424788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  bool OnlyLocalDecls
42532038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    = !AU->isMainFileAST() && AU->getOnlyLocalDecls();
42632038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor
42732038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor  if (OnlyLocalDecls && RegionOfInterest.isValid()) {
42832038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    // If we would only look at local declarations but we have a region of
42932038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    // interest, check whether that region of interest is in the main file.
43032038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    // If not, we should traverse all declarations.
43132038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    // FIXME: My kingdom for a proper binary search approach to finding
43232038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    // cursors!
43332038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    std::pair<FileID, unsigned> Location
43432038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor      = AU->getSourceManager().getDecomposedInstantiationLoc(
43532038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor                                                   RegionOfInterest.getBegin());
43632038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    if (Location.first != AU->getSourceManager().getMainFileID())
43732038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor      OnlyLocalDecls = false;
43832038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor  }
439788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
44089d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  PreprocessingRecord::iterator StartEntity, EndEntity;
44189d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  if (OnlyLocalDecls) {
44289d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor    StartEntity = AU->pp_entity_begin();
44389d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor    EndEntity = AU->pp_entity_end();
44489d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  } else {
44589d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor    StartEntity = PPRec.begin();
44689d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor    EndEntity = PPRec.end();
44789d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor  }
44889d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor
449788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  // There is no region of interest; we have to walk everything.
450788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  if (RegionOfInterest.isInvalid())
45189d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor    return std::make_pair(StartEntity, EndEntity);
452788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
453788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  // Find the file in which the region of interest lands.
454a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  SourceManager &SM = AU->getSourceManager();
455788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  std::pair<FileID, unsigned> Begin
456788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    = SM.getDecomposedInstantiationLoc(RegionOfInterest.getBegin());
457788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  std::pair<FileID, unsigned> End
458788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    = SM.getDecomposedInstantiationLoc(RegionOfInterest.getEnd());
459788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
460788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  // The region of interest spans files; we have to walk everything.
461788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  if (Begin.first != End.first)
46289d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor    return std::make_pair(StartEntity, EndEntity);
463788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
464788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  ASTUnit::PreprocessedEntitiesByFileMap &ByFileMap
465a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    = AU->getPreprocessedEntitiesByFile();
466788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  if (ByFileMap.empty()) {
467788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    // Build the mapping from files to sets of preprocessed entities.
46889d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor    for (PreprocessingRecord::iterator E = StartEntity; E != EndEntity; ++E) {
469788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor      std::pair<FileID, unsigned> P
470788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor        = SM.getDecomposedInstantiationLoc((*E)->getSourceRange().getBegin());
47189d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor
472788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor      ByFileMap[P.first].push_back(*E);
473788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    }
474788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  }
475788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
476788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  return std::make_pair(ByFileMap[Begin.first].begin(),
477788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor                        ByFileMap[Begin.first].end());
478788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor}
479788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
480b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \brief Visit the children of the given cursor.
481a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek///
482b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \returns true if the visitation should be aborted, false if it
483b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// should continue.
484f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekbool CursorVisitor::VisitChildren(CXCursor Cursor) {
485c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  if (clang_isReference(Cursor.kind) &&
486c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      Cursor.kind != CXCursor_CXXBaseSpecifier) {
487a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    // By definition, references have no children.
488a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return false;
489a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  }
490f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
491f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // Set the Parent field to Cursor, then back to its old value once we're
492b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  // done.
4930f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek  SetParentRAII SetParent(Parent, StmtParent, Cursor);
494f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
495b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isDeclaration(Cursor.kind)) {
496b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Decl *D = getCursorDecl(Cursor);
497b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    assert(D && "Invalid declaration cursor");
498539311e0221df256c70c1c3080c8af847cd29dffTed Kremenek    return VisitAttributes(D) || Visit(D);
499b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
500f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
501a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  if (clang_isStatement(Cursor.kind))
502a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return Visit(getCursorStmt(Cursor));
503a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  if (clang_isExpression(Cursor.kind))
504a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return Visit(getCursorExpr(Cursor));
505f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
506b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isTranslationUnit(Cursor.kind)) {
507a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    CXTranslationUnit tu = getCursorTU(Cursor);
508a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    ASTUnit *CXXUnit = static_cast<ASTUnit*>(tu->TUData);
50904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
51004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor    int VisitOrder[2] = { VisitPreprocessorLast, !VisitPreprocessorLast };
51104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor    for (unsigned I = 0; I != 2; ++I) {
51204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      if (VisitOrder[I]) {
51304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        if (!CXXUnit->isMainFileAST() && CXXUnit->getOnlyLocalDecls() &&
51404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            RegionOfInterest.isInvalid()) {
51504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          for (ASTUnit::top_level_iterator TL = CXXUnit->top_level_begin(),
51604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                                        TLEnd = CXXUnit->top_level_end();
51704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor               TL != TLEnd; ++TL) {
51804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            if (Visit(MakeCXCursor(*TL, tu), true))
51904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor              return true;
52004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          }
52104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        } else if (VisitDeclContext(
52204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                                CXXUnit->getASTContext().getTranslationUnitDecl()))
5237b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor          return true;
52404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        continue;
5257b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor      }
5263178cb674ac8c3b59e1791e14d38d48619a1b621Bob Wilson
52704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      // Walk the preprocessing record.
52804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      if (CXXUnit->getPreprocessor().getPreprocessingRecord()) {
52904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        // FIXME: Once we have the ability to deserialize a preprocessing record,
53004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        // do so.
53104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        PreprocessingRecord::iterator E, EEnd;
53204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        for (llvm::tie(E, EEnd) = getPreprocessedEntities(); E != EEnd; ++E) {
53304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          if (MacroInstantiation *MI = dyn_cast<MacroInstantiation>(*E)) {
53404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            if (Visit(MakeMacroInstantiationCursor(MI, tu)))
53504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor              return true;
53604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
53704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            continue;
53804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          }
5390396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor
54004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          if (MacroDefinition *MD = dyn_cast<MacroDefinition>(*E)) {
54104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            if (Visit(MakeMacroDefinitionCursor(MD, tu)))
54204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor              return true;
54304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
54404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            continue;
54504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          }
546ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
54704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          if (InclusionDirective *ID = dyn_cast<InclusionDirective>(*E)) {
54804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            if (Visit(MakeInclusionDirectiveCursor(ID, tu)))
54904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor              return true;
55004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
55104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            continue;
55204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          }
553ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor        }
5540396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor      }
5550396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    }
55604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
5577b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor    return false;
558b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
559f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
560c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  if (Cursor.kind == CXCursor_CXXBaseSpecifier) {
561c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor    if (CXXBaseSpecifier *Base = getCursorCXXBaseSpecifier(Cursor)) {
562c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      if (TypeSourceInfo *BaseTSInfo = Base->getTypeSourceInfo()) {
563c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor        return Visit(BaseTSInfo->getTypeLoc());
564c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      }
565c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor    }
566c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  }
567c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor
568b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  // Nothing to visit at the moment.
569b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
570dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
571dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
5721ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenekbool CursorVisitor::VisitBlockDecl(BlockDecl *B) {
573fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall  if (Visit(B->getSignatureAsWritten()->getTypeLoc()))
574fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall    return true;
5751ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
576664cffd330611d78fc0286f539589920a37ca328Ted Kremenek  if (Stmt *Body = B->getBody())
577664cffd330611d78fc0286f539589920a37ca328Ted Kremenek    return Visit(MakeCXCursor(Body, StmtParent, TU));
578664cffd330611d78fc0286f539589920a37ca328Ted Kremenek
579664cffd330611d78fc0286f539589920a37ca328Ted Kremenek  return false;
5801ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek}
5811ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
582d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenekllvm::Optional<bool> CursorVisitor::shouldVisitCursor(CXCursor Cursor) {
583d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (RegionOfInterest.isValid()) {
5846653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    SourceRange Range = getFullCursorExtent(Cursor, AU->getSourceManager());
585d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (Range.isInvalid())
586d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return llvm::Optional<bool>();
5876653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
588d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    switch (CompareRegionOfInterest(Range)) {
589d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeBefore:
590d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration comes before the region of interest; skip it.
591d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return llvm::Optional<bool>();
59223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
593d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeAfter:
594d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration comes after the region of interest; we're done.
595d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
596d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar
597d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeOverlap:
598d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration overlaps the region of interest; visit it.
599d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      break;
600d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
601d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
602d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  return true;
603d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek}
604f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
605d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenekbool CursorVisitor::VisitDeclContext(DeclContext *DC) {
606d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  DeclContext::decl_iterator I = DC->decls_begin(), E = DC->decls_end();
607f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
608d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually remove.  This part of a hack to support proper
609d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // iteration over all Decls contained lexically within an ObjC container.
610d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SaveAndRestore<DeclContext::decl_iterator*> DI_saved(DI_current, &I);
611d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SaveAndRestore<DeclContext::decl_iterator> DE_saved(DE_current, E);
612f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
613d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for ( ; I != E; ++I) {
614d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    Decl *D = *I;
615d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (D->getLexicalDeclContext() != DC)
616d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
617d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    CXCursor Cursor = MakeCXCursor(D, TU);
618d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    const llvm::Optional<bool> &V = shouldVisitCursor(Cursor);
619d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.hasValue())
620d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
621d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.getValue())
622d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
623d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar    if (Visit(Cursor, true))
624b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
625b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
626b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
627dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
628dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
6291ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTranslationUnitDecl(TranslationUnitDecl *D) {
6301ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  llvm_unreachable("Translation units are visited directly by Visit()");
6311ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
6321ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6331ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6341ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTypedefDecl(TypedefDecl *D) {
6351ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
6361ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(TSInfo->getTypeLoc());
637f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
6381ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
6391ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6401ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6411ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTagDecl(TagDecl *D) {
6421ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitDeclContext(D);
6431ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6441ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6450ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregorbool CursorVisitor::VisitClassTemplateSpecializationDecl(
6460ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor                                          ClassTemplateSpecializationDecl *D) {
6470ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  bool ShouldVisitBody = false;
6480ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  switch (D->getSpecializationKind()) {
6490ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_Undeclared:
6500ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ImplicitInstantiation:
6510ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    // Nothing to visit
6520ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    return false;
6530ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6540ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitInstantiationDeclaration:
6550ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitInstantiationDefinition:
6560ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    break;
6570ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6580ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitSpecialization:
6590ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    ShouldVisitBody = true;
6600ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    break;
6610ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  }
6620ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6630ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  // Visit the template arguments used in the specialization.
6640ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  if (TypeSourceInfo *SpecType = D->getTypeAsWritten()) {
6650ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    TypeLoc TL = SpecType->getTypeLoc();
6660ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    if (TemplateSpecializationTypeLoc *TSTLoc
6670ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor          = dyn_cast<TemplateSpecializationTypeLoc>(&TL)) {
6680ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor      for (unsigned I = 0, N = TSTLoc->getNumArgs(); I != N; ++I)
6690ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor        if (VisitTemplateArgumentLoc(TSTLoc->getArgLoc(I)))
6700ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor          return true;
6710ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    }
6720ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  }
6730ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6740ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  if (ShouldVisitBody && VisitCXXRecordDecl(D))
6750ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    return true;
6760ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6770ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  return false;
6780ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor}
6790ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
68074dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregorbool CursorVisitor::VisitClassTemplatePartialSpecializationDecl(
68174dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor                                   ClassTemplatePartialSpecializationDecl *D) {
68274dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // FIXME: Visit the "outer" template parameter lists on the TagDecl
68374dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // before visiting these template parameters.
68474dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
68574dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    return true;
68674dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
68774dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // Visit the partial specialization arguments.
68874dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  const TemplateArgumentLoc *TemplateArgs = D->getTemplateArgsAsWritten();
68974dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  for (unsigned I = 0, N = D->getNumTemplateArgsAsWritten(); I != N; ++I)
69074dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    if (VisitTemplateArgumentLoc(TemplateArgs[I]))
69174dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor      return true;
69274dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
69374dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  return VisitCXXRecordDecl(D);
69474dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor}
69574dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
696fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateTypeParmDecl(TemplateTypeParmDecl *D) {
69784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  // Visit the default argument.
69884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
69984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    if (TypeSourceInfo *DefArg = D->getDefaultArgumentInfo())
70084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      if (Visit(DefArg->getTypeLoc()))
70184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor        return true;
70284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
703fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
704fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
705fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
7061ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitEnumConstantDecl(EnumConstantDecl *D) {
7071ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *Init = D->getInitExpr())
7081ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(MakeCXCursor(Init, StmtParent, TU));
7091ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
7101ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
7111ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
7127d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorbool CursorVisitor::VisitDeclaratorDecl(DeclaratorDecl *DD) {
7137d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  if (TypeSourceInfo *TSInfo = DD->getTypeSourceInfo())
7147d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    if (Visit(TSInfo->getTypeLoc()))
7157d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      return true;
7167d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
717c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  // Visit the nested-name-specifier, if present.
718c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = DD->getQualifierLoc())
719c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
720c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      return true;
721c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor
7227d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  return false;
7237d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor}
7247d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
725a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor/// \brief Compare two base or member initializers based on their source order.
726cbb67480094b3bcb5b715acd827cbad55e2a204cSean Huntstatic int CompareCXXCtorInitializers(const void* Xp, const void *Yp) {
727cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt  CXXCtorInitializer const * const *X
728cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt    = static_cast<CXXCtorInitializer const * const *>(Xp);
729cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt  CXXCtorInitializer const * const *Y
730cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt    = static_cast<CXXCtorInitializer const * const *>(Yp);
731a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
732a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  if ((*X)->getSourceOrder() < (*Y)->getSourceOrder())
733a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return -1;
734a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  else if ((*X)->getSourceOrder() > (*Y)->getSourceOrder())
735a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return 1;
736a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  else
737a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return 0;
738a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor}
739a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
740b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitFunctionDecl(FunctionDecl *ND) {
74101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  if (TypeSourceInfo *TSInfo = ND->getTypeSourceInfo()) {
74201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the function declaration's syntactic components in the order
74301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // written. This requires a bit of work.
744723df245307a530da5433dfb43accf187dc3e243Abramo Bagnara    TypeLoc TL = TSInfo->getTypeLoc().IgnoreParens();
74501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    FunctionTypeLoc *FTL = dyn_cast<FunctionTypeLoc>(&TL);
74601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
74701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // If we have a function declared directly (without the use of a typedef),
74801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // visit just the return type. Otherwise, just visit the function's type
74901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // now.
75001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if ((FTL && !isa<CXXConversionDecl>(ND) && Visit(FTL->getResultLoc())) ||
75101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor        (!FTL && Visit(TL)))
75201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
75301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
754c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // Visit the nested-name-specifier, if present.
755c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (NestedNameSpecifierLoc QualifierLoc = ND->getQualifierLoc())
756c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      if (VisitNestedNameSpecifierLoc(QualifierLoc))
757c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor        return true;
75801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
75901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the declaration name.
76001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (VisitDeclarationNameInfo(ND->getNameInfo()))
76101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
76201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
76301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Visit explicitly-specified template arguments!
76401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
76501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the function parameters, if we have a function type.
76601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (FTL && VisitFunctionTypeLoc(*FTL, true))
76701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
76801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
76901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Attributes?
77001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  }
77101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
772a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  if (ND->isThisDeclarationADefinition()) {
773a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    if (CXXConstructorDecl *Constructor = dyn_cast<CXXConstructorDecl>(ND)) {
774a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Find the initializers that were written in the source.
775cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt      llvm::SmallVector<CXXCtorInitializer *, 4> WrittenInits;
776a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      for (CXXConstructorDecl::init_iterator I = Constructor->init_begin(),
777a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor                                          IEnd = Constructor->init_end();
778a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor           I != IEnd; ++I) {
779a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        if (!(*I)->isWritten())
780a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor          continue;
781a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
782a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        WrittenInits.push_back(*I);
783a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      }
784a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
785a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Sort the initializers in source order
786a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      llvm::array_pod_sort(WrittenInits.begin(), WrittenInits.end(),
787cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt                           &CompareCXXCtorInitializers);
788a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
789a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Visit the initializers in source order
790a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      for (unsigned I = 0, N = WrittenInits.size(); I != N; ++I) {
791cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt        CXXCtorInitializer *Init = WrittenInits[I];
79200eb3f9c5b33e3d99aee1f8b75dd9c9678fdd66bFrancois Pichet        if (Init->isAnyMemberInitializer()) {
79300eb3f9c5b33e3d99aee1f8b75dd9c9678fdd66bFrancois Pichet          if (Visit(MakeCursorMemberRef(Init->getAnyMember(),
794a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor                                        Init->getMemberLocation(), TU)))
795a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
796a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        } else if (TypeSourceInfo *BaseInfo = Init->getBaseClassInfo()) {
797a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor          if (Visit(BaseInfo->getTypeLoc()))
798a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
799a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        }
800a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
801a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        // Visit the initializer value.
802a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        if (Expr *Initializer = Init->getInit())
803a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor          if (Visit(MakeCXCursor(Initializer, ND, TU)))
804a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
805a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      }
806a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
807a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
808a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    if (Visit(MakeCXCursor(ND->getBody(), StmtParent, TU)))
809a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return true;
810a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  }
811f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
812b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
813b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
814dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
8151ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitFieldDecl(FieldDecl *D) {
8161ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (VisitDeclaratorDecl(D))
8171ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
818f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8191ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *BitWidth = D->getBitWidth())
8201ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(MakeCXCursor(BitWidth, StmtParent, TU));
821f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8221ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
8231ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8241ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
8251ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitVarDecl(VarDecl *D) {
8261ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (VisitDeclaratorDecl(D))
8271ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
828f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8291ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *Init = D->getInit())
8301ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(MakeCXCursor(Init, StmtParent, TU));
831f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8321ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
8331ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8341ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
83584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregorbool CursorVisitor::VisitNonTypeTemplateParmDecl(NonTypeTemplateParmDecl *D) {
83684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (VisitDeclaratorDecl(D))
83784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
83884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
83984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
84084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    if (Expr *DefArg = D->getDefaultArgument())
84184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      return Visit(MakeCXCursor(DefArg, StmtParent, TU));
84284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
84384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  return false;
84484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor}
84584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
846fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitFunctionTemplateDecl(FunctionTemplateDecl *D) {
847fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // FIXME: Visit the "outer" template parameter lists on the FunctionDecl
848fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // before visiting these template parameters.
849fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
850fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return true;
851fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
852fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return VisitFunctionDecl(D->getTemplatedDecl());
853fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
854fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
85539d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregorbool CursorVisitor::VisitClassTemplateDecl(ClassTemplateDecl *D) {
85639d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  // FIXME: Visit the "outer" template parameter lists on the TagDecl
85739d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  // before visiting these template parameters.
85839d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
85939d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor    return true;
86039d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor
86139d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  return VisitCXXRecordDecl(D->getTemplatedDecl());
86239d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor}
86339d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor
86484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregorbool CursorVisitor::VisitTemplateTemplateParmDecl(TemplateTemplateParmDecl *D) {
86584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
86684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
86784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
86884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited() &&
86984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      VisitTemplateArgumentLoc(D->getDefaultArgument()))
87084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
87184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
87284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  return false;
87384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor}
87484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
8751ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCMethodDecl(ObjCMethodDecl *ND) {
8764bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor  if (TypeSourceInfo *TSInfo = ND->getResultTypeSourceInfo())
8774bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor    if (Visit(TSInfo->getTypeLoc()))
8784bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor      return true;
8794bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor
880f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  for (ObjCMethodDecl::param_iterator P = ND->param_begin(),
8811ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       PEnd = ND->param_end();
8821ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       P != PEnd; ++P) {
8831ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    if (Visit(MakeCXCursor(*P, TU)))
8841ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
8851ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  }
886f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8871ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (ND->isThisDeclarationADefinition() &&
8881ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      Visit(MakeCXCursor(ND->getBody(), StmtParent, TU)))
8891ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
890f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8911ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
8921ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8931ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
894d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremeneknamespace {
895d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  struct ContainerDeclsSort {
896d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    SourceManager &SM;
897d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    ContainerDeclsSort(SourceManager &sm) : SM(sm) {}
898d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    bool operator()(Decl *A, Decl *B) {
899d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L_A = A->getLocStart();
900d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L_B = B->getLocStart();
901d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      assert(L_A.isValid() && L_B.isValid());
902d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return SM.isBeforeInTranslationUnit(L_A, L_B);
903d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
904d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  };
905d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek}
906d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
907a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregorbool CursorVisitor::VisitObjCContainerDecl(ObjCContainerDecl *D) {
908d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually convert back to just 'VisitDeclContext()'.  Essentially
909d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // an @implementation can lexically contain Decls that are not properly
910d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // nested in the AST.  When we identify such cases, we need to retrofit
911d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // this nesting here.
912d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (!DI_current)
913d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    return VisitDeclContext(D);
914d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
915d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Scan the Decls that immediately come after the container
916d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // in the current DeclContext.  If any fall within the
917d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // container's lexical region, stash them into a vector
918d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // for later processing.
919d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  llvm::SmallVector<Decl *, 24> DeclsInContainer;
920d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SourceLocation EndLoc = D->getSourceRange().getEnd();
921a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  SourceManager &SM = AU->getSourceManager();
922d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (EndLoc.isValid()) {
923d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    DeclContext::decl_iterator next = *DI_current;
924d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    while (++next != DE_current) {
925d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      Decl *D_next = *next;
926d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      if (!D_next)
927d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        break;
928d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L = D_next->getLocStart();
929d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      if (!L.isValid())
930d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        break;
931d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      if (SM.isBeforeInTranslationUnit(L, EndLoc)) {
932d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        *DI_current = next;
933d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        DeclsInContainer.push_back(D_next);
934d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        continue;
935d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      }
936d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      break;
937d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
938d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
939d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
940d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // The common case.
941d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (DeclsInContainer.empty())
942d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    return VisitDeclContext(D);
943d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
944d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Get all the Decls in the DeclContext, and sort them with the
945d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // additional ones we've collected.  Then visit them.
946d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for (DeclContext::decl_iterator I = D->decls_begin(), E = D->decls_end();
947d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek       I!=E; ++I) {
948d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    Decl *subDecl = *I;
9490582c897ec7261b4c6af0fe26dc2a0b6b54d266cTed Kremenek    if (!subDecl || subDecl->getLexicalDeclContext() != D ||
9500582c897ec7261b4c6af0fe26dc2a0b6b54d266cTed Kremenek        subDecl->getLocStart().isInvalid())
951d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
952d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    DeclsInContainer.push_back(subDecl);
953d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
954d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
955d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Now sort the Decls so that they appear in lexical order.
956d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  std::sort(DeclsInContainer.begin(), DeclsInContainer.end(),
957d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek            ContainerDeclsSort(SM));
958d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
959d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Now visit the decls.
960d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for (llvm::SmallVectorImpl<Decl*>::iterator I = DeclsInContainer.begin(),
961d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek         E = DeclsInContainer.end(); I != E; ++I) {
962d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    CXCursor Cursor = MakeCXCursor(*I, TU);
963d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    const llvm::Optional<bool> &V = shouldVisitCursor(Cursor);
964d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.hasValue())
965d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
966d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.getValue())
967d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
968d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (Visit(Cursor, true))
969d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return true;
970d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
971d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  return false;
972a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor}
973a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor
974b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitObjCCategoryDecl(ObjCCategoryDecl *ND) {
975b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (Visit(MakeCursorObjCClassRef(ND->getClassInterface(), ND->getLocation(),
976b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                   TU)))
977b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
978f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
97978db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  ObjCCategoryDecl::protocol_loc_iterator PL = ND->protocol_loc_begin();
98078db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  for (ObjCCategoryDecl::protocol_iterator I = ND->protocol_begin(),
98178db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor         E = ND->protocol_end(); I != E; ++I, ++PL)
982b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
983b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
984f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
985a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitObjCContainerDecl(ND);
986dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
987dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
9881ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCProtocolDecl(ObjCProtocolDecl *PID) {
9891ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  ObjCProtocolDecl::protocol_loc_iterator PL = PID->protocol_loc_begin();
9901ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCProtocolDecl::protocol_iterator I = PID->protocol_begin(),
9911ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       E = PID->protocol_end(); I != E; ++I, ++PL)
9921ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
9931ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
994f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
9951ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCContainerDecl(PID);
9961ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
9971ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
99823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenekbool CursorVisitor::VisitObjCPropertyDecl(ObjCPropertyDecl *PD) {
99983cb94269015bf2770ade71e616c5322ea7e76e1Douglas Gregor  if (PD->getTypeSourceInfo() && Visit(PD->getTypeSourceInfo()->getTypeLoc()))
1000fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall    return true;
1001fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall
100223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // FIXME: This implements a workaround with @property declarations also being
100323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // installed in the DeclContext for the @interface.  Eventually this code
100423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // should be removed.
100523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCCategoryDecl *CDecl = dyn_cast<ObjCCategoryDecl>(PD->getDeclContext());
100623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!CDecl || !CDecl->IsClassExtension())
100723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
100823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
100923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCInterfaceDecl *ID = CDecl->getClassInterface();
101023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!ID)
101123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
101223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
101323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  IdentifierInfo *PropertyId = PD->getIdentifier();
101423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCPropertyDecl *prevDecl =
101523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(ID), PropertyId);
101623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
101723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!prevDecl)
101823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
101923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
102023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // Visit synthesized methods since they will be skipped when visiting
102123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // the @interface.
102223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (ObjCMethodDecl *MD = prevDecl->getGetterMethodDecl())
1023a054fb46b1fb596d1719b89d2d9a5be3c32a4b0dTed Kremenek    if (MD->isSynthesized() && MD->getLexicalDeclContext() == CDecl)
102423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek      if (Visit(MakeCXCursor(MD, TU)))
102523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return true;
102623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
102723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (ObjCMethodDecl *MD = prevDecl->getSetterMethodDecl())
1028a054fb46b1fb596d1719b89d2d9a5be3c32a4b0dTed Kremenek    if (MD->isSynthesized() && MD->getLexicalDeclContext() == CDecl)
102923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek      if (Visit(MakeCXCursor(MD, TU)))
103023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return true;
103123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
103223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  return false;
103323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek}
103423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
1035b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitObjCInterfaceDecl(ObjCInterfaceDecl *D) {
1036dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek  // Issue callbacks for super class.
1037b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (D->getSuperClass() &&
1038b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1039f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        D->getSuperClassLoc(),
1040b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                        TU)))
1041b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
1042f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
104378db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  ObjCInterfaceDecl::protocol_loc_iterator PL = D->protocol_loc_begin();
104478db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  for (ObjCInterfaceDecl::protocol_iterator I = D->protocol_begin(),
104578db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor         E = D->protocol_end(); I != E; ++I, ++PL)
1046b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1047b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
1048f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1049a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitObjCContainerDecl(D);
1050dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1051dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
10521ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCImplDecl(ObjCImplDecl *D) {
10531ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCContainerDecl(D);
10541ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
10551ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
10561ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D) {
1057ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  // 'ID' could be null when dealing with invalid code.
1058ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  if (ObjCInterfaceDecl *ID = D->getClassInterface())
1059ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek    if (Visit(MakeCursorObjCClassRef(ID, D->getLocation(), TU)))
1060ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek      return true;
1061f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10621ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCImplDecl(D);
10631ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
10641ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
10651ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCImplementationDecl(ObjCImplementationDecl *D) {
10661ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor#if 0
10671ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // Issue callbacks for super class.
10681ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // FIXME: No source location information!
10691ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (D->getSuperClass() &&
10701ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1071f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        D->getSuperClassLoc(),
10721ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor                                        TU)))
1073a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return true;
10741ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor#endif
1075f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10761ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCImplDecl(D);
1077dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1078dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
10791ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCForwardProtocolDecl(ObjCForwardProtocolDecl *D) {
10801ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  ObjCForwardProtocolDecl::protocol_loc_iterator PL = D->protocol_loc_begin();
10811ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCForwardProtocolDecl::protocol_iterator I = D->protocol_begin(),
10821ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor                                                  E = D->protocol_end();
10831ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       I != E; ++I, ++PL)
1084b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1085b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
1086f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1087f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return false;
1088dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1089dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
10901ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCClassDecl(ObjCClassDecl *D) {
10911ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCClassDecl::iterator C = D->begin(), CEnd = D->end(); C != CEnd; ++C)
10921ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    if (Visit(MakeCursorObjCClassRef(C->getInterface(), C->getLocation(), TU)))
10931ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
1094f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10951ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
1096dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
10975e4bc590b0ea010e38372d0b4a0aab578a746fe6Benjamin Kramer
1098a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregorbool CursorVisitor::VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *PD) {
1099a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  if (ObjCIvarDecl *Ivar = PD->getPropertyIvarDecl())
1100a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor    return Visit(MakeCursorMemberRef(Ivar, PD->getPropertyIvarDeclLoc(), TU));
1101a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor
1102a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  return false;
1103a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor}
1104a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor
11058f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenekbool CursorVisitor::VisitNamespaceDecl(NamespaceDecl *D) {
11068f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  return VisitDeclContext(D);
11078f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek}
11088f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek
11096931900f43cea558c6974075256c07728dbfecc6Douglas Gregorbool CursorVisitor::VisitNamespaceAliasDecl(NamespaceAliasDecl *D) {
1110c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
11110cfaf6a270ecd0f5c7e541a8047c87948317548bDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
11120cfaf6a270ecd0f5c7e541a8047c87948317548bDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1113c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
11146931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
11156931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  return Visit(MakeCursorNamespaceRef(D->getAliasedNamespace(),
11166931900f43cea558c6974075256c07728dbfecc6Douglas Gregor                                      D->getTargetNameLoc(), TU));
11176931900f43cea558c6974075256c07728dbfecc6Douglas Gregor}
11186931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
11197e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUsingDecl(UsingDecl *D) {
1120c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1121dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1122dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1123c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1124dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
11257e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
11261f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (Visit(MakeCursorOverloadedDeclRef(D, D->getLocation(), TU)))
11271f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return true;
11281f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
11297e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return VisitDeclarationNameInfo(D->getNameInfo());
11307e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11317e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
11320a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregorbool CursorVisitor::VisitUsingDirectiveDecl(UsingDirectiveDecl *D) {
1133c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1134db9924191092b4d426cc066637d81698211846aaDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1135db9924191092b4d426cc066637d81698211846aaDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1136c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
11370a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
11380a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  return Visit(MakeCursorNamespaceRef(D->getNominatedNamespaceAsWritten(),
11390a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor                                      D->getIdentLocation(), TU));
11400a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor}
11410a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
11427e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D) {
1143c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1144dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1145dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1146c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1147dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
1148c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
11497e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return VisitDeclarationNameInfo(D->getNameInfo());
11507e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11517e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
11527e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUnresolvedUsingTypenameDecl(
11537e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor                                               UnresolvedUsingTypenameDecl *D) {
1154c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1155dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1156dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1157c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1158c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
11597e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return false;
11607e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11617e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
116201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitDeclarationNameInfo(DeclarationNameInfo Name) {
116301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  switch (Name.getName().getNameKind()) {
116401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::Identifier:
116501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXLiteralOperatorName:
116601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXOperatorName:
116701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXUsingDirective:
116801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
116901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
117001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXConstructorName:
117101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXDestructorName:
117201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXConversionFunctionName:
117301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (TypeSourceInfo *TSInfo = Name.getNamedTypeInfo())
117401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return Visit(TSInfo->getTypeLoc());
117501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
117601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
117701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCZeroArgSelector:
117801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCOneArgSelector:
117901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCMultiArgSelector:
118001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Per-identifier location info?
118101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
118201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  }
118301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
118401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  return false;
118501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor}
118601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
1187c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregorbool CursorVisitor::VisitNestedNameSpecifier(NestedNameSpecifier *NNS,
1188c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                             SourceRange Range) {
1189c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // FIXME: This whole routine is a hack to work around the lack of proper
1190c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // source information in nested-name-specifiers (PR5791). Since we do have
1191c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // a beginning source location, we can visit the first component of the
1192c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // nested-name-specifier, if it's a single-token component.
1193c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  if (!NNS)
1194c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    return false;
1195c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1196c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Get the first component in the nested-name-specifier.
1197c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  while (NestedNameSpecifier *Prefix = NNS->getPrefix())
1198c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    NNS = Prefix;
1199c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1200c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  switch (NNS->getKind()) {
1201c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Namespace:
1202c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    return Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(), Range.getBegin(),
1203c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                        TU));
1204c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
120514aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor  case NestedNameSpecifier::NamespaceAlias:
120614aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor    return Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
120714aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor                                        Range.getBegin(), TU));
120814aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor
1209c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::TypeSpec: {
1210c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // If the type has a form where we know that the beginning of the source
1211c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // range matches up with a reference cursor. Visit the appropriate reference
1212c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // cursor.
1213f4c7371fb1d3cebcfb40abad4537bb82515704eaJohn McCall    const Type *T = NNS->getAsType();
1214c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TypedefType *Typedef = dyn_cast<TypedefType>(T))
1215c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return Visit(MakeCursorTypeRef(Typedef->getDecl(), Range.getBegin(), TU));
1216c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TagType *Tag = dyn_cast<TagType>(T))
1217c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return Visit(MakeCursorTypeRef(Tag->getDecl(), Range.getBegin(), TU));
1218c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TemplateSpecializationType *TST
1219c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                      = dyn_cast<TemplateSpecializationType>(T))
1220c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return VisitTemplateName(TST->getTemplateName(), Range.getBegin());
1221c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    break;
1222c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  }
1223c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1224c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::TypeSpecWithTemplate:
1225c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Global:
1226c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Identifier:
1227c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    break;
1228c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  }
1229c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1230c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  return false;
1231c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor}
1232c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1233dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregorbool
1234dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas GregorCursorVisitor::VisitNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
1235dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  llvm::SmallVector<NestedNameSpecifierLoc, 4> Qualifiers;
1236dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  for (; Qualifier; Qualifier = Qualifier.getPrefix())
1237dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    Qualifiers.push_back(Qualifier);
1238dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1239dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  while (!Qualifiers.empty()) {
1240dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    NestedNameSpecifierLoc Q = Qualifiers.pop_back_val();
1241dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    NestedNameSpecifier *NNS = Q.getNestedNameSpecifier();
1242dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    switch (NNS->getKind()) {
1243dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Namespace:
1244dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(),
1245c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor                                       Q.getLocalBeginLoc(),
1246dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor                                       TU)))
1247dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1248dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1249dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1250dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1251dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::NamespaceAlias:
1252dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
1253c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor                                       Q.getLocalBeginLoc(),
1254dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor                                       TU)))
1255dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1256dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1257dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1258dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1259dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::TypeSpec:
1260dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::TypeSpecWithTemplate:
1261dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(Q.getTypeLoc()))
1262dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1263dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1264dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1265dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1266dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Global:
1267dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Identifier:
1268dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1269dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    }
1270dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
1271dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1272dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  return false;
1273dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor}
1274dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1275fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateParameters(
1276fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                          const TemplateParameterList *Params) {
1277fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  if (!Params)
1278fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1279fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1280fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  for (TemplateParameterList::const_iterator P = Params->begin(),
1281fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                          PEnd = Params->end();
1282fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor       P != PEnd; ++P) {
1283fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Visit(MakeCXCursor(*P, TU)))
1284fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return true;
1285fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  }
1286fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1287fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1288fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1289fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
12900b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregorbool CursorVisitor::VisitTemplateName(TemplateName Name, SourceLocation Loc) {
12910b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  switch (Name.getKind()) {
12920b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::Template:
12930b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return Visit(MakeCursorTemplateRef(Name.getAsTemplateDecl(), Loc, TU));
12940b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
12950b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::OverloadedTemplate:
12961f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    // Visit the overloaded template set.
12971f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (Visit(MakeCursorOverloadedDeclRef(Name, Loc, TU)))
12981f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return true;
12991f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
13000b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return false;
13010b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
13020b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::DependentTemplate:
13030b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    // FIXME: Visit nested-name-specifier.
13040b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return false;
13050b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
13060b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::QualifiedTemplate:
13070b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    // FIXME: Visit nested-name-specifier.
13080b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return Visit(MakeCursorTemplateRef(
13090b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                                  Name.getAsQualifiedTemplateName()->getDecl(),
13100b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                                       Loc, TU));
13111aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor
13121aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor  case TemplateName::SubstTemplateTemplateParmPack:
13131aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor    return Visit(MakeCursorTemplateRef(
13141aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor                  Name.getAsSubstTemplateTemplateParmPack()->getParameterPack(),
13151aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor                                       Loc, TU));
13160b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  }
13170b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
13180b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  return false;
13190b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor}
13200b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
1321fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateArgumentLoc(const TemplateArgumentLoc &TAL) {
1322fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  switch (TAL.getArgument().getKind()) {
1323fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Null:
1324fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Integral:
1325fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Pack:
1326fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
132787dd697dcc8ecb64df73ae64d61b8c80ff0c157cDouglas Gregor
1328fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Type:
1329fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (TypeSourceInfo *TSInfo = TAL.getTypeSourceInfo())
1330fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return Visit(TSInfo->getTypeLoc());
1331fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1332fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1333fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Declaration:
1334fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Expr *E = TAL.getSourceDeclExpression())
1335fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return Visit(MakeCXCursor(E, StmtParent, TU));
1336fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1337fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1338fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Expression:
1339fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Expr *E = TAL.getSourceExpression())
1340fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return Visit(MakeCXCursor(E, StmtParent, TU));
1341fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1342fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1343fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Template:
1344a7fc901a2e39bfe55bfcff5934b2d9fdf9656491Douglas Gregor  case TemplateArgument::TemplateExpansion:
1345b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor    if (VisitNestedNameSpecifierLoc(TAL.getTemplateQualifierLoc()))
1346b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor      return true;
1347b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor
1348a7fc901a2e39bfe55bfcff5934b2d9fdf9656491Douglas Gregor    return VisitTemplateName(TAL.getArgument().getAsTemplateOrTemplatePattern(),
13490b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                             TAL.getTemplateNameLoc());
1350fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  }
1351fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1352fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1353fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1354fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1355a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenekbool CursorVisitor::VisitLinkageSpecDecl(LinkageSpecDecl *D) {
1356a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  return VisitDeclContext(D);
1357a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek}
1358a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek
135901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitQualifiedTypeLoc(QualifiedTypeLoc TL) {
136001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  return Visit(TL.getUnqualifiedLoc());
136101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor}
136201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
1363f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitBuiltinTypeLoc(BuiltinTypeLoc TL) {
1364a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTContext &Context = AU->getASTContext();
1365f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1366f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // Some builtin types (such as Objective-C's "id", "sel", and
1367f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // "Class") have associated declarations. Create cursors for those.
1368f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  QualType VisitType;
1369f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  switch (TL.getType()->getAs<BuiltinType>()->getKind()) {
13706b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Void:
1371f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Bool:
13726b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Char_U:
13736b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UChar:
1374f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Char16:
1375f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Char32:
13766b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UShort:
13776b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UInt:
13786b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ULong:
13796b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ULongLong:
13806b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UInt128:
1381f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Char_S:
13826b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::SChar:
13833f59c975aa5d047f7edd1b900b5e885c38af0ef7Chris Lattner  case BuiltinType::WChar_U:
13843f59c975aa5d047f7edd1b900b5e885c38af0ef7Chris Lattner  case BuiltinType::WChar_S:
13856b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Short:
1386f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Int:
1387f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Long:
1388c4174cc4b9b657abb77d0825de473ea29cf48297Ted Kremenek  case BuiltinType::LongLong:
13896b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Int128:
13906b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Float:
13916b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Double:
13926b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::LongDouble:
1393f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::NullPtr:
1394f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Overload:
13956b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Dependent:
1396f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
13976b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
1398f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::ObjCId:
1399f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    VisitType = Context.getObjCIdType();
1400f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
14016b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
14026b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ObjCClass:
14036b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek    VisitType = Context.getObjCClassType();
14046b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek    break;
14056b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
1406f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::ObjCSel:
1407f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    VisitType = Context.getObjCSelType();
1408f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
1409f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1410f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1411f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (!VisitType.isNull()) {
1412f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    if (const TypedefType *Typedef = VisitType->getAs<TypedefType>())
1413f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek      return Visit(MakeCursorTypeRef(Typedef->getDecl(), TL.getBuiltinLoc(),
1414f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                     TU));
1415f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1416f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1417f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1418f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1419f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
14207d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorbool CursorVisitor::VisitTypedefTypeLoc(TypedefTypeLoc TL) {
14217d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getTypedefDecl(), TL.getNameLoc(), TU));
14227d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor}
14237d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
1424f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitUnresolvedUsingTypeLoc(UnresolvedUsingTypeLoc TL) {
1425f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1426f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1427f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1428f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitTagTypeLoc(TagTypeLoc TL) {
1429f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1430f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1431f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1432fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateTypeParmTypeLoc(TemplateTypeParmTypeLoc TL) {
14331f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  // FIXME: We can't visit the template type parameter, because there's
1434fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // no context information with which we can match up the depth/index in the
1435fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // type to the appropriate
1436fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1437fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1438fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1439f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL) {
1440f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(MakeCursorObjCClassRef(TL.getIFaceDecl(), TL.getNameLoc(), TU)))
1441f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1442f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1443c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  return false;
1444c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall}
1445c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall
1446c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCallbool CursorVisitor::VisitObjCObjectTypeLoc(ObjCObjectTypeLoc TL) {
1447c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  if (TL.hasBaseTypeAsWritten() && Visit(TL.getBaseLoc()))
1448c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall    return true;
1449c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall
1450f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  for (unsigned I = 0, N = TL.getNumProtocols(); I != N; ++I) {
1451f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    if (Visit(MakeCursorObjCProtocolRef(TL.getProtocol(I), TL.getProtocolLoc(I),
1452f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                        TU)))
1453f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor      return true;
1454f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1455f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1456f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1457f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1458f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1459f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitObjCObjectPointerTypeLoc(ObjCObjectPointerTypeLoc TL) {
1460c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  return Visit(TL.getPointeeLoc());
1461f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1462f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1463075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnarabool CursorVisitor::VisitParenTypeLoc(ParenTypeLoc TL) {
1464075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara  return Visit(TL.getInnerLoc());
1465075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara}
1466075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara
1467f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitPointerTypeLoc(PointerTypeLoc TL) {
1468f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1469f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1470f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1471f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitBlockPointerTypeLoc(BlockPointerTypeLoc TL) {
1472f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1473f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1474f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1475f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitMemberPointerTypeLoc(MemberPointerTypeLoc TL) {
1476f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1477f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1478f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1479f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitLValueReferenceTypeLoc(LValueReferenceTypeLoc TL) {
1480f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return Visit(TL.getPointeeLoc());
1481f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1482f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1483f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitRValueReferenceTypeLoc(RValueReferenceTypeLoc TL) {
1484f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return Visit(TL.getPointeeLoc());
1485f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1486f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
148701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitFunctionTypeLoc(FunctionTypeLoc TL,
148801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor                                         bool SkipResultType) {
148901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  if (!SkipResultType && Visit(TL.getResultLoc()))
1490f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1491f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1492f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
14935dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek    if (Decl *D = TL.getArg(I))
14945dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek      if (Visit(MakeCXCursor(D, TU)))
14955dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek        return true;
1496f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1497f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1498f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1499f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1500f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitArrayTypeLoc(ArrayTypeLoc TL) {
1501f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(TL.getElementLoc()))
1502f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1503f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1504f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Expr *Size = TL.getSizeExpr())
1505f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return Visit(MakeCXCursor(Size, StmtParent, TU));
1506f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1507f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1508f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1509f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1510fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateSpecializationTypeLoc(
1511fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                             TemplateSpecializationTypeLoc TL) {
15120b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  // Visit the template name.
15130b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  if (VisitTemplateName(TL.getTypePtr()->getTemplateName(),
15140b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                        TL.getTemplateNameLoc()))
15150b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return true;
1516fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1517fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // Visit the template arguments.
1518fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
1519fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
1520fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return true;
1521fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1522fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1523fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1524fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
15252332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregorbool CursorVisitor::VisitTypeOfExprTypeLoc(TypeOfExprTypeLoc TL) {
15262332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  return Visit(MakeCXCursor(TL.getUnderlyingExpr(), StmtParent, TU));
15272332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor}
15282332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15292332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregorbool CursorVisitor::VisitTypeOfTypeLoc(TypeOfTypeLoc TL) {
15302332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
15312332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor    return Visit(TSInfo->getTypeLoc());
15322332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15332332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  return false;
15342332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor}
15352332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15362494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregorbool CursorVisitor::VisitDependentNameTypeLoc(DependentNameTypeLoc TL) {
15372494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor  if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
15382494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    return true;
15392494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
15402494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor  return false;
15412494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor}
15422494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
154394fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregorbool CursorVisitor::VisitDependentTemplateSpecializationTypeLoc(
154494fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor                                    DependentTemplateSpecializationTypeLoc TL) {
154594fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  // Visit the nested-name-specifier, if there is one.
154694fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  if (TL.getQualifierLoc() &&
154794fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor      VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
154894fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor    return true;
154994fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
155094fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  // Visit the template arguments.
155194fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
155294fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor    if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
155394fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor      return true;
155494fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
155594fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  return false;
155694fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor}
155794fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
15589e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregorbool CursorVisitor::VisitElaboratedTypeLoc(ElaboratedTypeLoc TL) {
15599e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor  if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
15609e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor    return true;
15619e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor
15629e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor  return Visit(TL.getNamedTypeLoc());
15639e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor}
15649e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor
15657536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregorbool CursorVisitor::VisitPackExpansionTypeLoc(PackExpansionTypeLoc TL) {
15667536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor  return Visit(TL.getPatternLoc());
15677536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor}
15687536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor
15693064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenekbool CursorVisitor::VisitCXXRecordDecl(CXXRecordDecl *D) {
1570c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  // Visit the nested-name-specifier, if present.
1571c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1572c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1573c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      return true;
1574c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor
15753064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  if (D->isDefinition()) {
15763064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    for (CXXRecordDecl::base_class_iterator I = D->bases_begin(),
15773064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek         E = D->bases_end(); I != E; ++I) {
15783064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      if (Visit(cxcursor::MakeCursorCXXBaseSpecifier(I, TU)))
15793064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek        return true;
15803064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
15813064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  }
15823064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek
15833064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  return VisitTagDecl(D);
15843064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek}
15853064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek
158609dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenekbool CursorVisitor::VisitAttributes(Decl *D) {
1587cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt  for (AttrVec::const_iterator i = D->attr_begin(), e = D->attr_end();
1588cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt       i != e; ++i)
1589cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt    if (Visit(MakeCXCursor(*i, D, TU)))
159009dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek        return true;
159109dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
159209dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek  return false;
159309dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek}
159409dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
1595c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
1596c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek// Data-recursive visitor methods.
1597c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
1598c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
159928a719433411ef782b582946823bc648ddcc4533Ted Kremeneknamespace {
1600035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek#define DEF_JOB(NAME, DATA, KIND)\
1601035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass NAME : public VisitorJob {\
1602035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:\
1603035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  NAME(DATA *d, CXCursor parent) : VisitorJob(parent, VisitorJob::KIND, d) {} \
1604035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) { return VJ->getKind() == KIND; }\
1605f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DATA *get() const { return static_cast<DATA*>(data[0]); }\
1606035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1607035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1608035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(StmtVisit, Stmt, StmtVisitKind)
1609035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(MemberExprParts, MemberExpr, MemberExprPartsKind)
1610e4979ccb5960608edce73f3b274eb7c2de15dac5Ted KremenekDEF_JOB(DeclRefExprParts, DeclRefExpr, DeclRefExprPartsKind)
1611035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(OverloadExprParts, OverloadExpr, OverloadExprPartsKind)
161260608ec12d17168a3d1f415409a6a6eaf6d94508Ted KremenekDEF_JOB(ExplicitTemplateArgsVisit, ExplicitTemplateArgumentList,
161360608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        ExplicitTemplateArgsVisitKind)
161494d96291cd041adc5731a2294828a9c20e450b74Douglas GregorDEF_JOB(SizeOfPackExprParts, SizeOfPackExpr, SizeOfPackExprPartsKind)
1615035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek#undef DEF_JOB
1616035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1617035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass DeclVisit : public VisitorJob {
1618035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:
1619035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  DeclVisit(Decl *d, CXCursor parent, bool isFirst) :
1620035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    VisitorJob(parent, VisitorJob::DeclVisitKind,
1621035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek               d, isFirst ? (void*) 1 : (void*) 0) {}
1622035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) {
162382f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    return VJ->getKind() == DeclVisitKind;
1624035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1625f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  Decl *get() const { return static_cast<Decl*>(data[0]); }
1626f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  bool isFirst() const { return data[1] ? true : false; }
1627035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1628035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass TypeLocVisit : public VisitorJob {
1629035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:
1630035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  TypeLocVisit(TypeLoc tl, CXCursor parent) :
1631035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    VisitorJob(parent, VisitorJob::TypeLocVisitKind,
1632035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek               tl.getType().getAsOpaquePtr(), tl.getOpaqueData()) {}
1633035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1634035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1635035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    return VJ->getKind() == TypeLocVisitKind;
1636035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1637035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
163882f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek  TypeLoc get() const {
1639f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    QualType T = QualType::getFromOpaquePtr(data[0]);
1640f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return TypeLoc(T, data[1]);
1641035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1642035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1643035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1644ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekclass LabelRefVisit : public VisitorJob {
1645ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekpublic:
1646ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  LabelRefVisit(LabelDecl *LD, SourceLocation labelLoc, CXCursor parent)
1647ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner    : VisitorJob(parent, VisitorJob::LabelRefVisitKind, LD,
1648dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 labelLoc.getPtrEncoding()) {}
1649ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek
1650ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1651ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek    return VJ->getKind() == VisitorJob::LabelRefVisitKind;
1652ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  }
1653ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  LabelDecl *get() const { return static_cast<LabelDecl*>(data[0]); }
1654ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  SourceLocation getLoc() const {
1655dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin    return SourceLocation::getFromPtrEncoding(data[1]); }
1656f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek};
1657f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekclass NestedNameSpecifierVisit : public VisitorJob {
1658f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekpublic:
1659f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  NestedNameSpecifierVisit(NestedNameSpecifier *NS, SourceRange R,
1660f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek                           CXCursor parent)
1661f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    : VisitorJob(parent, VisitorJob::NestedNameSpecifierVisitKind,
1662dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 NS, R.getBegin().getPtrEncoding(),
1663dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 R.getEnd().getPtrEncoding()) {}
1664f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1665f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return VJ->getKind() == VisitorJob::NestedNameSpecifierVisitKind;
1666f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1667f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  NestedNameSpecifier *get() const {
1668f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return static_cast<NestedNameSpecifier*>(data[0]);
1669f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1670f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  SourceRange getSourceRange() const {
1671f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    SourceLocation A =
1672f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      SourceLocation::getFromRawEncoding((unsigned)(uintptr_t) data[1]);
1673f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    SourceLocation B =
1674f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      SourceLocation::getFromRawEncoding((unsigned)(uintptr_t) data[2]);
1675f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return SourceRange(A, B);
1676f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1677f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek};
1678f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1679f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorclass NestedNameSpecifierLocVisit : public VisitorJob {
1680f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorpublic:
1681f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  NestedNameSpecifierLocVisit(NestedNameSpecifierLoc Qualifier, CXCursor parent)
1682f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    : VisitorJob(parent, VisitorJob::NestedNameSpecifierLocVisitKind,
1683f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                 Qualifier.getNestedNameSpecifier(),
1684f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                 Qualifier.getOpaqueData()) { }
1685f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1686f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  static bool classof(const VisitorJob *VJ) {
1687f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    return VJ->getKind() == VisitorJob::NestedNameSpecifierLocVisitKind;
1688f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  }
1689f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1690f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  NestedNameSpecifierLoc get() const {
1691f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    return NestedNameSpecifierLoc(static_cast<NestedNameSpecifier*>(data[0]),
1692f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                                  data[1]);
1693f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  }
1694f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor};
1695f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1696f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekclass DeclarationNameInfoVisit : public VisitorJob {
1697f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekpublic:
1698f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DeclarationNameInfoVisit(Stmt *S, CXCursor parent)
1699f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    : VisitorJob(parent, VisitorJob::DeclarationNameInfoVisitKind, S) {}
1700f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1701f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return VJ->getKind() == VisitorJob::DeclarationNameInfoVisitKind;
1702f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1703f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DeclarationNameInfo get() const {
1704f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    Stmt *S = static_cast<Stmt*>(data[0]);
1705f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    switch (S->getStmtClass()) {
1706f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    default:
1707f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      llvm_unreachable("Unhandled Stmt");
1708f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    case Stmt::CXXDependentScopeMemberExprClass:
1709f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      return cast<CXXDependentScopeMemberExpr>(S)->getMemberNameInfo();
1710f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    case Stmt::DependentScopeDeclRefExprClass:
1711f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      return cast<DependentScopeDeclRefExpr>(S)->getNameInfo();
1712f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    }
1713f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1714ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek};
1715cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekclass MemberRefVisit : public VisitorJob {
1716cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekpublic:
1717cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  MemberRefVisit(FieldDecl *D, SourceLocation L, CXCursor parent)
1718cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    : VisitorJob(parent, VisitorJob::MemberRefVisitKind, D,
1719dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 L.getPtrEncoding()) {}
1720cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  static bool classof(const VisitorJob *VJ) {
1721cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return VJ->getKind() == VisitorJob::MemberRefVisitKind;
1722cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1723cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  FieldDecl *get() const {
1724cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return static_cast<FieldDecl*>(data[0]);
1725cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1726cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  SourceLocation getLoc() const {
1727cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return SourceLocation::getFromRawEncoding((unsigned)(uintptr_t) data[1]);
1728cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1729cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek};
173028a719433411ef782b582946823bc648ddcc4533Ted Kremenekclass EnqueueVisitor : public StmtVisitor<EnqueueVisitor, void> {
173128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  VisitorWorkList &WL;
173228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  CXCursor Parent;
173328a719433411ef782b582946823bc648ddcc4533Ted Kremenekpublic:
173428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueVisitor(VisitorWorkList &wl, CXCursor parent)
173528a719433411ef782b582946823bc648ddcc4533Ted Kremenek    : WL(wl), Parent(parent) {}
173628a719433411ef782b582946823bc648ddcc4533Ted Kremenek
1737ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  void VisitAddrLabelExpr(AddrLabelExpr *E);
173873d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitBlockExpr(BlockExpr *B);
173928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
1740083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  void VisitCompoundStmt(CompoundStmt *S);
174111b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  void VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) { /* Do nothing. */ }
1742f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E);
174311b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  void VisitCXXNewExpr(CXXNewExpr *E);
17446d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  void VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E);
174528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E);
1746cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E);
174773d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E);
1748b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  void VisitCXXTypeidExpr(CXXTypeidExpr *E);
174955b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  void VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E);
17501e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  void VisitCXXUuidofExpr(CXXUuidofExpr *E);
1751e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek  void VisitDeclRefExpr(DeclRefExpr *D);
1752035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  void VisitDeclStmt(DeclStmt *S);
1753f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E);
1754cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitDesignatedInitExpr(DesignatedInitExpr *E);
175528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitExplicitCastExpr(ExplicitCastExpr *E);
175628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitForStmt(ForStmt *FS);
1757ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  void VisitGotoStmt(GotoStmt *GS);
175828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitIfStmt(IfStmt *If);
175928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitInitListExpr(InitListExpr *IE);
176028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitMemberExpr(MemberExpr *M);
1761cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitOffsetOfExpr(OffsetOfExpr *E);
176273d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitObjCEncodeExpr(ObjCEncodeExpr *E);
176328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitObjCMessageExpr(ObjCMessageExpr *M);
176428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitOverloadExpr(OverloadExpr *E);
1765f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbourne  void VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E);
176628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitStmt(Stmt *S);
176728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitSwitchStmt(SwitchStmt *S);
176828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitWhileStmt(WhileStmt *W);
17692939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek  void VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E);
17706ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  void VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E);
177128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitUnresolvedMemberExpr(UnresolvedMemberExpr *U);
17729d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  void VisitVAArgExpr(VAArgExpr *E);
177394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  void VisitSizeOfPackExpr(SizeOfPackExpr *E);
1774ee8aff06f6a96214731de17b2cb6df407c6c1820Douglas Gregor
177528a719433411ef782b582946823bc648ddcc4533Ted Kremenekprivate:
1776f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void AddDeclarationNameInfo(Stmt *S);
1777f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void AddNestedNameSpecifier(NestedNameSpecifier *NS, SourceRange R);
1778f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  void AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier);
177960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  void AddExplicitTemplateArgs(const ExplicitTemplateArgumentList *A);
1780cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void AddMemberRef(FieldDecl *D, SourceLocation L);
178128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void AddStmt(Stmt *S);
1782035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  void AddDecl(Decl *D, bool isFirst = true);
178328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void AddTypeLoc(TypeSourceInfo *TI);
178428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void EnqueueChildren(Stmt *S);
178528a719433411ef782b582946823bc648ddcc4533Ted Kremenek};
178628a719433411ef782b582946823bc648ddcc4533Ted Kremenek} // end anonyous namespace
178728a719433411ef782b582946823bc648ddcc4533Ted Kremenek
1788f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::AddDeclarationNameInfo(Stmt *S) {
1789f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  // 'S' should always be non-null, since it comes from the
1790f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  // statement we are visiting.
1791f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  WL.push_back(DeclarationNameInfoVisit(S, Parent));
1792f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
1793f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::AddNestedNameSpecifier(NestedNameSpecifier *N,
1794f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek                                            SourceRange R) {
1795f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  if (N)
1796f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    WL.push_back(NestedNameSpecifierVisit(N, R, Parent));
1797f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
1798f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1799f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorvoid
1800f3db29fff6a583ecda823cf909ab7737d8d30129Douglas GregorEnqueueVisitor::AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
1801f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  if (Qualifier)
1802f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    WL.push_back(NestedNameSpecifierLocVisit(Qualifier, Parent));
1803f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor}
1804f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
180528a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::AddStmt(Stmt *S) {
180628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (S)
180728a719433411ef782b582946823bc648ddcc4533Ted Kremenek    WL.push_back(StmtVisit(S, Parent));
180828a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1809035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekvoid EnqueueVisitor::AddDecl(Decl *D, bool isFirst) {
181028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (D)
1811035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    WL.push_back(DeclVisit(D, Parent, isFirst));
181228a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
181360608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenekvoid EnqueueVisitor::
181460608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  AddExplicitTemplateArgs(const ExplicitTemplateArgumentList *A) {
181560608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  if (A)
181660608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek    WL.push_back(ExplicitTemplateArgsVisit(
181760608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek                        const_cast<ExplicitTemplateArgumentList*>(A), Parent));
181860608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek}
1819cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::AddMemberRef(FieldDecl *D, SourceLocation L) {
1820cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  if (D)
1821cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    WL.push_back(MemberRefVisit(D, L, Parent));
1822cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
182328a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::AddTypeLoc(TypeSourceInfo *TI) {
182428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (TI)
182528a719433411ef782b582946823bc648ddcc4533Ted Kremenek    WL.push_back(TypeLocVisit(TI->getTypeLoc(), Parent));
182628a719433411ef782b582946823bc648ddcc4533Ted Kremenek }
182728a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::EnqueueChildren(Stmt *S) {
1828a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  unsigned size = WL.size();
18297502c1d3ce8bb97bcc4f7bebef507040bd93b26fJohn McCall  for (Stmt::child_range Child = S->children(); Child; ++Child) {
183028a719433411ef782b582946823bc648ddcc4533Ted Kremenek    AddStmt(*Child);
1831a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  }
1832a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  if (size == WL.size())
1833a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek    return;
1834a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  // Now reverse the entries we just added.  This will match the DFS
1835a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  // ordering performed by the worklist.
1836a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
1837a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  std::reverse(I, E);
1838a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek}
1839ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekvoid EnqueueVisitor::VisitAddrLabelExpr(AddrLabelExpr *E) {
1840ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  WL.push_back(LabelRefVisit(E->getLabel(), E->getLabelLoc(), Parent));
1841ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek}
184273d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitBlockExpr(BlockExpr *B) {
184373d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddDecl(B->getBlockDecl());
184473d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
184528a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
184628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(E);
184728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(E->getTypeSourceInfo());
184828a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1849083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenekvoid EnqueueVisitor::VisitCompoundStmt(CompoundStmt *S) {
1850083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  for (CompoundStmt::reverse_body_iterator I = S->body_rbegin(),
1851083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek        E = S->body_rend(); I != E; ++I) {
1852083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek    AddStmt(*I);
1853083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  }
185411b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek}
1855f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::
1856f64d80306144f978148ba92f36f7cea7b671dd34Ted KremenekVisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E) {
1857f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
1858f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddDeclarationNameInfo(E);
18597c3179cf463c3b3b8c21dbb955f933ba50b74f28Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
18607c3179cf463c3b3b8c21dbb955f933ba50b74f28Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
1861f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  if (!E->isImplicitAccess())
1862f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    AddStmt(E->getBase());
1863f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
186411b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenekvoid EnqueueVisitor::VisitCXXNewExpr(CXXNewExpr *E) {
186511b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the initializer or constructor arguments.
186611b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  for (unsigned I = E->getNumConstructorArgs(); I > 0; --I)
186711b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek    AddStmt(E->getConstructorArg(I-1));
186811b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the array size, if any.
186911b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  AddStmt(E->getArraySize());
187011b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the allocated type.
187111b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  AddTypeLoc(E->getAllocatedTypeSourceInfo());
187211b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the placement arguments.
187311b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  for (unsigned I = E->getNumPlacementArgs(); I > 0; --I)
187411b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek    AddStmt(E->getPlacementArg(I-1));
187511b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek}
187628a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *CE) {
18778b8d8c90f2d8ac651d14b57f116d20b3c911ac7fTed Kremenek  for (unsigned I = CE->getNumArgs(); I > 1 /* Yes, this is 1 */; --I)
18788b8d8c90f2d8ac651d14b57f116d20b3c911ac7fTed Kremenek    AddStmt(CE->getArg(I-1));
187928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(CE->getCallee());
188028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(CE->getArg(0));
188128a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1882cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
1883cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the name of the type being destroyed.
1884cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getDestroyedTypeInfo());
1885cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the scope type that looks disturbingly like the nested-name-specifier
1886cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // but isn't.
1887cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getScopeTypeInfo());
1888cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the nested-name-specifier.
1889f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
1890f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
1891cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit base expression.
1892cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddStmt(E->getBase());
1893cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
18946d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenekvoid EnqueueVisitor::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
18956d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
18966d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek}
189773d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
189873d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  EnqueueChildren(E);
189973d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
190073d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
1901b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenekvoid EnqueueVisitor::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
1902b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  EnqueueChildren(E);
1903b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  if (E->isTypeOperand())
1904b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek    AddTypeLoc(E->getTypeOperandSourceInfo());
1905b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek}
190655b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek
190755b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenekvoid EnqueueVisitor::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr
190855b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek                                                     *E) {
190955b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  EnqueueChildren(E);
191055b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
191155b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek}
19121e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenekvoid EnqueueVisitor::VisitCXXUuidofExpr(CXXUuidofExpr *E) {
19131e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  EnqueueChildren(E);
19141e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  if (E->isTypeOperand())
19151e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek    AddTypeLoc(E->getTypeOperandSourceInfo());
19161e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek}
1917e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenekvoid EnqueueVisitor::VisitDeclRefExpr(DeclRefExpr *DR) {
191860608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  if (DR->hasExplicitTemplateArgs()) {
191960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek    AddExplicitTemplateArgs(&DR->getExplicitTemplateArgs());
192060608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  }
1921e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek  WL.push_back(DeclRefExprParts(DR, Parent));
1922e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek}
1923f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
1924f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
1925f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddDeclarationNameInfo(E);
192600cf3cc2718671aa48e8da264a523b0058a8591eDouglas Gregor  AddNestedNameSpecifierLoc(E->getQualifierLoc());
1927f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
1928035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekvoid EnqueueVisitor::VisitDeclStmt(DeclStmt *S) {
1929035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  unsigned size = WL.size();
1930035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  bool isFirst = true;
1931035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  for (DeclStmt::decl_iterator D = S->decl_begin(), DEnd = S->decl_end();
1932035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek       D != DEnd; ++D) {
1933035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    AddDecl(*D, isFirst);
1934035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    isFirst = false;
1935035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1936035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  if (size == WL.size())
1937035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    return;
1938035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  // Now reverse the entries we just added.  This will match the DFS
1939035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  // ordering performed by the worklist.
1940035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
1941035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  std::reverse(I, E);
1942035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek}
1943cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
1944cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddStmt(E->getInit());
1945cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  typedef DesignatedInitExpr::Designator Designator;
1946cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  for (DesignatedInitExpr::reverse_designators_iterator
1947cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek         D = E->designators_rbegin(), DEnd = E->designators_rend();
1948cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek         D != DEnd; ++D) {
1949cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    if (D->isFieldDesignator()) {
1950cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      if (FieldDecl *Field = D->getField())
1951cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        AddMemberRef(Field, D->getFieldLoc());
1952cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
1953cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
1954cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    if (D->isArrayDesignator()) {
1955cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      AddStmt(E->getArrayIndex(*D));
1956cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
1957cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
1958cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    assert(D->isArrayRangeDesignator() && "Unknown designator kind");
1959cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    AddStmt(E->getArrayRangeEnd(*D));
1960cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    AddStmt(E->getArrayRangeStart(*D));
1961cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1962cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
196328a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitExplicitCastExpr(ExplicitCastExpr *E) {
196428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(E);
196528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(E->getTypeInfoAsWritten());
196628a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
196728a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitForStmt(ForStmt *FS) {
196828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getBody());
196928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getInc());
197028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getCond());
197128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(FS->getConditionVariable());
197228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getInit());
197328a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1974ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekvoid EnqueueVisitor::VisitGotoStmt(GotoStmt *GS) {
1975ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  WL.push_back(LabelRefVisit(GS->getLabel(), GS->getLabelLoc(), Parent));
1976ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek}
197728a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitIfStmt(IfStmt *If) {
197828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getElse());
197928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getThen());
198028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getCond());
198128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(If->getConditionVariable());
198228a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
198328a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitInitListExpr(InitListExpr *IE) {
198428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  // We care about the syntactic form of the initializer list, only.
198528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (InitListExpr *Syntactic = IE->getSyntacticForm())
198628a719433411ef782b582946823bc648ddcc4533Ted Kremenek    IE = Syntactic;
198728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(IE);
198828a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
198928a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitMemberExpr(MemberExpr *M) {
199089629a746019a42797495b091711a1d68467e88aDouglas Gregor  WL.push_back(MemberExprParts(M, Parent));
199189629a746019a42797495b091711a1d68467e88aDouglas Gregor
199289629a746019a42797495b091711a1d68467e88aDouglas Gregor  // If the base of the member access expression is an implicit 'this', don't
199389629a746019a42797495b091711a1d68467e88aDouglas Gregor  // visit it.
199489629a746019a42797495b091711a1d68467e88aDouglas Gregor  // FIXME: If we ever want to show these implicit accesses, this will be
199589629a746019a42797495b091711a1d68467e88aDouglas Gregor  // unfortunate. However, clang_getCursor() relies on this behavior.
199675e85048e73fcde2ce9d8a48dfdb1220e132eb59Douglas Gregor  if (!M->isImplicitAccess())
199775e85048e73fcde2ce9d8a48dfdb1220e132eb59Douglas Gregor    AddStmt(M->getBase());
199828a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
199973d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
200073d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddTypeLoc(E->getEncodedTypeSourceInfo());
200173d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
200228a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitObjCMessageExpr(ObjCMessageExpr *M) {
200328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(M);
200428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(M->getClassReceiverTypeInfo());
200528a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
2006cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitOffsetOfExpr(OffsetOfExpr *E) {
2007cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the components of the offsetof expression.
2008cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  for (unsigned N = E->getNumComponents(), I = N; I > 0; --I) {
2009cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    typedef OffsetOfExpr::OffsetOfNode OffsetOfNode;
2010cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    const OffsetOfNode &Node = E->getComponent(I-1);
2011cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    switch (Node.getKind()) {
2012cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Array:
2013cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      AddStmt(E->getIndexExpr(Node.getArrayExprIndex()));
2014cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      break;
2015cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Field:
201606dec892b5300b43263d25c5476b506c9d6cfbadAbramo Bagnara      AddMemberRef(Node.getField(), Node.getSourceRange().getEnd());
2017cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      break;
2018cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Identifier:
2019cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Base:
2020cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
2021cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
2022cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
2023cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the type into which we're computing the offset.
2024cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
2025cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
202628a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitOverloadExpr(OverloadExpr *E) {
202760608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
20286045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek  WL.push_back(OverloadExprParts(E, Parent));
20296045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek}
2030f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbournevoid EnqueueVisitor::VisitUnaryExprOrTypeTraitExpr(
2031f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbourne                                              UnaryExprOrTypeTraitExpr *E) {
20326d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  EnqueueChildren(E);
20336d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  if (E->isArgumentType())
20346d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek    AddTypeLoc(E->getArgumentTypeInfo());
20356d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek}
203628a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitStmt(Stmt *S) {
203728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(S);
203828a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
203928a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitSwitchStmt(SwitchStmt *S) {
204028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(S->getBody());
204128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(S->getCond());
204228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(S->getConditionVariable());
204328a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
2044fafa75aebadef8d6b44a920e3f40529f150a5574Ted Kremenek
204528a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitWhileStmt(WhileStmt *W) {
204628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(W->getBody());
204728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(W->getCond());
204828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(W->getConditionVariable());
204928a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
20502939b6f356161f572712d4d6310b65f9599e3675Ted Kremenekvoid EnqueueVisitor::VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E) {
20512939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek  AddTypeLoc(E->getQueriedTypeSourceInfo());
20522939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek}
20536ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet
20546ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichetvoid EnqueueVisitor::VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E) {
20556ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  AddTypeLoc(E->getRhsTypeSourceInfo());
20560a03a3f98b14006a54bcac9e8908a7c9f50e519fFrancois Pichet  AddTypeLoc(E->getLhsTypeSourceInfo());
20576ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet}
20586ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet
205928a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *U) {
206028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  VisitOverloadExpr(U);
206128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (!U->isImplicitAccess())
206228a719433411ef782b582946823bc648ddcc4533Ted Kremenek    AddStmt(U->getBase());
206328a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
20649d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenekvoid EnqueueVisitor::VisitVAArgExpr(VAArgExpr *E) {
20659d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  AddStmt(E->getSubExpr());
20669d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  AddTypeLoc(E->getWrittenTypeInfo());
20679d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek}
206894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregorvoid EnqueueVisitor::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
206994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  WL.push_back(SizeOfPackExprParts(E, Parent));
207094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor}
20716045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek
2072c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekvoid CursorVisitor::EnqueueWorkList(VisitorWorkList &WL, Stmt *S) {
207328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueVisitor(WL, MakeCXCursor(S, StmtParent, TU)).Visit(S);
2074c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2075c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2076c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekbool CursorVisitor::IsInRegionOfInterest(CXCursor C) {
2077c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  if (RegionOfInterest.isValid()) {
2078c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    SourceRange Range = getRawCursorExtent(C);
2079c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    if (Range.isInvalid() || CompareRegionOfInterest(Range))
2080c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      return false;
2081c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  }
2082c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  return true;
2083c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2084c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2085c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekbool CursorVisitor::RunVisitorWorkList(VisitorWorkList &WL) {
2086c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  while (!WL.empty()) {
2087c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    // Dequeue the worklist item.
208882f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    VisitorJob LI = WL.back();
208982f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    WL.pop_back();
209082f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek
2091c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    // Set the Parent field, then back to its old value once we're done.
2092c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    SetParentRAII SetParent(Parent, StmtParent, LI.getParent());
2093c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2094c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    switch (LI.getKind()) {
2095f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek      case VisitorJob::DeclVisitKind: {
209682f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        Decl *D = cast<DeclVisit>(&LI)->get();
2097f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        if (!D)
2098f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek          continue;
2099f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek
2100f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        // For now, perform default visitation for Decls.
210182f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        if (Visit(MakeCXCursor(D, TU, cast<DeclVisit>(&LI)->isFirst())))
2102f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek            return true;
2103f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek
2104f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        continue;
2105f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek      }
210660608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek      case VisitorJob::ExplicitTemplateArgsVisitKind: {
210760608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        const ExplicitTemplateArgumentList *ArgList =
210860608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek          cast<ExplicitTemplateArgsVisit>(&LI)->get();
210960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        for (const TemplateArgumentLoc *Arg = ArgList->getTemplateArgs(),
211060608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek               *ArgEnd = Arg + ArgList->NumTemplateArgs;
211160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek               Arg != ArgEnd; ++Arg) {
211260608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek          if (VisitTemplateArgumentLoc(*Arg))
211360608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek            return true;
211460608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        }
211560608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        continue;
211660608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek      }
2117cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek      case VisitorJob::TypeLocVisitKind: {
2118cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek        // Perform default visitation for TypeLocs.
211982f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        if (Visit(cast<TypeLocVisit>(&LI)->get()))
2120cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek          return true;
2121cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek        continue;
2122cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek      }
2123ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek      case VisitorJob::LabelRefVisitKind: {
2124ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner        LabelDecl *LS = cast<LabelRefVisit>(&LI)->get();
2125e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek        if (LabelStmt *stmt = LS->getStmt()) {
2126e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek          if (Visit(MakeCursorLabelRef(stmt, cast<LabelRefVisit>(&LI)->getLoc(),
2127e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek                                       TU))) {
2128e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek            return true;
2129e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek          }
2130e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek        }
2131ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek        continue;
2132ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek      }
2133f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
2134f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      case VisitorJob::NestedNameSpecifierVisitKind: {
2135f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        NestedNameSpecifierVisit *V = cast<NestedNameSpecifierVisit>(&LI);
2136f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        if (VisitNestedNameSpecifier(V->get(), V->getSourceRange()))
2137f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek          return true;
2138f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        continue;
2139f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      }
2140f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
2141f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor      case VisitorJob::NestedNameSpecifierLocVisitKind: {
2142f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        NestedNameSpecifierLocVisit *V = cast<NestedNameSpecifierLocVisit>(&LI);
2143f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        if (VisitNestedNameSpecifierLoc(V->get()))
2144f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor          return true;
2145f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        continue;
2146f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor      }
2147f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
2148f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      case VisitorJob::DeclarationNameInfoVisitKind: {
2149f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        if (VisitDeclarationNameInfo(cast<DeclarationNameInfoVisit>(&LI)
2150f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek                                     ->get()))
2151f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek          return true;
2152f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        continue;
2153f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      }
2154cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      case VisitorJob::MemberRefVisitKind: {
2155cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        MemberRefVisit *V = cast<MemberRefVisit>(&LI);
2156cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        if (Visit(MakeCursorMemberRef(V->get(), V->getLoc(), TU)))
2157cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          return true;
2158cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        continue;
2159cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      }
2160c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      case VisitorJob::StmtVisitKind: {
216182f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        Stmt *S = cast<StmtVisit>(&LI)->get();
21628c269ac75569454a049385b1246140db5f2b6faaTed Kremenek        if (!S)
21638c269ac75569454a049385b1246140db5f2b6faaTed Kremenek          continue;
21648c269ac75569454a049385b1246140db5f2b6faaTed Kremenek
2165f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        // Update the current cursor.
2166c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        CXCursor Cursor = MakeCXCursor(S, StmtParent, TU);
2167cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        if (!IsInRegionOfInterest(Cursor))
2168cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          continue;
2169cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        switch (Visitor(Cursor, Parent, ClientData)) {
2170cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Break: return true;
2171cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Continue: break;
2172cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Recurse:
2173cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek            EnqueueWorkList(WL, S);
217482f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek            break;
2175c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        }
217682f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        continue;
2177c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      }
2178c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      case VisitorJob::MemberExprPartsKind: {
2179c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Handle the other pieces in the MemberExpr besides the base.
218082f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        MemberExpr *M = cast<MemberExprParts>(&LI)->get();
2181c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2182c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the nested-name-specifier
218340d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = M->getQualifierLoc())
218440d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
2185c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek            return true;
2186c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2187c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the declaration name.
2188c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        if (VisitDeclarationNameInfo(M->getMemberNameInfo()))
2189c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          return true;
2190c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2191c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the explicitly-specified template arguments, if any.
2192c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        if (M->hasExplicitTemplateArgs()) {
2193c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          for (const TemplateArgumentLoc *Arg = M->getTemplateArgs(),
2194c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek               *ArgEnd = Arg + M->getNumTemplateArgs();
2195c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek               Arg != ArgEnd; ++Arg) {
2196c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek            if (VisitTemplateArgumentLoc(*Arg))
2197c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek              return true;
2198c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          }
2199c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        }
2200c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        continue;
2201c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      }
2202e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek      case VisitorJob::DeclRefExprPartsKind: {
220382f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        DeclRefExpr *DR = cast<DeclRefExprParts>(&LI)->get();
2204e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        // Visit nested-name-specifier, if present.
220540d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = DR->getQualifierLoc())
220640d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
2207e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek            return true;
2208e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        // Visit declaration name.
2209e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        if (VisitDeclarationNameInfo(DR->getNameInfo()))
2210e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek          return true;
2211e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        continue;
2212e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek      }
22136045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek      case VisitorJob::OverloadExprPartsKind: {
221482f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        OverloadExpr *O = cast<OverloadExprParts>(&LI)->get();
22156045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the nested-name-specifier.
22164c9be89bb615ec07eb3ed507c8fa9d0baa8a5ad7Douglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = O->getQualifierLoc())
22174c9be89bb615ec07eb3ed507c8fa9d0baa8a5ad7Douglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
22186045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek            return true;
22196045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the declaration name.
22206045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        if (VisitDeclarationNameInfo(O->getNameInfo()))
22216045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek          return true;
22226045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the overloaded declaration reference.
22236045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        if (Visit(MakeCursorOverloadedDeclRef(O, TU)))
22246045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek          return true;
22256045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        continue;
22266045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek      }
222794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      case VisitorJob::SizeOfPackExprPartsKind: {
222894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        SizeOfPackExpr *E = cast<SizeOfPackExprParts>(&LI)->get();
222994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        NamedDecl *Pack = E->getPack();
223094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        if (isa<TemplateTypeParmDecl>(Pack)) {
223194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          if (Visit(MakeCursorTypeRef(cast<TemplateTypeParmDecl>(Pack),
223294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor                                      E->getPackLoc(), TU)))
223394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor            return true;
223494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
223594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          continue;
223694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        }
223794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
223894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        if (isa<TemplateTemplateParmDecl>(Pack)) {
223994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          if (Visit(MakeCursorTemplateRef(cast<TemplateTemplateParmDecl>(Pack),
224094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor                                          E->getPackLoc(), TU)))
224194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor            return true;
224294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
224394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          continue;
224494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        }
224594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
224694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        // Non-type template parameter packs and function parameter packs are
224794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        // treated like DeclRefExpr cursors.
224894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        continue;
224994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      }
2250c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    }
2251c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  }
2252c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  return false;
2253c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2254c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2255cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekbool CursorVisitor::Visit(Stmt *S) {
2256d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  VisitorWorkList *WL = 0;
2257d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  if (!WorkListFreeList.empty()) {
2258d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL = WorkListFreeList.back();
2259d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL->clear();
2260d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WorkListFreeList.pop_back();
2261d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
2262d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  else {
2263d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL = new VisitorWorkList();
2264d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WorkListCache.push_back(WL);
2265d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
2266d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  EnqueueWorkList(*WL, S);
2267d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  bool result = RunVisitorWorkList(*WL);
2268d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  WorkListFreeList.push_back(WL);
2269d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  return result;
2270c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2271c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2272c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
2273c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek// Misc. API hooks.
2274c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
2275c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
22768c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregorstatic llvm::sys::Mutex EnableMultithreadingMutex;
22778c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregorstatic bool EnabledMultithreading;
22788c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor
22795e4bc590b0ea010e38372d0b4a0aab578a746fe6Benjamin Kramerextern "C" {
22800a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas GregorCXIndex clang_createIndex(int excludeDeclarationsFromPCH,
22810a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor                          int displayDiagnostics) {
228248615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  // Disable pretty stack trace functionality, which will otherwise be a very
228348615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  // poor citizen of the world and set up all sorts of signal handlers.
228448615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  llvm::DisablePrettyStackTrace = true;
228548615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar
2286c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  // We use crash recovery to make some of our APIs more reliable, implicitly
2287c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  // enable it.
2288c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  llvm::CrashRecoveryContext::Enable();
2289c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar
22908c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  // Enable support for multithreading in LLVM.
22918c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  {
22928c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    llvm::sys::ScopedLock L(EnableMultithreadingMutex);
22938c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    if (!EnabledMultithreading) {
22948c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor      llvm::llvm_start_multithreaded();
22958c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor      EnabledMultithreading = true;
22968c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    }
22978c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  }
22988c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor
2299a030b7cf5e6aad5889b1b662b6979840bc75f87fDouglas Gregor  CIndexer *CIdxr = new CIndexer();
2300e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  if (excludeDeclarationsFromPCH)
2301e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff    CIdxr->setOnlyLocalDecls();
23020a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  if (displayDiagnostics)
23030a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    CIdxr->setDisplayDiagnostics();
2304e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  return CIdxr;
2305600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
2306600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
23079ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarvoid clang_disposeIndex(CXIndex CIdx) {
23082b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (CIdx)
23092b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    delete static_cast<CIndexer *>(CIdx);
23102bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff}
23112bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff
2312d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenekvoid clang_toggleCrashRecovery(unsigned isEnabled) {
2313d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek  if (isEnabled)
2314d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek    llvm::CrashRecoveryContext::Enable();
2315d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek  else
2316d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek    llvm::CrashRecoveryContext::Disable();
2317d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek}
2318d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek
23199ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXTranslationUnit clang_createTranslationUnit(CXIndex CIdx,
2320a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                              const char *ast_filename) {
23212b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CIdx)
23222b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    return 0;
2323f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
23247d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
2325389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  FileSystemOptions FileSystemOpts;
2326389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  FileSystemOpts.WorkingDir = CXXIdx->getWorkingDirectory();
23270d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
232828019772db70d4547be05a042eb950bc910f134fDouglas Gregor  llvm::IntrusiveRefCntPtr<Diagnostic> Diags;
2329a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *TU = ASTUnit::LoadFromASTFile(ast_filename, Diags, FileSystemOpts,
2330a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                  CXXIdx->getOnlyLocalDecls(),
2331a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                  0, 0, true);
2332a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return MakeCXTranslationUnit(TU);
2333600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
2334600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
2335b1c031be513705d924038f497279b9b599868ba1Douglas Gregorunsigned clang_defaultEditingTranslationUnitOptions() {
23362a2c50b330e7754499f42173616a36865b5f313bDouglas Gregor  return CXTranslationUnit_PrecompiledPreamble |
233799ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor         CXTranslationUnit_CacheCompletionResults |
233899ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor         CXTranslationUnit_CXXPrecompiledPreamble;
2339b1c031be513705d924038f497279b9b599868ba1Douglas Gregor}
2340b1c031be513705d924038f497279b9b599868ba1Douglas Gregor
23419ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXTranslationUnit
23429ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarclang_createTranslationUnitFromSourceFile(CXIndex CIdx,
23439ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          const char *source_filename,
23449ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          int num_command_line_args,
23452ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor                                          const char * const *command_line_args,
23464db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor                                          unsigned num_unsaved_files,
2347a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                          struct CXUnsavedFile *unsaved_files) {
23485a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor  return clang_parseTranslationUnit(CIdx, source_filename,
23495a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                    command_line_args, num_command_line_args,
23505a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                    unsaved_files, num_unsaved_files,
23515a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                 CXTranslationUnit_DetailedPreprocessingRecord);
23525a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor}
235319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
235419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbarstruct ParseTranslationUnitInfo {
235519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXIndex CIdx;
235619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  const char *source_filename;
23572ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor  const char *const *command_line_args;
235819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  int num_command_line_args;
235919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  struct CXUnsavedFile *unsaved_files;
236019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned num_unsaved_files;
236119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned options;
236219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXTranslationUnit result;
236319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar};
2364b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbarstatic void clang_parseTranslationUnit_Impl(void *UserData) {
236519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  ParseTranslationUnitInfo *PTUI =
236619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    static_cast<ParseTranslationUnitInfo*>(UserData);
236719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXIndex CIdx = PTUI->CIdx;
236819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  const char *source_filename = PTUI->source_filename;
23692ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor  const char * const *command_line_args = PTUI->command_line_args;
237019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  int num_command_line_args = PTUI->num_command_line_args;
237119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  struct CXUnsavedFile *unsaved_files = PTUI->unsaved_files;
237219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned num_unsaved_files = PTUI->num_unsaved_files;
237319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned options = PTUI->options;
237419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  PTUI->result = 0;
23755a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor
23762b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CIdx)
237719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    return;
2378f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2379e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
2380e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff
238144c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  bool PrecompilePreamble = options & CXTranslationUnit_PrecompiledPreamble;
2382df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor  bool CompleteTranslationUnit
2383df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor    = ((options & CXTranslationUnit_Incomplete) == 0);
238487c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  bool CacheCodeCompetionResults
238587c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    = options & CXTranslationUnit_CacheCompletionResults;
238699ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor  bool CXXPrecompilePreamble
238799ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor    = options & CXTranslationUnit_CXXPrecompiledPreamble;
238899ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor  bool CXXChainedPCH
238999ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor    = options & CXTranslationUnit_CXXChainedPCH;
239087c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor
23915352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  // Configure the diagnostics.
23925352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  DiagnosticOptions DiagOpts;
239325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::IntrusiveRefCntPtr<Diagnostic>
239425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Diags(CompilerInstance::createDiagnostics(DiagOpts, num_command_line_args,
239525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                                command_line_args));
239625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
239725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
239825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<Diagnostic,
239925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    llvm::CrashRecoveryContextReleaseRefCleanup<Diagnostic> >
240025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    DiagCleanup(Diags.getPtr());
240125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
240225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::OwningPtr<std::vector<ASTUnit::RemappedFile> >
240325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles(new std::vector<ASTUnit::RemappedFile>());
240425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
240525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
240625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<
240725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
2408f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
24094db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  for (unsigned I = 0; I != num_unsaved_files; ++I) {
2410a0a270c0f1c0a4e3482438bdc5f4a7bd3d25f0a6Chris Lattner    llvm::StringRef Data(unsaved_files[I].Contents, unsaved_files[I].Length);
2411f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    const llvm::MemoryBuffer *Buffer
2412a0a270c0f1c0a4e3482438bdc5f4a7bd3d25f0a6Chris Lattner      = llvm::MemoryBuffer::getMemBufferCopy(Data, unsaved_files[I].Filename);
241325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles->push_back(std::make_pair(unsaved_files[I].Filename,
241425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                            Buffer));
24154db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  }
2416f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
241725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::OwningPtr<std::vector<const char *> >
241825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args(new std::vector<const char*>());
241925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
242025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this method.
242125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<std::vector<const char*> >
242225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    ArgsCleanup(Args.get());
242325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
242452ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // Since the Clang C library is primarily used by batch tools dealing with
242552ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // (often very broken) source code, where spell-checking can have a
242652ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // significant negative impact on performance (particularly when
242752ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // precompiled headers are involved), we disable it by default.
2428b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  // Only do this if we haven't found a spell-checking-related argument.
2429b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  bool FoundSpellCheckingArgument = false;
2430b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  for (int I = 0; I != num_command_line_args; ++I) {
2431b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    if (strcmp(command_line_args[I], "-fno-spell-checking") == 0 ||
2432b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        strcmp(command_line_args[I], "-fspell-checking") == 0) {
2433b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      FoundSpellCheckingArgument = true;
2434b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      break;
2435e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff    }
2436b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  }
2437b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  if (!FoundSpellCheckingArgument)
243825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-fno-spell-checking");
2439b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor
244025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  Args->insert(Args->end(), command_line_args,
244125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek               command_line_args + num_command_line_args);
2442d93256e55673a17d18543397ec462416acb13792Douglas Gregor
2443c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // The 'source_filename' argument is optional.  If the caller does not
2444c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // specify it then it is assumed that the source file is specified
2445c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // in the actual argument list.
2446c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // Put the source file after command_line_args otherwise if '-x' flag is
2447c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // present it will be unused.
2448c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  if (source_filename)
244925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back(source_filename);
2450c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis
245144c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  // Do we need the detailed preprocessing record?
245244c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  if (options & CXTranslationUnit_DetailedPreprocessingRecord) {
245325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-Xclang");
245425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-detailed-preprocessing-record");
245544c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  }
245644c181aec37789f25f6c15543c164416f72e562aDouglas Gregor
2457026f6911bb985c800a54446de9f6da8745ae025aArgyrios Kyrtzidis  unsigned NumErrors = Diags->getClient()->getNumErrors();
2458b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  llvm::OwningPtr<ASTUnit> Unit(
24594ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek    ASTUnit::LoadFromCommandLine(Args->size() ? &(*Args)[0] : 0
24604ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 /* vector::data() not portable */,
24614ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 Args->size() ? (&(*Args)[0] + Args->size()) :0,
2462b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 Diags,
2463b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CXXIdx->getClangResourcesPath(),
2464b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CXXIdx->getOnlyLocalDecls(),
2465e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor                                 /*CaptureDiagnostics=*/true,
24664ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 RemappedFiles->size() ? &(*RemappedFiles)[0]:0,
246725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                 RemappedFiles->size(),
2468299a4a967b02c9f0d0d94ad8560e3ced893f9116Argyrios Kyrtzidis                                 /*RemappedFilesKeepOriginalName=*/true,
2469b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 PrecompilePreamble,
2470b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CompleteTranslationUnit,
247199ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor                                 CacheCodeCompetionResults,
247299ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor                                 CXXPrecompilePreamble,
247399ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor                                 CXXChainedPCH));
2474b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor
2475026f6911bb985c800a54446de9f6da8745ae025aArgyrios Kyrtzidis  if (NumErrors != Diags->getClient()->getNumErrors()) {
2476b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    // Make sure to check that 'Unit' is non-NULL.
2477b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    if (CXXIdx->getDisplayDiagnostics() && Unit.get()) {
2478b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      for (ASTUnit::stored_diag_iterator D = Unit->stored_diag_begin(),
2479b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                      DEnd = Unit->stored_diag_end();
2480b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor           D != DEnd; ++D) {
2481b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        CXStoredDiagnostic Diag(*D, Unit->getASTContext().getLangOptions());
2482b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        CXString Msg = clang_formatDiagnostic(&Diag,
2483b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                    clang_defaultDiagnosticDisplayOptions());
2484b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        fprintf(stderr, "%s\n", clang_getCString(Msg));
2485b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        clang_disposeString(Msg);
2486b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      }
2487274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor#ifdef LLVM_ON_WIN32
2488b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      // On Windows, force a flush, since there may be multiple copies of
2489b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      // stderr and stdout in the file system, all with different buffers
2490b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      // but writing to the same device.
2491b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      fflush(stderr);
2492b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor#endif
2493b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    }
2494a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  }
2495d93256e55673a17d18543397ec462416acb13792Douglas Gregor
2496a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  PTUI->result = MakeCXTranslationUnit(Unit.take());
249719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar}
249819ffd492a31a25fb691098bf79f317e5f3edf177Daniel DunbarCXTranslationUnit clang_parseTranslationUnit(CXIndex CIdx,
249919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             const char *source_filename,
25002ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor                                         const char * const *command_line_args,
250119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             int num_command_line_args,
25029e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                            struct CXUnsavedFile *unsaved_files,
250319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             unsigned num_unsaved_files,
250419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             unsigned options) {
250519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  ParseTranslationUnitInfo PTUI = { CIdx, source_filename, command_line_args,
25069e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                    num_command_line_args, unsaved_files,
25079e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                    num_unsaved_files, options, 0 };
250819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  llvm::CrashRecoveryContext CRC;
250919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
2510bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  if (!RunSafely(CRC, clang_parseTranslationUnit_Impl, &PTUI)) {
251160a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "libclang: crash detected during parsing: {\n");
251260a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'source_filename' : '%s'\n", source_filename);
251360a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'command_line_args' : [");
251460a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    for (int i = 0; i != num_command_line_args; ++i) {
251560a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      if (i)
251660a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar        fprintf(stderr, ", ");
251760a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      fprintf(stderr, "'%s'", command_line_args[i]);
251860a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    }
251960a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "],\n");
252060a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'unsaved_files' : [");
252160a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    for (unsigned i = 0; i != num_unsaved_files; ++i) {
252260a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      if (i)
252360a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar        fprintf(stderr, ", ");
252460a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      fprintf(stderr, "('%s', '...', %ld)", unsaved_files[i].Filename,
252560a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar              unsaved_files[i].Length);
252660a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    }
252760a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "],\n");
252860a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'options' : %d,\n", options);
252960a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "}\n");
253060a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar
253119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    return 0;
253219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  }
253319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
253419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  return PTUI.result;
25355b7d8e254f6c2855b37b5521c0aee0a560dab237Steve Naroff}
25365b7d8e254f6c2855b37b5521c0aee0a560dab237Steve Naroff
25371999844e7a18786e61e619e1dc6c789827541863Douglas Gregorunsigned clang_defaultSaveOptions(CXTranslationUnit TU) {
25381999844e7a18786e61e619e1dc6c789827541863Douglas Gregor  return CXSaveTranslationUnit_None;
25391999844e7a18786e61e619e1dc6c789827541863Douglas Gregor}
25401999844e7a18786e61e619e1dc6c789827541863Douglas Gregor
25411999844e7a18786e61e619e1dc6c789827541863Douglas Gregorint clang_saveTranslationUnit(CXTranslationUnit TU, const char *FileName,
25421999844e7a18786e61e619e1dc6c789827541863Douglas Gregor                              unsigned options) {
25437ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor  if (!TU)
25447ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor    return 1;
25457ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor
2546a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return static_cast<ASTUnit *>(TU->TUData)->Save(FileName);
25477ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor}
254819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
25499ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarvoid clang_disposeTranslationUnit(CXTranslationUnit CTUnit) {
2550ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  if (CTUnit) {
2551ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    // If the translation unit has been marked as unsafe to free, just discard
2552ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    // it.
2553a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    if (static_cast<ASTUnit *>(CTUnit->TUData)->isUnsafeToFree())
2554ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar      return;
2555ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2556a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    delete static_cast<ASTUnit *>(CTUnit->TUData);
2557a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    disposeCXStringPool(CTUnit->StringPool);
2558a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    delete CTUnit;
2559ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  }
25602bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff}
25610d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
2562e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregorunsigned clang_defaultReparseOptions(CXTranslationUnit TU) {
2563e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor  return CXReparse_None;
2564e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor}
2565e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor
2566ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbarstruct ReparseTranslationUnitInfo {
2567ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  CXTranslationUnit TU;
2568ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned num_unsaved_files;
2569ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  struct CXUnsavedFile *unsaved_files;
2570ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned options;
2571ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  int result;
2572ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar};
2573593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2574b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbarstatic void clang_reparseTranslationUnit_Impl(void *UserData) {
2575ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  ReparseTranslationUnitInfo *RTUI =
2576ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    static_cast<ReparseTranslationUnitInfo*>(UserData);
2577ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  CXTranslationUnit TU = RTUI->TU;
2578ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned num_unsaved_files = RTUI->num_unsaved_files;
2579ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  struct CXUnsavedFile *unsaved_files = RTUI->unsaved_files;
2580ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned options = RTUI->options;
2581ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  (void) options;
2582ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  RTUI->result = 1;
2583ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2584abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  if (!TU)
2585ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    return;
2586593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2587a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
2588593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
2589abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor
259025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::OwningPtr<std::vector<ASTUnit::RemappedFile> >
259125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles(new std::vector<ASTUnit::RemappedFile>());
259225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
259325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
259425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<
259525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
259625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
2597abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  for (unsigned I = 0; I != num_unsaved_files; ++I) {
2598abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor    llvm::StringRef Data(unsaved_files[I].Contents, unsaved_files[I].Length);
2599abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor    const llvm::MemoryBuffer *Buffer
26001abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor      = llvm::MemoryBuffer::getMemBufferCopy(Data, unsaved_files[I].Filename);
260125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles->push_back(std::make_pair(unsaved_files[I].Filename,
260225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                            Buffer));
2603abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  }
2604abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor
26054ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek  if (!CXXUnit->Reparse(RemappedFiles->size() ? &(*RemappedFiles)[0] : 0,
26064ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                        RemappedFiles->size()))
2607593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor    RTUI->result = 0;
2608abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor}
2609593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2610ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbarint clang_reparseTranslationUnit(CXTranslationUnit TU,
2611ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 unsigned num_unsaved_files,
2612ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 struct CXUnsavedFile *unsaved_files,
2613ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 unsigned options) {
2614ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  ReparseTranslationUnitInfo RTUI = { TU, num_unsaved_files, unsaved_files,
2615ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                      options, 0 };
2616ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  llvm::CrashRecoveryContext CRC;
2617ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2618bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  if (!RunSafely(CRC, clang_reparseTranslationUnit_Impl, &RTUI)) {
2619b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbar    fprintf(stderr, "libclang: crash detected during reparsing\n");
2620a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    static_cast<ASTUnit *>(TU->TUData)->setUnsafeToFree(true);
2621ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    return 1;
2622ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  }
2623ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
26241dfb26af4d6aa4f7818e256659a79f1ec2cba784Ted Kremenek
2625ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  return RTUI.result;
2626ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar}
2627ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2628df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor
26299ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXString clang_getTranslationUnitSpelling(CXTranslationUnit CTUnit) {
26302b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CTUnit)
2631ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
2632f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2633a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit->TUData);
2634ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString(CXXUnit->getOriginalSourceFileName(), true);
2635af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff}
26361eb79b58e56b99cf557d5d353586a10c5360364dDaniel Dunbar
26377eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas GregorCXCursor clang_getTranslationUnitCursor(CXTranslationUnit TU) {
2638b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  CXCursor Result = { CXCursor_TranslationUnit, { 0, 0, TU } };
26397eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  return Result;
26407eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor}
26417eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
2642fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
2643600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
2644fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
26451db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor// CXSourceLocation and CXSourceRange Operations.
26461db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor//===----------------------------------------------------------------------===//
26471db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
2648b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregorextern "C" {
2649b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXSourceLocation clang_getNullLocation() {
26505352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  CXSourceLocation Result = { { 0, 0 }, 0 };
2651b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  return Result;
2652b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2653b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
2654b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregorunsigned clang_equalLocations(CXSourceLocation loc1, CXSourceLocation loc2) {
265590a6b9e1e4d4d1995ada044e319d6e722b07a6b4Daniel Dunbar  return (loc1.ptr_data[0] == loc2.ptr_data[0] &&
265690a6b9e1e4d4d1995ada044e319d6e722b07a6b4Daniel Dunbar          loc1.ptr_data[1] == loc2.ptr_data[1] &&
265790a6b9e1e4d4d1995ada044e319d6e722b07a6b4Daniel Dunbar          loc1.int_data == loc2.int_data);
2658b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2659b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
2660b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXSourceLocation clang_getLocation(CXTranslationUnit tu,
2661b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor                                   CXFile file,
2662b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor                                   unsigned line,
2663b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor                                   unsigned column) {
266442748ec5cb2d75fe0dbb3a6db5aee6c11b5dc190Douglas Gregor  if (!tu || !file)
2665b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor    return clang_getNullLocation();
266642748ec5cb2d75fe0dbb3a6db5aee6c11b5dc190Douglas Gregor
266786a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  bool Logging = ::getenv("LIBCLANG_LOGGING");
2668a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
266986a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  const FileEntry *File = static_cast<const FileEntry *>(file);
2670b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  SourceLocation SLoc
267186a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor    = CXXUnit->getSourceManager().getLocation(File, line, column);
267286a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  if (SLoc.isInvalid()) {
267386a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor    if (Logging)
267486a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor      llvm::errs() << "clang_getLocation(\"" << File->getName()
267586a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor                   << "\", " << line << ", " << column << ") = invalid\n";
267686a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor    return clang_getNullLocation();
267786a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  }
267886a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor
267986a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  if (Logging)
268086a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor    llvm::errs() << "clang_getLocation(\"" << File->getName()
268186a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor                 << "\", " << line << ", " << column << ") = "
268286a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor                 << SLoc.getRawEncoding() << "\n";
268383889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall
268483889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  return cxloc::translateSourceLocation(CXXUnit->getASTContext(), SLoc);
268583889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall}
268683889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall
268783889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid ChisnallCXSourceLocation clang_getLocationForOffset(CXTranslationUnit tu,
268883889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall                                            CXFile file,
268983889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall                                            unsigned offset) {
269083889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  if (!tu || !file)
269183889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall    return clang_getNullLocation();
269283889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall
2693a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
269483889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  SourceLocation Start
269583889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall    = CXXUnit->getSourceManager().getLocation(
269683889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall                                        static_cast<const FileEntry *>(file),
269783889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall                                              1, 1);
269883889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  if (Start.isInvalid()) return clang_getNullLocation();
269983889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall
270083889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  SourceLocation SLoc = Start.getFileLocWithOffset(offset);
270183889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall
270283889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  if (SLoc.isInvalid()) return clang_getNullLocation();
2703f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
27041a9a0bc472ee4fec72ee8be8b575fb66ca600d1bTed Kremenek  return cxloc::translateSourceLocation(CXXUnit->getASTContext(), SLoc);
2705b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2706b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
27075352ac06d8f6194825bb2a99ffa009b61bafb503Douglas GregorCXSourceRange clang_getNullRange() {
27085352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  CXSourceRange Result = { { 0, 0 }, 0, 0 };
27095352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  return Result;
27105352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor}
2711d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar
2712b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXSourceRange clang_getRange(CXSourceLocation begin, CXSourceLocation end) {
27135352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  if (begin.ptr_data[0] != end.ptr_data[0] ||
27145352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor      begin.ptr_data[1] != end.ptr_data[1])
27155352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
2716f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2717f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  CXSourceRange Result = { { begin.ptr_data[0], begin.ptr_data[1] },
27185352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor                           begin.int_data, end.int_data };
2719b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  return Result;
2720b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2721b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
272246766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregorvoid clang_getInstantiationLocation(CXSourceLocation location,
272346766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor                                    CXFile *file,
272446766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor                                    unsigned *line,
272546766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor                                    unsigned *column,
272646766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor                                    unsigned *offset) {
27271db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  SourceLocation Loc = SourceLocation::getFromRawEncoding(location.int_data);
27281db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
2729bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar  if (!location.ptr_data[0] || Loc.isInvalid()) {
273046766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor    if (file)
273146766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor      *file = 0;
273246766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor    if (line)
273346766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor      *line = 0;
273446766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor    if (column)
273546766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor      *column = 0;
273646766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor    if (offset)
273746766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor      *offset = 0;
273846766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor    return;
273946766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor  }
274046766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor
2741bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar  const SourceManager &SM =
2742bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar    *static_cast<const SourceManager*>(location.ptr_data[0]);
27431db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  SourceLocation InstLoc = SM.getInstantiationLoc(Loc);
27441db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
27451db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  if (file)
27461db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor    *file = (void *)SM.getFileEntryForID(SM.getFileID(InstLoc));
27471db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  if (line)
27481db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor    *line = SM.getInstantiationLineNumber(InstLoc);
27491db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  if (column)
27501db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor    *column = SM.getInstantiationColumnNumber(InstLoc);
2751e69517ce61638f12c9abe4605753a45275ac4e37Douglas Gregor  if (offset)
275246766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor    *offset = SM.getDecomposedLoc(InstLoc).second;
2753e69517ce61638f12c9abe4605753a45275ac4e37Douglas Gregor}
2754e69517ce61638f12c9abe4605753a45275ac4e37Douglas Gregor
2755a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregorvoid clang_getSpellingLocation(CXSourceLocation location,
2756a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor                               CXFile *file,
2757a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor                               unsigned *line,
2758a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor                               unsigned *column,
2759a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor                               unsigned *offset) {
2760a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  SourceLocation Loc = SourceLocation::getFromRawEncoding(location.int_data);
2761a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
2762a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (!location.ptr_data[0] || Loc.isInvalid()) {
2763a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    if (file)
2764a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      *file = 0;
2765a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    if (line)
2766a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      *line = 0;
2767a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    if (column)
2768a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      *column = 0;
2769a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    if (offset)
2770a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      *offset = 0;
2771a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    return;
2772a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  }
2773a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
2774a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  const SourceManager &SM =
2775a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *static_cast<const SourceManager*>(location.ptr_data[0]);
2776a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  SourceLocation SpellLoc = Loc;
2777a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (SpellLoc.isMacroID()) {
2778a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    SourceLocation SimpleSpellingLoc = SM.getImmediateSpellingLoc(SpellLoc);
2779a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    if (SimpleSpellingLoc.isFileID() &&
2780a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor        SM.getFileEntryForID(SM.getDecomposedLoc(SimpleSpellingLoc).first))
2781a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      SpellLoc = SimpleSpellingLoc;
2782a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    else
2783a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      SpellLoc = SM.getInstantiationLoc(SpellLoc);
2784a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  }
2785a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
2786a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  std::pair<FileID, unsigned> LocInfo = SM.getDecomposedLoc(SpellLoc);
2787a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  FileID FID = LocInfo.first;
2788a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  unsigned FileOffset = LocInfo.second;
2789a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
2790a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (file)
2791a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *file = (void *)SM.getFileEntryForID(FID);
2792a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (line)
2793a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *line = SM.getLineNumber(FID, FileOffset);
2794a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (column)
2795a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *column = SM.getColumnNumber(FID, FileOffset);
2796a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (offset)
2797a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *offset = FileOffset;
2798a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor}
2799a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
28001db19dea8d221f27be46332d668d1e2decb7f1abDouglas GregorCXSourceLocation clang_getRangeStart(CXSourceRange range) {
2801f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  CXSourceLocation Result = { { range.ptr_data[0], range.ptr_data[1] },
28025352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor                              range.begin_int_data };
28031db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  return Result;
28041db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor}
28051db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
28061db19dea8d221f27be46332d668d1e2decb7f1abDouglas GregorCXSourceLocation clang_getRangeEnd(CXSourceRange range) {
2807bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar  CXSourceLocation Result = { { range.ptr_data[0], range.ptr_data[1] },
28085352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor                              range.end_int_data };
28091db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  return Result;
28101db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor}
28111db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
2812b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor} // end: extern "C"
2813b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
28141db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor//===----------------------------------------------------------------------===//
2815fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXFile Operations.
2816fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2817fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2818fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
281974844072411bae91d5dbb89955d200cbe1e0a1c8Ted KremenekCXString clang_getFileName(CXFile SFile) {
282098258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (!SFile)
2821a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return createCXString((const char*)NULL);
2822f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
282388145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  FileEntry *FEnt = static_cast<FileEntry *>(SFile);
282474844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek  return createCXString(FEnt->getName());
282588145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff}
282688145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff
282788145034694ed5267fa6fa5febc54fadc02bd479Steve Narofftime_t clang_getFileTime(CXFile SFile) {
282898258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (!SFile)
282998258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor    return 0;
2830f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
283188145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  FileEntry *FEnt = static_cast<FileEntry *>(SFile);
283288145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  return FEnt->getModificationTime();
2833ee9405e807d7c447c0143c2bd865b759192e97b3Steve Naroff}
2834f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2835b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXFile clang_getFile(CXTranslationUnit tu, const char *file_name) {
2836b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  if (!tu)
2837b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor    return 0;
2838f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2839a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
2840f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2841b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  FileManager &FMgr = CXXUnit->getFileManager();
284239b49bcaaddb1049234fca9500c0ac02c088e23dChris Lattner  return const_cast<FileEntry *>(FMgr.getFile(file_name));
2843b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2844f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2845fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
2846fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2847fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2848fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXCursor Operations.
2849fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2850fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2851fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekstatic Decl *getDeclFromExpr(Stmt *E) {
2852db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor  if (CastExpr *CE = dyn_cast<CastExpr>(E))
2853db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor    return getDeclFromExpr(CE->getSubExpr());
2854db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2855fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (DeclRefExpr *RefExpr = dyn_cast<DeclRefExpr>(E))
2856fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return RefExpr->getDecl();
285738f28c1189142429384996409fffbc57f67b5c60Douglas Gregor  if (BlockDeclRefExpr *RefExpr = dyn_cast<BlockDeclRefExpr>(E))
285838f28c1189142429384996409fffbc57f67b5c60Douglas Gregor    return RefExpr->getDecl();
2859fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (MemberExpr *ME = dyn_cast<MemberExpr>(E))
2860fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return ME->getMemberDecl();
2861fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (ObjCIvarRefExpr *RE = dyn_cast<ObjCIvarRefExpr>(E))
2862fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return RE->getDecl();
2863db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor  if (ObjCPropertyRefExpr *PRE = dyn_cast<ObjCPropertyRefExpr>(E))
286412f78a6741a4cb3d904340f8d3d2714568b50e7aJohn McCall    return PRE->isExplicitProperty() ? PRE->getExplicitProperty() : 0;
2865db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2866fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (CallExpr *CE = dyn_cast<CallExpr>(E))
2867fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return getDeclFromExpr(CE->getCallee());
286893798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  if (CXXConstructExpr *CE = llvm::dyn_cast<CXXConstructExpr>(E))
286993798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    if (!CE->isElidable())
287093798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    return CE->getConstructor();
2871fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (ObjCMessageExpr *OME = dyn_cast<ObjCMessageExpr>(E))
2872fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return OME->getMethodDecl();
2873f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2874db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor  if (ObjCProtocolExpr *PE = dyn_cast<ObjCProtocolExpr>(E))
2875db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor    return PE->getProtocol();
2876c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor  if (SubstNonTypeTemplateParmPackExpr *NTTP
2877c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor                              = dyn_cast<SubstNonTypeTemplateParmPackExpr>(E))
2878c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor    return NTTP->getParameterPack();
287994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  if (SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
288094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor    if (isa<NonTypeTemplateParmDecl>(SizeOfPack->getPack()) ||
288194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        isa<ParmVarDecl>(SizeOfPack->getPack()))
288294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      return SizeOfPack->getPack();
2883db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2884fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  return 0;
2885fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek}
2886ee9405e807d7c447c0143c2bd865b759192e97b3Steve Naroff
2887c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbarstatic SourceLocation getLocationFromExpr(Expr *E) {
2888c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (ObjCMessageExpr *Msg = dyn_cast<ObjCMessageExpr>(E))
2889c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return /*FIXME:*/Msg->getLeftLoc();
2890c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
2891c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return DRE->getLocation();
289238f28c1189142429384996409fffbc57f67b5c60Douglas Gregor  if (BlockDeclRefExpr *RefExpr = dyn_cast<BlockDeclRefExpr>(E))
289338f28c1189142429384996409fffbc57f67b5c60Douglas Gregor    return RefExpr->getLocation();
2894c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (MemberExpr *Member = dyn_cast<MemberExpr>(E))
2895c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return Member->getMemberLoc();
2896c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (ObjCIvarRefExpr *Ivar = dyn_cast<ObjCIvarRefExpr>(E))
2897c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return Ivar->getLocation();
289894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  if (SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
289994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor    return SizeOfPack->getPackLoc();
290094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
2901c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  return E->getLocStart();
2902c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar}
2903c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar
2904fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
2905f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2906f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekunsigned clang_visitChildren(CXCursor parent,
2907b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor                             CXCursorVisitor visitor,
2908b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor                             CXClientData client_data) {
2909a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CursorVisitor CursorVis(getCursorTU(parent), visitor, client_data,
291004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                          getCursorASTUnit(parent)->getMaxPCHLevel(),
291104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                          false);
2912b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return CursorVis.VisitChildren(parent);
2913b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
2914b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor
29153387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#ifndef __has_feature
29163387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#define __has_feature(x) 0
29173387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#endif
29183387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#if __has_feature(blocks)
29193387c65a094a02b2a94c05111d035a97d3d5c794David Chisnalltypedef enum CXChildVisitResult
29203387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall     (^CXCursorVisitorBlock)(CXCursor cursor, CXCursor parent);
29213387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
29223387c65a094a02b2a94c05111d035a97d3d5c794David Chisnallstatic enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
29233387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall    CXClientData client_data) {
29243387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
29253387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return block(cursor, parent);
29263387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
29273387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#else
29283387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall// If we are compiled with a compiler that doesn't have native blocks support,
29293387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall// define and call the block manually, so the
29303387c65a094a02b2a94c05111d035a97d3d5c794David Chisnalltypedef struct _CXChildVisitResult
29313387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall{
29323387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	void *isa;
29333387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	int flags;
29343387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	int reserved;
29359e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar	enum CXChildVisitResult(*invoke)(struct _CXChildVisitResult*, CXCursor,
29369e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                         CXCursor);
29373387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall} *CXCursorVisitorBlock;
29383387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
29393387c65a094a02b2a94c05111d035a97d3d5c794David Chisnallstatic enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
29403387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall    CXClientData client_data) {
29413387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
29423387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return block->invoke(block, cursor, parent);
29433387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
29443387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#endif
29453387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
29463387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
29479e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbarunsigned clang_visitChildrenWithBlock(CXCursor parent,
29489e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                      CXCursorVisitorBlock block) {
29493387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return clang_visitChildren(parent, visitWithBlock, block);
29503387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
29513387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
295278205d4bada39d95097e766af9eb30cdd0159461Douglas Gregorstatic CXString getDeclSpelling(Decl *D) {
295378205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D);
2954e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor  if (!ND) {
2955e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor    if (ObjCPropertyImplDecl *PropImpl =llvm::dyn_cast<ObjCPropertyImplDecl>(D))
2956e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor      if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
2957e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor        return createCXString(Property->getIdentifier()->getName());
2958e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
2959ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
2960e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor  }
2961e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
296278205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
2963ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(OMD->getSelector().getAsString());
2964f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
296578205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ObjCCategoryImplDecl *CIMP = dyn_cast<ObjCCategoryImplDecl>(ND))
296678205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // No, this isn't the same as the code below. getIdentifier() is non-virtual
296778205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // and returns different names. NamedDecl returns the class name and
296878205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // ObjCCategoryImplDecl returns the category name.
2969ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(CIMP->getIdentifier()->getNameStart());
2970f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
29710a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  if (isa<UsingDirectiveDecl>(D))
29720a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor    return createCXString("");
29730a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
297450aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  llvm::SmallString<1024> S;
297550aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  llvm::raw_svector_ostream os(S);
297650aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  ND->printName(os);
297750aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek
297850aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  return createCXString(os.str());
297978205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor}
2980f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
29819ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXString clang_getCursorSpelling(CXCursor C) {
29827eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  if (clang_isTranslationUnit(C.kind))
2983a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return clang_getTranslationUnitSpelling(
2984a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                            static_cast<CXTranslationUnit>(C.data[2]));
29857eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
2986f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  if (clang_isReference(C.kind)) {
2987f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff    switch (C.kind) {
2988acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCSuperClassRef: {
29892e331b938b38057e333fab0ba841130ea8467794Douglas Gregor      ObjCInterfaceDecl *Super = getCursorObjCSuperClassRef(C).first;
2990ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(Super->getIdentifier()->getNameStart());
2991acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
2992acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCClassRef: {
29931adb082a709f7b588f03672999294e061234b2cfDouglas Gregor      ObjCInterfaceDecl *Class = getCursorObjCClassRef(C).first;
2994ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(Class->getIdentifier()->getNameStart());
2995acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
2996acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCProtocolRef: {
299778db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor      ObjCProtocolDecl *OID = getCursorObjCProtocolRef(C).first;
2998f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      assert(OID && "getCursorSpelling(): Missing protocol decl");
2999ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(OID->getIdentifier()->getNameStart());
3000acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
30013064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
30023064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      CXXBaseSpecifier *B = getCursorCXXBaseSpecifier(C);
30033064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      return createCXString(B->getType().getAsString());
30043064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
30057d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    case CXCursor_TypeRef: {
30067d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      TypeDecl *Type = getCursorTypeRef(C).first;
30077d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      assert(Type && "Missing type decl");
30087d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3009ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(getCursorContext(C).getTypeDeclType(Type).
3010ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek                              getAsString());
30117d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    }
30120b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef: {
30130b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      TemplateDecl *Template = getCursorTemplateRef(C).first;
30146931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      assert(Template && "Missing template decl");
30150b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
30160b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return createCXString(Template->getNameAsString());
30170b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    }
30186931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
30196931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef: {
30206931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      NamedDecl *NS = getCursorNamespaceRef(C).first;
30216931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      assert(NS && "Missing namespace decl");
30226931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
30236931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return createCXString(NS->getNameAsString());
30246931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    }
30257d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3026a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef: {
3027a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      FieldDecl *Field = getCursorMemberRef(C).first;
3028a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      assert(Field && "Missing member decl");
3029a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
3030a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return createCXString(Field->getNameAsString());
3031a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
3032a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
303336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef: {
303436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      LabelStmt *Label = getCursorLabelRef(C).first;
303536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      assert(Label && "Missing label");
303636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
3037ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner      return createCXString(Label->getName());
303836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    }
303936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
30401f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef: {
30411f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
30421f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (Decl *D = Storage.dyn_cast<Decl *>()) {
30431f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        if (NamedDecl *ND = dyn_cast<NamedDecl>(D))
30441f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor          return createCXString(ND->getNameAsString());
30451f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString("");
30461f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      }
30471f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
30481f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString(E->getName().getAsString());
30491f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      OverloadedTemplateStorage *Ovl
30501f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        = Storage.get<OverloadedTemplateStorage*>();
30511f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (Ovl->size() == 0)
30521f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString("");
30531f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return createCXString((*Ovl->begin())->getNameAsString());
30541f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    }
30551f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3056acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    default:
3057ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString("<not implemented>");
3058f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff    }
3059f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  }
306097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
306197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind)) {
306297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    Decl *D = getDeclFromExpr(getCursorExpr(C));
306397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    if (D)
306478205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor      return getDeclSpelling(D);
3065ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
306697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  }
306797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
306836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind)) {
306936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    Stmt *S = getCursorStmt(C);
307036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    if (LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S))
3071ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner      return createCXString(Label->getName());
307236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
307336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return createCXString("");
307436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  }
307536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
30764ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor  if (C.kind == CXCursor_MacroInstantiation)
30774ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor    return createCXString(getCursorMacroInstantiation(C)->getName()
30784ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor                                                           ->getNameStart());
30794ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor
3080572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition)
3081572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return createCXString(getCursorMacroDefinition(C)->getName()
3082572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor                                                           ->getNameStart());
3083572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3084ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective)
3085ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return createCXString(getCursorInclusionDirective(C)->getFileName());
3086ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
308760cbfacd947590f83257a4191566dda92fbbde69Douglas Gregor  if (clang_isDeclaration(C.kind))
308860cbfacd947590f83257a4191566dda92fbbde69Douglas Gregor    return getDeclSpelling(getCursorDecl(C));
3089e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3090ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString("");
3091f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff}
3092f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff
3093358559d8d7b458c5f64941842383a16e61f0828dDouglas GregorCXString clang_getCursorDisplayName(CXCursor C) {
3094358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (!clang_isDeclaration(C.kind))
3095358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return clang_getCursorSpelling(C);
3096358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3097358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  Decl *D = getCursorDecl(C);
3098358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (!D)
3099358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString("");
3100358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3101358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  PrintingPolicy &Policy = getCursorContext(C).PrintingPolicy;
3102358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(D))
3103358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    D = FunTmpl->getTemplatedDecl();
3104358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3105358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (FunctionDecl *Function = dyn_cast<FunctionDecl>(D)) {
3106358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::SmallString<64> Str;
3107358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3108358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << Function->getNameAsString();
3109358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (Function->getPrimaryTemplate())
3110358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << "<>";
3111358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << "(";
3112358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    for (unsigned I = 0, N = Function->getNumParams(); I != N; ++I) {
3113358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (I)
3114358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3115358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << Function->getParamDecl(I)->getType().getAsString(Policy);
3116358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3117358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3118358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (Function->isVariadic()) {
3119358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (Function->getNumParams())
3120358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3121358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << "...";
3122358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3123358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ")";
3124358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3125358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3126358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3127358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D)) {
3128358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::SmallString<64> Str;
3129358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3130358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ClassTemplate->getNameAsString();
3131358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << "<";
3132358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    TemplateParameterList *Params = ClassTemplate->getTemplateParameters();
3133358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    for (unsigned I = 0, N = Params->size(); I != N; ++I) {
3134358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (I)
3135358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3136358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3137358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      NamedDecl *Param = Params->getParam(I);
3138358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (Param->getIdentifier()) {
3139358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << Param->getIdentifier()->getName();
3140358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        continue;
3141358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      }
3142358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3143358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      // There is no parameter name, which makes this tricky. Try to come up
3144358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      // with something useful that isn't too long.
3145358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
3146358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << (TTP->wasDeclaredWithTypename()? "typename" : "class");
3147358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      else if (NonTypeTemplateParmDecl *NTTP
3148358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                                    = dyn_cast<NonTypeTemplateParmDecl>(Param))
3149358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << NTTP->getType().getAsString(Policy);
3150358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      else
3151358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << "template<...> class";
3152358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3153358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3154358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ">";
3155358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3156358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3157358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3158358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (ClassTemplateSpecializationDecl *ClassSpec
3159358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                              = dyn_cast<ClassTemplateSpecializationDecl>(D)) {
3160358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    // If the type was explicitly written, use that.
3161358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (TypeSourceInfo *TSInfo = ClassSpec->getTypeAsWritten())
3162358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      return createCXString(TSInfo->getType().getAsString(Policy));
3163358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3164358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::SmallString<64> Str;
3165358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3166358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ClassSpec->getNameAsString();
3167358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << TemplateSpecializationType::PrintTemplateArgumentList(
3168910f8008fea79120489a53593fe971b0b8a4a740Douglas Gregor                                      ClassSpec->getTemplateArgs().data(),
3169910f8008fea79120489a53593fe971b0b8a4a740Douglas Gregor                                      ClassSpec->getTemplateArgs().size(),
3170358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                                                                Policy);
3171358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3172358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3173358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3174358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  return clang_getCursorSpelling(C);
3175358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor}
3176358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3177e68fff6fc083c6270d835216a3de0b82c6ef0310Ted KremenekCXString clang_getCursorKindSpelling(enum CXCursorKind Kind) {
317889922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  switch (Kind) {
3179e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_FunctionDecl:
3180e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("FunctionDecl");
3181e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TypedefDecl:
3182e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TypedefDecl");
3183e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_EnumDecl:
3184e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("EnumDecl");
3185e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_EnumConstantDecl:
3186e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("EnumConstantDecl");
3187e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_StructDecl:
3188e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("StructDecl");
3189e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnionDecl:
3190e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnionDecl");
3191e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ClassDecl:
3192e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ClassDecl");
3193e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_FieldDecl:
3194e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("FieldDecl");
3195e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_VarDecl:
3196e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("VarDecl");
3197e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ParmDecl:
3198e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ParmDecl");
3199e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCInterfaceDecl:
3200e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCInterfaceDecl");
3201e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCCategoryDecl:
3202e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCCategoryDecl");
3203e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCProtocolDecl:
3204e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCProtocolDecl");
3205e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCPropertyDecl:
3206e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCPropertyDecl");
3207e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCIvarDecl:
3208e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCIvarDecl");
3209e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCInstanceMethodDecl:
3210e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCInstanceMethodDecl");
3211e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCClassMethodDecl:
3212e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCClassMethodDecl");
3213e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCImplementationDecl:
3214e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCImplementationDecl");
3215e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCCategoryImplDecl:
3216e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCCategoryImplDecl");
32178bd5a69999cfd06b6b5a58fdd04e4f802b2df5a4Ted Kremenek  case CXCursor_CXXMethod:
32188bd5a69999cfd06b6b5a58fdd04e4f802b2df5a4Ted Kremenek      return createCXString("CXXMethod");
3219e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedDecl:
3220e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedDecl");
3221e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCSuperClassRef:
3222e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCSuperClassRef");
3223e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCProtocolRef:
3224e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCProtocolRef");
3225e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCClassRef:
3226e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCClassRef");
3227e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TypeRef:
3228e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TypeRef");
32290b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case CXCursor_TemplateRef:
32300b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return createCXString("TemplateRef");
32316931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  case CXCursor_NamespaceRef:
32326931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    return createCXString("NamespaceRef");
3233a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  case CXCursor_MemberRef:
3234a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return createCXString("MemberRef");
323536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  case CXCursor_LabelRef:
323636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return createCXString("LabelRef");
32371f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case CXCursor_OverloadedDeclRef:
32381f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return createCXString("OverloadedDeclRef");
3239e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedExpr:
3240e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedExpr");
32411ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  case CXCursor_BlockExpr:
32421ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek      return createCXString("BlockExpr");
3243e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_DeclRefExpr:
3244e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("DeclRefExpr");
3245e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_MemberRefExpr:
3246e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("MemberRefExpr");
3247e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_CallExpr:
3248e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("CallExpr");
3249e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCMessageExpr:
3250e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCMessageExpr");
3251e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedStmt:
3252e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedStmt");
325336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  case CXCursor_LabelStmt:
325436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return createCXString("LabelStmt");
3255e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_InvalidFile:
3256e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("InvalidFile");
3257292db6401f040795db3ea4e00fc02622d6c3ba1dTed Kremenek  case CXCursor_InvalidCode:
3258292db6401f040795db3ea4e00fc02622d6c3ba1dTed Kremenek    return createCXString("InvalidCode");
3259e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_NoDeclFound:
3260e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("NoDeclFound");
3261e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_NotImplemented:
3262e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("NotImplemented");
3263e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TranslationUnit:
3264e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TranslationUnit");
3265e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek  case CXCursor_UnexposedAttr:
3266e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek      return createCXString("UnexposedAttr");
3267e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek  case CXCursor_IBActionAttr:
3268e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek      return createCXString("attribute(ibaction)");
32699f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  case CXCursor_IBOutletAttr:
32709f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor     return createCXString("attribute(iboutlet)");
3271857e918a8a40deb128840308a318bf623d68295fTed Kremenek  case CXCursor_IBOutletCollectionAttr:
3272857e918a8a40deb128840308a318bf623d68295fTed Kremenek      return createCXString("attribute(iboutletcollection)");
32739f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  case CXCursor_PreprocessingDirective:
32749f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    return createCXString("preprocessing directive");
3275572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  case CXCursor_MacroDefinition:
3276572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return createCXString("macro definition");
32774807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  case CXCursor_MacroInstantiation:
32784807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor    return createCXString("macro instantiation");
3279ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  case CXCursor_InclusionDirective:
3280ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return createCXString("inclusion directive");
32818f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  case CXCursor_Namespace:
32828f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek    return createCXString("Namespace");
3283a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  case CXCursor_LinkageSpec:
3284a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek    return createCXString("LinkageSpec");
32853064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  case CXCursor_CXXBaseSpecifier:
32863064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    return createCXString("C++ base class specifier");
328701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_Constructor:
328801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXConstructor");
328901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_Destructor:
329001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXDestructor");
329101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_ConversionFunction:
329201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXConversion");
3293fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_TemplateTypeParameter:
3294fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("TemplateTypeParameter");
3295fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_NonTypeTemplateParameter:
3296fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("NonTypeTemplateParameter");
3297fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_TemplateTemplateParameter:
3298fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("TemplateTemplateParameter");
3299fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_FunctionTemplate:
3300fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("FunctionTemplate");
330139d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  case CXCursor_ClassTemplate:
330239d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor    return createCXString("ClassTemplate");
330374dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  case CXCursor_ClassTemplatePartialSpecialization:
330474dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    return createCXString("ClassTemplatePartialSpecialization");
33056931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  case CXCursor_NamespaceAlias:
33066931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    return createCXString("NamespaceAlias");
33070a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  case CXCursor_UsingDirective:
33080a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor    return createCXString("UsingDirective");
33097e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  case CXCursor_UsingDeclaration:
33107e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor    return createCXString("UsingDeclaration");
331189922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  }
3312e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3313deb06bd3566e18f677e76bc435d478b033fe328bTed Kremenek  llvm_unreachable("Unhandled CXCursorKind");
3314a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return createCXString((const char*) 0);
3315600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
331689922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff
3317e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenekenum CXChildVisitResult GetCursorVisitor(CXCursor cursor,
3318e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek                                         CXCursor parent,
331933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor                                         CXClientData client_data) {
332033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  CXCursor *BestCursor = static_cast<CXCursor *>(client_data);
332193798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor
332293798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // If our current best cursor is the construction of a temporary object,
332393798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // don't replace that cursor with a type reference, because we want
332493798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // clang_getCursor() to point at the constructor.
332593798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  if (clang_isExpression(BestCursor->kind) &&
332693798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor      isa<CXXTemporaryObjectExpr>(getCursorExpr(*BestCursor)) &&
332793798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor      cursor.kind == CXCursor_TypeRef)
332893798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    return CXChildVisit_Recurse;
332993798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor
333085fe1560b061b5f93a52dbd07cddd6e808854710Douglas Gregor  // Don't override a preprocessing cursor with another preprocessing
333185fe1560b061b5f93a52dbd07cddd6e808854710Douglas Gregor  // cursor; we want the outermost preprocessing cursor.
333285fe1560b061b5f93a52dbd07cddd6e808854710Douglas Gregor  if (clang_isPreprocessing(cursor.kind) &&
333385fe1560b061b5f93a52dbd07cddd6e808854710Douglas Gregor      clang_isPreprocessing(BestCursor->kind))
333485fe1560b061b5f93a52dbd07cddd6e808854710Douglas Gregor    return CXChildVisit_Recurse;
333585fe1560b061b5f93a52dbd07cddd6e808854710Douglas Gregor
333633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  *BestCursor = cursor;
333733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  return CXChildVisit_Recurse;
333833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
3339e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3340b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXCursor clang_getCursor(CXTranslationUnit TU, CXSourceLocation Loc) {
3341b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  if (!TU)
3342f462989fe8d6f59ab2d7d0fe2b4b96292ce706eaTed Kremenek    return clang_getNullCursor();
3343e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3344a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
3345bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
3346bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor
3347a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  // Translate the given source location to make it point at the beginning of
3348a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  // the token under the cursor.
3349a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek  SourceLocation SLoc = cxloc::translateSourceLocation(Loc);
3350a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek
3351a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek  // Guard against an invalid SourceLocation, or we may assert in one
3352a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek  // of the following calls.
3353a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek  if (SLoc.isInvalid())
3354a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek    return clang_getNullCursor();
3355a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek
335640749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  bool Logging = getenv("LIBCLANG_LOGGING");
3357a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  SLoc = Lexer::GetBeginningOfToken(SLoc, CXXUnit->getSourceManager(),
3358a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor                                    CXXUnit->getASTContext().getLangOptions());
3359a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
336033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  CXCursor Result = MakeCXCursorInvalid(CXCursor_NoDeclFound);
336133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (SLoc.isValid()) {
336233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    // FIXME: Would be great to have a "hint" cursor, then walk from that
336333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    // hint cursor upward until we find a cursor whose source range encloses
336433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    // the region of interest, rather than starting from the translation unit.
3365a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    CXCursor Parent = clang_getTranslationUnitCursor(TU);
3366a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    CursorVisitor CursorVis(TU, GetCursorVisitor, &Result,
336704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                            Decl::MaxPCHLevel, true, SourceLocation(SLoc));
336833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    CursorVis.VisitChildren(Parent);
336977128ddd3077fc045751a55bb3226802b15d5510Steve Naroff  }
337040749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
337140749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  if (Logging) {
337240749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXFile SearchFile;
337340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    unsigned SearchLine, SearchColumn;
337440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXFile ResultFile;
337540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    unsigned ResultLine, ResultColumn;
33766653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    CXString SearchFileName, ResultFileName, KindSpelling, USR;
33776653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    const char *IsDef = clang_isCursorDefinition(Result)? " (Definition)" : "";
337840749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXSourceLocation ResultLoc = clang_getCursorLocation(Result);
337940749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
338040749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_getInstantiationLocation(Loc, &SearchFile, &SearchLine, &SearchColumn,
338140749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor                                   0);
338240749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_getInstantiationLocation(ResultLoc, &ResultFile, &ResultLine,
338340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor                                   &ResultColumn, 0);
338440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    SearchFileName = clang_getFileName(SearchFile);
338540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    ResultFileName = clang_getFileName(ResultFile);
338640749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    KindSpelling = clang_getCursorKindSpelling(Result.kind);
33876653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    USR = clang_getCursorUSR(Result);
33886653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    fprintf(stderr, "clang_getCursor(%s:%d:%d) = %s(%s:%d:%d):%s%s\n",
338940749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor            clang_getCString(SearchFileName), SearchLine, SearchColumn,
339040749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor            clang_getCString(KindSpelling),
33916653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor            clang_getCString(ResultFileName), ResultLine, ResultColumn,
33926653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor            clang_getCString(USR), IsDef);
339340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(SearchFileName);
339440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(ResultFileName);
339540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(KindSpelling);
33966653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    clang_disposeString(USR);
33970aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor
33980aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    CXCursor Definition = clang_getCursorDefinition(Result);
33990aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    if (!clang_equalCursors(Definition, clang_getNullCursor())) {
34000aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXSourceLocation DefinitionLoc = clang_getCursorLocation(Definition);
34010aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXString DefinitionKindSpelling
34020aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor                                = clang_getCursorKindSpelling(Definition.kind);
34030aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXFile DefinitionFile;
34040aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      unsigned DefinitionLine, DefinitionColumn;
34050aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_getInstantiationLocation(DefinitionLoc, &DefinitionFile,
34060aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor                                     &DefinitionLine, &DefinitionColumn, 0);
34070aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXString DefinitionFileName = clang_getFileName(DefinitionFile);
34080aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      fprintf(stderr, "  -> %s(%s:%d:%d)\n",
34090aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              clang_getCString(DefinitionKindSpelling),
34100aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              clang_getCString(DefinitionFileName),
34110aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              DefinitionLine, DefinitionColumn);
34120aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_disposeString(DefinitionFileName);
34130aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_disposeString(DefinitionKindSpelling);
34140aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    }
341540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  }
341640749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
3417e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  return Result;
341877128ddd3077fc045751a55bb3226802b15d5510Steve Naroff}
341977128ddd3077fc045751a55bb3226802b15d5510Steve Naroff
3420738855554394a6afcf39cc8345fd22c3756b8dd0Ted KremenekCXCursor clang_getNullCursor(void) {
34215bfb8c128c2ac8eb4032afc180cdc400a0f953caDouglas Gregor  return MakeCXCursorInvalid(CXCursor_InvalidFile);
3422738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek}
3423738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek
3424738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenekunsigned clang_equalCursors(CXCursor X, CXCursor Y) {
3425283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  return X == Y;
3426738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek}
34270d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
34289ce5584553054d0cb934940586aca0186e87fa57Douglas Gregorunsigned clang_hashCursor(CXCursor C) {
34299ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  unsigned Index = 0;
34309ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  if (clang_isExpression(C.kind) || clang_isStatement(C.kind))
34319ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor    Index = 1;
34329ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor
34339ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  return llvm::DenseMapInfo<std::pair<unsigned, void*> >::getHashValue(
34349ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor                                        std::make_pair(C.kind, C.data[Index]));
34359ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor}
34369ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor
34379ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isInvalid(enum CXCursorKind K) {
343877128ddd3077fc045751a55bb3226802b15d5510Steve Naroff  return K >= CXCursor_FirstInvalid && K <= CXCursor_LastInvalid;
343977128ddd3077fc045751a55bb3226802b15d5510Steve Naroff}
344077128ddd3077fc045751a55bb3226802b15d5510Steve Naroff
34419ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isDeclaration(enum CXCursorKind K) {
344289922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  return K >= CXCursor_FirstDecl && K <= CXCursor_LastDecl;
344389922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff}
34442d4d629d8a0de5112c7ae9d05c03ddbf6dcd956aSteve Naroff
34459ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isReference(enum CXCursorKind K) {
3446f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  return K >= CXCursor_FirstRef && K <= CXCursor_LastRef;
3447f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff}
3448f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff
344997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregorunsigned clang_isExpression(enum CXCursorKind K) {
345097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  return K >= CXCursor_FirstExpr && K <= CXCursor_LastExpr;
345197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor}
345297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
345397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregorunsigned clang_isStatement(enum CXCursorKind K) {
345497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  return K >= CXCursor_FirstStmt && K <= CXCursor_LastStmt;
345597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor}
345697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
34577eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregorunsigned clang_isTranslationUnit(enum CXCursorKind K) {
34587eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  return K == CXCursor_TranslationUnit;
34597eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor}
34607eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
34619f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregorunsigned clang_isPreprocessing(enum CXCursorKind K) {
34629f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  return K >= CXCursor_FirstPreprocessing && K <= CXCursor_LastPreprocessing;
34639f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor}
34649f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor
3465ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenekunsigned clang_isUnexposed(enum CXCursorKind K) {
3466ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek  switch (K) {
3467ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedDecl:
3468ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedExpr:
3469ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedStmt:
3470ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedAttr:
3471ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek      return true;
3472ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    default:
3473ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek      return false;
3474ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek  }
3475ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek}
3476ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek
34779ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXCursorKind clang_getCursorKind(CXCursor C) {
34789efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff  return C.kind;
34799efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff}
34809efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff
348198258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas GregorCXSourceLocation clang_getCursorLocation(CXCursor C) {
348298258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (clang_isReference(C.kind)) {
3483f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    switch (C.kind) {
3484f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCSuperClassRef: {
3485f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCInterfaceDecl *, SourceLocation> P
3486f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCSuperClassRef(C);
3487a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3488f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
3489f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor
3490f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCProtocolRef: {
3491f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCProtocolDecl *, SourceLocation> P
3492f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCProtocolRef(C);
3493a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3494f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
3495f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor
3496f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCClassRef: {
3497f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCInterfaceDecl *, SourceLocation> P
3498f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCClassRef(C);
3499a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3500f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
35017d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3502f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_TypeRef: {
35037d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      std::pair<TypeDecl *, SourceLocation> P = getCursorTypeRef(C);
3504a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
35057d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    }
35060b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
35070b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef: {
35080b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      std::pair<TemplateDecl *, SourceLocation> P = getCursorTemplateRef(C);
35090b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
35100b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    }
35110b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
35126931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef: {
35136931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      std::pair<NamedDecl *, SourceLocation> P = getCursorNamespaceRef(C);
35146931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
35156931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    }
35166931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
3517a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef: {
3518a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      std::pair<FieldDecl *, SourceLocation> P = getCursorMemberRef(C);
3519a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3520a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
3521a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
35223064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
35231b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      CXXBaseSpecifier *BaseSpec = getCursorCXXBaseSpecifier(C);
35241b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      if (!BaseSpec)
35251b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor        return clang_getNullLocation();
35261b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor
35271b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      if (TypeSourceInfo *TSInfo = BaseSpec->getTypeSourceInfo())
35281b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor        return cxloc::translateSourceLocation(getCursorContext(C),
35291b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor                                            TSInfo->getTypeLoc().getBeginLoc());
35301b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor
35311b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C),
35321b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor                                        BaseSpec->getSourceRange().getBegin());
35333064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
3534f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
353536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef: {
353636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      std::pair<LabelStmt *, SourceLocation> P = getCursorLabelRef(C);
353736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C), P.second);
353836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    }
353936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
35401f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
35411f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C),
35421f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                                          getCursorOverloadedDeclRef(C).second);
35431f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3544f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    default:
3545f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      // FIXME: Need a way to enumerate all non-reference cases.
3546f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      llvm_unreachable("Missed a reference kind");
3547f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
354898258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  }
354997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
355097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind))
3551f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    return cxloc::translateSourceLocation(getCursorContext(C),
355297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor                                   getLocationFromExpr(getCursorExpr(C)));
355397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
355436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind))
355536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C),
355636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor                                          getCursorStmt(C)->getLocStart());
355736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
35589f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  if (C.kind == CXCursor_PreprocessingDirective) {
35599f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    SourceLocation L = cxcursor::getCursorPreprocessingDirective(C).getBegin();
35609f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
35619f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  }
35624807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor
35634807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  if (C.kind == CXCursor_MacroInstantiation) {
35644ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor    SourceLocation L
35654ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor      = cxcursor::getCursorMacroInstantiation(C)->getSourceRange().getBegin();
35664807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
35674807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  }
3568572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3569572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition) {
3570572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    SourceLocation L = cxcursor::getCursorMacroDefinition(C)->getLocation();
3571572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
3572572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  }
3573ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3574ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective) {
3575ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    SourceLocation L
3576ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor      = cxcursor::getCursorInclusionDirective(C)->getSourceRange().getBegin();
3577ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
3578ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  }
3579ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
35809a700d277c38d9afaa7cb3fe93a714bfe9b62eecTed Kremenek  if (C.kind < CXCursor_FirstDecl || C.kind > CXCursor_LastDecl)
35815352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullLocation();
358298258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor
3583f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  Decl *D = getCursorDecl(C);
3584f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  SourceLocation Loc = D->getLocation();
3585f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(D))
3586f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    Loc = Class->getClassLoc();
3587007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // FIXME: Multiple variables declared in a single declaration
3588007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // currently lack the information needed to correctly determine their
3589007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // ranges when accounting for the type-specifier.  We use context
3590007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
3591007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // and if so, whether it is the first decl.
3592007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
3593007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    if (!cxcursor::isFirstInDeclGroup(C))
3594007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek      Loc = VD->getLocation();
3595007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  }
3596007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek
35972ca54feee89d7277fb967e3247a64f40ef155a82Douglas Gregor  return cxloc::translateSourceLocation(getCursorContext(C), Loc);
359888145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff}
3599a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor
3600a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor} // end extern "C"
3601a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
3602a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorstatic SourceRange getRawCursorExtent(CXCursor C) {
3603a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  if (clang_isReference(C.kind)) {
3604a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor    switch (C.kind) {
3605a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCSuperClassRef:
3606a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return  getCursorObjCSuperClassRef(C).second;
3607f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3608a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCProtocolRef:
3609a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorObjCProtocolRef(C).second;
3610f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3611a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCClassRef:
3612a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorObjCClassRef(C).second;
36137d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3614a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_TypeRef:
3615a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorTypeRef(C).second;
36160b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
36170b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef:
36180b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return getCursorTemplateRef(C).second;
36190b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
36206931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef:
36216931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return getCursorNamespaceRef(C).second;
3622a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
3623a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef:
3624a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return getCursorMemberRef(C).second;
3625a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
36263064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier:
36271b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      return getCursorCXXBaseSpecifier(C)->getSourceRange();
3628f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
362936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef:
363036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return getCursorLabelRef(C).second;
363136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
36321f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
36331f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return getCursorOverloadedDeclRef(C).second;
36341f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3635a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    default:
3636a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      // FIXME: Need a way to enumerate all non-reference cases.
3637a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      llvm_unreachable("Missed a reference kind");
3638a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor    }
3639a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  }
364097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
364197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind))
3642a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return getCursorExpr(C)->getSourceRange();
364333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
364433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (clang_isStatement(C.kind))
3645a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return getCursorStmt(C)->getSourceRange();
3646f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3647a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (C.kind == CXCursor_PreprocessingDirective)
3648a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return cxcursor::getCursorPreprocessingDirective(C);
36494807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor
3650a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (C.kind == CXCursor_MacroInstantiation)
3651a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return cxcursor::getCursorMacroInstantiation(C)->getSourceRange();
3652572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3653a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (C.kind == CXCursor_MacroDefinition)
3654a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return cxcursor::getCursorMacroDefinition(C)->getSourceRange();
3655ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3656ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective)
3657ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return cxcursor::getCursorInclusionDirective(C)->getSourceRange();
3658ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3659007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  if (C.kind >= CXCursor_FirstDecl && C.kind <= CXCursor_LastDecl) {
3660007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    Decl *D = cxcursor::getCursorDecl(C);
3661007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    SourceRange R = D->getSourceRange();
3662007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // FIXME: Multiple variables declared in a single declaration
3663007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // currently lack the information needed to correctly determine their
3664007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // ranges when accounting for the type-specifier.  We use context
3665007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
3666007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // and if so, whether it is the first decl.
3667007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
3668007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek      if (!cxcursor::isFirstInDeclGroup(C))
3669007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek        R.setBegin(VD->getLocation());
3670007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    }
3671007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    return R;
3672007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  }
36736653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  return SourceRange();
36746653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor}
36756653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
36766653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor/// \brief Retrieves the "raw" cursor extent, which is then extended to include
36776653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor/// the decl-specifier-seq for declarations.
36786653798ff5ce6deb58112777e21307ccc453133dDouglas Gregorstatic SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr) {
36796653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  if (C.kind >= CXCursor_FirstDecl && C.kind <= CXCursor_LastDecl) {
36806653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    Decl *D = cxcursor::getCursorDecl(C);
36816653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    SourceRange R = D->getSourceRange();
36822494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
36832494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // Adjust the start of the location for declarations preceded by
36842494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // declaration specifiers.
36852494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    SourceLocation StartLoc;
36866653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    if (const DeclaratorDecl *DD = dyn_cast<DeclaratorDecl>(D)) {
36872494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
36882494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
36892494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    } else if (TypedefDecl *Typedef = dyn_cast<TypedefDecl>(D)) {
36902494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
36912494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
36922494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    }
36936653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
36942494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (StartLoc.isValid() && R.getBegin().isValid() &&
36952494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        SrcMgr.isBeforeInTranslationUnit(StartLoc, R.getBegin()))
36962494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      R.setBegin(StartLoc);
36972494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
36982494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // FIXME: Multiple variables declared in a single declaration
36992494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // currently lack the information needed to correctly determine their
37002494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // ranges when accounting for the type-specifier.  We use context
37012494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
37022494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // and if so, whether it is the first decl.
37032494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
37042494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (!cxcursor::isFirstInDeclGroup(C))
37052494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        R.setBegin(VD->getLocation());
37066653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    }
37076653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
37086653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    return R;
37096653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  }
37106653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
37116653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  return getRawCursorExtent(C);
37126653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor}
3713a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
3714a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorextern "C" {
3715a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
3716a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas GregorCXSourceRange clang_getCursorExtent(CXCursor C) {
3717a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  SourceRange R = getRawCursorExtent(C);
3718a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (R.isInvalid())
37195352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
3720f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3721a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  return cxloc::translateSourceRange(getCursorContext(C), R);
3722a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor}
3723c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor
3724c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas GregorCXCursor clang_getCursorReferenced(CXCursor C) {
3725b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (clang_isInvalid(C.kind))
3726b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getNullCursor();
3727f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3728a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit tu = getCursorTU(C);
37291f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (clang_isDeclaration(C.kind)) {
37301f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Decl *D = getCursorDecl(C);
37311f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (UsingDecl *Using = dyn_cast<UsingDecl>(D))
3732a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Using, D->getLocation(), tu);
37331f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (ObjCClassDecl *Classes = dyn_cast<ObjCClassDecl>(D))
3734a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Classes, D->getLocation(), tu);
37351f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (ObjCForwardProtocolDecl *Protocols
37361f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                                        = dyn_cast<ObjCForwardProtocolDecl>(D))
3737a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Protocols, D->getLocation(), tu);
3738e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor    if (ObjCPropertyImplDecl *PropImpl =llvm::dyn_cast<ObjCPropertyImplDecl>(D))
3739e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor      if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
3740e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor        return MakeCXCursor(Property, tu);
3741e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
3742c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    return C;
37431f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  }
37441f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
374597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind)) {
37461f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Expr *E = getCursorExpr(C);
37471f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Decl *D = getDeclFromExpr(E);
374897b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    if (D)
3749a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(D, tu);
37501f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
37511f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (OverloadExpr *Ovl = dyn_cast_or_null<OverloadExpr>(E))
3752a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Ovl, tu);
37531f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
375497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    return clang_getNullCursor();
375597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  }
375697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
375736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind)) {
375836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    Stmt *S = getCursorStmt(C);
375936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    if (GotoStmt *Goto = dyn_cast_or_null<GotoStmt>(S))
376037c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek      if (LabelDecl *label = Goto->getLabel())
376137c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek        if (LabelStmt *labelS = label->getStmt())
376237c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek        return MakeCXCursor(labelS, getCursorDecl(C), tu);
376336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
376436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return clang_getNullCursor();
376536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  }
376636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
3767bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor  if (C.kind == CXCursor_MacroInstantiation) {
3768bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor    if (MacroDefinition *Def = getCursorMacroInstantiation(C)->getDefinition())
3769a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeMacroDefinitionCursor(Def, tu);
3770bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor  }
3771bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor
3772c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  if (!clang_isReference(C.kind))
3773c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    return clang_getNullCursor();
3774f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3775c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  switch (C.kind) {
3776c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    case CXCursor_ObjCSuperClassRef:
3777a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorObjCSuperClassRef(C).first, tu);
3778f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3779f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCProtocolRef: {
3780a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorObjCProtocolRef(C).first, tu);
3781f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3782f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCClassRef:
3783a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorObjCClassRef(C).first, tu );
37847d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3785f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_TypeRef:
3786a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorTypeRef(C).first, tu );
37870b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
37880b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef:
3789a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorTemplateRef(C).first, tu );
37900b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
37916931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef:
3792a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorNamespaceRef(C).first, tu );
37936931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
3794a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef:
3795a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorMemberRef(C).first, tu );
3796a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
37973064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
37983064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      CXXBaseSpecifier *B = cxcursor::getCursorCXXBaseSpecifier(C);
37993064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      return clang_getTypeDeclaration(cxtype::MakeCXType(B->getType(),
3800a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                                         tu ));
38013064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
3802f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
380336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef:
380436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      // FIXME: We end up faking the "parent" declaration here because we
380536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      // don't want to make CXCursor larger.
380636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return MakeCXCursor(getCursorLabelRef(C).first,
3807a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek               static_cast<ASTUnit*>(tu->TUData)->getASTContext()
3808a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          .getTranslationUnitDecl(),
3809a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          tu);
381036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
38111f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
38121f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return C;
38131f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3814c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    default:
3815c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      // We would prefer to enumerate all non-reference cursor kinds here.
3816c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      llvm_unreachable("Unhandled reference cursor kind");
3817c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      break;
3818c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    }
3819c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  }
3820f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3821c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  return clang_getNullCursor();
3822c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor}
3823c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor
3824b699866820102a69d83d6ac6941985c5ef4e8c40Douglas GregorCXCursor clang_getCursorDefinition(CXCursor C) {
3825b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (clang_isInvalid(C.kind))
3826b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getNullCursor();
3827f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3828a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(C);
3829f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3830b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  bool WasReference = false;
383197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isReference(C.kind) || clang_isExpression(C.kind)) {
3832b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    C = clang_getCursorReferenced(C);
3833b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    WasReference = true;
3834b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
3835b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3836bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor  if (C.kind == CXCursor_MacroInstantiation)
3837bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor    return clang_getCursorReferenced(C);
3838bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor
3839b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!clang_isDeclaration(C.kind))
3840b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3841b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3842b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  Decl *D = getCursorDecl(C);
3843b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!D)
3844b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3845f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3846b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  switch (D->getKind()) {
3847b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds that don't really separate the notions of
3848b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // declaration and definition.
3849b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Namespace:
3850b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Typedef:
3851b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TemplateTypeParm:
3852b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::EnumConstant:
3853b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Field:
3854d98114647e16796a976b04af79975b4f0eacf22bBenjamin Kramer  case Decl::IndirectField:
3855b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCIvar:
3856b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCAtDefsField:
3857b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ImplicitParam:
3858b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ParmVar:
3859b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::NonTypeTemplateParm:
3860b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TemplateTemplateParm:
3861b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCategoryImpl:
3862b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCImplementation:
38636206d53f67613958ae1b023aba337ebb46f11a8bAbramo Bagnara  case Decl::AccessSpec:
3864b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::LinkageSpec:
3865b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCPropertyImpl:
3866b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FileScopeAsm:
3867b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::StaticAssert:
3868b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Block:
3869ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  case Decl::Label:  // FIXME: Is this right??
3870b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return C;
3871b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3872b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds that don't make any sense here, but are
3873b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // nonetheless harmless.
3874b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TranslationUnit:
3875b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    break;
3876b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3877b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds for which the definition is not resolvable.
3878b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UnresolvedUsingTypename:
3879b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UnresolvedUsingValue:
3880b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    break;
3881b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3882b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UsingDirective:
3883b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return MakeCXCursor(cast<UsingDirectiveDecl>(D)->getNominatedNamespace(),
3884a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                        TU);
3885b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3886b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::NamespaceAlias:
3887a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(cast<NamespaceAliasDecl>(D)->getNamespace(), TU);
3888b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3889b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Enum:
3890b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Record:
3891b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXRecord:
3892b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplateSpecialization:
3893b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplatePartialSpecialization:
3894952b017601f9c82b51119c3a1600f1312a833db9Douglas Gregor    if (TagDecl *Def = cast<TagDecl>(D)->getDefinition())
3895a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def, TU);
3896b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3897b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3898b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Function:
3899b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXMethod:
3900b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXConstructor:
3901b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXDestructor:
3902b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXConversion: {
3903b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    const FunctionDecl *Def = 0;
3904b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (cast<FunctionDecl>(D)->getBody(Def))
3905a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(const_cast<FunctionDecl *>(Def), TU);
3906b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3907b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
3908b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3909b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Var: {
391031310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    // Ask the variable if it has a definition.
391131310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    if (VarDecl *Def = cast<VarDecl>(D)->getDefinition())
3912a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def, TU);
391331310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    return clang_getNullCursor();
3914b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
3915f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3916b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FunctionTemplate: {
3917b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    const FunctionDecl *Def = 0;
3918b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (cast<FunctionTemplateDecl>(D)->getTemplatedDecl()->getBody(Def))
3919a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def->getDescribedFunctionTemplate(), TU);
3920b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3921b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
3922f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3923b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplate: {
3924b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (RecordDecl *Def = cast<ClassTemplateDecl>(D)->getTemplatedDecl()
3925952b017601f9c82b51119c3a1600f1312a833db9Douglas Gregor                                                            ->getDefinition())
39260b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return MakeCXCursor(cast<CXXRecordDecl>(Def)->getDescribedClassTemplate(),
3927a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          TU);
3928b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3929b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
3930b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
39311f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case Decl::Using:
39321f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return MakeCursorOverloadedDeclRef(cast<UsingDecl>(D),
3933a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                       D->getLocation(), TU);
3934b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3935b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UsingShadow:
3936b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getCursorDefinition(
3937f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                       MakeCXCursor(cast<UsingShadowDecl>(D)->getTargetDecl(),
3938a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                    TU));
3939b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3940b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCMethod: {
3941b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    ObjCMethodDecl *Method = cast<ObjCMethodDecl>(D);
3942b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (Method->isThisDeclarationADefinition())
3943b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      return C;
3944b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3945b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // Dig out the method definition in the associated
3946b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // @implementation, if we have it.
3947b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: The ASTs should make finding the definition easier.
3948b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCInterfaceDecl *Class
3949b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                       = dyn_cast<ObjCInterfaceDecl>(Method->getDeclContext()))
3950b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (ObjCImplementationDecl *ClassImpl = Class->getImplementation())
3951b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor        if (ObjCMethodDecl *Def = ClassImpl->getMethod(Method->getSelector(),
3952b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                                                  Method->isInstanceMethod()))
3953b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor          if (Def->isThisDeclarationADefinition())
3954a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek            return MakeCXCursor(Def, TU);
3955b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3956b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3957b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
3958b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3959b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCategory:
3960b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCCategoryImplDecl *Impl
3961b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                               = cast<ObjCCategoryDecl>(D)->getImplementation())
3962a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Impl, TU);
3963b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3964b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3965b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCProtocol:
3966b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (!cast<ObjCProtocolDecl>(D)->isForwardDecl())
3967b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      return C;
3968b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3969b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3970b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCInterface:
3971b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // There are two notions of a "definition" for an Objective-C
3972b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // class: the interface and its implementation. When we resolved a
3973b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // reference to an Objective-C class, produce the @interface as
3974b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // the definition; when we were provided with the interface,
3975b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // produce the @implementation as the definition.
3976b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (WasReference) {
3977b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (!cast<ObjCInterfaceDecl>(D)->isForwardDecl())
3978b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor        return C;
3979b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    } else if (ObjCImplementationDecl *Impl
3980b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                              = cast<ObjCInterfaceDecl>(D)->getImplementation())
3981a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Impl, TU);
3982b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3983f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3984b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCProperty:
3985b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: We don't really know where to find the
3986b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // ObjCPropertyImplDecls that implement this property.
3987b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3988b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3989b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCompatibleAlias:
3990b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCInterfaceDecl *Class
3991b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor          = cast<ObjCCompatibleAliasDecl>(D)->getClassInterface())
3992b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (!Class->isForwardDecl())
3993a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek        return MakeCXCursor(Class, TU);
3994f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3995b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3996b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
39971f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case Decl::ObjCForwardProtocol:
39981f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return MakeCursorOverloadedDeclRef(cast<ObjCForwardProtocolDecl>(D),
3999a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                       D->getLocation(), TU);
4000b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
40011f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case Decl::ObjCClass:
40029e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar    return MakeCursorOverloadedDeclRef(cast<ObjCClassDecl>(D), D->getLocation(),
4003a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                       TU);
4004b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4005b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Friend:
4006b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (NamedDecl *Friend = cast<FriendDecl>(D)->getFriendDecl())
4007a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
4008b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4009b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4010b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FriendTemplate:
4011b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (NamedDecl *Friend = cast<FriendTemplateDecl>(D)->getFriendDecl())
4012a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
4013b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4014b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4015b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4016b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  return clang_getNullCursor();
4017b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor}
4018b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4019b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregorunsigned clang_isCursorDefinition(CXCursor C) {
4020b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!clang_isDeclaration(C.kind))
4021b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return 0;
4022b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4023b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  return clang_getCursorDefinition(C) == C;
4024b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor}
4025b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
40261a9d0503b67a499797141af0fd6d315d5045f0eaDouglas GregorCXCursor clang_getCanonicalCursor(CXCursor C) {
40271a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  if (!clang_isDeclaration(C.kind))
40281a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor    return C;
40291a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
40301a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  if (Decl *D = getCursorDecl(C))
40311a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor    return MakeCXCursor(D->getCanonicalDecl(), getCursorTU(C));
40321a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
40331a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  return C;
40341a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor}
40351a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
40361f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregorunsigned clang_getNumOverloadedDecls(CXCursor C) {
40377c432dd959609a3689c2e4406450c092e6d76d6dDouglas Gregor  if (C.kind != CXCursor_OverloadedDeclRef)
40381f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return 0;
40391f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40401f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
40411f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
40421f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return E->getNumDecls();
40431f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40441f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadedTemplateStorage *S
40451f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                              = Storage.dyn_cast<OverloadedTemplateStorage*>())
40461f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return S->size();
40471f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40481f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  Decl *D = Storage.get<Decl*>();
40491f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (UsingDecl *Using = dyn_cast<UsingDecl>(D))
4050826faa22bae112e01293a58534a40711043cce65Argyrios Kyrtzidis    return Using->shadow_size();
40511f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (ObjCClassDecl *Classes = dyn_cast<ObjCClassDecl>(D))
40521f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return Classes->size();
40531f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (ObjCForwardProtocolDecl *Protocols =dyn_cast<ObjCForwardProtocolDecl>(D))
40541f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return Protocols->protocol_size();
40551f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40561f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  return 0;
40571f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor}
40581f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40591f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas GregorCXCursor clang_getOverloadedDecl(CXCursor cursor, unsigned index) {
40607c432dd959609a3689c2e4406450c092e6d76d6dDouglas Gregor  if (cursor.kind != CXCursor_OverloadedDeclRef)
40611f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return clang_getNullCursor();
40621f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40631f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (index >= clang_getNumOverloadedDecls(cursor))
40641f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return clang_getNullCursor();
40651f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
4066a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(cursor);
40671f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(cursor).first;
40681f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
4069a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(E->decls_begin()[index], TU);
40701f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40711f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadedTemplateStorage *S
40721f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                              = Storage.dyn_cast<OverloadedTemplateStorage*>())
4073a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(S->begin()[index], TU);
40741f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40751f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  Decl *D = Storage.get<Decl*>();
40761f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (UsingDecl *Using = dyn_cast<UsingDecl>(D)) {
40771f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    // FIXME: This is, unfortunately, linear time.
40781f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    UsingDecl::shadow_iterator Pos = Using->shadow_begin();
40791f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    std::advance(Pos, index);
4080a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(cast<UsingShadowDecl>(*Pos)->getTargetDecl(), TU);
40811f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  }
40821f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40831f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (ObjCClassDecl *Classes = dyn_cast<ObjCClassDecl>(D))
4084a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(Classes->begin()[index].getInterface(), TU);
40851f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40861f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (ObjCForwardProtocolDecl *Protocols = dyn_cast<ObjCForwardProtocolDecl>(D))
4087a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(Protocols->protocol_begin()[index], TU);
40881f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40891f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  return clang_getNullCursor();
40901f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor}
40911f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40920d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbarvoid clang_getDefinitionSpellingAndExtent(CXCursor C,
40934ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          const char **startBuf,
40944ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          const char **endBuf,
40954ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *startLine,
40964ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *startColumn,
40974ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *endLine,
40989ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          unsigned *endColumn) {
4099283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  assert(getCursorDecl(C) && "CXCursor has null decl");
4100283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  NamedDecl *ND = static_cast<NamedDecl *>(getCursorDecl(C));
41014ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  FunctionDecl *FD = dyn_cast<FunctionDecl>(ND);
41024ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  CompoundStmt *Body = dyn_cast<CompoundStmt>(FD->getBody());
4103f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
41044ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  SourceManager &SM = FD->getASTContext().getSourceManager();
41054ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startBuf = SM.getCharacterData(Body->getLBracLoc());
41064ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endBuf = SM.getCharacterData(Body->getRBracLoc());
41074ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startLine = SM.getSpellingLineNumber(Body->getLBracLoc());
41084ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startColumn = SM.getSpellingColumnNumber(Body->getLBracLoc());
41094ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endLine = SM.getSpellingLineNumber(Body->getRBracLoc());
41104ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endColumn = SM.getSpellingColumnNumber(Body->getRBracLoc());
41114ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff}
4112f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
41130a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregorvoid clang_enableStackTraces(void) {
41140a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  llvm::sys::PrintStackTraceOnErrorSignal();
41150a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor}
41160a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor
4117995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbarvoid clang_executeOnThread(void (*fn)(void*), void *user_data,
4118995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar                           unsigned stack_size) {
4119995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar  llvm::llvm_execute_on_thread(fn, user_data, stack_size);
4120995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar}
4121995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar
4122fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
4123fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
4124fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
4125fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor// Token-based Operations.
4126fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor//===----------------------------------------------------------------------===//
4127fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4128fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor/* CXToken layout:
4129fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[0]: a CXTokenKind
4130fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[1]: starting token location
4131fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[2]: token length
4132fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[3]: reserved
4133f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek *   ptr_data: for identifiers and keywords, an IdentifierInfo*.
4134fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   otherwise unused.
4135fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor */
4136fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregorextern "C" {
4137fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4138fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXTokenKind clang_getTokenKind(CXToken CXTok) {
4139fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  return static_cast<CXTokenKind>(CXTok.int_data[0]);
4140fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4141fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4142fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXString clang_getTokenSpelling(CXTranslationUnit TU, CXToken CXTok) {
4143fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  switch (clang_getTokenKind(CXTok)) {
4144fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Identifier:
4145fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Keyword:
4146fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // We know we have an IdentifierInfo*, so use that.
4147ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(static_cast<IdentifierInfo *>(CXTok.ptr_data)
4148ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek                            ->getNameStart());
4149fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4150fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Literal: {
4151fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // We have stashed the starting pointer in the ptr_data field. Use it.
4152fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    const char *Text = static_cast<const char *>(CXTok.ptr_data);
4153ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(llvm::StringRef(Text, CXTok.int_data[2]));
4154fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  }
4155f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4156fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Punctuation:
4157fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Comment:
4158fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    break;
4159fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  }
4160f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4161f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // We have to find the starting buffer pointer the hard way, by
4162fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // deconstructing the source location.
4163a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4164fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit)
4165ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
4166f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4167fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  SourceLocation Loc = SourceLocation::getFromRawEncoding(CXTok.int_data[1]);
4168fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> LocInfo
4169fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    = CXXUnit->getSourceManager().getDecomposedLoc(Loc);
4170f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  bool Invalid = false;
4171f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  llvm::StringRef Buffer
4172f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    = CXXUnit->getSourceManager().getBufferData(LocInfo.first, &Invalid);
4173f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  if (Invalid)
4174aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor    return createCXString("");
4175fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4176f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  return createCXString(Buffer.substr(LocInfo.second, CXTok.int_data[2]));
4177fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4178f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4179fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXSourceLocation clang_getTokenLocation(CXTranslationUnit TU, CXToken CXTok) {
4180a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4181fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit)
4182fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return clang_getNullLocation();
4183f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4184fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  return cxloc::translateSourceLocation(CXXUnit->getASTContext(),
4185fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                        SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
4186fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4187fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4188fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXSourceRange clang_getTokenExtent(CXTranslationUnit TU, CXToken CXTok) {
4189a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
41905352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  if (!CXXUnit)
41915352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
4192f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4193f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return cxloc::translateSourceRange(CXXUnit->getASTContext(),
4194fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                        SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
4195fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4196f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4197fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregorvoid clang_tokenize(CXTranslationUnit TU, CXSourceRange Range,
4198fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                    CXToken **Tokens, unsigned *NumTokens) {
4199fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (Tokens)
4200fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    *Tokens = 0;
4201fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (NumTokens)
4202fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    *NumTokens = 0;
4203f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4204a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4205fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit || !Tokens || !NumTokens)
4206fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4207f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4208bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
4209bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor
421085b988fdfa6adab6d43e16efd19ad4f3f7e2b49bDaniel Dunbar  SourceRange R = cxloc::translateCXSourceRange(Range);
4211fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (R.isInvalid())
4212fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4213f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4214fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  SourceManager &SourceMgr = CXXUnit->getSourceManager();
4215fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> BeginLocInfo
4216fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    = SourceMgr.getDecomposedLoc(R.getBegin());
4217fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> EndLocInfo
4218fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    = SourceMgr.getDecomposedLoc(R.getEnd());
4219f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4220fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // Cannot tokenize across files.
4221fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (BeginLocInfo.first != EndLocInfo.first)
4222fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4223f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4224f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // Create a lexer
4225f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  bool Invalid = false;
4226f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  llvm::StringRef Buffer
4227f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
422847a3fcd4afe122b23f9e7b6148f147bfa460cfe8Douglas Gregor  if (Invalid)
422947a3fcd4afe122b23f9e7b6148f147bfa460cfe8Douglas Gregor    return;
4230aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor
4231fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
4232fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor            CXXUnit->getASTContext().getLangOptions(),
4233f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer            Buffer.begin(), Buffer.data() + BeginLocInfo.second, Buffer.end());
4234fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Lex.SetCommentRetentionState(true);
4235f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4236fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // Lex tokens until we hit the end of the range.
4237f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  const char *EffectiveBufferEnd = Buffer.data() + EndLocInfo.second;
4238fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  llvm::SmallVector<CXToken, 32> CXTokens;
4239fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Token Tok;
4240096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall  bool previousWasAt = false;
4241fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  do {
4242fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // Lex the next token
4243fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    Lex.LexFromRawLexer(Tok);
4244fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    if (Tok.is(tok::eof))
4245fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      break;
4246f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4247fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // Initialize the CXToken.
4248fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXToken CXTok;
4249f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4250fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    //   - Common fields
4251fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[1] = Tok.getLocation().getRawEncoding();
4252fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[2] = Tok.getLength();
4253fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[3] = 0;
4254f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4255fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    //   - Kind-specific fields
4256fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    if (Tok.isLiteral()) {
4257fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Literal;
4258fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = (void *)Tok.getLiteralData();
4259c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara    } else if (Tok.is(tok::raw_identifier)) {
4260aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor      // Lookup the identifier to determine whether we have a keyword.
4261fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      IdentifierInfo *II
4262c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara        = CXXUnit->getPreprocessor().LookUpIdentifierInfo(Tok);
4263aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek
4264096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall      if ((II->getObjCKeywordID() != tok::objc_not_keyword) && previousWasAt) {
4265aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek        CXTok.int_data[0] = CXToken_Keyword;
4266aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      }
4267aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      else {
4268c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara        CXTok.int_data[0] = Tok.is(tok::identifier)
4269c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara          ? CXToken_Identifier
4270c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara          : CXToken_Keyword;
4271aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      }
4272fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = II;
4273fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else if (Tok.is(tok::comment)) {
4274fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Comment;
4275fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = 0;
4276fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else {
4277fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Punctuation;
4278fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = 0;
4279fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    }
4280fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTokens.push_back(CXTok);
4281096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall    previousWasAt = Tok.is(tok::at);
4282fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  } while (Lex.getBufferLocation() <= EffectiveBufferEnd);
4283f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4284fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (CXTokens.empty())
4285fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4286f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4287fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  *Tokens = (CXToken *)malloc(sizeof(CXToken) * CXTokens.size());
4288fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  memmove(*Tokens, CXTokens.data(), sizeof(CXToken) * CXTokens.size());
4289fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  *NumTokens = CXTokens.size();
4290fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
42910045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
42926db610934bedc6896393c1e1099525b35380acd6Ted Kremenekvoid clang_disposeTokens(CXTranslationUnit TU,
42936db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                         CXToken *Tokens, unsigned NumTokens) {
42946db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  free(Tokens);
42956db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
42966db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
42976db610934bedc6896393c1e1099525b35380acd6Ted Kremenek} // end: extern "C"
42986db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
42996db610934bedc6896393c1e1099525b35380acd6Ted Kremenek//===----------------------------------------------------------------------===//
43006db610934bedc6896393c1e1099525b35380acd6Ted Kremenek// Token annotation APIs.
43016db610934bedc6896393c1e1099525b35380acd6Ted Kremenek//===----------------------------------------------------------------------===//
43026db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
43030045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregortypedef llvm::DenseMap<unsigned, CXCursor> AnnotateTokensData;
4304fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekstatic enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
4305fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                     CXCursor parent,
4306fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                     CXClientData client_data);
43076db610934bedc6896393c1e1099525b35380acd6Ted Kremeneknamespace {
43086db610934bedc6896393c1e1099525b35380acd6Ted Kremenekclass AnnotateTokensWorker {
43096db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  AnnotateTokensData &Annotated;
431011949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  CXToken *Tokens;
431111949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  CXCursor *Cursors;
431211949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  unsigned NumTokens;
4313fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  unsigned TokIdx;
43144419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  unsigned PreprocessingTokIdx;
4315fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CursorVisitor AnnotateVis;
4316fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceManager &SrcMgr;
4317f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  bool HasContextSensitiveKeywords;
4318f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4319fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  bool MoreTokens() const { return TokIdx < NumTokens; }
4320fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  unsigned NextToken() const { return TokIdx; }
4321fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void AdvanceToken() { ++TokIdx; }
4322fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceLocation GetTokenLoc(unsigned tokI) {
4323fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[1]);
4324fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4325fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
43266db610934bedc6896393c1e1099525b35380acd6Ted Kremenekpublic:
432711949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  AnnotateTokensWorker(AnnotateTokensData &annotated,
4328fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                       CXToken *tokens, CXCursor *cursors, unsigned numTokens,
4329a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                       CXTranslationUnit tu, SourceRange RegionOfInterest)
433011949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek    : Annotated(annotated), Tokens(tokens), Cursors(cursors),
43314419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      NumTokens(numTokens), TokIdx(0), PreprocessingTokIdx(0),
4332a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      AnnotateVis(tu,
4333a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                  AnnotateTokensVisitor, this,
433404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                  Decl::MaxPCHLevel, true, RegionOfInterest),
4335f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      SrcMgr(static_cast<ASTUnit*>(tu->TUData)->getSourceManager()),
4336f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      HasContextSensitiveKeywords(false) { }
433711949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4338fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void VisitChildren(CXCursor C) { AnnotateVis.VisitChildren(C); }
43396db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  enum CXChildVisitResult Visit(CXCursor cursor, CXCursor parent);
4340fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void AnnotateTokens(CXCursor parent);
4341ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek  void AnnotateTokens() {
4342a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    AnnotateTokens(clang_getTranslationUnitCursor(AnnotateVis.getTU()));
4343ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek  }
4344f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4345f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  /// \brief Determine whether the annotator saw any cursors that have
4346f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  /// context-sensitive keywords.
4347f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  bool hasContextSensitiveKeywords() const {
4348f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    return HasContextSensitiveKeywords;
4349f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
43506db610934bedc6896393c1e1099525b35380acd6Ted Kremenek};
43516db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
43520045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
4353fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekvoid AnnotateTokensWorker::AnnotateTokens(CXCursor parent) {
4354fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Walk the AST within the region of interest, annotating tokens
4355fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // along the way.
4356fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  VisitChildren(parent);
4357fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4358fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = 0 ; I < TokIdx ; ++I) {
435911949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek    AnnotateTokensData::iterator Pos = Annotated.find(Tokens[I].int_data[1]);
43604419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    if (Pos != Annotated.end() &&
43614419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        (clang_isInvalid(Cursors[I].kind) ||
43624419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor         Pos->second.kind != CXCursor_PreprocessingDirective))
4363fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      Cursors[I] = Pos->second;
4364fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4365fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4366fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Finish up annotating any tokens left.
4367fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (!MoreTokens())
4368fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return;
436911949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4370fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const CXCursor &C = clang_getNullCursor();
4371fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = TokIdx ; I < NumTokens ; ++I) {
4372fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    AnnotateTokensData::iterator Pos = Annotated.find(Tokens[I].int_data[1]);
4373fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = (Pos == Annotated.end()) ? C : Pos->second;
437411949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  }
437511949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek}
437611949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
43776db610934bedc6896393c1e1099525b35380acd6Ted Kremenekenum CXChildVisitResult
43784419b675577d7c281a659fab1fec10e1bfbe04c5Douglas GregorAnnotateTokensWorker::Visit(CXCursor cursor, CXCursor parent) {
4379fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CXSourceLocation Loc = clang_getCursorLocation(cursor);
43804419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  SourceRange cursorRange = getRawCursorExtent(cursor);
438181d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor  if (cursorRange.isInvalid())
438281d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor    return CXChildVisit_Recurse;
4383f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4384f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  if (!HasContextSensitiveKeywords) {
4385f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // Objective-C properties can have context-sensitive keywords.
4386f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    if (cursor.kind == CXCursor_ObjCPropertyDecl) {
4387f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (ObjCPropertyDecl *Property
4388f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                  = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(cursor)))
4389f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        HasContextSensitiveKeywords = Property->getPropertyAttributesAsWritten() != 0;
4390f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4391f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // Objective-C methods can have context-sensitive keywords.
4392f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_ObjCInstanceMethodDecl ||
4393f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ObjCClassMethodDecl) {
4394f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (ObjCMethodDecl *Method
4395f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(cursor))) {
4396f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (Method->getObjCDeclQualifier())
4397f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4398f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        else {
4399f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
4400f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                                           PEnd = Method->param_end();
4401f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor               P != PEnd; ++P) {
4402f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            if ((*P)->getObjCDeclQualifier()) {
4403f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              HasContextSensitiveKeywords = true;
4404f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              break;
4405f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            }
4406f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          }
4407f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
4408f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4409f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4410f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // C++ methods can have context-sensitive keywords.
4411f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_CXXMethod) {
4412f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (CXXMethodDecl *Method
4413f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                  = dyn_cast_or_null<CXXMethodDecl>(getCursorDecl(cursor))) {
4414f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (Method->hasAttr<FinalAttr>() || Method->hasAttr<OverrideAttr>())
4415f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4416f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4417f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4418f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // C++ classes can have context-sensitive keywords.
4419f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_StructDecl ||
4420f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassDecl ||
4421f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassTemplate ||
4422f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassTemplatePartialSpecialization) {
4423f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Decl *D = getCursorDecl(cursor))
4424f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (D->hasAttr<FinalAttr>())
4425f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4426f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4427f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
4428f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
44294419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  if (clang_isPreprocessing(cursor.kind)) {
44304419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // For macro instantiations, just note where the beginning of the macro
44314419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // instantiation occurs.
44324419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    if (cursor.kind == CXCursor_MacroInstantiation) {
44334419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      Annotated[Loc.int_data] = cursor;
44344419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      return CXChildVisit_Recurse;
44354419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
44364419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
44374419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Items in the preprocessing record are kept separate from items in
44384419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // declarations, so we keep a separate token index.
44394419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    unsigned SavedTokIdx = TokIdx;
44404419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    TokIdx = PreprocessingTokIdx;
44414419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
44424419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Skip tokens up until we catch up to the beginning of the preprocessing
44434419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // entry.
44444419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    while (MoreTokens()) {
44454419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      const unsigned I = NextToken();
44464419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      SourceLocation TokLoc = GetTokenLoc(I);
44474419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
44484419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeBefore:
44494419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        AdvanceToken();
44504419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        continue;
44514419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeAfter:
44524419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeOverlap:
44534419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        break;
44544419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      }
44554419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      break;
44564419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
44574419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
44584419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Look at all of the tokens within this range.
44594419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    while (MoreTokens()) {
44604419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      const unsigned I = NextToken();
44614419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      SourceLocation TokLoc = GetTokenLoc(I);
44624419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
44634419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeBefore:
44644419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        assert(0 && "Infeasible");
44654419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeAfter:
44664419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        break;
44674419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeOverlap:
44684419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        Cursors[I] = cursor;
44694419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        AdvanceToken();
44704419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        continue;
44714419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      }
44724419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      break;
44734419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
44744419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
44754419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Save the preprocessing token index; restore the non-preprocessing
44764419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // token index.
44774419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    PreprocessingTokIdx = TokIdx;
44784419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    TokIdx = SavedTokIdx;
44790045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor    return CXChildVisit_Recurse;
44800045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  }
4481fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4482fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (cursorRange.isInvalid())
4483fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return CXChildVisit_Continue;
4484a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek
4485fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceLocation L = SourceLocation::getFromRawEncoding(Loc.int_data);
4486fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4487a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  // Adjust the annotated range based specific declarations.
4488a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  const enum CXCursorKind cursorK = clang_getCursorKind(cursor);
4489a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  if (cursorK >= CXCursor_FirstDecl && cursorK <= CXCursor_LastDecl) {
449023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    Decl *D = cxcursor::getCursorDecl(cursor);
449123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    // Don't visit synthesized ObjC methods, since they have no syntatic
449223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    // representation in the source.
449323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D)) {
449423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek      if (MD->isSynthesized())
449523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return CXChildVisit_Continue;
449623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    }
44972494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
44982494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    SourceLocation StartLoc;
449923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    if (const DeclaratorDecl *DD = dyn_cast<DeclaratorDecl>(D)) {
45002494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
45012494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
45022494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    } else if (TypedefDecl *Typedef = dyn_cast<TypedefDecl>(D)) {
45032494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
45042494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
4505a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek    }
45062494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
45072494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (StartLoc.isValid() && L.isValid() &&
45082494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        SrcMgr.isBeforeInTranslationUnit(StartLoc, L))
45092494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      cursorRange.setBegin(StartLoc);
4510a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  }
451181d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor
45123f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // If the location of the cursor occurs within a macro instantiation, record
45133f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // the spelling location of the cursor in our annotation map.  We can then
45143f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // paper over the token labelings during a post-processing step to try and
45153f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // get cursor mappings for tokens that are the *arguments* of a macro
45163f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // instantiation.
45173f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  if (L.isMacroID()) {
45183f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    unsigned rawEncoding = SrcMgr.getSpellingLoc(L).getRawEncoding();
45193f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // Only invalidate the old annotation if it isn't part of a preprocessing
45203f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // directive.  Here we assume that the default construction of CXCursor
45213f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // results in CXCursor.kind being an initialized value (i.e., 0).  If
45223f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // this isn't the case, we can fix by doing lookup + insertion.
45234419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
45243f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    CXCursor &oldC = Annotated[rawEncoding];
45253f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    if (!clang_isPreprocessing(oldC.kind))
45263f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek      oldC = cursor;
45273f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  }
45283f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek
4529fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const enum CXCursorKind K = clang_getCursorKind(parent);
4530fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const CXCursor updateC =
4531d8b0a84d586bc0a08695968acf2f169c9d01da69Ted Kremenek    (clang_isInvalid(K) || K == CXCursor_TranslationUnit)
4532d8b0a84d586bc0a08695968acf2f169c9d01da69Ted Kremenek     ? clang_getNullCursor() : parent;
4533fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4534fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  while (MoreTokens()) {
4535fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    const unsigned I = NextToken();
4536fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    SourceLocation TokLoc = GetTokenLoc(I);
4537fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
4538fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeBefore:
4539fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        Cursors[I] = updateC;
4540fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        AdvanceToken();
4541fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        continue;
4542fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeAfter:
4543fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeOverlap:
4544fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        break;
4545fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    }
4546fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    break;
4547fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4548fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4549fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Visit children to get their cursor information.
4550fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned BeforeChildren = NextToken();
4551fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  VisitChildren(cursor);
4552fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned AfterChildren = NextToken();
4553fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4554fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Adjust 'Last' to the last token within the extent of the cursor.
4555fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  while (MoreTokens()) {
4556fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    const unsigned I = NextToken();
4557fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    SourceLocation TokLoc = GetTokenLoc(I);
4558fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
4559fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeBefore:
4560fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        assert(0 && "Infeasible");
4561fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeAfter:
4562fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        break;
4563fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeOverlap:
4564fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        Cursors[I] = updateC;
4565fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        AdvanceToken();
4566fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        continue;
4567fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    }
4568fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    break;
4569fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4570fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned Last = NextToken();
45716db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
4572fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Scan the tokens that are at the beginning of the cursor, but are not
4573fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // capture by the child cursors.
4574fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4575fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // For AST elements within macros, rely on a post-annotate pass to
4576fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // to correctly annotate the tokens with cursors.  Otherwise we can
4577fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // get confusing results of having tokens that map to cursors that really
4578fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // are expanded by an instantiation.
4579fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (L.isMacroID())
4580fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    cursor = clang_getNullCursor();
4581fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4582fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = BeforeChildren; I != AfterChildren; ++I) {
4583fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    if (!clang_isInvalid(clang_getCursorKind(Cursors[I])))
4584fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      break;
45854419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
4586fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = cursor;
4587fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4588fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Scan the tokens that are at the end of the cursor, but are not captured
4589fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // but the child cursors.
4590fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = AfterChildren; I != Last; ++I)
4591fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = cursor;
4592fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4593fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  TokIdx = Last;
4594fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  return CXChildVisit_Continue;
45950045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor}
45960045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
45976db610934bedc6896393c1e1099525b35380acd6Ted Kremenekstatic enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
45986db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                                     CXCursor parent,
45996db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                                     CXClientData client_data) {
46006db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  return static_cast<AnnotateTokensWorker*>(client_data)->Visit(cursor, parent);
46016db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
46026db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
46036628a614c504263ae539462f049d523dd07ac1baTed Kremeneknamespace {
46046628a614c504263ae539462f049d523dd07ac1baTed Kremenek  struct clang_annotateTokens_Data {
46056628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXTranslationUnit TU;
46066628a614c504263ae539462f049d523dd07ac1baTed Kremenek    ASTUnit *CXXUnit;
46076628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXToken *Tokens;
46086628a614c504263ae539462f049d523dd07ac1baTed Kremenek    unsigned NumTokens;
46096628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXCursor *Cursors;
46106628a614c504263ae539462f049d523dd07ac1baTed Kremenek  };
4611ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek}
4612ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek
46136628a614c504263ae539462f049d523dd07ac1baTed Kremenek// This gets run a separate thread to avoid stack blowout.
46146628a614c504263ae539462f049d523dd07ac1baTed Kremenekstatic void clang_annotateTokensImpl(void *UserData) {
46156628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXTranslationUnit TU = ((clang_annotateTokens_Data*)UserData)->TU;
46166628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit *CXXUnit = ((clang_annotateTokens_Data*)UserData)->CXXUnit;
46176628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXToken *Tokens = ((clang_annotateTokens_Data*)UserData)->Tokens;
46186628a614c504263ae539462f049d523dd07ac1baTed Kremenek  const unsigned NumTokens = ((clang_annotateTokens_Data*)UserData)->NumTokens;
46196628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXCursor *Cursors = ((clang_annotateTokens_Data*)UserData)->Cursors;
4620fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
46210396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // Determine the region of interest, which contains all of the tokens.
46220045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  SourceRange RegionOfInterest;
46236628a614c504263ae539462f049d523dd07ac1baTed Kremenek  RegionOfInterest.setBegin(
46246628a614c504263ae539462f049d523dd07ac1baTed Kremenek    cxloc::translateSourceLocation(clang_getTokenLocation(TU, Tokens[0])));
46256628a614c504263ae539462f049d523dd07ac1baTed Kremenek  RegionOfInterest.setEnd(
46266628a614c504263ae539462f049d523dd07ac1baTed Kremenek    cxloc::translateSourceLocation(clang_getTokenLocation(TU,
46276628a614c504263ae539462f049d523dd07ac1baTed Kremenek                                                         Tokens[NumTokens-1])));
4628fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
46290396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // A mapping from the source locations found when re-lexing or traversing the
46300396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // region of interest to the corresponding cursors.
46310045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  AnnotateTokensData Annotated;
46326628a614c504263ae539462f049d523dd07ac1baTed Kremenek
4633fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Relex the tokens within the source range to look for preprocessing
46340396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // directives.
46359f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  SourceManager &SourceMgr = CXXUnit->getSourceManager();
46369f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  std::pair<FileID, unsigned> BeginLocInfo
46379f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    = SourceMgr.getDecomposedLoc(RegionOfInterest.getBegin());
46389f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  std::pair<FileID, unsigned> EndLocInfo
46399f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    = SourceMgr.getDecomposedLoc(RegionOfInterest.getEnd());
46406628a614c504263ae539462f049d523dd07ac1baTed Kremenek
46419f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  llvm::StringRef Buffer;
46420396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  bool Invalid = false;
46430396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  if (BeginLocInfo.first == EndLocInfo.first &&
46440396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor      ((Buffer = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid)),true) &&
46450396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor      !Invalid) {
46469f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
46479f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor              CXXUnit->getASTContext().getLangOptions(),
4648fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek              Buffer.begin(), Buffer.data() + BeginLocInfo.second,
46494ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor              Buffer.end());
46509f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    Lex.SetCommentRetentionState(true);
46516628a614c504263ae539462f049d523dd07ac1baTed Kremenek
4652fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    // Lex tokens in raw mode until we hit the end of the range, to avoid
46539f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    // entering #includes or expanding macros.
46544807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor    while (true) {
46559f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor      Token Tok;
46569f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor      Lex.LexFromRawLexer(Tok);
46576628a614c504263ae539462f049d523dd07ac1baTed Kremenek
46589f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    reprocess:
46599f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor      if (Tok.is(tok::hash) && Tok.isAtStartOfLine()) {
46609f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        // We have found a preprocessing directive. Gobble it up so that we
46619e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar        // don't see it while preprocessing these tokens later, but keep track
46629e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar        // of all of the token locations inside this preprocessing directive so
46639e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar        // that we can annotate them appropriately.
46649f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        //
46659f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        // FIXME: Some simple tests here could identify macro definitions and
46669f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        // #undefs, to provide specific cursor kinds for those.
46676628a614c504263ae539462f049d523dd07ac1baTed Kremenek        llvm::SmallVector<SourceLocation, 32> Locations;
46689f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        do {
46699f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor          Locations.push_back(Tok.getLocation());
4670fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek          Lex.LexFromRawLexer(Tok);
46719f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        } while (!Tok.isAtStartOfLine() && !Tok.is(tok::eof));
46726628a614c504263ae539462f049d523dd07ac1baTed Kremenek
46739f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        using namespace cxcursor;
46749f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        CXCursor Cursor
46756628a614c504263ae539462f049d523dd07ac1baTed Kremenek        = MakePreprocessingDirectiveCursor(SourceRange(Locations.front(),
46766628a614c504263ae539462f049d523dd07ac1baTed Kremenek                                                       Locations.back()),
4677a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                           TU);
46789f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        for (unsigned I = 0, N = Locations.size(); I != N; ++I) {
46799f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor          Annotated[Locations[I].getRawEncoding()] = Cursor;
46809f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        }
46816628a614c504263ae539462f049d523dd07ac1baTed Kremenek
46829f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        if (Tok.isAtStartOfLine())
46839f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor          goto reprocess;
46846628a614c504263ae539462f049d523dd07ac1baTed Kremenek
46859f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        continue;
46869f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor      }
46876628a614c504263ae539462f049d523dd07ac1baTed Kremenek
46884807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor      if (Tok.is(tok::eof))
46899f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        break;
46909f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    }
46914ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor  }
46926628a614c504263ae539462f049d523dd07ac1baTed Kremenek
46930396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // Annotate all of the source locations in the region of interest that map to
4694fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // a specific cursor.
4695fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  AnnotateTokensWorker W(Annotated, Tokens, Cursors, NumTokens,
4696a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                         TU, RegionOfInterest);
46976628a614c504263ae539462f049d523dd07ac1baTed Kremenek
46986c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // FIXME: We use a ridiculous stack size here because the data-recursion
46996c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // algorithm uses a large stack frame than the non-data recursive version,
47006c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // and AnnotationTokensWorker currently transforms the data-recursion
47016c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // algorithm back into a traditional recursion by explicitly calling
47026c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // VisitChildren().  We will need to remove this explicit recursive call.
47036628a614c504263ae539462f049d523dd07ac1baTed Kremenek  W.AnnotateTokens();
47046628a614c504263ae539462f049d523dd07ac1baTed Kremenek
4705f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  // If we ran into any entities that involve context-sensitive keywords,
4706f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  // take another pass through the tokens to mark them as such.
4707f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  if (W.hasContextSensitiveKeywords()) {
4708f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    for (unsigned I = 0; I != NumTokens; ++I) {
4709f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (clang_getTokenKind(Tokens[I]) != CXToken_Identifier)
4710f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
4711f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4712f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ObjCPropertyDecl) {
4713f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
4714f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (ObjCPropertyDecl *Property
47156628a614c504263ae539462f049d523dd07ac1baTed Kremenek            = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(Cursors[I]))) {
4716f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          if (Property->getPropertyAttributesAsWritten() != 0 &&
4717f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              llvm::StringSwitch<bool>(II->getName())
47186628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("readonly", true)
47196628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("assign", true)
47206628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("readwrite", true)
47216628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("retain", true)
47226628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("copy", true)
47236628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("nonatomic", true)
47246628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("atomic", true)
47256628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("getter", true)
47266628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("setter", true)
47276628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Default(false))
4728f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            Tokens[I].int_data[0] = CXToken_Keyword;
4729f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
4730f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
4731f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4732f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4733f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ObjCInstanceMethodDecl ||
4734f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Cursors[I].kind == CXCursor_ObjCClassMethodDecl) {
4735f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
4736f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (llvm::StringSwitch<bool>(II->getName())
47376628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("in", true)
47386628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("out", true)
47396628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("inout", true)
47406628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("oneway", true)
47416628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("bycopy", true)
47426628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("byref", true)
47436628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Default(false))
4744f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Tokens[I].int_data[0] = CXToken_Keyword;
4745f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
4746f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4747f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4748f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_CXXMethod) {
4749f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
4750f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (CXXMethodDecl *Method
47516628a614c504263ae539462f049d523dd07ac1baTed Kremenek            = dyn_cast_or_null<CXXMethodDecl>(getCursorDecl(Cursors[I]))) {
4752f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          if ((Method->hasAttr<FinalAttr>() ||
4753f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor               Method->hasAttr<OverrideAttr>()) &&
4754f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              Method->getLocation().getRawEncoding() != Tokens[I].int_data[1] &&
4755f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              llvm::StringSwitch<bool>(II->getName())
47566628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("final", true)
47576628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("override", true)
47586628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Default(false))
4759f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            Tokens[I].int_data[0] = CXToken_Keyword;
4760f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
4761f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
4762f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4763f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4764f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ClassDecl ||
4765f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Cursors[I].kind == CXCursor_StructDecl ||
4766f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Cursors[I].kind == CXCursor_ClassTemplate) {
4767f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
4768f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (II->getName() == "final") {
4769f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          // We have to be careful with 'final', since it could be the name
4770f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          // of a member class rather than the context-sensitive keyword.
4771f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          // So, check whether the cursor associated with this
4772f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Decl *D = getCursorDecl(Cursors[I]);
4773f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          if (CXXRecordDecl *Record = dyn_cast_or_null<CXXRecordDecl>(D)) {
4774f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            if ((Record->hasAttr<FinalAttr>()) &&
4775f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                Record->getIdentifier() != II)
4776f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              Tokens[I].int_data[0] = CXToken_Keyword;
4777f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          } else if (ClassTemplateDecl *ClassTemplate
47786628a614c504263ae539462f049d523dd07ac1baTed Kremenek                     = dyn_cast_or_null<ClassTemplateDecl>(D)) {
4779f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            CXXRecordDecl *Record = ClassTemplate->getTemplatedDecl();
4780f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            if ((Record->hasAttr<FinalAttr>()) &&
4781f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                Record->getIdentifier() != II)
47826628a614c504263ae539462f049d523dd07ac1baTed Kremenek              Tokens[I].int_data[0] = CXToken_Keyword;
4783f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          }
4784f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
47856628a614c504263ae539462f049d523dd07ac1baTed Kremenek        continue;
4786f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4787f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4788f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
4789fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
47906628a614c504263ae539462f049d523dd07ac1baTed Kremenek
47916628a614c504263ae539462f049d523dd07ac1baTed Kremenekextern "C" {
47926628a614c504263ae539462f049d523dd07ac1baTed Kremenek
47936628a614c504263ae539462f049d523dd07ac1baTed Kremenekvoid clang_annotateTokens(CXTranslationUnit TU,
47946628a614c504263ae539462f049d523dd07ac1baTed Kremenek                          CXToken *Tokens, unsigned NumTokens,
47956628a614c504263ae539462f049d523dd07ac1baTed Kremenek                          CXCursor *Cursors) {
47966628a614c504263ae539462f049d523dd07ac1baTed Kremenek
47976628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (NumTokens == 0 || !Tokens || !Cursors)
47986628a614c504263ae539462f049d523dd07ac1baTed Kremenek    return;
47996628a614c504263ae539462f049d523dd07ac1baTed Kremenek
48006628a614c504263ae539462f049d523dd07ac1baTed Kremenek  // Any token we don't specifically annotate will have a NULL cursor.
48016628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXCursor C = clang_getNullCursor();
48026628a614c504263ae539462f049d523dd07ac1baTed Kremenek  for (unsigned I = 0; I != NumTokens; ++I)
48036628a614c504263ae539462f049d523dd07ac1baTed Kremenek    Cursors[I] = C;
48046628a614c504263ae539462f049d523dd07ac1baTed Kremenek
48056628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
48066628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (!CXXUnit)
48076628a614c504263ae539462f049d523dd07ac1baTed Kremenek    return;
48086628a614c504263ae539462f049d523dd07ac1baTed Kremenek
48096628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
48106628a614c504263ae539462f049d523dd07ac1baTed Kremenek
48116628a614c504263ae539462f049d523dd07ac1baTed Kremenek  clang_annotateTokens_Data data = { TU, CXXUnit, Tokens, NumTokens, Cursors };
48126628a614c504263ae539462f049d523dd07ac1baTed Kremenek  llvm::CrashRecoveryContext CRC;
48136628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (!RunSafely(CRC, clang_annotateTokensImpl, &data,
48146628a614c504263ae539462f049d523dd07ac1baTed Kremenek                 GetSafetyThreadStackSize() * 2)) {
48156628a614c504263ae539462f049d523dd07ac1baTed Kremenek    fprintf(stderr, "libclang: crash detected while annotating tokens\n");
48166628a614c504263ae539462f049d523dd07ac1baTed Kremenek  }
48176628a614c504263ae539462f049d523dd07ac1baTed Kremenek}
48186628a614c504263ae539462f049d523dd07ac1baTed Kremenek
4819fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor} // end: extern "C"
4820fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4821fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor//===----------------------------------------------------------------------===//
482216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek// Operations for querying linkage of a cursor.
482316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek//===----------------------------------------------------------------------===//
482416b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
482516b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenekextern "C" {
482616b4259aecaa22b642d35d36fd89965ed700c1e0Ted KremenekCXLinkageKind clang_getCursorLinkage(CXCursor cursor) {
48270396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  if (!clang_isDeclaration(cursor.kind))
48280396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    return CXLinkage_Invalid;
48290396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor
483016b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  Decl *D = cxcursor::getCursorDecl(cursor);
483116b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  if (NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D))
483216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek    switch (ND->getLinkage()) {
483316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case NoLinkage: return CXLinkage_NoLinkage;
483416b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case InternalLinkage: return CXLinkage_Internal;
483516b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case UniqueExternalLinkage: return CXLinkage_UniqueExternal;
483616b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case ExternalLinkage: return CXLinkage_External;
483716b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek    };
483816b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
483916b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  return CXLinkage_Invalid;
484016b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek}
484116b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek} // end: extern "C"
484216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
484316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek//===----------------------------------------------------------------------===//
484445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek// Operations for querying language of a cursor.
484545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek//===----------------------------------------------------------------------===//
484645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
484745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenekstatic CXLanguageKind getDeclLanguage(const Decl *D) {
484845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  switch (D->getKind()) {
484945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    default:
485045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      break;
485145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ImplicitParam:
485245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCAtDefsField:
485345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCategory:
485445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCategoryImpl:
485545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCClass:
485645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCompatibleAlias:
485745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCForwardProtocol:
485845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCImplementation:
485945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCInterface:
486045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCIvar:
486145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCMethod:
486245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCProperty:
486345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCPropertyImpl:
486445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCProtocol:
486545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      return CXLanguage_ObjC;
486645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXConstructor:
486745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXConversion:
486845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXDestructor:
486945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXMethod:
487045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXRecord:
487145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplate:
487245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplatePartialSpecialization:
487345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplateSpecialization:
487445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Friend:
487545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::FriendTemplate:
487645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::FunctionTemplate:
487745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::LinkageSpec:
487845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Namespace:
487945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::NamespaceAlias:
488045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::NonTypeTemplateParm:
488145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::StaticAssert:
488245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::TemplateTemplateParm:
488345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::TemplateTypeParm:
488445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UnresolvedUsingTypename:
488545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UnresolvedUsingValue:
488645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Using:
488745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UsingDirective:
488845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UsingShadow:
488945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      return CXLanguage_CPlusPlus;
489045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  }
489145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
489245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  return CXLanguage_C;
489345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek}
489445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
489545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenekextern "C" {
489658ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
489758ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregorenum CXAvailabilityKind clang_getCursorAvailability(CXCursor cursor) {
489858ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor  if (clang_isDeclaration(cursor.kind))
489958ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor    if (Decl *D = cxcursor::getCursorDecl(cursor)) {
490058ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor      if (D->hasAttr<UnavailableAttr>() ||
490158ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor          (isa<FunctionDecl>(D) && cast<FunctionDecl>(D)->isDeleted()))
490258ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor        return CXAvailability_Available;
490358ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
490458ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor      if (D->hasAttr<DeprecatedAttr>())
490558ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor        return CXAvailability_Deprecated;
490658ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor    }
490758ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
490858ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor  return CXAvailability_Available;
490958ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor}
491058ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
491145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted KremenekCXLanguageKind clang_getCursorLanguage(CXCursor cursor) {
491245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  if (clang_isDeclaration(cursor.kind))
491345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    return getDeclLanguage(cxcursor::getCursorDecl(cursor));
491445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
491545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  return CXLanguage_Invalid;
491645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek}
49173910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49183910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// \brief If the given cursor is the "templated" declaration
49193910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// descibing a class or function template, return the class or
49203910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// function template.
49213910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregorstatic Decl *maybeGetTemplateCursor(Decl *D) {
49223910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (!D)
49233910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    return 0;
49243910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49253910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
49263910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    if (FunctionTemplateDecl *FunTmpl = FD->getDescribedFunctionTemplate())
49273910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return FunTmpl;
49283910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49293910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(D))
49303910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    if (ClassTemplateDecl *ClassTmpl = RD->getDescribedClassTemplate())
49313910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return ClassTmpl;
49323910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49333910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  return D;
49343910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor}
49353910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49362be5bc9ad3981347a000742f81b91ab3080f1214Douglas GregorCXCursor clang_getCursorSemanticParent(CXCursor cursor) {
49372be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isDeclaration(cursor.kind)) {
49382be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor)) {
49392be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor      DeclContext *DC = D->getDeclContext();
49403910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      if (!DC)
49413910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor        return clang_getNullCursor();
49423910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49433910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
49443910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor                          getCursorTU(cursor));
49452be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    }
49462be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
49472be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
49482be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isStatement(cursor.kind) || clang_isExpression(cursor.kind)) {
49492be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor))
4950a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(D, getCursorTU(cursor));
49512be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
49522be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
49532be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  return clang_getNullCursor();
49542be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor}
49552be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
49562be5bc9ad3981347a000742f81b91ab3080f1214Douglas GregorCXCursor clang_getCursorLexicalParent(CXCursor cursor) {
49572be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isDeclaration(cursor.kind)) {
49582be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor)) {
49592be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor      DeclContext *DC = D->getLexicalDeclContext();
49603910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      if (!DC)
49613910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor        return clang_getNullCursor();
49623910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49633910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
49643910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor                          getCursorTU(cursor));
49652be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    }
49662be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
49672be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
49682be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  // FIXME: Note that we can't easily compute the lexical context of a
49692be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  // statement or expression, so we return nothing.
49702be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  return clang_getNullCursor();
49712be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor}
49722be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
49739f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregorstatic void CollectOverriddenMethods(DeclContext *Ctx,
49749f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                     ObjCMethodDecl *Method,
49759f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                            llvm::SmallVectorImpl<ObjCMethodDecl *> &Methods) {
49769f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!Ctx)
49779f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
49789f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
49799f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  // If we have a class or category implementation, jump straight to the
49809f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  // interface.
49819f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (ObjCImplDecl *Impl = dyn_cast<ObjCImplDecl>(Ctx))
49829f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return CollectOverriddenMethods(Impl->getClassInterface(), Method, Methods);
49839f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
49849f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  ObjCContainerDecl *Container = dyn_cast<ObjCContainerDecl>(Ctx);
49859f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!Container)
49869f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
49879f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
49889f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  // Check whether we have a matching method at this level.
49899f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (ObjCMethodDecl *Overridden = Container->getMethod(Method->getSelector(),
49909f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                                    Method->isInstanceMethod()))
49919f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    if (Method != Overridden) {
49929f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      // We found an override at this level; there is no need to look
49939f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      // into other protocols or categories.
49949f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      Methods.push_back(Overridden);
49959f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      return;
49969f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    }
49979f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
49989f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
49999f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    for (ObjCProtocolDecl::protocol_iterator P = Protocol->protocol_begin(),
50009f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                          PEnd = Protocol->protocol_end();
50019f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor         P != PEnd; ++P)
50029f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      CollectOverriddenMethods(*P, Method, Methods);
50039f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  }
50049f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50059f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(Container)) {
50069f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    for (ObjCCategoryDecl::protocol_iterator P = Category->protocol_begin(),
50079f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                          PEnd = Category->protocol_end();
50089f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor         P != PEnd; ++P)
50099f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      CollectOverriddenMethods(*P, Method, Methods);
50109f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  }
50119f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50129f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (ObjCInterfaceDecl *Interface = dyn_cast<ObjCInterfaceDecl>(Container)) {
50139f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    for (ObjCInterfaceDecl::protocol_iterator P = Interface->protocol_begin(),
50149f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                           PEnd = Interface->protocol_end();
50159f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor         P != PEnd; ++P)
50169f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      CollectOverriddenMethods(*P, Method, Methods);
50179f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50189f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    for (ObjCCategoryDecl *Category = Interface->getCategoryList();
50199f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor         Category; Category = Category->getNextClassCategory())
50209f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      CollectOverriddenMethods(Category, Method, Methods);
50219f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50229f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    // We only look into the superclass if we haven't found anything yet.
50239f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    if (Methods.empty())
50249f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      if (ObjCInterfaceDecl *Super = Interface->getSuperClass())
50259f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor        return CollectOverriddenMethods(Super, Method, Methods);
50269f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  }
50279f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor}
50289f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50299f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregorvoid clang_getOverriddenCursors(CXCursor cursor,
50309f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                CXCursor **overridden,
50319f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                unsigned *num_overridden) {
50329f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (overridden)
50339f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    *overridden = 0;
50349f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (num_overridden)
50359f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    *num_overridden = 0;
50369f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!overridden || !num_overridden)
50379f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
50389f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50399f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!clang_isDeclaration(cursor.kind))
50409f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
50419f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50429f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  Decl *D = getCursorDecl(cursor);
50439f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!D)
50449f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
50459f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50469f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  // Handle C++ member functions.
5047a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(cursor);
50489f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (CXXMethodDecl *CXXMethod = dyn_cast<CXXMethodDecl>(D)) {
50499f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    *num_overridden = CXXMethod->size_overridden_methods();
50509f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    if (!*num_overridden)
50519f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      return;
50529f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50539f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    *overridden = new CXCursor [*num_overridden];
50549f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    unsigned I = 0;
50559f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    for (CXXMethodDecl::method_iterator
50569f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor              M = CXXMethod->begin_overridden_methods(),
50579f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor           MEnd = CXXMethod->end_overridden_methods();
50589f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor         M != MEnd; (void)++M, ++I)
5059a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      (*overridden)[I] = MakeCXCursor(const_cast<CXXMethodDecl*>(*M), TU);
50609f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
50619f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  }
50629f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50639f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(D);
50649f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!Method)
50659f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
50669f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50679f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  // Handle Objective-C methods.
50689f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  llvm::SmallVector<ObjCMethodDecl *, 4> Methods;
50699f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  CollectOverriddenMethods(Method->getDeclContext(), Method, Methods);
50709f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50719f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (Methods.empty())
50729f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
50739f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50749f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  *num_overridden = Methods.size();
50759f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  *overridden = new CXCursor [Methods.size()];
50769f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  for (unsigned I = 0, N = Methods.size(); I != N; ++I)
5077a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    (*overridden)[I] = MakeCXCursor(Methods[I], TU);
50789f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor}
50799f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50809f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregorvoid clang_disposeOverriddenCursors(CXCursor *overridden) {
50819f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  delete [] overridden;
50829f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor}
50839f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
5084ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas GregorCXFile clang_getIncludedFile(CXCursor cursor) {
5085ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (cursor.kind != CXCursor_InclusionDirective)
5086ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return 0;
5087ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
5088ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  InclusionDirective *ID = getCursorInclusionDirective(cursor);
5089ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  return (void *)ID->getFile();
5090ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor}
5091ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
509245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek} // end: extern "C"
509345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
50949ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
50959ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek//===----------------------------------------------------------------------===//
50969ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek// C++ AST instrospection.
50979ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek//===----------------------------------------------------------------------===//
50989ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
50999ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenekextern "C" {
51009ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenekunsigned clang_CXXMethod_isStatic(CXCursor C) {
51019ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek  if (!clang_isDeclaration(C.kind))
51029ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek    return 0;
510349f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor
510449f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  CXXMethodDecl *Method = 0;
510549f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  Decl *D = cxcursor::getCursorDecl(C);
510649f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast_or_null<FunctionTemplateDecl>(D))
510749f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor    Method = dyn_cast<CXXMethodDecl>(FunTmpl->getTemplatedDecl());
510849f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  else
510949f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor    Method = dyn_cast_or_null<CXXMethodDecl>(D);
511049f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  return (Method && Method->isStatic()) ? 1 : 0;
511140b492a43bac3ed0c465772aa6921d011cfc273fTed Kremenek}
5112b12903e1a4b8d1b611b8c7e4f910665d628e68cdTed Kremenek
51139ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek} // end: extern "C"
51149ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
511545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek//===----------------------------------------------------------------------===//
511695f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek// Attribute introspection.
511795f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek//===----------------------------------------------------------------------===//
511895f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
511995f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenekextern "C" {
512095f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted KremenekCXType clang_getIBOutletCollectionType(CXCursor C) {
512195f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek  if (C.kind != CXCursor_IBOutletCollectionAttr)
5122a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return cxtype::MakeCXType(QualType(), cxcursor::getCursorTU(C));
512395f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
512495f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek  IBOutletCollectionAttr *A =
512595f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek    cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(C));
512695f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
5127841b238087d6cdb21c2443b7429cb85bd1f9fce2Douglas Gregor  return cxtype::MakeCXType(A->getInterFace(), cxcursor::getCursorTU(C));
512895f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek}
512995f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek} // end: extern "C"
513095f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
513195f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek//===----------------------------------------------------------------------===//
513204bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek// Misc. utility functions.
513304bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek//===----------------------------------------------------------------------===//
5134f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
5135abdce7abc8a22dd2fe79a05c0b71864039bd8296Daniel Dunbar/// Default to using an 8 MB stack size on "safety" threads.
5136abdce7abc8a22dd2fe79a05c0b71864039bd8296Daniel Dunbarstatic unsigned SafetyStackThreadSize = 8 << 20;
5137bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5138bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarnamespace clang {
5139bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5140bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarbool RunSafely(llvm::CrashRecoveryContext &CRC,
51416c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek               void (*Fn)(void*), void *UserData,
51426c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek               unsigned Size) {
51436c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  if (!Size)
51446c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek    Size = GetSafetyThreadStackSize();
51456c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  if (Size)
5146bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar    return CRC.RunSafelyOnThread(Fn, UserData, Size);
5147bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  return CRC.RunSafely(Fn, UserData);
5148bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5149bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5150bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarunsigned GetSafetyThreadStackSize() {
5151bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  return SafetyStackThreadSize;
5152bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5153bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5154bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarvoid SetSafetyThreadStackSize(unsigned Value) {
5155bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  SafetyStackThreadSize = Value;
5156bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5157bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5158bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5159bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
516004bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenekextern "C" {
516104bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
5162a2a9d6e4e5b6001b86b7dfc5db1ea296ce29a3d3Ted KremenekCXString clang_getClangVersion() {
5163ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString(getClangFullVersion());
516404bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek}
516504bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
516604bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek} // end: extern "C"
5167