CIndex.cpp revision 06d9b1ad0bca7230cbae57e3e3207dda77a9eac0
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);
49706d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (!D)
49806d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return false;
49906d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
500539311e0221df256c70c1c3080c8af847cd29dffTed Kremenek    return VisitAttributes(D) || Visit(D);
501b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
502f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
50306d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  if (clang_isStatement(Cursor.kind)) {
50406d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (Stmt *S = getCursorStmt(Cursor))
50506d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return Visit(S);
50606d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
50706d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    return false;
50806d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  }
50906d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
51006d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  if (clang_isExpression(Cursor.kind)) {
51106d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (Expr *E = getCursorExpr(Cursor))
51206d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return Visit(E);
51306d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
51406d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    return false;
51506d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  }
516f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
517b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isTranslationUnit(Cursor.kind)) {
518a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    CXTranslationUnit tu = getCursorTU(Cursor);
519a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    ASTUnit *CXXUnit = static_cast<ASTUnit*>(tu->TUData);
52004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
52104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor    int VisitOrder[2] = { VisitPreprocessorLast, !VisitPreprocessorLast };
52204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor    for (unsigned I = 0; I != 2; ++I) {
52304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      if (VisitOrder[I]) {
52404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        if (!CXXUnit->isMainFileAST() && CXXUnit->getOnlyLocalDecls() &&
52504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            RegionOfInterest.isInvalid()) {
52604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          for (ASTUnit::top_level_iterator TL = CXXUnit->top_level_begin(),
52704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                                        TLEnd = CXXUnit->top_level_end();
52804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor               TL != TLEnd; ++TL) {
52904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            if (Visit(MakeCXCursor(*TL, tu), true))
53004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor              return true;
53104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          }
53204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        } else if (VisitDeclContext(
53304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                                CXXUnit->getASTContext().getTranslationUnitDecl()))
5347b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor          return true;
53504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        continue;
5367b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor      }
5373178cb674ac8c3b59e1791e14d38d48619a1b621Bob Wilson
53804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      // Walk the preprocessing record.
53904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      if (CXXUnit->getPreprocessor().getPreprocessingRecord()) {
54004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        // FIXME: Once we have the ability to deserialize a preprocessing record,
54104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        // do so.
54204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        PreprocessingRecord::iterator E, EEnd;
54304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        for (llvm::tie(E, EEnd) = getPreprocessedEntities(); E != EEnd; ++E) {
54404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          if (MacroInstantiation *MI = dyn_cast<MacroInstantiation>(*E)) {
54504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            if (Visit(MakeMacroInstantiationCursor(MI, tu)))
54604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor              return true;
54704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
54804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            continue;
54904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          }
5500396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor
55104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          if (MacroDefinition *MD = dyn_cast<MacroDefinition>(*E)) {
55204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            if (Visit(MakeMacroDefinitionCursor(MD, tu)))
55304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor              return true;
55404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
55504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            continue;
55604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          }
557ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
55804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          if (InclusionDirective *ID = dyn_cast<InclusionDirective>(*E)) {
55904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            if (Visit(MakeInclusionDirectiveCursor(ID, tu)))
56004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor              return true;
56104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
56204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            continue;
56304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          }
564ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor        }
5650396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor      }
5660396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    }
56704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
5687b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor    return false;
569b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
570f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
571c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  if (Cursor.kind == CXCursor_CXXBaseSpecifier) {
572c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor    if (CXXBaseSpecifier *Base = getCursorCXXBaseSpecifier(Cursor)) {
573c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      if (TypeSourceInfo *BaseTSInfo = Base->getTypeSourceInfo()) {
574c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor        return Visit(BaseTSInfo->getTypeLoc());
575c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      }
576c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor    }
577c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  }
578c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor
579b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  // Nothing to visit at the moment.
580b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
581dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
582dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
5831ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenekbool CursorVisitor::VisitBlockDecl(BlockDecl *B) {
584fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall  if (Visit(B->getSignatureAsWritten()->getTypeLoc()))
585fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall    return true;
5861ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
587664cffd330611d78fc0286f539589920a37ca328Ted Kremenek  if (Stmt *Body = B->getBody())
588664cffd330611d78fc0286f539589920a37ca328Ted Kremenek    return Visit(MakeCXCursor(Body, StmtParent, TU));
589664cffd330611d78fc0286f539589920a37ca328Ted Kremenek
590664cffd330611d78fc0286f539589920a37ca328Ted Kremenek  return false;
5911ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek}
5921ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
593d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenekllvm::Optional<bool> CursorVisitor::shouldVisitCursor(CXCursor Cursor) {
594d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (RegionOfInterest.isValid()) {
5956653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    SourceRange Range = getFullCursorExtent(Cursor, AU->getSourceManager());
596d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (Range.isInvalid())
597d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return llvm::Optional<bool>();
5986653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
599d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    switch (CompareRegionOfInterest(Range)) {
600d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeBefore:
601d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration comes before the region of interest; skip it.
602d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return llvm::Optional<bool>();
60323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
604d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeAfter:
605d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration comes after the region of interest; we're done.
606d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
607d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar
608d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeOverlap:
609d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration overlaps the region of interest; visit it.
610d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      break;
611d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
612d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
613d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  return true;
614d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek}
615f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
616d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenekbool CursorVisitor::VisitDeclContext(DeclContext *DC) {
617d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  DeclContext::decl_iterator I = DC->decls_begin(), E = DC->decls_end();
618f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
619d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually remove.  This part of a hack to support proper
620d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // iteration over all Decls contained lexically within an ObjC container.
621d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SaveAndRestore<DeclContext::decl_iterator*> DI_saved(DI_current, &I);
622d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SaveAndRestore<DeclContext::decl_iterator> DE_saved(DE_current, E);
623f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
624d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for ( ; I != E; ++I) {
625d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    Decl *D = *I;
626d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (D->getLexicalDeclContext() != DC)
627d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
628d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    CXCursor Cursor = MakeCXCursor(D, TU);
629d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    const llvm::Optional<bool> &V = shouldVisitCursor(Cursor);
630d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.hasValue())
631d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
632d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.getValue())
633d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
634d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar    if (Visit(Cursor, true))
635b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
636b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
637b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
638dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
639dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
6401ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTranslationUnitDecl(TranslationUnitDecl *D) {
6411ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  llvm_unreachable("Translation units are visited directly by Visit()");
6421ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
6431ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6441ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6451ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTypedefDecl(TypedefDecl *D) {
6461ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
6471ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(TSInfo->getTypeLoc());
648f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
6491ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
6501ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6511ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6521ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTagDecl(TagDecl *D) {
6531ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitDeclContext(D);
6541ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6551ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6560ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregorbool CursorVisitor::VisitClassTemplateSpecializationDecl(
6570ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor                                          ClassTemplateSpecializationDecl *D) {
6580ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  bool ShouldVisitBody = false;
6590ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  switch (D->getSpecializationKind()) {
6600ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_Undeclared:
6610ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ImplicitInstantiation:
6620ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    // Nothing to visit
6630ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    return false;
6640ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6650ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitInstantiationDeclaration:
6660ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitInstantiationDefinition:
6670ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    break;
6680ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6690ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitSpecialization:
6700ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    ShouldVisitBody = true;
6710ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    break;
6720ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  }
6730ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6740ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  // Visit the template arguments used in the specialization.
6750ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  if (TypeSourceInfo *SpecType = D->getTypeAsWritten()) {
6760ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    TypeLoc TL = SpecType->getTypeLoc();
6770ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    if (TemplateSpecializationTypeLoc *TSTLoc
6780ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor          = dyn_cast<TemplateSpecializationTypeLoc>(&TL)) {
6790ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor      for (unsigned I = 0, N = TSTLoc->getNumArgs(); I != N; ++I)
6800ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor        if (VisitTemplateArgumentLoc(TSTLoc->getArgLoc(I)))
6810ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor          return true;
6820ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    }
6830ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  }
6840ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6850ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  if (ShouldVisitBody && VisitCXXRecordDecl(D))
6860ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    return true;
6870ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6880ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  return false;
6890ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor}
6900ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
69174dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregorbool CursorVisitor::VisitClassTemplatePartialSpecializationDecl(
69274dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor                                   ClassTemplatePartialSpecializationDecl *D) {
69374dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // FIXME: Visit the "outer" template parameter lists on the TagDecl
69474dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // before visiting these template parameters.
69574dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
69674dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    return true;
69774dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
69874dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // Visit the partial specialization arguments.
69974dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  const TemplateArgumentLoc *TemplateArgs = D->getTemplateArgsAsWritten();
70074dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  for (unsigned I = 0, N = D->getNumTemplateArgsAsWritten(); I != N; ++I)
70174dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    if (VisitTemplateArgumentLoc(TemplateArgs[I]))
70274dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor      return true;
70374dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
70474dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  return VisitCXXRecordDecl(D);
70574dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor}
70674dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
707fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateTypeParmDecl(TemplateTypeParmDecl *D) {
70884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  // Visit the default argument.
70984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
71084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    if (TypeSourceInfo *DefArg = D->getDefaultArgumentInfo())
71184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      if (Visit(DefArg->getTypeLoc()))
71284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor        return true;
71384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
714fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
715fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
716fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
7171ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitEnumConstantDecl(EnumConstantDecl *D) {
7181ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *Init = D->getInitExpr())
7191ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(MakeCXCursor(Init, StmtParent, TU));
7201ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
7211ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
7221ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
7237d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorbool CursorVisitor::VisitDeclaratorDecl(DeclaratorDecl *DD) {
7247d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  if (TypeSourceInfo *TSInfo = DD->getTypeSourceInfo())
7257d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    if (Visit(TSInfo->getTypeLoc()))
7267d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      return true;
7277d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
728c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  // Visit the nested-name-specifier, if present.
729c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = DD->getQualifierLoc())
730c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
731c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      return true;
732c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor
7337d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  return false;
7347d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor}
7357d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
736a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor/// \brief Compare two base or member initializers based on their source order.
737cbb67480094b3bcb5b715acd827cbad55e2a204cSean Huntstatic int CompareCXXCtorInitializers(const void* Xp, const void *Yp) {
738cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt  CXXCtorInitializer const * const *X
739cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt    = static_cast<CXXCtorInitializer const * const *>(Xp);
740cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt  CXXCtorInitializer const * const *Y
741cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt    = static_cast<CXXCtorInitializer const * const *>(Yp);
742a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
743a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  if ((*X)->getSourceOrder() < (*Y)->getSourceOrder())
744a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return -1;
745a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  else if ((*X)->getSourceOrder() > (*Y)->getSourceOrder())
746a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return 1;
747a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  else
748a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return 0;
749a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor}
750a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
751b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitFunctionDecl(FunctionDecl *ND) {
75201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  if (TypeSourceInfo *TSInfo = ND->getTypeSourceInfo()) {
75301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the function declaration's syntactic components in the order
75401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // written. This requires a bit of work.
755723df245307a530da5433dfb43accf187dc3e243Abramo Bagnara    TypeLoc TL = TSInfo->getTypeLoc().IgnoreParens();
75601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    FunctionTypeLoc *FTL = dyn_cast<FunctionTypeLoc>(&TL);
75701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
75801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // If we have a function declared directly (without the use of a typedef),
75901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // visit just the return type. Otherwise, just visit the function's type
76001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // now.
76101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if ((FTL && !isa<CXXConversionDecl>(ND) && Visit(FTL->getResultLoc())) ||
76201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor        (!FTL && Visit(TL)))
76301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
76401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
765c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // Visit the nested-name-specifier, if present.
766c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (NestedNameSpecifierLoc QualifierLoc = ND->getQualifierLoc())
767c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      if (VisitNestedNameSpecifierLoc(QualifierLoc))
768c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor        return true;
76901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
77001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the declaration name.
77101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (VisitDeclarationNameInfo(ND->getNameInfo()))
77201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
77301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
77401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Visit explicitly-specified template arguments!
77501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
77601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the function parameters, if we have a function type.
77701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (FTL && VisitFunctionTypeLoc(*FTL, true))
77801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
77901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
78001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Attributes?
78101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  }
78201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
783a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  if (ND->isThisDeclarationADefinition()) {
784a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    if (CXXConstructorDecl *Constructor = dyn_cast<CXXConstructorDecl>(ND)) {
785a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Find the initializers that were written in the source.
786cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt      llvm::SmallVector<CXXCtorInitializer *, 4> WrittenInits;
787a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      for (CXXConstructorDecl::init_iterator I = Constructor->init_begin(),
788a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor                                          IEnd = Constructor->init_end();
789a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor           I != IEnd; ++I) {
790a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        if (!(*I)->isWritten())
791a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor          continue;
792a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
793a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        WrittenInits.push_back(*I);
794a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      }
795a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
796a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Sort the initializers in source order
797a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      llvm::array_pod_sort(WrittenInits.begin(), WrittenInits.end(),
798cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt                           &CompareCXXCtorInitializers);
799a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
800a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Visit the initializers in source order
801a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      for (unsigned I = 0, N = WrittenInits.size(); I != N; ++I) {
802cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt        CXXCtorInitializer *Init = WrittenInits[I];
80300eb3f9c5b33e3d99aee1f8b75dd9c9678fdd66bFrancois Pichet        if (Init->isAnyMemberInitializer()) {
80400eb3f9c5b33e3d99aee1f8b75dd9c9678fdd66bFrancois Pichet          if (Visit(MakeCursorMemberRef(Init->getAnyMember(),
805a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor                                        Init->getMemberLocation(), TU)))
806a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
807a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        } else if (TypeSourceInfo *BaseInfo = Init->getBaseClassInfo()) {
808a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor          if (Visit(BaseInfo->getTypeLoc()))
809a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
810a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        }
811a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
812a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        // Visit the initializer value.
813a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        if (Expr *Initializer = Init->getInit())
814a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor          if (Visit(MakeCXCursor(Initializer, ND, TU)))
815a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
816a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      }
817a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
818a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
819a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    if (Visit(MakeCXCursor(ND->getBody(), StmtParent, TU)))
820a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return true;
821a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  }
822f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
823b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
824b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
825dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
8261ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitFieldDecl(FieldDecl *D) {
8271ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (VisitDeclaratorDecl(D))
8281ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
829f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8301ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *BitWidth = D->getBitWidth())
8311ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(MakeCXCursor(BitWidth, StmtParent, TU));
832f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8331ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
8341ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8351ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
8361ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitVarDecl(VarDecl *D) {
8371ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (VisitDeclaratorDecl(D))
8381ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
839f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8401ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *Init = D->getInit())
8411ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(MakeCXCursor(Init, StmtParent, TU));
842f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8431ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
8441ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8451ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
84684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregorbool CursorVisitor::VisitNonTypeTemplateParmDecl(NonTypeTemplateParmDecl *D) {
84784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (VisitDeclaratorDecl(D))
84884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
84984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
85084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
85184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    if (Expr *DefArg = D->getDefaultArgument())
85284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      return Visit(MakeCXCursor(DefArg, StmtParent, TU));
85384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
85484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  return false;
85584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor}
85684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
857fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitFunctionTemplateDecl(FunctionTemplateDecl *D) {
858fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // FIXME: Visit the "outer" template parameter lists on the FunctionDecl
859fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // before visiting these template parameters.
860fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
861fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return true;
862fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
863fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return VisitFunctionDecl(D->getTemplatedDecl());
864fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
865fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
86639d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregorbool CursorVisitor::VisitClassTemplateDecl(ClassTemplateDecl *D) {
86739d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  // FIXME: Visit the "outer" template parameter lists on the TagDecl
86839d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  // before visiting these template parameters.
86939d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
87039d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor    return true;
87139d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor
87239d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  return VisitCXXRecordDecl(D->getTemplatedDecl());
87339d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor}
87439d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor
87584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregorbool CursorVisitor::VisitTemplateTemplateParmDecl(TemplateTemplateParmDecl *D) {
87684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
87784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
87884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
87984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited() &&
88084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      VisitTemplateArgumentLoc(D->getDefaultArgument()))
88184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
88284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
88384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  return false;
88484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor}
88584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
8861ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCMethodDecl(ObjCMethodDecl *ND) {
8874bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor  if (TypeSourceInfo *TSInfo = ND->getResultTypeSourceInfo())
8884bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor    if (Visit(TSInfo->getTypeLoc()))
8894bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor      return true;
8904bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor
891f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  for (ObjCMethodDecl::param_iterator P = ND->param_begin(),
8921ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       PEnd = ND->param_end();
8931ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       P != PEnd; ++P) {
8941ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    if (Visit(MakeCXCursor(*P, TU)))
8951ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
8961ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  }
897f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8981ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (ND->isThisDeclarationADefinition() &&
8991ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      Visit(MakeCXCursor(ND->getBody(), StmtParent, TU)))
9001ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
901f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
9021ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
9031ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
9041ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
905d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremeneknamespace {
906d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  struct ContainerDeclsSort {
907d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    SourceManager &SM;
908d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    ContainerDeclsSort(SourceManager &sm) : SM(sm) {}
909d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    bool operator()(Decl *A, Decl *B) {
910d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L_A = A->getLocStart();
911d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L_B = B->getLocStart();
912d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      assert(L_A.isValid() && L_B.isValid());
913d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return SM.isBeforeInTranslationUnit(L_A, L_B);
914d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
915d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  };
916d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek}
917d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
918a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregorbool CursorVisitor::VisitObjCContainerDecl(ObjCContainerDecl *D) {
919d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually convert back to just 'VisitDeclContext()'.  Essentially
920d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // an @implementation can lexically contain Decls that are not properly
921d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // nested in the AST.  When we identify such cases, we need to retrofit
922d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // this nesting here.
923d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (!DI_current)
924d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    return VisitDeclContext(D);
925d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
926d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Scan the Decls that immediately come after the container
927d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // in the current DeclContext.  If any fall within the
928d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // container's lexical region, stash them into a vector
929d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // for later processing.
930d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  llvm::SmallVector<Decl *, 24> DeclsInContainer;
931d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SourceLocation EndLoc = D->getSourceRange().getEnd();
932a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  SourceManager &SM = AU->getSourceManager();
933d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (EndLoc.isValid()) {
934d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    DeclContext::decl_iterator next = *DI_current;
935d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    while (++next != DE_current) {
936d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      Decl *D_next = *next;
937d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      if (!D_next)
938d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        break;
939d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L = D_next->getLocStart();
940d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      if (!L.isValid())
941d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        break;
942d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      if (SM.isBeforeInTranslationUnit(L, EndLoc)) {
943d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        *DI_current = next;
944d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        DeclsInContainer.push_back(D_next);
945d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        continue;
946d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      }
947d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      break;
948d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
949d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
950d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
951d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // The common case.
952d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (DeclsInContainer.empty())
953d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    return VisitDeclContext(D);
954d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
955d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Get all the Decls in the DeclContext, and sort them with the
956d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // additional ones we've collected.  Then visit them.
957d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for (DeclContext::decl_iterator I = D->decls_begin(), E = D->decls_end();
958d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek       I!=E; ++I) {
959d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    Decl *subDecl = *I;
9600582c897ec7261b4c6af0fe26dc2a0b6b54d266cTed Kremenek    if (!subDecl || subDecl->getLexicalDeclContext() != D ||
9610582c897ec7261b4c6af0fe26dc2a0b6b54d266cTed Kremenek        subDecl->getLocStart().isInvalid())
962d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
963d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    DeclsInContainer.push_back(subDecl);
964d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
965d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
966d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Now sort the Decls so that they appear in lexical order.
967d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  std::sort(DeclsInContainer.begin(), DeclsInContainer.end(),
968d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek            ContainerDeclsSort(SM));
969d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
970d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Now visit the decls.
971d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for (llvm::SmallVectorImpl<Decl*>::iterator I = DeclsInContainer.begin(),
972d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek         E = DeclsInContainer.end(); I != E; ++I) {
973d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    CXCursor Cursor = MakeCXCursor(*I, TU);
974d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    const llvm::Optional<bool> &V = shouldVisitCursor(Cursor);
975d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.hasValue())
976d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
977d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.getValue())
978d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
979d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (Visit(Cursor, true))
980d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return true;
981d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
982d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  return false;
983a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor}
984a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor
985b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitObjCCategoryDecl(ObjCCategoryDecl *ND) {
986b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (Visit(MakeCursorObjCClassRef(ND->getClassInterface(), ND->getLocation(),
987b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                   TU)))
988b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
989f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
99078db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  ObjCCategoryDecl::protocol_loc_iterator PL = ND->protocol_loc_begin();
99178db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  for (ObjCCategoryDecl::protocol_iterator I = ND->protocol_begin(),
99278db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor         E = ND->protocol_end(); I != E; ++I, ++PL)
993b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
994b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
995f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
996a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitObjCContainerDecl(ND);
997dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
998dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
9991ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCProtocolDecl(ObjCProtocolDecl *PID) {
10001ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  ObjCProtocolDecl::protocol_loc_iterator PL = PID->protocol_loc_begin();
10011ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCProtocolDecl::protocol_iterator I = PID->protocol_begin(),
10021ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       E = PID->protocol_end(); I != E; ++I, ++PL)
10031ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
10041ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
1005f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10061ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCContainerDecl(PID);
10071ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
10081ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
100923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenekbool CursorVisitor::VisitObjCPropertyDecl(ObjCPropertyDecl *PD) {
101083cb94269015bf2770ade71e616c5322ea7e76e1Douglas Gregor  if (PD->getTypeSourceInfo() && Visit(PD->getTypeSourceInfo()->getTypeLoc()))
1011fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall    return true;
1012fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall
101323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // FIXME: This implements a workaround with @property declarations also being
101423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // installed in the DeclContext for the @interface.  Eventually this code
101523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // should be removed.
101623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCCategoryDecl *CDecl = dyn_cast<ObjCCategoryDecl>(PD->getDeclContext());
101723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!CDecl || !CDecl->IsClassExtension())
101823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
101923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
102023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCInterfaceDecl *ID = CDecl->getClassInterface();
102123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!ID)
102223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
102323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
102423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  IdentifierInfo *PropertyId = PD->getIdentifier();
102523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCPropertyDecl *prevDecl =
102623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(ID), PropertyId);
102723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
102823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!prevDecl)
102923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
103023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
103123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // Visit synthesized methods since they will be skipped when visiting
103223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // the @interface.
103323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (ObjCMethodDecl *MD = prevDecl->getGetterMethodDecl())
1034a054fb46b1fb596d1719b89d2d9a5be3c32a4b0dTed Kremenek    if (MD->isSynthesized() && MD->getLexicalDeclContext() == CDecl)
103523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek      if (Visit(MakeCXCursor(MD, TU)))
103623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return true;
103723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
103823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (ObjCMethodDecl *MD = prevDecl->getSetterMethodDecl())
1039a054fb46b1fb596d1719b89d2d9a5be3c32a4b0dTed Kremenek    if (MD->isSynthesized() && MD->getLexicalDeclContext() == CDecl)
104023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek      if (Visit(MakeCXCursor(MD, TU)))
104123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return true;
104223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
104323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  return false;
104423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek}
104523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
1046b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitObjCInterfaceDecl(ObjCInterfaceDecl *D) {
1047dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek  // Issue callbacks for super class.
1048b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (D->getSuperClass() &&
1049b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1050f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        D->getSuperClassLoc(),
1051b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                        TU)))
1052b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
1053f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
105478db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  ObjCInterfaceDecl::protocol_loc_iterator PL = D->protocol_loc_begin();
105578db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  for (ObjCInterfaceDecl::protocol_iterator I = D->protocol_begin(),
105678db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor         E = D->protocol_end(); I != E; ++I, ++PL)
1057b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1058b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
1059f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1060a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitObjCContainerDecl(D);
1061dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1062dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
10631ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCImplDecl(ObjCImplDecl *D) {
10641ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCContainerDecl(D);
10651ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
10661ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
10671ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D) {
1068ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  // 'ID' could be null when dealing with invalid code.
1069ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  if (ObjCInterfaceDecl *ID = D->getClassInterface())
1070ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek    if (Visit(MakeCursorObjCClassRef(ID, D->getLocation(), TU)))
1071ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek      return true;
1072f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10731ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCImplDecl(D);
10741ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
10751ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
10761ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCImplementationDecl(ObjCImplementationDecl *D) {
10771ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor#if 0
10781ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // Issue callbacks for super class.
10791ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // FIXME: No source location information!
10801ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (D->getSuperClass() &&
10811ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1082f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        D->getSuperClassLoc(),
10831ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor                                        TU)))
1084a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return true;
10851ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor#endif
1086f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10871ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCImplDecl(D);
1088dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1089dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
10901ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCForwardProtocolDecl(ObjCForwardProtocolDecl *D) {
10911ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  ObjCForwardProtocolDecl::protocol_loc_iterator PL = D->protocol_loc_begin();
10921ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCForwardProtocolDecl::protocol_iterator I = D->protocol_begin(),
10931ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor                                                  E = D->protocol_end();
10941ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       I != E; ++I, ++PL)
1095b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1096b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
1097f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1098f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return false;
1099dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1100dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
11011ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCClassDecl(ObjCClassDecl *D) {
11021ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCClassDecl::iterator C = D->begin(), CEnd = D->end(); C != CEnd; ++C)
11031ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    if (Visit(MakeCursorObjCClassRef(C->getInterface(), C->getLocation(), TU)))
11041ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
1105f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
11061ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
1107dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
11085e4bc590b0ea010e38372d0b4a0aab578a746fe6Benjamin Kramer
1109a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregorbool CursorVisitor::VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *PD) {
1110a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  if (ObjCIvarDecl *Ivar = PD->getPropertyIvarDecl())
1111a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor    return Visit(MakeCursorMemberRef(Ivar, PD->getPropertyIvarDeclLoc(), TU));
1112a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor
1113a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  return false;
1114a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor}
1115a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor
11168f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenekbool CursorVisitor::VisitNamespaceDecl(NamespaceDecl *D) {
11178f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  return VisitDeclContext(D);
11188f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek}
11198f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek
11206931900f43cea558c6974075256c07728dbfecc6Douglas Gregorbool CursorVisitor::VisitNamespaceAliasDecl(NamespaceAliasDecl *D) {
1121c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
11220cfaf6a270ecd0f5c7e541a8047c87948317548bDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
11230cfaf6a270ecd0f5c7e541a8047c87948317548bDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1124c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
11256931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
11266931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  return Visit(MakeCursorNamespaceRef(D->getAliasedNamespace(),
11276931900f43cea558c6974075256c07728dbfecc6Douglas Gregor                                      D->getTargetNameLoc(), TU));
11286931900f43cea558c6974075256c07728dbfecc6Douglas Gregor}
11296931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
11307e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUsingDecl(UsingDecl *D) {
1131c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1132dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1133dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1134c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1135dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
11367e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
11371f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (Visit(MakeCursorOverloadedDeclRef(D, D->getLocation(), TU)))
11381f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return true;
11391f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
11407e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return VisitDeclarationNameInfo(D->getNameInfo());
11417e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11427e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
11430a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregorbool CursorVisitor::VisitUsingDirectiveDecl(UsingDirectiveDecl *D) {
1144c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1145db9924191092b4d426cc066637d81698211846aaDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1146db9924191092b4d426cc066637d81698211846aaDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1147c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
11480a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
11490a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  return Visit(MakeCursorNamespaceRef(D->getNominatedNamespaceAsWritten(),
11500a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor                                      D->getIdentLocation(), TU));
11510a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor}
11520a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
11537e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D) {
1154c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1155dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1156dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1157c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1158dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
1159c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
11607e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return VisitDeclarationNameInfo(D->getNameInfo());
11617e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11627e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
11637e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUnresolvedUsingTypenameDecl(
11647e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor                                               UnresolvedUsingTypenameDecl *D) {
1165c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1166dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1167dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1168c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1169c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
11707e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return false;
11717e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11727e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
117301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitDeclarationNameInfo(DeclarationNameInfo Name) {
117401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  switch (Name.getName().getNameKind()) {
117501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::Identifier:
117601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXLiteralOperatorName:
117701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXOperatorName:
117801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXUsingDirective:
117901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
118001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
118101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXConstructorName:
118201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXDestructorName:
118301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXConversionFunctionName:
118401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (TypeSourceInfo *TSInfo = Name.getNamedTypeInfo())
118501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return Visit(TSInfo->getTypeLoc());
118601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
118701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
118801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCZeroArgSelector:
118901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCOneArgSelector:
119001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCMultiArgSelector:
119101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Per-identifier location info?
119201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
119301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  }
119401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
119501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  return false;
119601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor}
119701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
1198c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregorbool CursorVisitor::VisitNestedNameSpecifier(NestedNameSpecifier *NNS,
1199c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                             SourceRange Range) {
1200c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // FIXME: This whole routine is a hack to work around the lack of proper
1201c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // source information in nested-name-specifiers (PR5791). Since we do have
1202c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // a beginning source location, we can visit the first component of the
1203c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // nested-name-specifier, if it's a single-token component.
1204c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  if (!NNS)
1205c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    return false;
1206c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1207c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Get the first component in the nested-name-specifier.
1208c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  while (NestedNameSpecifier *Prefix = NNS->getPrefix())
1209c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    NNS = Prefix;
1210c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1211c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  switch (NNS->getKind()) {
1212c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Namespace:
1213c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    return Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(), Range.getBegin(),
1214c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                        TU));
1215c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
121614aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor  case NestedNameSpecifier::NamespaceAlias:
121714aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor    return Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
121814aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor                                        Range.getBegin(), TU));
121914aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor
1220c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::TypeSpec: {
1221c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // If the type has a form where we know that the beginning of the source
1222c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // range matches up with a reference cursor. Visit the appropriate reference
1223c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // cursor.
1224f4c7371fb1d3cebcfb40abad4537bb82515704eaJohn McCall    const Type *T = NNS->getAsType();
1225c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TypedefType *Typedef = dyn_cast<TypedefType>(T))
1226c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return Visit(MakeCursorTypeRef(Typedef->getDecl(), Range.getBegin(), TU));
1227c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TagType *Tag = dyn_cast<TagType>(T))
1228c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return Visit(MakeCursorTypeRef(Tag->getDecl(), Range.getBegin(), TU));
1229c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TemplateSpecializationType *TST
1230c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                      = dyn_cast<TemplateSpecializationType>(T))
1231c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return VisitTemplateName(TST->getTemplateName(), Range.getBegin());
1232c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    break;
1233c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  }
1234c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1235c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::TypeSpecWithTemplate:
1236c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Global:
1237c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Identifier:
1238c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    break;
1239c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  }
1240c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1241c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  return false;
1242c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor}
1243c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1244dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregorbool
1245dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas GregorCursorVisitor::VisitNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
1246dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  llvm::SmallVector<NestedNameSpecifierLoc, 4> Qualifiers;
1247dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  for (; Qualifier; Qualifier = Qualifier.getPrefix())
1248dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    Qualifiers.push_back(Qualifier);
1249dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1250dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  while (!Qualifiers.empty()) {
1251dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    NestedNameSpecifierLoc Q = Qualifiers.pop_back_val();
1252dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    NestedNameSpecifier *NNS = Q.getNestedNameSpecifier();
1253dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    switch (NNS->getKind()) {
1254dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Namespace:
1255dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(),
1256c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor                                       Q.getLocalBeginLoc(),
1257dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor                                       TU)))
1258dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1259dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1260dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1261dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1262dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::NamespaceAlias:
1263dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
1264c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor                                       Q.getLocalBeginLoc(),
1265dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor                                       TU)))
1266dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1267dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1268dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1269dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1270dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::TypeSpec:
1271dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::TypeSpecWithTemplate:
1272dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(Q.getTypeLoc()))
1273dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1274dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1275dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1276dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1277dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Global:
1278dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Identifier:
1279dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1280dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    }
1281dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
1282dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1283dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  return false;
1284dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor}
1285dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1286fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateParameters(
1287fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                          const TemplateParameterList *Params) {
1288fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  if (!Params)
1289fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1290fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1291fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  for (TemplateParameterList::const_iterator P = Params->begin(),
1292fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                          PEnd = Params->end();
1293fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor       P != PEnd; ++P) {
1294fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Visit(MakeCXCursor(*P, TU)))
1295fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return true;
1296fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  }
1297fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1298fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1299fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1300fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
13010b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregorbool CursorVisitor::VisitTemplateName(TemplateName Name, SourceLocation Loc) {
13020b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  switch (Name.getKind()) {
13030b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::Template:
13040b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return Visit(MakeCursorTemplateRef(Name.getAsTemplateDecl(), Loc, TU));
13050b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
13060b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::OverloadedTemplate:
13071f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    // Visit the overloaded template set.
13081f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (Visit(MakeCursorOverloadedDeclRef(Name, Loc, TU)))
13091f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return true;
13101f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
13110b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return false;
13120b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
13130b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::DependentTemplate:
13140b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    // FIXME: Visit nested-name-specifier.
13150b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return false;
13160b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
13170b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::QualifiedTemplate:
13180b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    // FIXME: Visit nested-name-specifier.
13190b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return Visit(MakeCursorTemplateRef(
13200b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                                  Name.getAsQualifiedTemplateName()->getDecl(),
13210b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                                       Loc, TU));
13221aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor
13231aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor  case TemplateName::SubstTemplateTemplateParmPack:
13241aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor    return Visit(MakeCursorTemplateRef(
13251aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor                  Name.getAsSubstTemplateTemplateParmPack()->getParameterPack(),
13261aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor                                       Loc, TU));
13270b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  }
13280b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
13290b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  return false;
13300b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor}
13310b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
1332fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateArgumentLoc(const TemplateArgumentLoc &TAL) {
1333fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  switch (TAL.getArgument().getKind()) {
1334fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Null:
1335fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Integral:
1336fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Pack:
1337fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
133887dd697dcc8ecb64df73ae64d61b8c80ff0c157cDouglas Gregor
1339fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Type:
1340fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (TypeSourceInfo *TSInfo = TAL.getTypeSourceInfo())
1341fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return Visit(TSInfo->getTypeLoc());
1342fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1343fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1344fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Declaration:
1345fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Expr *E = TAL.getSourceDeclExpression())
1346fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return Visit(MakeCXCursor(E, StmtParent, TU));
1347fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1348fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1349fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Expression:
1350fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Expr *E = TAL.getSourceExpression())
1351fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return Visit(MakeCXCursor(E, StmtParent, TU));
1352fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1353fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1354fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Template:
1355a7fc901a2e39bfe55bfcff5934b2d9fdf9656491Douglas Gregor  case TemplateArgument::TemplateExpansion:
1356b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor    if (VisitNestedNameSpecifierLoc(TAL.getTemplateQualifierLoc()))
1357b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor      return true;
1358b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor
1359a7fc901a2e39bfe55bfcff5934b2d9fdf9656491Douglas Gregor    return VisitTemplateName(TAL.getArgument().getAsTemplateOrTemplatePattern(),
13600b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                             TAL.getTemplateNameLoc());
1361fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  }
1362fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1363fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1364fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1365fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1366a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenekbool CursorVisitor::VisitLinkageSpecDecl(LinkageSpecDecl *D) {
1367a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  return VisitDeclContext(D);
1368a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek}
1369a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek
137001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitQualifiedTypeLoc(QualifiedTypeLoc TL) {
137101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  return Visit(TL.getUnqualifiedLoc());
137201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor}
137301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
1374f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitBuiltinTypeLoc(BuiltinTypeLoc TL) {
1375a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTContext &Context = AU->getASTContext();
1376f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1377f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // Some builtin types (such as Objective-C's "id", "sel", and
1378f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // "Class") have associated declarations. Create cursors for those.
1379f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  QualType VisitType;
1380f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  switch (TL.getType()->getAs<BuiltinType>()->getKind()) {
13816b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Void:
1382f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Bool:
13836b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Char_U:
13846b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UChar:
1385f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Char16:
1386f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Char32:
13876b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UShort:
13886b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UInt:
13896b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ULong:
13906b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ULongLong:
13916b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UInt128:
1392f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Char_S:
13936b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::SChar:
13943f59c975aa5d047f7edd1b900b5e885c38af0ef7Chris Lattner  case BuiltinType::WChar_U:
13953f59c975aa5d047f7edd1b900b5e885c38af0ef7Chris Lattner  case BuiltinType::WChar_S:
13966b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Short:
1397f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Int:
1398f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Long:
1399c4174cc4b9b657abb77d0825de473ea29cf48297Ted Kremenek  case BuiltinType::LongLong:
14006b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Int128:
14016b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Float:
14026b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Double:
14036b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::LongDouble:
1404f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::NullPtr:
1405f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Overload:
14066b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Dependent:
14071de4d4e8cb2e9c88809fea8092bc6e835a5473d2John McCall  case BuiltinType::UnknownAny:
1408f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
14096b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
1410f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::ObjCId:
1411f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    VisitType = Context.getObjCIdType();
1412f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
14136b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
14146b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ObjCClass:
14156b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek    VisitType = Context.getObjCClassType();
14166b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek    break;
14176b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
1418f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::ObjCSel:
1419f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    VisitType = Context.getObjCSelType();
1420f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
1421f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1422f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1423f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (!VisitType.isNull()) {
1424f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    if (const TypedefType *Typedef = VisitType->getAs<TypedefType>())
1425f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek      return Visit(MakeCursorTypeRef(Typedef->getDecl(), TL.getBuiltinLoc(),
1426f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                     TU));
1427f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1428f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1429f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1430f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1431f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
14327d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorbool CursorVisitor::VisitTypedefTypeLoc(TypedefTypeLoc TL) {
14337d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getTypedefDecl(), TL.getNameLoc(), TU));
14347d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor}
14357d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
1436f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitUnresolvedUsingTypeLoc(UnresolvedUsingTypeLoc TL) {
1437f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1438f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1439f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1440f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitTagTypeLoc(TagTypeLoc TL) {
1441f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1442f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1443f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1444fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateTypeParmTypeLoc(TemplateTypeParmTypeLoc TL) {
14451f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  // FIXME: We can't visit the template type parameter, because there's
1446fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // no context information with which we can match up the depth/index in the
1447fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // type to the appropriate
1448fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1449fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1450fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1451f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL) {
1452f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(MakeCursorObjCClassRef(TL.getIFaceDecl(), TL.getNameLoc(), TU)))
1453f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1454f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1455c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  return false;
1456c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall}
1457c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall
1458c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCallbool CursorVisitor::VisitObjCObjectTypeLoc(ObjCObjectTypeLoc TL) {
1459c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  if (TL.hasBaseTypeAsWritten() && Visit(TL.getBaseLoc()))
1460c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall    return true;
1461c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall
1462f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  for (unsigned I = 0, N = TL.getNumProtocols(); I != N; ++I) {
1463f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    if (Visit(MakeCursorObjCProtocolRef(TL.getProtocol(I), TL.getProtocolLoc(I),
1464f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                        TU)))
1465f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor      return true;
1466f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1467f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1468f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1469f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1470f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1471f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitObjCObjectPointerTypeLoc(ObjCObjectPointerTypeLoc TL) {
1472c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  return Visit(TL.getPointeeLoc());
1473f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1474f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1475075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnarabool CursorVisitor::VisitParenTypeLoc(ParenTypeLoc TL) {
1476075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara  return Visit(TL.getInnerLoc());
1477075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara}
1478075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara
1479f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitPointerTypeLoc(PointerTypeLoc TL) {
1480f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1481f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1482f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1483f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitBlockPointerTypeLoc(BlockPointerTypeLoc TL) {
1484f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1485f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1486f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1487f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitMemberPointerTypeLoc(MemberPointerTypeLoc TL) {
1488f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1489f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1490f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1491f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitLValueReferenceTypeLoc(LValueReferenceTypeLoc TL) {
1492f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return Visit(TL.getPointeeLoc());
1493f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1494f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1495f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitRValueReferenceTypeLoc(RValueReferenceTypeLoc TL) {
1496f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return Visit(TL.getPointeeLoc());
1497f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1498f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
149901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitFunctionTypeLoc(FunctionTypeLoc TL,
150001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor                                         bool SkipResultType) {
150101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  if (!SkipResultType && Visit(TL.getResultLoc()))
1502f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1503f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1504f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
15055dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek    if (Decl *D = TL.getArg(I))
15065dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek      if (Visit(MakeCXCursor(D, TU)))
15075dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek        return true;
1508f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1509f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1510f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1511f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1512f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitArrayTypeLoc(ArrayTypeLoc TL) {
1513f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(TL.getElementLoc()))
1514f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1515f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1516f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Expr *Size = TL.getSizeExpr())
1517f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return Visit(MakeCXCursor(Size, StmtParent, TU));
1518f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1519f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1520f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1521f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1522fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateSpecializationTypeLoc(
1523fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                             TemplateSpecializationTypeLoc TL) {
15240b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  // Visit the template name.
15250b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  if (VisitTemplateName(TL.getTypePtr()->getTemplateName(),
15260b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                        TL.getTemplateNameLoc()))
15270b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return true;
1528fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1529fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // Visit the template arguments.
1530fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
1531fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
1532fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return true;
1533fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1534fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1535fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1536fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
15372332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregorbool CursorVisitor::VisitTypeOfExprTypeLoc(TypeOfExprTypeLoc TL) {
15382332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  return Visit(MakeCXCursor(TL.getUnderlyingExpr(), StmtParent, TU));
15392332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor}
15402332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15412332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregorbool CursorVisitor::VisitTypeOfTypeLoc(TypeOfTypeLoc TL) {
15422332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
15432332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor    return Visit(TSInfo->getTypeLoc());
15442332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15452332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  return false;
15462332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor}
15472332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15482494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregorbool CursorVisitor::VisitDependentNameTypeLoc(DependentNameTypeLoc TL) {
15492494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor  if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
15502494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    return true;
15512494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
15522494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor  return false;
15532494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor}
15542494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
155594fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregorbool CursorVisitor::VisitDependentTemplateSpecializationTypeLoc(
155694fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor                                    DependentTemplateSpecializationTypeLoc TL) {
155794fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  // Visit the nested-name-specifier, if there is one.
155894fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  if (TL.getQualifierLoc() &&
155994fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor      VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
156094fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor    return true;
156194fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
156294fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  // Visit the template arguments.
156394fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
156494fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor    if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
156594fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor      return true;
156694fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
156794fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  return false;
156894fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor}
156994fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
15709e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregorbool CursorVisitor::VisitElaboratedTypeLoc(ElaboratedTypeLoc TL) {
15719e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor  if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
15729e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor    return true;
15739e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor
15749e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor  return Visit(TL.getNamedTypeLoc());
15759e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor}
15769e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor
15777536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregorbool CursorVisitor::VisitPackExpansionTypeLoc(PackExpansionTypeLoc TL) {
15787536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor  return Visit(TL.getPatternLoc());
15797536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor}
15807536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor
15813064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenekbool CursorVisitor::VisitCXXRecordDecl(CXXRecordDecl *D) {
1582c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  // Visit the nested-name-specifier, if present.
1583c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1584c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1585c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      return true;
1586c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor
15873064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  if (D->isDefinition()) {
15883064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    for (CXXRecordDecl::base_class_iterator I = D->bases_begin(),
15893064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek         E = D->bases_end(); I != E; ++I) {
15903064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      if (Visit(cxcursor::MakeCursorCXXBaseSpecifier(I, TU)))
15913064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek        return true;
15923064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
15933064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  }
15943064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek
15953064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  return VisitTagDecl(D);
15963064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek}
15973064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek
159809dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenekbool CursorVisitor::VisitAttributes(Decl *D) {
1599cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt  for (AttrVec::const_iterator i = D->attr_begin(), e = D->attr_end();
1600cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt       i != e; ++i)
1601cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt    if (Visit(MakeCXCursor(*i, D, TU)))
160209dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek        return true;
160309dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
160409dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek  return false;
160509dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek}
160609dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
1607c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
1608c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek// Data-recursive visitor methods.
1609c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
1610c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
161128a719433411ef782b582946823bc648ddcc4533Ted Kremeneknamespace {
1612035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek#define DEF_JOB(NAME, DATA, KIND)\
1613035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass NAME : public VisitorJob {\
1614035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:\
1615035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  NAME(DATA *d, CXCursor parent) : VisitorJob(parent, VisitorJob::KIND, d) {} \
1616035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) { return VJ->getKind() == KIND; }\
1617f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DATA *get() const { return static_cast<DATA*>(data[0]); }\
1618035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1619035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1620035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(StmtVisit, Stmt, StmtVisitKind)
1621035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(MemberExprParts, MemberExpr, MemberExprPartsKind)
1622e4979ccb5960608edce73f3b274eb7c2de15dac5Ted KremenekDEF_JOB(DeclRefExprParts, DeclRefExpr, DeclRefExprPartsKind)
1623035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(OverloadExprParts, OverloadExpr, OverloadExprPartsKind)
162460608ec12d17168a3d1f415409a6a6eaf6d94508Ted KremenekDEF_JOB(ExplicitTemplateArgsVisit, ExplicitTemplateArgumentList,
162560608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        ExplicitTemplateArgsVisitKind)
162694d96291cd041adc5731a2294828a9c20e450b74Douglas GregorDEF_JOB(SizeOfPackExprParts, SizeOfPackExpr, SizeOfPackExprPartsKind)
1627035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek#undef DEF_JOB
1628035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1629035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass DeclVisit : public VisitorJob {
1630035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:
1631035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  DeclVisit(Decl *d, CXCursor parent, bool isFirst) :
1632035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    VisitorJob(parent, VisitorJob::DeclVisitKind,
1633035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek               d, isFirst ? (void*) 1 : (void*) 0) {}
1634035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) {
163582f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    return VJ->getKind() == DeclVisitKind;
1636035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1637f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  Decl *get() const { return static_cast<Decl*>(data[0]); }
1638f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  bool isFirst() const { return data[1] ? true : false; }
1639035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1640035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass TypeLocVisit : public VisitorJob {
1641035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:
1642035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  TypeLocVisit(TypeLoc tl, CXCursor parent) :
1643035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    VisitorJob(parent, VisitorJob::TypeLocVisitKind,
1644035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek               tl.getType().getAsOpaquePtr(), tl.getOpaqueData()) {}
1645035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1646035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1647035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    return VJ->getKind() == TypeLocVisitKind;
1648035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1649035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
165082f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek  TypeLoc get() const {
1651f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    QualType T = QualType::getFromOpaquePtr(data[0]);
1652f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return TypeLoc(T, data[1]);
1653035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1654035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1655035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1656ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekclass LabelRefVisit : public VisitorJob {
1657ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekpublic:
1658ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  LabelRefVisit(LabelDecl *LD, SourceLocation labelLoc, CXCursor parent)
1659ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner    : VisitorJob(parent, VisitorJob::LabelRefVisitKind, LD,
1660dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 labelLoc.getPtrEncoding()) {}
1661ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek
1662ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1663ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek    return VJ->getKind() == VisitorJob::LabelRefVisitKind;
1664ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  }
1665ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  LabelDecl *get() const { return static_cast<LabelDecl*>(data[0]); }
1666ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  SourceLocation getLoc() const {
1667dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin    return SourceLocation::getFromPtrEncoding(data[1]); }
1668f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek};
1669f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekclass NestedNameSpecifierVisit : public VisitorJob {
1670f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekpublic:
1671f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  NestedNameSpecifierVisit(NestedNameSpecifier *NS, SourceRange R,
1672f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek                           CXCursor parent)
1673f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    : VisitorJob(parent, VisitorJob::NestedNameSpecifierVisitKind,
1674dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 NS, R.getBegin().getPtrEncoding(),
1675dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 R.getEnd().getPtrEncoding()) {}
1676f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1677f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return VJ->getKind() == VisitorJob::NestedNameSpecifierVisitKind;
1678f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1679f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  NestedNameSpecifier *get() const {
1680f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return static_cast<NestedNameSpecifier*>(data[0]);
1681f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1682f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  SourceRange getSourceRange() const {
1683f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    SourceLocation A =
1684f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      SourceLocation::getFromRawEncoding((unsigned)(uintptr_t) data[1]);
1685f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    SourceLocation B =
1686f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      SourceLocation::getFromRawEncoding((unsigned)(uintptr_t) data[2]);
1687f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return SourceRange(A, B);
1688f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1689f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek};
1690f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1691f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorclass NestedNameSpecifierLocVisit : public VisitorJob {
1692f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorpublic:
1693f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  NestedNameSpecifierLocVisit(NestedNameSpecifierLoc Qualifier, CXCursor parent)
1694f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    : VisitorJob(parent, VisitorJob::NestedNameSpecifierLocVisitKind,
1695f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                 Qualifier.getNestedNameSpecifier(),
1696f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                 Qualifier.getOpaqueData()) { }
1697f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1698f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  static bool classof(const VisitorJob *VJ) {
1699f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    return VJ->getKind() == VisitorJob::NestedNameSpecifierLocVisitKind;
1700f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  }
1701f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1702f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  NestedNameSpecifierLoc get() const {
1703f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    return NestedNameSpecifierLoc(static_cast<NestedNameSpecifier*>(data[0]),
1704f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                                  data[1]);
1705f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  }
1706f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor};
1707f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1708f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekclass DeclarationNameInfoVisit : public VisitorJob {
1709f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekpublic:
1710f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DeclarationNameInfoVisit(Stmt *S, CXCursor parent)
1711f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    : VisitorJob(parent, VisitorJob::DeclarationNameInfoVisitKind, S) {}
1712f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1713f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return VJ->getKind() == VisitorJob::DeclarationNameInfoVisitKind;
1714f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1715f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DeclarationNameInfo get() const {
1716f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    Stmt *S = static_cast<Stmt*>(data[0]);
1717f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    switch (S->getStmtClass()) {
1718f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    default:
1719f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      llvm_unreachable("Unhandled Stmt");
1720f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    case Stmt::CXXDependentScopeMemberExprClass:
1721f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      return cast<CXXDependentScopeMemberExpr>(S)->getMemberNameInfo();
1722f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    case Stmt::DependentScopeDeclRefExprClass:
1723f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      return cast<DependentScopeDeclRefExpr>(S)->getNameInfo();
1724f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    }
1725f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1726ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek};
1727cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekclass MemberRefVisit : public VisitorJob {
1728cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekpublic:
1729cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  MemberRefVisit(FieldDecl *D, SourceLocation L, CXCursor parent)
1730cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    : VisitorJob(parent, VisitorJob::MemberRefVisitKind, D,
1731dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 L.getPtrEncoding()) {}
1732cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  static bool classof(const VisitorJob *VJ) {
1733cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return VJ->getKind() == VisitorJob::MemberRefVisitKind;
1734cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1735cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  FieldDecl *get() const {
1736cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return static_cast<FieldDecl*>(data[0]);
1737cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1738cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  SourceLocation getLoc() const {
1739cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return SourceLocation::getFromRawEncoding((unsigned)(uintptr_t) data[1]);
1740cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1741cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek};
174228a719433411ef782b582946823bc648ddcc4533Ted Kremenekclass EnqueueVisitor : public StmtVisitor<EnqueueVisitor, void> {
174328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  VisitorWorkList &WL;
174428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  CXCursor Parent;
174528a719433411ef782b582946823bc648ddcc4533Ted Kremenekpublic:
174628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueVisitor(VisitorWorkList &wl, CXCursor parent)
174728a719433411ef782b582946823bc648ddcc4533Ted Kremenek    : WL(wl), Parent(parent) {}
174828a719433411ef782b582946823bc648ddcc4533Ted Kremenek
1749ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  void VisitAddrLabelExpr(AddrLabelExpr *E);
175073d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitBlockExpr(BlockExpr *B);
175128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
1752083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  void VisitCompoundStmt(CompoundStmt *S);
175311b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  void VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) { /* Do nothing. */ }
1754f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E);
175511b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  void VisitCXXNewExpr(CXXNewExpr *E);
17566d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  void VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E);
175728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E);
1758cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E);
175973d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E);
1760b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  void VisitCXXTypeidExpr(CXXTypeidExpr *E);
176155b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  void VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E);
17621e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  void VisitCXXUuidofExpr(CXXUuidofExpr *E);
1763e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek  void VisitDeclRefExpr(DeclRefExpr *D);
1764035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  void VisitDeclStmt(DeclStmt *S);
1765f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E);
1766cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitDesignatedInitExpr(DesignatedInitExpr *E);
176728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitExplicitCastExpr(ExplicitCastExpr *E);
176828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitForStmt(ForStmt *FS);
1769ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  void VisitGotoStmt(GotoStmt *GS);
177028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitIfStmt(IfStmt *If);
177128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitInitListExpr(InitListExpr *IE);
177228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitMemberExpr(MemberExpr *M);
1773cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitOffsetOfExpr(OffsetOfExpr *E);
177473d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitObjCEncodeExpr(ObjCEncodeExpr *E);
177528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitObjCMessageExpr(ObjCMessageExpr *M);
177628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitOverloadExpr(OverloadExpr *E);
1777f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbourne  void VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E);
177828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitStmt(Stmt *S);
177928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitSwitchStmt(SwitchStmt *S);
178028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitWhileStmt(WhileStmt *W);
17812939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek  void VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E);
17826ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  void VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E);
178328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitUnresolvedMemberExpr(UnresolvedMemberExpr *U);
17849d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  void VisitVAArgExpr(VAArgExpr *E);
178594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  void VisitSizeOfPackExpr(SizeOfPackExpr *E);
1786ee8aff06f6a96214731de17b2cb6df407c6c1820Douglas Gregor
178728a719433411ef782b582946823bc648ddcc4533Ted Kremenekprivate:
1788f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void AddDeclarationNameInfo(Stmt *S);
1789f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void AddNestedNameSpecifier(NestedNameSpecifier *NS, SourceRange R);
1790f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  void AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier);
179160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  void AddExplicitTemplateArgs(const ExplicitTemplateArgumentList *A);
1792cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void AddMemberRef(FieldDecl *D, SourceLocation L);
179328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void AddStmt(Stmt *S);
1794035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  void AddDecl(Decl *D, bool isFirst = true);
179528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void AddTypeLoc(TypeSourceInfo *TI);
179628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void EnqueueChildren(Stmt *S);
179728a719433411ef782b582946823bc648ddcc4533Ted Kremenek};
179828a719433411ef782b582946823bc648ddcc4533Ted Kremenek} // end anonyous namespace
179928a719433411ef782b582946823bc648ddcc4533Ted Kremenek
1800f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::AddDeclarationNameInfo(Stmt *S) {
1801f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  // 'S' should always be non-null, since it comes from the
1802f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  // statement we are visiting.
1803f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  WL.push_back(DeclarationNameInfoVisit(S, Parent));
1804f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
1805f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::AddNestedNameSpecifier(NestedNameSpecifier *N,
1806f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek                                            SourceRange R) {
1807f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  if (N)
1808f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    WL.push_back(NestedNameSpecifierVisit(N, R, Parent));
1809f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
1810f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1811f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorvoid
1812f3db29fff6a583ecda823cf909ab7737d8d30129Douglas GregorEnqueueVisitor::AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
1813f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  if (Qualifier)
1814f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    WL.push_back(NestedNameSpecifierLocVisit(Qualifier, Parent));
1815f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor}
1816f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
181728a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::AddStmt(Stmt *S) {
181828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (S)
181928a719433411ef782b582946823bc648ddcc4533Ted Kremenek    WL.push_back(StmtVisit(S, Parent));
182028a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1821035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekvoid EnqueueVisitor::AddDecl(Decl *D, bool isFirst) {
182228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (D)
1823035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    WL.push_back(DeclVisit(D, Parent, isFirst));
182428a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
182560608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenekvoid EnqueueVisitor::
182660608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  AddExplicitTemplateArgs(const ExplicitTemplateArgumentList *A) {
182760608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  if (A)
182860608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek    WL.push_back(ExplicitTemplateArgsVisit(
182960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek                        const_cast<ExplicitTemplateArgumentList*>(A), Parent));
183060608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek}
1831cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::AddMemberRef(FieldDecl *D, SourceLocation L) {
1832cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  if (D)
1833cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    WL.push_back(MemberRefVisit(D, L, Parent));
1834cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
183528a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::AddTypeLoc(TypeSourceInfo *TI) {
183628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (TI)
183728a719433411ef782b582946823bc648ddcc4533Ted Kremenek    WL.push_back(TypeLocVisit(TI->getTypeLoc(), Parent));
183828a719433411ef782b582946823bc648ddcc4533Ted Kremenek }
183928a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::EnqueueChildren(Stmt *S) {
1840a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  unsigned size = WL.size();
18417502c1d3ce8bb97bcc4f7bebef507040bd93b26fJohn McCall  for (Stmt::child_range Child = S->children(); Child; ++Child) {
184228a719433411ef782b582946823bc648ddcc4533Ted Kremenek    AddStmt(*Child);
1843a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  }
1844a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  if (size == WL.size())
1845a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek    return;
1846a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  // Now reverse the entries we just added.  This will match the DFS
1847a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  // ordering performed by the worklist.
1848a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
1849a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  std::reverse(I, E);
1850a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek}
1851ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekvoid EnqueueVisitor::VisitAddrLabelExpr(AddrLabelExpr *E) {
1852ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  WL.push_back(LabelRefVisit(E->getLabel(), E->getLabelLoc(), Parent));
1853ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek}
185473d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitBlockExpr(BlockExpr *B) {
185573d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddDecl(B->getBlockDecl());
185673d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
185728a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
185828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(E);
185928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(E->getTypeSourceInfo());
186028a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1861083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenekvoid EnqueueVisitor::VisitCompoundStmt(CompoundStmt *S) {
1862083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  for (CompoundStmt::reverse_body_iterator I = S->body_rbegin(),
1863083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek        E = S->body_rend(); I != E; ++I) {
1864083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek    AddStmt(*I);
1865083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  }
186611b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek}
1867f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::
1868f64d80306144f978148ba92f36f7cea7b671dd34Ted KremenekVisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E) {
1869f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
1870f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddDeclarationNameInfo(E);
18717c3179cf463c3b3b8c21dbb955f933ba50b74f28Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
18727c3179cf463c3b3b8c21dbb955f933ba50b74f28Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
1873f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  if (!E->isImplicitAccess())
1874f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    AddStmt(E->getBase());
1875f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
187611b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenekvoid EnqueueVisitor::VisitCXXNewExpr(CXXNewExpr *E) {
187711b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the initializer or constructor arguments.
187811b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  for (unsigned I = E->getNumConstructorArgs(); I > 0; --I)
187911b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek    AddStmt(E->getConstructorArg(I-1));
188011b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the array size, if any.
188111b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  AddStmt(E->getArraySize());
188211b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the allocated type.
188311b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  AddTypeLoc(E->getAllocatedTypeSourceInfo());
188411b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the placement arguments.
188511b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  for (unsigned I = E->getNumPlacementArgs(); I > 0; --I)
188611b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek    AddStmt(E->getPlacementArg(I-1));
188711b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek}
188828a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *CE) {
18898b8d8c90f2d8ac651d14b57f116d20b3c911ac7fTed Kremenek  for (unsigned I = CE->getNumArgs(); I > 1 /* Yes, this is 1 */; --I)
18908b8d8c90f2d8ac651d14b57f116d20b3c911ac7fTed Kremenek    AddStmt(CE->getArg(I-1));
189128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(CE->getCallee());
189228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(CE->getArg(0));
189328a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1894cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
1895cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the name of the type being destroyed.
1896cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getDestroyedTypeInfo());
1897cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the scope type that looks disturbingly like the nested-name-specifier
1898cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // but isn't.
1899cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getScopeTypeInfo());
1900cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the nested-name-specifier.
1901f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
1902f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
1903cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit base expression.
1904cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddStmt(E->getBase());
1905cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
19066d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenekvoid EnqueueVisitor::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
19076d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
19086d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek}
190973d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
191073d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  EnqueueChildren(E);
191173d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
191273d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
1913b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenekvoid EnqueueVisitor::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
1914b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  EnqueueChildren(E);
1915b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  if (E->isTypeOperand())
1916b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek    AddTypeLoc(E->getTypeOperandSourceInfo());
1917b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek}
191855b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek
191955b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenekvoid EnqueueVisitor::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr
192055b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek                                                     *E) {
192155b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  EnqueueChildren(E);
192255b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
192355b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek}
19241e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenekvoid EnqueueVisitor::VisitCXXUuidofExpr(CXXUuidofExpr *E) {
19251e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  EnqueueChildren(E);
19261e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  if (E->isTypeOperand())
19271e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek    AddTypeLoc(E->getTypeOperandSourceInfo());
19281e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek}
1929e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenekvoid EnqueueVisitor::VisitDeclRefExpr(DeclRefExpr *DR) {
193060608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  if (DR->hasExplicitTemplateArgs()) {
193160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek    AddExplicitTemplateArgs(&DR->getExplicitTemplateArgs());
193260608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  }
1933e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek  WL.push_back(DeclRefExprParts(DR, Parent));
1934e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek}
1935f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
1936f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
1937f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddDeclarationNameInfo(E);
193800cf3cc2718671aa48e8da264a523b0058a8591eDouglas Gregor  AddNestedNameSpecifierLoc(E->getQualifierLoc());
1939f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
1940035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekvoid EnqueueVisitor::VisitDeclStmt(DeclStmt *S) {
1941035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  unsigned size = WL.size();
1942035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  bool isFirst = true;
1943035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  for (DeclStmt::decl_iterator D = S->decl_begin(), DEnd = S->decl_end();
1944035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek       D != DEnd; ++D) {
1945035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    AddDecl(*D, isFirst);
1946035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    isFirst = false;
1947035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1948035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  if (size == WL.size())
1949035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    return;
1950035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  // Now reverse the entries we just added.  This will match the DFS
1951035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  // ordering performed by the worklist.
1952035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
1953035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  std::reverse(I, E);
1954035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek}
1955cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
1956cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddStmt(E->getInit());
1957cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  typedef DesignatedInitExpr::Designator Designator;
1958cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  for (DesignatedInitExpr::reverse_designators_iterator
1959cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek         D = E->designators_rbegin(), DEnd = E->designators_rend();
1960cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek         D != DEnd; ++D) {
1961cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    if (D->isFieldDesignator()) {
1962cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      if (FieldDecl *Field = D->getField())
1963cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        AddMemberRef(Field, D->getFieldLoc());
1964cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
1965cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
1966cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    if (D->isArrayDesignator()) {
1967cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      AddStmt(E->getArrayIndex(*D));
1968cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
1969cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
1970cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    assert(D->isArrayRangeDesignator() && "Unknown designator kind");
1971cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    AddStmt(E->getArrayRangeEnd(*D));
1972cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    AddStmt(E->getArrayRangeStart(*D));
1973cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1974cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
197528a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitExplicitCastExpr(ExplicitCastExpr *E) {
197628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(E);
197728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(E->getTypeInfoAsWritten());
197828a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
197928a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitForStmt(ForStmt *FS) {
198028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getBody());
198128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getInc());
198228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getCond());
198328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(FS->getConditionVariable());
198428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getInit());
198528a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1986ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekvoid EnqueueVisitor::VisitGotoStmt(GotoStmt *GS) {
1987ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  WL.push_back(LabelRefVisit(GS->getLabel(), GS->getLabelLoc(), Parent));
1988ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek}
198928a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitIfStmt(IfStmt *If) {
199028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getElse());
199128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getThen());
199228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getCond());
199328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(If->getConditionVariable());
199428a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
199528a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitInitListExpr(InitListExpr *IE) {
199628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  // We care about the syntactic form of the initializer list, only.
199728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (InitListExpr *Syntactic = IE->getSyntacticForm())
199828a719433411ef782b582946823bc648ddcc4533Ted Kremenek    IE = Syntactic;
199928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(IE);
200028a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
200128a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitMemberExpr(MemberExpr *M) {
200289629a746019a42797495b091711a1d68467e88aDouglas Gregor  WL.push_back(MemberExprParts(M, Parent));
200389629a746019a42797495b091711a1d68467e88aDouglas Gregor
200489629a746019a42797495b091711a1d68467e88aDouglas Gregor  // If the base of the member access expression is an implicit 'this', don't
200589629a746019a42797495b091711a1d68467e88aDouglas Gregor  // visit it.
200689629a746019a42797495b091711a1d68467e88aDouglas Gregor  // FIXME: If we ever want to show these implicit accesses, this will be
200789629a746019a42797495b091711a1d68467e88aDouglas Gregor  // unfortunate. However, clang_getCursor() relies on this behavior.
200875e85048e73fcde2ce9d8a48dfdb1220e132eb59Douglas Gregor  if (!M->isImplicitAccess())
200975e85048e73fcde2ce9d8a48dfdb1220e132eb59Douglas Gregor    AddStmt(M->getBase());
201028a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
201173d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
201273d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddTypeLoc(E->getEncodedTypeSourceInfo());
201373d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
201428a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitObjCMessageExpr(ObjCMessageExpr *M) {
201528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(M);
201628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(M->getClassReceiverTypeInfo());
201728a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
2018cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitOffsetOfExpr(OffsetOfExpr *E) {
2019cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the components of the offsetof expression.
2020cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  for (unsigned N = E->getNumComponents(), I = N; I > 0; --I) {
2021cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    typedef OffsetOfExpr::OffsetOfNode OffsetOfNode;
2022cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    const OffsetOfNode &Node = E->getComponent(I-1);
2023cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    switch (Node.getKind()) {
2024cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Array:
2025cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      AddStmt(E->getIndexExpr(Node.getArrayExprIndex()));
2026cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      break;
2027cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Field:
202806dec892b5300b43263d25c5476b506c9d6cfbadAbramo Bagnara      AddMemberRef(Node.getField(), Node.getSourceRange().getEnd());
2029cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      break;
2030cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Identifier:
2031cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Base:
2032cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
2033cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
2034cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
2035cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the type into which we're computing the offset.
2036cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
2037cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
203828a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitOverloadExpr(OverloadExpr *E) {
203960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
20406045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek  WL.push_back(OverloadExprParts(E, Parent));
20416045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek}
2042f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbournevoid EnqueueVisitor::VisitUnaryExprOrTypeTraitExpr(
2043f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbourne                                              UnaryExprOrTypeTraitExpr *E) {
20446d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  EnqueueChildren(E);
20456d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  if (E->isArgumentType())
20466d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek    AddTypeLoc(E->getArgumentTypeInfo());
20476d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek}
204828a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitStmt(Stmt *S) {
204928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(S);
205028a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
205128a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitSwitchStmt(SwitchStmt *S) {
205228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(S->getBody());
205328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(S->getCond());
205428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(S->getConditionVariable());
205528a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
2056fafa75aebadef8d6b44a920e3f40529f150a5574Ted Kremenek
205728a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitWhileStmt(WhileStmt *W) {
205828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(W->getBody());
205928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(W->getCond());
206028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(W->getConditionVariable());
206128a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
20622939b6f356161f572712d4d6310b65f9599e3675Ted Kremenekvoid EnqueueVisitor::VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E) {
20632939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek  AddTypeLoc(E->getQueriedTypeSourceInfo());
20642939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek}
20656ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet
20666ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichetvoid EnqueueVisitor::VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E) {
20676ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  AddTypeLoc(E->getRhsTypeSourceInfo());
20680a03a3f98b14006a54bcac9e8908a7c9f50e519fFrancois Pichet  AddTypeLoc(E->getLhsTypeSourceInfo());
20696ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet}
20706ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet
207128a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *U) {
207228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  VisitOverloadExpr(U);
207328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (!U->isImplicitAccess())
207428a719433411ef782b582946823bc648ddcc4533Ted Kremenek    AddStmt(U->getBase());
207528a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
20769d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenekvoid EnqueueVisitor::VisitVAArgExpr(VAArgExpr *E) {
20779d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  AddStmt(E->getSubExpr());
20789d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  AddTypeLoc(E->getWrittenTypeInfo());
20799d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek}
208094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregorvoid EnqueueVisitor::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
208194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  WL.push_back(SizeOfPackExprParts(E, Parent));
208294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor}
20836045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek
2084c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekvoid CursorVisitor::EnqueueWorkList(VisitorWorkList &WL, Stmt *S) {
208528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueVisitor(WL, MakeCXCursor(S, StmtParent, TU)).Visit(S);
2086c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2087c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2088c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekbool CursorVisitor::IsInRegionOfInterest(CXCursor C) {
2089c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  if (RegionOfInterest.isValid()) {
2090c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    SourceRange Range = getRawCursorExtent(C);
2091c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    if (Range.isInvalid() || CompareRegionOfInterest(Range))
2092c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      return false;
2093c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  }
2094c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  return true;
2095c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2096c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2097c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekbool CursorVisitor::RunVisitorWorkList(VisitorWorkList &WL) {
2098c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  while (!WL.empty()) {
2099c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    // Dequeue the worklist item.
210082f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    VisitorJob LI = WL.back();
210182f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    WL.pop_back();
210282f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek
2103c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    // Set the Parent field, then back to its old value once we're done.
2104c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    SetParentRAII SetParent(Parent, StmtParent, LI.getParent());
2105c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2106c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    switch (LI.getKind()) {
2107f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek      case VisitorJob::DeclVisitKind: {
210882f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        Decl *D = cast<DeclVisit>(&LI)->get();
2109f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        if (!D)
2110f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek          continue;
2111f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek
2112f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        // For now, perform default visitation for Decls.
211382f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        if (Visit(MakeCXCursor(D, TU, cast<DeclVisit>(&LI)->isFirst())))
2114f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek            return true;
2115f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek
2116f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        continue;
2117f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek      }
211860608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek      case VisitorJob::ExplicitTemplateArgsVisitKind: {
211960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        const ExplicitTemplateArgumentList *ArgList =
212060608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek          cast<ExplicitTemplateArgsVisit>(&LI)->get();
212160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        for (const TemplateArgumentLoc *Arg = ArgList->getTemplateArgs(),
212260608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek               *ArgEnd = Arg + ArgList->NumTemplateArgs;
212360608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek               Arg != ArgEnd; ++Arg) {
212460608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek          if (VisitTemplateArgumentLoc(*Arg))
212560608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek            return true;
212660608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        }
212760608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        continue;
212860608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek      }
2129cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek      case VisitorJob::TypeLocVisitKind: {
2130cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek        // Perform default visitation for TypeLocs.
213182f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        if (Visit(cast<TypeLocVisit>(&LI)->get()))
2132cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek          return true;
2133cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek        continue;
2134cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek      }
2135ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek      case VisitorJob::LabelRefVisitKind: {
2136ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner        LabelDecl *LS = cast<LabelRefVisit>(&LI)->get();
2137e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek        if (LabelStmt *stmt = LS->getStmt()) {
2138e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek          if (Visit(MakeCursorLabelRef(stmt, cast<LabelRefVisit>(&LI)->getLoc(),
2139e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek                                       TU))) {
2140e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek            return true;
2141e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek          }
2142e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek        }
2143ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek        continue;
2144ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek      }
2145f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
2146f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      case VisitorJob::NestedNameSpecifierVisitKind: {
2147f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        NestedNameSpecifierVisit *V = cast<NestedNameSpecifierVisit>(&LI);
2148f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        if (VisitNestedNameSpecifier(V->get(), V->getSourceRange()))
2149f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek          return true;
2150f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        continue;
2151f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      }
2152f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
2153f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor      case VisitorJob::NestedNameSpecifierLocVisitKind: {
2154f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        NestedNameSpecifierLocVisit *V = cast<NestedNameSpecifierLocVisit>(&LI);
2155f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        if (VisitNestedNameSpecifierLoc(V->get()))
2156f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor          return true;
2157f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        continue;
2158f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor      }
2159f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
2160f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      case VisitorJob::DeclarationNameInfoVisitKind: {
2161f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        if (VisitDeclarationNameInfo(cast<DeclarationNameInfoVisit>(&LI)
2162f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek                                     ->get()))
2163f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek          return true;
2164f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        continue;
2165f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      }
2166cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      case VisitorJob::MemberRefVisitKind: {
2167cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        MemberRefVisit *V = cast<MemberRefVisit>(&LI);
2168cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        if (Visit(MakeCursorMemberRef(V->get(), V->getLoc(), TU)))
2169cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          return true;
2170cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        continue;
2171cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      }
2172c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      case VisitorJob::StmtVisitKind: {
217382f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        Stmt *S = cast<StmtVisit>(&LI)->get();
21748c269ac75569454a049385b1246140db5f2b6faaTed Kremenek        if (!S)
21758c269ac75569454a049385b1246140db5f2b6faaTed Kremenek          continue;
21768c269ac75569454a049385b1246140db5f2b6faaTed Kremenek
2177f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        // Update the current cursor.
2178c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        CXCursor Cursor = MakeCXCursor(S, StmtParent, TU);
2179cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        if (!IsInRegionOfInterest(Cursor))
2180cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          continue;
2181cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        switch (Visitor(Cursor, Parent, ClientData)) {
2182cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Break: return true;
2183cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Continue: break;
2184cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Recurse:
2185cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek            EnqueueWorkList(WL, S);
218682f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek            break;
2187c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        }
218882f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        continue;
2189c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      }
2190c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      case VisitorJob::MemberExprPartsKind: {
2191c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Handle the other pieces in the MemberExpr besides the base.
219282f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        MemberExpr *M = cast<MemberExprParts>(&LI)->get();
2193c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2194c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the nested-name-specifier
219540d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = M->getQualifierLoc())
219640d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
2197c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek            return true;
2198c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2199c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the declaration name.
2200c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        if (VisitDeclarationNameInfo(M->getMemberNameInfo()))
2201c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          return true;
2202c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2203c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the explicitly-specified template arguments, if any.
2204c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        if (M->hasExplicitTemplateArgs()) {
2205c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          for (const TemplateArgumentLoc *Arg = M->getTemplateArgs(),
2206c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek               *ArgEnd = Arg + M->getNumTemplateArgs();
2207c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek               Arg != ArgEnd; ++Arg) {
2208c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek            if (VisitTemplateArgumentLoc(*Arg))
2209c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek              return true;
2210c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          }
2211c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        }
2212c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        continue;
2213c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      }
2214e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek      case VisitorJob::DeclRefExprPartsKind: {
221582f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        DeclRefExpr *DR = cast<DeclRefExprParts>(&LI)->get();
2216e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        // Visit nested-name-specifier, if present.
221740d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = DR->getQualifierLoc())
221840d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
2219e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek            return true;
2220e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        // Visit declaration name.
2221e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        if (VisitDeclarationNameInfo(DR->getNameInfo()))
2222e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek          return true;
2223e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        continue;
2224e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek      }
22256045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek      case VisitorJob::OverloadExprPartsKind: {
222682f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        OverloadExpr *O = cast<OverloadExprParts>(&LI)->get();
22276045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the nested-name-specifier.
22284c9be89bb615ec07eb3ed507c8fa9d0baa8a5ad7Douglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = O->getQualifierLoc())
22294c9be89bb615ec07eb3ed507c8fa9d0baa8a5ad7Douglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
22306045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek            return true;
22316045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the declaration name.
22326045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        if (VisitDeclarationNameInfo(O->getNameInfo()))
22336045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek          return true;
22346045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the overloaded declaration reference.
22356045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        if (Visit(MakeCursorOverloadedDeclRef(O, TU)))
22366045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek          return true;
22376045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        continue;
22386045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek      }
223994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      case VisitorJob::SizeOfPackExprPartsKind: {
224094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        SizeOfPackExpr *E = cast<SizeOfPackExprParts>(&LI)->get();
224194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        NamedDecl *Pack = E->getPack();
224294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        if (isa<TemplateTypeParmDecl>(Pack)) {
224394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          if (Visit(MakeCursorTypeRef(cast<TemplateTypeParmDecl>(Pack),
224494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor                                      E->getPackLoc(), TU)))
224594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor            return true;
224694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
224794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          continue;
224894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        }
224994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
225094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        if (isa<TemplateTemplateParmDecl>(Pack)) {
225194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          if (Visit(MakeCursorTemplateRef(cast<TemplateTemplateParmDecl>(Pack),
225294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor                                          E->getPackLoc(), TU)))
225394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor            return true;
225494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
225594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          continue;
225694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        }
225794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
225894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        // Non-type template parameter packs and function parameter packs are
225994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        // treated like DeclRefExpr cursors.
226094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        continue;
226194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      }
2262c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    }
2263c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  }
2264c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  return false;
2265c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2266c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2267cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekbool CursorVisitor::Visit(Stmt *S) {
2268d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  VisitorWorkList *WL = 0;
2269d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  if (!WorkListFreeList.empty()) {
2270d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL = WorkListFreeList.back();
2271d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL->clear();
2272d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WorkListFreeList.pop_back();
2273d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
2274d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  else {
2275d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL = new VisitorWorkList();
2276d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WorkListCache.push_back(WL);
2277d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
2278d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  EnqueueWorkList(*WL, S);
2279d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  bool result = RunVisitorWorkList(*WL);
2280d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  WorkListFreeList.push_back(WL);
2281d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  return result;
2282c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2283c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2284c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
2285c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek// Misc. API hooks.
2286c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
2287c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
22888c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregorstatic llvm::sys::Mutex EnableMultithreadingMutex;
22898c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregorstatic bool EnabledMultithreading;
22908c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor
22915e4bc590b0ea010e38372d0b4a0aab578a746fe6Benjamin Kramerextern "C" {
22920a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas GregorCXIndex clang_createIndex(int excludeDeclarationsFromPCH,
22930a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor                          int displayDiagnostics) {
229448615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  // Disable pretty stack trace functionality, which will otherwise be a very
229548615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  // poor citizen of the world and set up all sorts of signal handlers.
229648615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  llvm::DisablePrettyStackTrace = true;
229748615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar
2298c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  // We use crash recovery to make some of our APIs more reliable, implicitly
2299c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  // enable it.
2300c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  llvm::CrashRecoveryContext::Enable();
2301c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar
23028c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  // Enable support for multithreading in LLVM.
23038c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  {
23048c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    llvm::sys::ScopedLock L(EnableMultithreadingMutex);
23058c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    if (!EnabledMultithreading) {
23068c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor      llvm::llvm_start_multithreaded();
23078c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor      EnabledMultithreading = true;
23088c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    }
23098c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  }
23108c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor
2311a030b7cf5e6aad5889b1b662b6979840bc75f87fDouglas Gregor  CIndexer *CIdxr = new CIndexer();
2312e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  if (excludeDeclarationsFromPCH)
2313e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff    CIdxr->setOnlyLocalDecls();
23140a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  if (displayDiagnostics)
23150a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    CIdxr->setDisplayDiagnostics();
2316e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  return CIdxr;
2317600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
2318600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
23199ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarvoid clang_disposeIndex(CXIndex CIdx) {
23202b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (CIdx)
23212b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    delete static_cast<CIndexer *>(CIdx);
23222bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff}
23232bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff
2324d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenekvoid clang_toggleCrashRecovery(unsigned isEnabled) {
2325d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek  if (isEnabled)
2326d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek    llvm::CrashRecoveryContext::Enable();
2327d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek  else
2328d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek    llvm::CrashRecoveryContext::Disable();
2329d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek}
2330d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek
23319ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXTranslationUnit clang_createTranslationUnit(CXIndex CIdx,
2332a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                              const char *ast_filename) {
23332b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CIdx)
23342b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    return 0;
2335f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
23367d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
2337389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  FileSystemOptions FileSystemOpts;
2338389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  FileSystemOpts.WorkingDir = CXXIdx->getWorkingDirectory();
23390d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
234028019772db70d4547be05a042eb950bc910f134fDouglas Gregor  llvm::IntrusiveRefCntPtr<Diagnostic> Diags;
2341a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *TU = ASTUnit::LoadFromASTFile(ast_filename, Diags, FileSystemOpts,
2342a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                  CXXIdx->getOnlyLocalDecls(),
2343a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                  0, 0, true);
2344a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return MakeCXTranslationUnit(TU);
2345600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
2346600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
2347b1c031be513705d924038f497279b9b599868ba1Douglas Gregorunsigned clang_defaultEditingTranslationUnitOptions() {
23482a2c50b330e7754499f42173616a36865b5f313bDouglas Gregor  return CXTranslationUnit_PrecompiledPreamble |
234999ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor         CXTranslationUnit_CacheCompletionResults |
235099ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor         CXTranslationUnit_CXXPrecompiledPreamble;
2351b1c031be513705d924038f497279b9b599868ba1Douglas Gregor}
2352b1c031be513705d924038f497279b9b599868ba1Douglas Gregor
23539ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXTranslationUnit
23549ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarclang_createTranslationUnitFromSourceFile(CXIndex CIdx,
23559ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          const char *source_filename,
23569ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          int num_command_line_args,
23572ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor                                          const char * const *command_line_args,
23584db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor                                          unsigned num_unsaved_files,
2359a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                          struct CXUnsavedFile *unsaved_files) {
23605a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor  return clang_parseTranslationUnit(CIdx, source_filename,
23615a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                    command_line_args, num_command_line_args,
23625a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                    unsaved_files, num_unsaved_files,
23635a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                 CXTranslationUnit_DetailedPreprocessingRecord);
23645a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor}
236519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
236619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbarstruct ParseTranslationUnitInfo {
236719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXIndex CIdx;
236819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  const char *source_filename;
23692ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor  const char *const *command_line_args;
237019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  int num_command_line_args;
237119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  struct CXUnsavedFile *unsaved_files;
237219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned num_unsaved_files;
237319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned options;
237419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXTranslationUnit result;
237519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar};
2376b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbarstatic void clang_parseTranslationUnit_Impl(void *UserData) {
237719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  ParseTranslationUnitInfo *PTUI =
237819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    static_cast<ParseTranslationUnitInfo*>(UserData);
237919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXIndex CIdx = PTUI->CIdx;
238019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  const char *source_filename = PTUI->source_filename;
23812ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor  const char * const *command_line_args = PTUI->command_line_args;
238219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  int num_command_line_args = PTUI->num_command_line_args;
238319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  struct CXUnsavedFile *unsaved_files = PTUI->unsaved_files;
238419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned num_unsaved_files = PTUI->num_unsaved_files;
238519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned options = PTUI->options;
238619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  PTUI->result = 0;
23875a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor
23882b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CIdx)
238919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    return;
2390f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2391e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
2392e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff
239344c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  bool PrecompilePreamble = options & CXTranslationUnit_PrecompiledPreamble;
2394df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor  bool CompleteTranslationUnit
2395df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor    = ((options & CXTranslationUnit_Incomplete) == 0);
239687c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  bool CacheCodeCompetionResults
239787c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    = options & CXTranslationUnit_CacheCompletionResults;
239899ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor  bool CXXPrecompilePreamble
239999ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor    = options & CXTranslationUnit_CXXPrecompiledPreamble;
240099ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor  bool CXXChainedPCH
240199ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor    = options & CXTranslationUnit_CXXChainedPCH;
240287c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor
24035352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  // Configure the diagnostics.
24045352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  DiagnosticOptions DiagOpts;
240525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::IntrusiveRefCntPtr<Diagnostic>
240625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Diags(CompilerInstance::createDiagnostics(DiagOpts, num_command_line_args,
240725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                                command_line_args));
240825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
240925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
241025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<Diagnostic,
241125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    llvm::CrashRecoveryContextReleaseRefCleanup<Diagnostic> >
241225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    DiagCleanup(Diags.getPtr());
241325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
241425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::OwningPtr<std::vector<ASTUnit::RemappedFile> >
241525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles(new std::vector<ASTUnit::RemappedFile>());
241625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
241725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
241825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<
241925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
2420f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
24214db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  for (unsigned I = 0; I != num_unsaved_files; ++I) {
2422a0a270c0f1c0a4e3482438bdc5f4a7bd3d25f0a6Chris Lattner    llvm::StringRef Data(unsaved_files[I].Contents, unsaved_files[I].Length);
2423f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    const llvm::MemoryBuffer *Buffer
2424a0a270c0f1c0a4e3482438bdc5f4a7bd3d25f0a6Chris Lattner      = llvm::MemoryBuffer::getMemBufferCopy(Data, unsaved_files[I].Filename);
242525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles->push_back(std::make_pair(unsaved_files[I].Filename,
242625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                            Buffer));
24274db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  }
2428f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
242925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::OwningPtr<std::vector<const char *> >
243025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args(new std::vector<const char*>());
243125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
243225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this method.
243325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<std::vector<const char*> >
243425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    ArgsCleanup(Args.get());
243525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
243652ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // Since the Clang C library is primarily used by batch tools dealing with
243752ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // (often very broken) source code, where spell-checking can have a
243852ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // significant negative impact on performance (particularly when
243952ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // precompiled headers are involved), we disable it by default.
2440b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  // Only do this if we haven't found a spell-checking-related argument.
2441b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  bool FoundSpellCheckingArgument = false;
2442b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  for (int I = 0; I != num_command_line_args; ++I) {
2443b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    if (strcmp(command_line_args[I], "-fno-spell-checking") == 0 ||
2444b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        strcmp(command_line_args[I], "-fspell-checking") == 0) {
2445b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      FoundSpellCheckingArgument = true;
2446b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      break;
2447e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff    }
2448b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  }
2449b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  if (!FoundSpellCheckingArgument)
245025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-fno-spell-checking");
2451b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor
245225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  Args->insert(Args->end(), command_line_args,
245325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek               command_line_args + num_command_line_args);
2454d93256e55673a17d18543397ec462416acb13792Douglas Gregor
2455c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // The 'source_filename' argument is optional.  If the caller does not
2456c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // specify it then it is assumed that the source file is specified
2457c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // in the actual argument list.
2458c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // Put the source file after command_line_args otherwise if '-x' flag is
2459c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // present it will be unused.
2460c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  if (source_filename)
246125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back(source_filename);
2462c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis
246344c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  // Do we need the detailed preprocessing record?
246444c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  if (options & CXTranslationUnit_DetailedPreprocessingRecord) {
246525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-Xclang");
246625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-detailed-preprocessing-record");
246744c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  }
246844c181aec37789f25f6c15543c164416f72e562aDouglas Gregor
2469026f6911bb985c800a54446de9f6da8745ae025aArgyrios Kyrtzidis  unsigned NumErrors = Diags->getClient()->getNumErrors();
2470b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  llvm::OwningPtr<ASTUnit> Unit(
24714ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek    ASTUnit::LoadFromCommandLine(Args->size() ? &(*Args)[0] : 0
24724ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 /* vector::data() not portable */,
24734ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 Args->size() ? (&(*Args)[0] + Args->size()) :0,
2474b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 Diags,
2475b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CXXIdx->getClangResourcesPath(),
2476b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CXXIdx->getOnlyLocalDecls(),
2477e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor                                 /*CaptureDiagnostics=*/true,
24784ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 RemappedFiles->size() ? &(*RemappedFiles)[0]:0,
247925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                 RemappedFiles->size(),
2480299a4a967b02c9f0d0d94ad8560e3ced893f9116Argyrios Kyrtzidis                                 /*RemappedFilesKeepOriginalName=*/true,
2481b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 PrecompilePreamble,
2482b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CompleteTranslationUnit,
248399ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor                                 CacheCodeCompetionResults,
248499ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor                                 CXXPrecompilePreamble,
248599ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor                                 CXXChainedPCH));
2486b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor
2487026f6911bb985c800a54446de9f6da8745ae025aArgyrios Kyrtzidis  if (NumErrors != Diags->getClient()->getNumErrors()) {
2488b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    // Make sure to check that 'Unit' is non-NULL.
2489b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    if (CXXIdx->getDisplayDiagnostics() && Unit.get()) {
2490b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      for (ASTUnit::stored_diag_iterator D = Unit->stored_diag_begin(),
2491b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                      DEnd = Unit->stored_diag_end();
2492b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor           D != DEnd; ++D) {
2493b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        CXStoredDiagnostic Diag(*D, Unit->getASTContext().getLangOptions());
2494b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        CXString Msg = clang_formatDiagnostic(&Diag,
2495b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                    clang_defaultDiagnosticDisplayOptions());
2496b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        fprintf(stderr, "%s\n", clang_getCString(Msg));
2497b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        clang_disposeString(Msg);
2498b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      }
2499274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor#ifdef LLVM_ON_WIN32
2500b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      // On Windows, force a flush, since there may be multiple copies of
2501b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      // stderr and stdout in the file system, all with different buffers
2502b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      // but writing to the same device.
2503b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      fflush(stderr);
2504b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor#endif
2505b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    }
2506a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  }
2507d93256e55673a17d18543397ec462416acb13792Douglas Gregor
2508a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  PTUI->result = MakeCXTranslationUnit(Unit.take());
250919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar}
251019ffd492a31a25fb691098bf79f317e5f3edf177Daniel DunbarCXTranslationUnit clang_parseTranslationUnit(CXIndex CIdx,
251119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             const char *source_filename,
25122ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor                                         const char * const *command_line_args,
251319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             int num_command_line_args,
25149e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                            struct CXUnsavedFile *unsaved_files,
251519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             unsigned num_unsaved_files,
251619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             unsigned options) {
251719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  ParseTranslationUnitInfo PTUI = { CIdx, source_filename, command_line_args,
25189e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                    num_command_line_args, unsaved_files,
25199e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                    num_unsaved_files, options, 0 };
252019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  llvm::CrashRecoveryContext CRC;
252119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
2522bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  if (!RunSafely(CRC, clang_parseTranslationUnit_Impl, &PTUI)) {
252360a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "libclang: crash detected during parsing: {\n");
252460a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'source_filename' : '%s'\n", source_filename);
252560a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'command_line_args' : [");
252660a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    for (int i = 0; i != num_command_line_args; ++i) {
252760a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      if (i)
252860a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar        fprintf(stderr, ", ");
252960a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      fprintf(stderr, "'%s'", command_line_args[i]);
253060a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    }
253160a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "],\n");
253260a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'unsaved_files' : [");
253360a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    for (unsigned i = 0; i != num_unsaved_files; ++i) {
253460a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      if (i)
253560a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar        fprintf(stderr, ", ");
253660a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      fprintf(stderr, "('%s', '...', %ld)", unsaved_files[i].Filename,
253760a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar              unsaved_files[i].Length);
253860a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    }
253960a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "],\n");
254060a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'options' : %d,\n", options);
254160a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "}\n");
254260a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar
254319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    return 0;
254419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  }
254519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
254619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  return PTUI.result;
25475b7d8e254f6c2855b37b5521c0aee0a560dab237Steve Naroff}
25485b7d8e254f6c2855b37b5521c0aee0a560dab237Steve Naroff
25491999844e7a18786e61e619e1dc6c789827541863Douglas Gregorunsigned clang_defaultSaveOptions(CXTranslationUnit TU) {
25501999844e7a18786e61e619e1dc6c789827541863Douglas Gregor  return CXSaveTranslationUnit_None;
25511999844e7a18786e61e619e1dc6c789827541863Douglas Gregor}
25521999844e7a18786e61e619e1dc6c789827541863Douglas Gregor
25531999844e7a18786e61e619e1dc6c789827541863Douglas Gregorint clang_saveTranslationUnit(CXTranslationUnit TU, const char *FileName,
25541999844e7a18786e61e619e1dc6c789827541863Douglas Gregor                              unsigned options) {
25557ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor  if (!TU)
25567ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor    return 1;
25577ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor
2558a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return static_cast<ASTUnit *>(TU->TUData)->Save(FileName);
25597ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor}
256019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
25619ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarvoid clang_disposeTranslationUnit(CXTranslationUnit CTUnit) {
2562ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  if (CTUnit) {
2563ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    // If the translation unit has been marked as unsafe to free, just discard
2564ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    // it.
2565a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    if (static_cast<ASTUnit *>(CTUnit->TUData)->isUnsafeToFree())
2566ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar      return;
2567ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2568a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    delete static_cast<ASTUnit *>(CTUnit->TUData);
2569a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    disposeCXStringPool(CTUnit->StringPool);
2570a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    delete CTUnit;
2571ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  }
25722bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff}
25730d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
2574e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregorunsigned clang_defaultReparseOptions(CXTranslationUnit TU) {
2575e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor  return CXReparse_None;
2576e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor}
2577e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor
2578ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbarstruct ReparseTranslationUnitInfo {
2579ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  CXTranslationUnit TU;
2580ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned num_unsaved_files;
2581ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  struct CXUnsavedFile *unsaved_files;
2582ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned options;
2583ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  int result;
2584ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar};
2585593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2586b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbarstatic void clang_reparseTranslationUnit_Impl(void *UserData) {
2587ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  ReparseTranslationUnitInfo *RTUI =
2588ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    static_cast<ReparseTranslationUnitInfo*>(UserData);
2589ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  CXTranslationUnit TU = RTUI->TU;
2590ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned num_unsaved_files = RTUI->num_unsaved_files;
2591ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  struct CXUnsavedFile *unsaved_files = RTUI->unsaved_files;
2592ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned options = RTUI->options;
2593ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  (void) options;
2594ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  RTUI->result = 1;
2595ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2596abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  if (!TU)
2597ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    return;
2598593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2599a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
2600593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
2601abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor
260225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::OwningPtr<std::vector<ASTUnit::RemappedFile> >
260325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles(new std::vector<ASTUnit::RemappedFile>());
260425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
260525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
260625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<
260725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
260825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
2609abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  for (unsigned I = 0; I != num_unsaved_files; ++I) {
2610abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor    llvm::StringRef Data(unsaved_files[I].Contents, unsaved_files[I].Length);
2611abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor    const llvm::MemoryBuffer *Buffer
26121abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor      = llvm::MemoryBuffer::getMemBufferCopy(Data, unsaved_files[I].Filename);
261325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles->push_back(std::make_pair(unsaved_files[I].Filename,
261425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                            Buffer));
2615abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  }
2616abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor
26174ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek  if (!CXXUnit->Reparse(RemappedFiles->size() ? &(*RemappedFiles)[0] : 0,
26184ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                        RemappedFiles->size()))
2619593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor    RTUI->result = 0;
2620abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor}
2621593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2622ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbarint clang_reparseTranslationUnit(CXTranslationUnit TU,
2623ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 unsigned num_unsaved_files,
2624ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 struct CXUnsavedFile *unsaved_files,
2625ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 unsigned options) {
2626ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  ReparseTranslationUnitInfo RTUI = { TU, num_unsaved_files, unsaved_files,
2627ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                      options, 0 };
2628ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  llvm::CrashRecoveryContext CRC;
2629ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2630bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  if (!RunSafely(CRC, clang_reparseTranslationUnit_Impl, &RTUI)) {
2631b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbar    fprintf(stderr, "libclang: crash detected during reparsing\n");
2632a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    static_cast<ASTUnit *>(TU->TUData)->setUnsafeToFree(true);
2633ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    return 1;
2634ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  }
2635ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
26361dfb26af4d6aa4f7818e256659a79f1ec2cba784Ted Kremenek
2637ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  return RTUI.result;
2638ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar}
2639ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2640df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor
26419ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXString clang_getTranslationUnitSpelling(CXTranslationUnit CTUnit) {
26422b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CTUnit)
2643ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
2644f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2645a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit->TUData);
2646ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString(CXXUnit->getOriginalSourceFileName(), true);
2647af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff}
26481eb79b58e56b99cf557d5d353586a10c5360364dDaniel Dunbar
26497eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas GregorCXCursor clang_getTranslationUnitCursor(CXTranslationUnit TU) {
2650b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  CXCursor Result = { CXCursor_TranslationUnit, { 0, 0, TU } };
26517eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  return Result;
26527eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor}
26537eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
2654fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
2655600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
2656fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
26571db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor// CXSourceLocation and CXSourceRange Operations.
26581db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor//===----------------------------------------------------------------------===//
26591db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
2660b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregorextern "C" {
2661b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXSourceLocation clang_getNullLocation() {
26625352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  CXSourceLocation Result = { { 0, 0 }, 0 };
2663b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  return Result;
2664b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2665b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
2666b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregorunsigned clang_equalLocations(CXSourceLocation loc1, CXSourceLocation loc2) {
266790a6b9e1e4d4d1995ada044e319d6e722b07a6b4Daniel Dunbar  return (loc1.ptr_data[0] == loc2.ptr_data[0] &&
266890a6b9e1e4d4d1995ada044e319d6e722b07a6b4Daniel Dunbar          loc1.ptr_data[1] == loc2.ptr_data[1] &&
266990a6b9e1e4d4d1995ada044e319d6e722b07a6b4Daniel Dunbar          loc1.int_data == loc2.int_data);
2670b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2671b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
2672b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXSourceLocation clang_getLocation(CXTranslationUnit tu,
2673b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor                                   CXFile file,
2674b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor                                   unsigned line,
2675b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor                                   unsigned column) {
267642748ec5cb2d75fe0dbb3a6db5aee6c11b5dc190Douglas Gregor  if (!tu || !file)
2677b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor    return clang_getNullLocation();
267842748ec5cb2d75fe0dbb3a6db5aee6c11b5dc190Douglas Gregor
267986a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  bool Logging = ::getenv("LIBCLANG_LOGGING");
2680a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
268186a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  const FileEntry *File = static_cast<const FileEntry *>(file);
2682b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  SourceLocation SLoc
268386a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor    = CXXUnit->getSourceManager().getLocation(File, line, column);
268486a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  if (SLoc.isInvalid()) {
268586a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor    if (Logging)
268686a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor      llvm::errs() << "clang_getLocation(\"" << File->getName()
268786a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor                   << "\", " << line << ", " << column << ") = invalid\n";
268886a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor    return clang_getNullLocation();
268986a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  }
269086a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor
269186a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  if (Logging)
269286a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor    llvm::errs() << "clang_getLocation(\"" << File->getName()
269386a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor                 << "\", " << line << ", " << column << ") = "
269486a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor                 << SLoc.getRawEncoding() << "\n";
269583889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall
269683889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  return cxloc::translateSourceLocation(CXXUnit->getASTContext(), SLoc);
269783889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall}
269883889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall
269983889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid ChisnallCXSourceLocation clang_getLocationForOffset(CXTranslationUnit tu,
270083889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall                                            CXFile file,
270183889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall                                            unsigned offset) {
270283889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  if (!tu || !file)
270383889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall    return clang_getNullLocation();
270483889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall
2705a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
270683889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  SourceLocation Start
270783889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall    = CXXUnit->getSourceManager().getLocation(
270883889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall                                        static_cast<const FileEntry *>(file),
270983889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall                                              1, 1);
271083889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  if (Start.isInvalid()) return clang_getNullLocation();
271183889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall
271283889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  SourceLocation SLoc = Start.getFileLocWithOffset(offset);
271383889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall
271483889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  if (SLoc.isInvalid()) return clang_getNullLocation();
2715f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
27161a9a0bc472ee4fec72ee8be8b575fb66ca600d1bTed Kremenek  return cxloc::translateSourceLocation(CXXUnit->getASTContext(), SLoc);
2717b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2718b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
27195352ac06d8f6194825bb2a99ffa009b61bafb503Douglas GregorCXSourceRange clang_getNullRange() {
27205352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  CXSourceRange Result = { { 0, 0 }, 0, 0 };
27215352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  return Result;
27225352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor}
2723d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar
2724b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXSourceRange clang_getRange(CXSourceLocation begin, CXSourceLocation end) {
27255352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  if (begin.ptr_data[0] != end.ptr_data[0] ||
27265352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor      begin.ptr_data[1] != end.ptr_data[1])
27275352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
2728f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2729f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  CXSourceRange Result = { { begin.ptr_data[0], begin.ptr_data[1] },
27305352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor                           begin.int_data, end.int_data };
2731b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  return Result;
2732b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
27339d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek} // end: extern "C"
2734b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
27359d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenekstatic void createNullLocation(CXFile *file, unsigned *line,
27369d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek                               unsigned *column, unsigned *offset) {
27379d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  if (file)
27389d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek   *file = 0;
27399d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  if (line)
27409d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek   *line = 0;
27419d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  if (column)
27429d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek   *column = 0;
27439d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  if (offset)
27449d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek   *offset = 0;
27459d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  return;
27469d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek}
27479d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek
27489d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenekextern "C" {
274946766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregorvoid clang_getInstantiationLocation(CXSourceLocation location,
275046766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor                                    CXFile *file,
275146766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor                                    unsigned *line,
275246766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor                                    unsigned *column,
275346766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor                                    unsigned *offset) {
27541db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  SourceLocation Loc = SourceLocation::getFromRawEncoding(location.int_data);
27551db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
2756bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar  if (!location.ptr_data[0] || Loc.isInvalid()) {
27579d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek    createNullLocation(file, line, column, offset);
275846766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor    return;
275946766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor  }
276046766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor
2761bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar  const SourceManager &SM =
2762bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar    *static_cast<const SourceManager*>(location.ptr_data[0]);
27631db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  SourceLocation InstLoc = SM.getInstantiationLoc(Loc);
27641db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
27659d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  // Check that the FileID is invalid on the instantiation location.
27669d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  // This can manifest in invalid code.
27679d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  FileID fileID = SM.getFileID(InstLoc);
27689d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  const SrcMgr::SLocEntry &sloc = SM.getSLocEntry(fileID);
27699d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  if (!sloc.isFile()) {
27709d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek    createNullLocation(file, line, column, offset);
27719d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek    return;
27729d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  }
27739d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek
27741db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  if (file)
27759d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek    *file = (void *)SM.getFileEntryForSLocEntry(sloc);
27761db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  if (line)
27771db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor    *line = SM.getInstantiationLineNumber(InstLoc);
27781db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  if (column)
27791db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor    *column = SM.getInstantiationColumnNumber(InstLoc);
2780e69517ce61638f12c9abe4605753a45275ac4e37Douglas Gregor  if (offset)
278146766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor    *offset = SM.getDecomposedLoc(InstLoc).second;
2782e69517ce61638f12c9abe4605753a45275ac4e37Douglas Gregor}
2783e69517ce61638f12c9abe4605753a45275ac4e37Douglas Gregor
2784a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregorvoid clang_getSpellingLocation(CXSourceLocation location,
2785a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor                               CXFile *file,
2786a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor                               unsigned *line,
2787a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor                               unsigned *column,
2788a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor                               unsigned *offset) {
2789a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  SourceLocation Loc = SourceLocation::getFromRawEncoding(location.int_data);
2790a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
2791a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (!location.ptr_data[0] || Loc.isInvalid()) {
2792a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    if (file)
2793a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      *file = 0;
2794a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    if (line)
2795a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      *line = 0;
2796a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    if (column)
2797a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      *column = 0;
2798a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    if (offset)
2799a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      *offset = 0;
2800a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    return;
2801a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  }
2802a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
2803a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  const SourceManager &SM =
2804a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *static_cast<const SourceManager*>(location.ptr_data[0]);
2805a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  SourceLocation SpellLoc = Loc;
2806a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (SpellLoc.isMacroID()) {
2807a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    SourceLocation SimpleSpellingLoc = SM.getImmediateSpellingLoc(SpellLoc);
2808a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    if (SimpleSpellingLoc.isFileID() &&
2809a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor        SM.getFileEntryForID(SM.getDecomposedLoc(SimpleSpellingLoc).first))
2810a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      SpellLoc = SimpleSpellingLoc;
2811a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    else
2812a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      SpellLoc = SM.getInstantiationLoc(SpellLoc);
2813a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  }
2814a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
2815a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  std::pair<FileID, unsigned> LocInfo = SM.getDecomposedLoc(SpellLoc);
2816a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  FileID FID = LocInfo.first;
2817a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  unsigned FileOffset = LocInfo.second;
2818a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
2819a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (file)
2820a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *file = (void *)SM.getFileEntryForID(FID);
2821a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (line)
2822a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *line = SM.getLineNumber(FID, FileOffset);
2823a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (column)
2824a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *column = SM.getColumnNumber(FID, FileOffset);
2825a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (offset)
2826a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *offset = FileOffset;
2827a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor}
2828a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
28291db19dea8d221f27be46332d668d1e2decb7f1abDouglas GregorCXSourceLocation clang_getRangeStart(CXSourceRange range) {
2830f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  CXSourceLocation Result = { { range.ptr_data[0], range.ptr_data[1] },
28315352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor                              range.begin_int_data };
28321db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  return Result;
28331db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor}
28341db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
28351db19dea8d221f27be46332d668d1e2decb7f1abDouglas GregorCXSourceLocation clang_getRangeEnd(CXSourceRange range) {
2836bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar  CXSourceLocation Result = { { range.ptr_data[0], range.ptr_data[1] },
28375352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor                              range.end_int_data };
28381db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  return Result;
28391db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor}
28401db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
2841b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor} // end: extern "C"
2842b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
28431db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor//===----------------------------------------------------------------------===//
2844fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXFile Operations.
2845fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2846fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2847fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
284874844072411bae91d5dbb89955d200cbe1e0a1c8Ted KremenekCXString clang_getFileName(CXFile SFile) {
284998258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (!SFile)
2850a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return createCXString((const char*)NULL);
2851f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
285288145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  FileEntry *FEnt = static_cast<FileEntry *>(SFile);
285374844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek  return createCXString(FEnt->getName());
285488145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff}
285588145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff
285688145034694ed5267fa6fa5febc54fadc02bd479Steve Narofftime_t clang_getFileTime(CXFile SFile) {
285798258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (!SFile)
285898258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor    return 0;
2859f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
286088145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  FileEntry *FEnt = static_cast<FileEntry *>(SFile);
286188145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  return FEnt->getModificationTime();
2862ee9405e807d7c447c0143c2bd865b759192e97b3Steve Naroff}
2863f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2864b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXFile clang_getFile(CXTranslationUnit tu, const char *file_name) {
2865b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  if (!tu)
2866b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor    return 0;
2867f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2868a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
2869f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2870b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  FileManager &FMgr = CXXUnit->getFileManager();
287139b49bcaaddb1049234fca9500c0ac02c088e23dChris Lattner  return const_cast<FileEntry *>(FMgr.getFile(file_name));
2872b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2873f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2874fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
2875fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2876fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2877fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXCursor Operations.
2878fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2879fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2880fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekstatic Decl *getDeclFromExpr(Stmt *E) {
2881db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor  if (CastExpr *CE = dyn_cast<CastExpr>(E))
2882db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor    return getDeclFromExpr(CE->getSubExpr());
2883db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2884fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (DeclRefExpr *RefExpr = dyn_cast<DeclRefExpr>(E))
2885fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return RefExpr->getDecl();
288638f28c1189142429384996409fffbc57f67b5c60Douglas Gregor  if (BlockDeclRefExpr *RefExpr = dyn_cast<BlockDeclRefExpr>(E))
288738f28c1189142429384996409fffbc57f67b5c60Douglas Gregor    return RefExpr->getDecl();
2888fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (MemberExpr *ME = dyn_cast<MemberExpr>(E))
2889fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return ME->getMemberDecl();
2890fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (ObjCIvarRefExpr *RE = dyn_cast<ObjCIvarRefExpr>(E))
2891fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return RE->getDecl();
2892db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor  if (ObjCPropertyRefExpr *PRE = dyn_cast<ObjCPropertyRefExpr>(E))
289312f78a6741a4cb3d904340f8d3d2714568b50e7aJohn McCall    return PRE->isExplicitProperty() ? PRE->getExplicitProperty() : 0;
2894db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2895fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (CallExpr *CE = dyn_cast<CallExpr>(E))
2896fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return getDeclFromExpr(CE->getCallee());
289793798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  if (CXXConstructExpr *CE = llvm::dyn_cast<CXXConstructExpr>(E))
289893798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    if (!CE->isElidable())
289993798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    return CE->getConstructor();
2900fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (ObjCMessageExpr *OME = dyn_cast<ObjCMessageExpr>(E))
2901fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return OME->getMethodDecl();
2902f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2903db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor  if (ObjCProtocolExpr *PE = dyn_cast<ObjCProtocolExpr>(E))
2904db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor    return PE->getProtocol();
2905c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor  if (SubstNonTypeTemplateParmPackExpr *NTTP
2906c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor                              = dyn_cast<SubstNonTypeTemplateParmPackExpr>(E))
2907c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor    return NTTP->getParameterPack();
290894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  if (SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
290994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor    if (isa<NonTypeTemplateParmDecl>(SizeOfPack->getPack()) ||
291094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        isa<ParmVarDecl>(SizeOfPack->getPack()))
291194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      return SizeOfPack->getPack();
2912db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2913fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  return 0;
2914fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek}
2915ee9405e807d7c447c0143c2bd865b759192e97b3Steve Naroff
2916c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbarstatic SourceLocation getLocationFromExpr(Expr *E) {
2917c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (ObjCMessageExpr *Msg = dyn_cast<ObjCMessageExpr>(E))
2918c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return /*FIXME:*/Msg->getLeftLoc();
2919c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
2920c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return DRE->getLocation();
292138f28c1189142429384996409fffbc57f67b5c60Douglas Gregor  if (BlockDeclRefExpr *RefExpr = dyn_cast<BlockDeclRefExpr>(E))
292238f28c1189142429384996409fffbc57f67b5c60Douglas Gregor    return RefExpr->getLocation();
2923c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (MemberExpr *Member = dyn_cast<MemberExpr>(E))
2924c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return Member->getMemberLoc();
2925c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (ObjCIvarRefExpr *Ivar = dyn_cast<ObjCIvarRefExpr>(E))
2926c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return Ivar->getLocation();
292794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  if (SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
292894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor    return SizeOfPack->getPackLoc();
292994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
2930c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  return E->getLocStart();
2931c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar}
2932c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar
2933fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
2934f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2935f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekunsigned clang_visitChildren(CXCursor parent,
2936b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor                             CXCursorVisitor visitor,
2937b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor                             CXClientData client_data) {
2938a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CursorVisitor CursorVis(getCursorTU(parent), visitor, client_data,
293904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                          getCursorASTUnit(parent)->getMaxPCHLevel(),
294004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                          false);
2941b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return CursorVis.VisitChildren(parent);
2942b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
2943b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor
29443387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#ifndef __has_feature
29453387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#define __has_feature(x) 0
29463387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#endif
29473387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#if __has_feature(blocks)
29483387c65a094a02b2a94c05111d035a97d3d5c794David Chisnalltypedef enum CXChildVisitResult
29493387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall     (^CXCursorVisitorBlock)(CXCursor cursor, CXCursor parent);
29503387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
29513387c65a094a02b2a94c05111d035a97d3d5c794David Chisnallstatic enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
29523387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall    CXClientData client_data) {
29533387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
29543387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return block(cursor, parent);
29553387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
29563387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#else
29573387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall// If we are compiled with a compiler that doesn't have native blocks support,
29583387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall// define and call the block manually, so the
29593387c65a094a02b2a94c05111d035a97d3d5c794David Chisnalltypedef struct _CXChildVisitResult
29603387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall{
29613387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	void *isa;
29623387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	int flags;
29633387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	int reserved;
29649e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar	enum CXChildVisitResult(*invoke)(struct _CXChildVisitResult*, CXCursor,
29659e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                         CXCursor);
29663387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall} *CXCursorVisitorBlock;
29673387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
29683387c65a094a02b2a94c05111d035a97d3d5c794David Chisnallstatic enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
29693387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall    CXClientData client_data) {
29703387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
29713387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return block->invoke(block, cursor, parent);
29723387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
29733387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#endif
29743387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
29753387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
29769e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbarunsigned clang_visitChildrenWithBlock(CXCursor parent,
29779e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                      CXCursorVisitorBlock block) {
29783387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return clang_visitChildren(parent, visitWithBlock, block);
29793387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
29803387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
298178205d4bada39d95097e766af9eb30cdd0159461Douglas Gregorstatic CXString getDeclSpelling(Decl *D) {
298278205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D);
2983e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor  if (!ND) {
2984e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor    if (ObjCPropertyImplDecl *PropImpl =llvm::dyn_cast<ObjCPropertyImplDecl>(D))
2985e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor      if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
2986e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor        return createCXString(Property->getIdentifier()->getName());
2987e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
2988ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
2989e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor  }
2990e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
299178205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
2992ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(OMD->getSelector().getAsString());
2993f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
299478205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ObjCCategoryImplDecl *CIMP = dyn_cast<ObjCCategoryImplDecl>(ND))
299578205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // No, this isn't the same as the code below. getIdentifier() is non-virtual
299678205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // and returns different names. NamedDecl returns the class name and
299778205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // ObjCCategoryImplDecl returns the category name.
2998ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(CIMP->getIdentifier()->getNameStart());
2999f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
30000a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  if (isa<UsingDirectiveDecl>(D))
30010a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor    return createCXString("");
30020a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
300350aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  llvm::SmallString<1024> S;
300450aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  llvm::raw_svector_ostream os(S);
300550aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  ND->printName(os);
300650aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek
300750aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  return createCXString(os.str());
300878205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor}
3009f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
30109ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXString clang_getCursorSpelling(CXCursor C) {
30117eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  if (clang_isTranslationUnit(C.kind))
3012a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return clang_getTranslationUnitSpelling(
3013a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                            static_cast<CXTranslationUnit>(C.data[2]));
30147eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
3015f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  if (clang_isReference(C.kind)) {
3016f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff    switch (C.kind) {
3017acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCSuperClassRef: {
30182e331b938b38057e333fab0ba841130ea8467794Douglas Gregor      ObjCInterfaceDecl *Super = getCursorObjCSuperClassRef(C).first;
3019ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(Super->getIdentifier()->getNameStart());
3020acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
3021acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCClassRef: {
30221adb082a709f7b588f03672999294e061234b2cfDouglas Gregor      ObjCInterfaceDecl *Class = getCursorObjCClassRef(C).first;
3023ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(Class->getIdentifier()->getNameStart());
3024acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
3025acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCProtocolRef: {
302678db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor      ObjCProtocolDecl *OID = getCursorObjCProtocolRef(C).first;
3027f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      assert(OID && "getCursorSpelling(): Missing protocol decl");
3028ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(OID->getIdentifier()->getNameStart());
3029acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
30303064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
30313064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      CXXBaseSpecifier *B = getCursorCXXBaseSpecifier(C);
30323064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      return createCXString(B->getType().getAsString());
30333064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
30347d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    case CXCursor_TypeRef: {
30357d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      TypeDecl *Type = getCursorTypeRef(C).first;
30367d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      assert(Type && "Missing type decl");
30377d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3038ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(getCursorContext(C).getTypeDeclType(Type).
3039ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek                              getAsString());
30407d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    }
30410b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef: {
30420b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      TemplateDecl *Template = getCursorTemplateRef(C).first;
30436931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      assert(Template && "Missing template decl");
30440b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
30450b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return createCXString(Template->getNameAsString());
30460b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    }
30476931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
30486931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef: {
30496931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      NamedDecl *NS = getCursorNamespaceRef(C).first;
30506931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      assert(NS && "Missing namespace decl");
30516931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
30526931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return createCXString(NS->getNameAsString());
30536931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    }
30547d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3055a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef: {
3056a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      FieldDecl *Field = getCursorMemberRef(C).first;
3057a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      assert(Field && "Missing member decl");
3058a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
3059a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return createCXString(Field->getNameAsString());
3060a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
3061a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
306236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef: {
306336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      LabelStmt *Label = getCursorLabelRef(C).first;
306436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      assert(Label && "Missing label");
306536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
3066ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner      return createCXString(Label->getName());
306736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    }
306836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
30691f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef: {
30701f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
30711f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (Decl *D = Storage.dyn_cast<Decl *>()) {
30721f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        if (NamedDecl *ND = dyn_cast<NamedDecl>(D))
30731f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor          return createCXString(ND->getNameAsString());
30741f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString("");
30751f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      }
30761f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
30771f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString(E->getName().getAsString());
30781f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      OverloadedTemplateStorage *Ovl
30791f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        = Storage.get<OverloadedTemplateStorage*>();
30801f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (Ovl->size() == 0)
30811f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString("");
30821f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return createCXString((*Ovl->begin())->getNameAsString());
30831f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    }
30841f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3085acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    default:
3086ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString("<not implemented>");
3087f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff    }
3088f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  }
308997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
309097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind)) {
309197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    Decl *D = getDeclFromExpr(getCursorExpr(C));
309297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    if (D)
309378205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor      return getDeclSpelling(D);
3094ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
309597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  }
309697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
309736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind)) {
309836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    Stmt *S = getCursorStmt(C);
309936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    if (LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S))
3100ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner      return createCXString(Label->getName());
310136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
310236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return createCXString("");
310336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  }
310436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
31054ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor  if (C.kind == CXCursor_MacroInstantiation)
31064ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor    return createCXString(getCursorMacroInstantiation(C)->getName()
31074ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor                                                           ->getNameStart());
31084ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor
3109572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition)
3110572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return createCXString(getCursorMacroDefinition(C)->getName()
3111572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor                                                           ->getNameStart());
3112572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3113ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective)
3114ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return createCXString(getCursorInclusionDirective(C)->getFileName());
3115ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
311660cbfacd947590f83257a4191566dda92fbbde69Douglas Gregor  if (clang_isDeclaration(C.kind))
311760cbfacd947590f83257a4191566dda92fbbde69Douglas Gregor    return getDeclSpelling(getCursorDecl(C));
3118e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3119ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString("");
3120f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff}
3121f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff
3122358559d8d7b458c5f64941842383a16e61f0828dDouglas GregorCXString clang_getCursorDisplayName(CXCursor C) {
3123358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (!clang_isDeclaration(C.kind))
3124358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return clang_getCursorSpelling(C);
3125358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3126358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  Decl *D = getCursorDecl(C);
3127358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (!D)
3128358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString("");
3129358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3130358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  PrintingPolicy &Policy = getCursorContext(C).PrintingPolicy;
3131358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(D))
3132358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    D = FunTmpl->getTemplatedDecl();
3133358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3134358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (FunctionDecl *Function = dyn_cast<FunctionDecl>(D)) {
3135358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::SmallString<64> Str;
3136358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3137358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << Function->getNameAsString();
3138358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (Function->getPrimaryTemplate())
3139358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << "<>";
3140358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << "(";
3141358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    for (unsigned I = 0, N = Function->getNumParams(); I != N; ++I) {
3142358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (I)
3143358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3144358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << Function->getParamDecl(I)->getType().getAsString(Policy);
3145358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3146358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3147358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (Function->isVariadic()) {
3148358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (Function->getNumParams())
3149358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3150358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << "...";
3151358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3152358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ")";
3153358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3154358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3155358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3156358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D)) {
3157358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::SmallString<64> Str;
3158358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3159358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ClassTemplate->getNameAsString();
3160358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << "<";
3161358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    TemplateParameterList *Params = ClassTemplate->getTemplateParameters();
3162358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    for (unsigned I = 0, N = Params->size(); I != N; ++I) {
3163358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (I)
3164358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3165358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3166358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      NamedDecl *Param = Params->getParam(I);
3167358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (Param->getIdentifier()) {
3168358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << Param->getIdentifier()->getName();
3169358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        continue;
3170358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      }
3171358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3172358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      // There is no parameter name, which makes this tricky. Try to come up
3173358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      // with something useful that isn't too long.
3174358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
3175358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << (TTP->wasDeclaredWithTypename()? "typename" : "class");
3176358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      else if (NonTypeTemplateParmDecl *NTTP
3177358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                                    = dyn_cast<NonTypeTemplateParmDecl>(Param))
3178358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << NTTP->getType().getAsString(Policy);
3179358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      else
3180358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << "template<...> class";
3181358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3182358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3183358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ">";
3184358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3185358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3186358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3187358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (ClassTemplateSpecializationDecl *ClassSpec
3188358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                              = dyn_cast<ClassTemplateSpecializationDecl>(D)) {
3189358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    // If the type was explicitly written, use that.
3190358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (TypeSourceInfo *TSInfo = ClassSpec->getTypeAsWritten())
3191358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      return createCXString(TSInfo->getType().getAsString(Policy));
3192358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3193358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::SmallString<64> Str;
3194358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3195358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ClassSpec->getNameAsString();
3196358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << TemplateSpecializationType::PrintTemplateArgumentList(
3197910f8008fea79120489a53593fe971b0b8a4a740Douglas Gregor                                      ClassSpec->getTemplateArgs().data(),
3198910f8008fea79120489a53593fe971b0b8a4a740Douglas Gregor                                      ClassSpec->getTemplateArgs().size(),
3199358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                                                                Policy);
3200358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3201358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3202358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3203358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  return clang_getCursorSpelling(C);
3204358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor}
3205358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3206e68fff6fc083c6270d835216a3de0b82c6ef0310Ted KremenekCXString clang_getCursorKindSpelling(enum CXCursorKind Kind) {
320789922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  switch (Kind) {
3208e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_FunctionDecl:
3209e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("FunctionDecl");
3210e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TypedefDecl:
3211e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TypedefDecl");
3212e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_EnumDecl:
3213e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("EnumDecl");
3214e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_EnumConstantDecl:
3215e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("EnumConstantDecl");
3216e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_StructDecl:
3217e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("StructDecl");
3218e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnionDecl:
3219e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnionDecl");
3220e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ClassDecl:
3221e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ClassDecl");
3222e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_FieldDecl:
3223e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("FieldDecl");
3224e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_VarDecl:
3225e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("VarDecl");
3226e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ParmDecl:
3227e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ParmDecl");
3228e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCInterfaceDecl:
3229e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCInterfaceDecl");
3230e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCCategoryDecl:
3231e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCCategoryDecl");
3232e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCProtocolDecl:
3233e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCProtocolDecl");
3234e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCPropertyDecl:
3235e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCPropertyDecl");
3236e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCIvarDecl:
3237e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCIvarDecl");
3238e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCInstanceMethodDecl:
3239e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCInstanceMethodDecl");
3240e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCClassMethodDecl:
3241e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCClassMethodDecl");
3242e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCImplementationDecl:
3243e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCImplementationDecl");
3244e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCCategoryImplDecl:
3245e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCCategoryImplDecl");
32468bd5a69999cfd06b6b5a58fdd04e4f802b2df5a4Ted Kremenek  case CXCursor_CXXMethod:
32478bd5a69999cfd06b6b5a58fdd04e4f802b2df5a4Ted Kremenek      return createCXString("CXXMethod");
3248e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedDecl:
3249e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedDecl");
3250e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCSuperClassRef:
3251e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCSuperClassRef");
3252e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCProtocolRef:
3253e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCProtocolRef");
3254e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCClassRef:
3255e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCClassRef");
3256e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TypeRef:
3257e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TypeRef");
32580b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case CXCursor_TemplateRef:
32590b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return createCXString("TemplateRef");
32606931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  case CXCursor_NamespaceRef:
32616931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    return createCXString("NamespaceRef");
3262a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  case CXCursor_MemberRef:
3263a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return createCXString("MemberRef");
326436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  case CXCursor_LabelRef:
326536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return createCXString("LabelRef");
32661f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case CXCursor_OverloadedDeclRef:
32671f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return createCXString("OverloadedDeclRef");
3268e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedExpr:
3269e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedExpr");
32701ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  case CXCursor_BlockExpr:
32711ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek      return createCXString("BlockExpr");
3272e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_DeclRefExpr:
3273e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("DeclRefExpr");
3274e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_MemberRefExpr:
3275e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("MemberRefExpr");
3276e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_CallExpr:
3277e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("CallExpr");
3278e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCMessageExpr:
3279e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCMessageExpr");
3280e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedStmt:
3281e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedStmt");
328236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  case CXCursor_LabelStmt:
328336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return createCXString("LabelStmt");
3284e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_InvalidFile:
3285e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("InvalidFile");
3286292db6401f040795db3ea4e00fc02622d6c3ba1dTed Kremenek  case CXCursor_InvalidCode:
3287292db6401f040795db3ea4e00fc02622d6c3ba1dTed Kremenek    return createCXString("InvalidCode");
3288e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_NoDeclFound:
3289e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("NoDeclFound");
3290e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_NotImplemented:
3291e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("NotImplemented");
3292e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TranslationUnit:
3293e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TranslationUnit");
3294e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek  case CXCursor_UnexposedAttr:
3295e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek      return createCXString("UnexposedAttr");
3296e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek  case CXCursor_IBActionAttr:
3297e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek      return createCXString("attribute(ibaction)");
32989f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  case CXCursor_IBOutletAttr:
32999f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor     return createCXString("attribute(iboutlet)");
3300857e918a8a40deb128840308a318bf623d68295fTed Kremenek  case CXCursor_IBOutletCollectionAttr:
3301857e918a8a40deb128840308a318bf623d68295fTed Kremenek      return createCXString("attribute(iboutletcollection)");
33029f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  case CXCursor_PreprocessingDirective:
33039f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    return createCXString("preprocessing directive");
3304572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  case CXCursor_MacroDefinition:
3305572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return createCXString("macro definition");
33064807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  case CXCursor_MacroInstantiation:
33074807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor    return createCXString("macro instantiation");
3308ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  case CXCursor_InclusionDirective:
3309ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return createCXString("inclusion directive");
33108f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  case CXCursor_Namespace:
33118f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek    return createCXString("Namespace");
3312a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  case CXCursor_LinkageSpec:
3313a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek    return createCXString("LinkageSpec");
33143064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  case CXCursor_CXXBaseSpecifier:
33153064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    return createCXString("C++ base class specifier");
331601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_Constructor:
331701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXConstructor");
331801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_Destructor:
331901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXDestructor");
332001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_ConversionFunction:
332101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXConversion");
3322fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_TemplateTypeParameter:
3323fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("TemplateTypeParameter");
3324fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_NonTypeTemplateParameter:
3325fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("NonTypeTemplateParameter");
3326fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_TemplateTemplateParameter:
3327fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("TemplateTemplateParameter");
3328fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_FunctionTemplate:
3329fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("FunctionTemplate");
333039d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  case CXCursor_ClassTemplate:
333139d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor    return createCXString("ClassTemplate");
333274dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  case CXCursor_ClassTemplatePartialSpecialization:
333374dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    return createCXString("ClassTemplatePartialSpecialization");
33346931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  case CXCursor_NamespaceAlias:
33356931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    return createCXString("NamespaceAlias");
33360a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  case CXCursor_UsingDirective:
33370a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor    return createCXString("UsingDirective");
33387e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  case CXCursor_UsingDeclaration:
33397e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor    return createCXString("UsingDeclaration");
334089922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  }
3341e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3342deb06bd3566e18f677e76bc435d478b033fe328bTed Kremenek  llvm_unreachable("Unhandled CXCursorKind");
3343a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return createCXString((const char*) 0);
3344600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
334589922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff
3346e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenekenum CXChildVisitResult GetCursorVisitor(CXCursor cursor,
3347e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek                                         CXCursor parent,
334833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor                                         CXClientData client_data) {
334933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  CXCursor *BestCursor = static_cast<CXCursor *>(client_data);
335093798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor
335193798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // If our current best cursor is the construction of a temporary object,
335293798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // don't replace that cursor with a type reference, because we want
335393798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // clang_getCursor() to point at the constructor.
335493798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  if (clang_isExpression(BestCursor->kind) &&
335593798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor      isa<CXXTemporaryObjectExpr>(getCursorExpr(*BestCursor)) &&
335693798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor      cursor.kind == CXCursor_TypeRef)
335793798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    return CXChildVisit_Recurse;
335893798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor
335985fe1560b061b5f93a52dbd07cddd6e808854710Douglas Gregor  // Don't override a preprocessing cursor with another preprocessing
336085fe1560b061b5f93a52dbd07cddd6e808854710Douglas Gregor  // cursor; we want the outermost preprocessing cursor.
336185fe1560b061b5f93a52dbd07cddd6e808854710Douglas Gregor  if (clang_isPreprocessing(cursor.kind) &&
336285fe1560b061b5f93a52dbd07cddd6e808854710Douglas Gregor      clang_isPreprocessing(BestCursor->kind))
336385fe1560b061b5f93a52dbd07cddd6e808854710Douglas Gregor    return CXChildVisit_Recurse;
336485fe1560b061b5f93a52dbd07cddd6e808854710Douglas Gregor
336533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  *BestCursor = cursor;
336633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  return CXChildVisit_Recurse;
336733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
3368e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3369b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXCursor clang_getCursor(CXTranslationUnit TU, CXSourceLocation Loc) {
3370b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  if (!TU)
3371f462989fe8d6f59ab2d7d0fe2b4b96292ce706eaTed Kremenek    return clang_getNullCursor();
3372e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3373a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
3374bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
3375bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor
3376a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  // Translate the given source location to make it point at the beginning of
3377a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  // the token under the cursor.
3378a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek  SourceLocation SLoc = cxloc::translateSourceLocation(Loc);
3379a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek
3380a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek  // Guard against an invalid SourceLocation, or we may assert in one
3381a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek  // of the following calls.
3382a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek  if (SLoc.isInvalid())
3383a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek    return clang_getNullCursor();
3384a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek
338540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  bool Logging = getenv("LIBCLANG_LOGGING");
3386a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  SLoc = Lexer::GetBeginningOfToken(SLoc, CXXUnit->getSourceManager(),
3387a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor                                    CXXUnit->getASTContext().getLangOptions());
3388a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
338933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  CXCursor Result = MakeCXCursorInvalid(CXCursor_NoDeclFound);
339033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (SLoc.isValid()) {
339133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    // FIXME: Would be great to have a "hint" cursor, then walk from that
339233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    // hint cursor upward until we find a cursor whose source range encloses
339333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    // the region of interest, rather than starting from the translation unit.
3394a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    CXCursor Parent = clang_getTranslationUnitCursor(TU);
3395a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    CursorVisitor CursorVis(TU, GetCursorVisitor, &Result,
339604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                            Decl::MaxPCHLevel, true, SourceLocation(SLoc));
339733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    CursorVis.VisitChildren(Parent);
339877128ddd3077fc045751a55bb3226802b15d5510Steve Naroff  }
339940749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
340040749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  if (Logging) {
340140749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXFile SearchFile;
340240749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    unsigned SearchLine, SearchColumn;
340340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXFile ResultFile;
340440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    unsigned ResultLine, ResultColumn;
34056653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    CXString SearchFileName, ResultFileName, KindSpelling, USR;
34066653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    const char *IsDef = clang_isCursorDefinition(Result)? " (Definition)" : "";
340740749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXSourceLocation ResultLoc = clang_getCursorLocation(Result);
340840749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
340940749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_getInstantiationLocation(Loc, &SearchFile, &SearchLine, &SearchColumn,
341040749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor                                   0);
341140749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_getInstantiationLocation(ResultLoc, &ResultFile, &ResultLine,
341240749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor                                   &ResultColumn, 0);
341340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    SearchFileName = clang_getFileName(SearchFile);
341440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    ResultFileName = clang_getFileName(ResultFile);
341540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    KindSpelling = clang_getCursorKindSpelling(Result.kind);
34166653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    USR = clang_getCursorUSR(Result);
34176653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    fprintf(stderr, "clang_getCursor(%s:%d:%d) = %s(%s:%d:%d):%s%s\n",
341840749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor            clang_getCString(SearchFileName), SearchLine, SearchColumn,
341940749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor            clang_getCString(KindSpelling),
34206653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor            clang_getCString(ResultFileName), ResultLine, ResultColumn,
34216653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor            clang_getCString(USR), IsDef);
342240749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(SearchFileName);
342340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(ResultFileName);
342440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(KindSpelling);
34256653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    clang_disposeString(USR);
34260aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor
34270aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    CXCursor Definition = clang_getCursorDefinition(Result);
34280aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    if (!clang_equalCursors(Definition, clang_getNullCursor())) {
34290aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXSourceLocation DefinitionLoc = clang_getCursorLocation(Definition);
34300aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXString DefinitionKindSpelling
34310aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor                                = clang_getCursorKindSpelling(Definition.kind);
34320aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXFile DefinitionFile;
34330aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      unsigned DefinitionLine, DefinitionColumn;
34340aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_getInstantiationLocation(DefinitionLoc, &DefinitionFile,
34350aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor                                     &DefinitionLine, &DefinitionColumn, 0);
34360aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXString DefinitionFileName = clang_getFileName(DefinitionFile);
34370aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      fprintf(stderr, "  -> %s(%s:%d:%d)\n",
34380aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              clang_getCString(DefinitionKindSpelling),
34390aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              clang_getCString(DefinitionFileName),
34400aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              DefinitionLine, DefinitionColumn);
34410aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_disposeString(DefinitionFileName);
34420aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_disposeString(DefinitionKindSpelling);
34430aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    }
344440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  }
344540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
3446e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  return Result;
344777128ddd3077fc045751a55bb3226802b15d5510Steve Naroff}
344877128ddd3077fc045751a55bb3226802b15d5510Steve Naroff
3449738855554394a6afcf39cc8345fd22c3756b8dd0Ted KremenekCXCursor clang_getNullCursor(void) {
34505bfb8c128c2ac8eb4032afc180cdc400a0f953caDouglas Gregor  return MakeCXCursorInvalid(CXCursor_InvalidFile);
3451738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek}
3452738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek
3453738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenekunsigned clang_equalCursors(CXCursor X, CXCursor Y) {
3454283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  return X == Y;
3455738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek}
34560d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
34579ce5584553054d0cb934940586aca0186e87fa57Douglas Gregorunsigned clang_hashCursor(CXCursor C) {
34589ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  unsigned Index = 0;
34599ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  if (clang_isExpression(C.kind) || clang_isStatement(C.kind))
34609ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor    Index = 1;
34619ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor
34629ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  return llvm::DenseMapInfo<std::pair<unsigned, void*> >::getHashValue(
34639ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor                                        std::make_pair(C.kind, C.data[Index]));
34649ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor}
34659ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor
34669ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isInvalid(enum CXCursorKind K) {
346777128ddd3077fc045751a55bb3226802b15d5510Steve Naroff  return K >= CXCursor_FirstInvalid && K <= CXCursor_LastInvalid;
346877128ddd3077fc045751a55bb3226802b15d5510Steve Naroff}
346977128ddd3077fc045751a55bb3226802b15d5510Steve Naroff
34709ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isDeclaration(enum CXCursorKind K) {
347189922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  return K >= CXCursor_FirstDecl && K <= CXCursor_LastDecl;
347289922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff}
34732d4d629d8a0de5112c7ae9d05c03ddbf6dcd956aSteve Naroff
34749ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isReference(enum CXCursorKind K) {
3475f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  return K >= CXCursor_FirstRef && K <= CXCursor_LastRef;
3476f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff}
3477f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff
347897b9872d5775446cb8aca1380e437649fe848d91Douglas Gregorunsigned clang_isExpression(enum CXCursorKind K) {
347997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  return K >= CXCursor_FirstExpr && K <= CXCursor_LastExpr;
348097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor}
348197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
348297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregorunsigned clang_isStatement(enum CXCursorKind K) {
348397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  return K >= CXCursor_FirstStmt && K <= CXCursor_LastStmt;
348497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor}
348597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
34867eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregorunsigned clang_isTranslationUnit(enum CXCursorKind K) {
34877eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  return K == CXCursor_TranslationUnit;
34887eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor}
34897eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
34909f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregorunsigned clang_isPreprocessing(enum CXCursorKind K) {
34919f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  return K >= CXCursor_FirstPreprocessing && K <= CXCursor_LastPreprocessing;
34929f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor}
34939f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor
3494ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenekunsigned clang_isUnexposed(enum CXCursorKind K) {
3495ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek  switch (K) {
3496ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedDecl:
3497ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedExpr:
3498ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedStmt:
3499ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedAttr:
3500ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek      return true;
3501ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    default:
3502ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek      return false;
3503ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek  }
3504ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek}
3505ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek
35069ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXCursorKind clang_getCursorKind(CXCursor C) {
35079efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff  return C.kind;
35089efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff}
35099efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff
351098258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas GregorCXSourceLocation clang_getCursorLocation(CXCursor C) {
351198258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (clang_isReference(C.kind)) {
3512f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    switch (C.kind) {
3513f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCSuperClassRef: {
3514f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCInterfaceDecl *, SourceLocation> P
3515f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCSuperClassRef(C);
3516a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3517f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
3518f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor
3519f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCProtocolRef: {
3520f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCProtocolDecl *, SourceLocation> P
3521f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCProtocolRef(C);
3522a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3523f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
3524f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor
3525f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCClassRef: {
3526f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCInterfaceDecl *, SourceLocation> P
3527f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCClassRef(C);
3528a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3529f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
35307d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3531f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_TypeRef: {
35327d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      std::pair<TypeDecl *, SourceLocation> P = getCursorTypeRef(C);
3533a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
35347d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    }
35350b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
35360b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef: {
35370b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      std::pair<TemplateDecl *, SourceLocation> P = getCursorTemplateRef(C);
35380b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
35390b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    }
35400b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
35416931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef: {
35426931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      std::pair<NamedDecl *, SourceLocation> P = getCursorNamespaceRef(C);
35436931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
35446931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    }
35456931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
3546a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef: {
3547a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      std::pair<FieldDecl *, SourceLocation> P = getCursorMemberRef(C);
3548a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3549a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
3550a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
35513064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
35521b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      CXXBaseSpecifier *BaseSpec = getCursorCXXBaseSpecifier(C);
35531b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      if (!BaseSpec)
35541b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor        return clang_getNullLocation();
35551b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor
35561b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      if (TypeSourceInfo *TSInfo = BaseSpec->getTypeSourceInfo())
35571b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor        return cxloc::translateSourceLocation(getCursorContext(C),
35581b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor                                            TSInfo->getTypeLoc().getBeginLoc());
35591b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor
35601b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C),
35611b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor                                        BaseSpec->getSourceRange().getBegin());
35623064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
3563f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
356436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef: {
356536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      std::pair<LabelStmt *, SourceLocation> P = getCursorLabelRef(C);
356636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C), P.second);
356736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    }
356836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
35691f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
35701f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C),
35711f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                                          getCursorOverloadedDeclRef(C).second);
35721f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3573f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    default:
3574f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      // FIXME: Need a way to enumerate all non-reference cases.
3575f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      llvm_unreachable("Missed a reference kind");
3576f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
357798258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  }
357897b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
357997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind))
3580f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    return cxloc::translateSourceLocation(getCursorContext(C),
358197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor                                   getLocationFromExpr(getCursorExpr(C)));
358297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
358336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind))
358436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C),
358536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor                                          getCursorStmt(C)->getLocStart());
358636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
35879f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  if (C.kind == CXCursor_PreprocessingDirective) {
35889f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    SourceLocation L = cxcursor::getCursorPreprocessingDirective(C).getBegin();
35899f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
35909f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  }
35914807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor
35924807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  if (C.kind == CXCursor_MacroInstantiation) {
35934ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor    SourceLocation L
35944ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor      = cxcursor::getCursorMacroInstantiation(C)->getSourceRange().getBegin();
35954807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
35964807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  }
3597572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3598572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition) {
3599572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    SourceLocation L = cxcursor::getCursorMacroDefinition(C)->getLocation();
3600572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
3601572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  }
3602ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3603ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective) {
3604ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    SourceLocation L
3605ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor      = cxcursor::getCursorInclusionDirective(C)->getSourceRange().getBegin();
3606ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
3607ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  }
3608ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
36099a700d277c38d9afaa7cb3fe93a714bfe9b62eecTed Kremenek  if (C.kind < CXCursor_FirstDecl || C.kind > CXCursor_LastDecl)
36105352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullLocation();
361198258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor
3612f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  Decl *D = getCursorDecl(C);
3613f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  SourceLocation Loc = D->getLocation();
3614f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(D))
3615f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    Loc = Class->getClassLoc();
3616007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // FIXME: Multiple variables declared in a single declaration
3617007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // currently lack the information needed to correctly determine their
3618007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // ranges when accounting for the type-specifier.  We use context
3619007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
3620007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // and if so, whether it is the first decl.
3621007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
3622007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    if (!cxcursor::isFirstInDeclGroup(C))
3623007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek      Loc = VD->getLocation();
3624007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  }
3625007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek
36262ca54feee89d7277fb967e3247a64f40ef155a82Douglas Gregor  return cxloc::translateSourceLocation(getCursorContext(C), Loc);
362788145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff}
3628a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor
3629a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor} // end extern "C"
3630a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
3631a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorstatic SourceRange getRawCursorExtent(CXCursor C) {
3632a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  if (clang_isReference(C.kind)) {
3633a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor    switch (C.kind) {
3634a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCSuperClassRef:
3635a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return  getCursorObjCSuperClassRef(C).second;
3636f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3637a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCProtocolRef:
3638a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorObjCProtocolRef(C).second;
3639f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3640a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCClassRef:
3641a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorObjCClassRef(C).second;
36427d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3643a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_TypeRef:
3644a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorTypeRef(C).second;
36450b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
36460b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef:
36470b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return getCursorTemplateRef(C).second;
36480b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
36496931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef:
36506931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return getCursorNamespaceRef(C).second;
3651a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
3652a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef:
3653a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return getCursorMemberRef(C).second;
3654a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
36553064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier:
36561b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      return getCursorCXXBaseSpecifier(C)->getSourceRange();
3657f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
365836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef:
365936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return getCursorLabelRef(C).second;
366036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
36611f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
36621f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return getCursorOverloadedDeclRef(C).second;
36631f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3664a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    default:
3665a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      // FIXME: Need a way to enumerate all non-reference cases.
3666a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      llvm_unreachable("Missed a reference kind");
3667a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor    }
3668a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  }
366997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
367097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind))
3671a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return getCursorExpr(C)->getSourceRange();
367233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
367333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (clang_isStatement(C.kind))
3674a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return getCursorStmt(C)->getSourceRange();
3675f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3676a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (C.kind == CXCursor_PreprocessingDirective)
3677a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return cxcursor::getCursorPreprocessingDirective(C);
36784807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor
3679a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (C.kind == CXCursor_MacroInstantiation)
3680a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return cxcursor::getCursorMacroInstantiation(C)->getSourceRange();
3681572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3682a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (C.kind == CXCursor_MacroDefinition)
3683a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return cxcursor::getCursorMacroDefinition(C)->getSourceRange();
3684ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3685ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective)
3686ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return cxcursor::getCursorInclusionDirective(C)->getSourceRange();
3687ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3688007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  if (C.kind >= CXCursor_FirstDecl && C.kind <= CXCursor_LastDecl) {
3689007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    Decl *D = cxcursor::getCursorDecl(C);
3690007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    SourceRange R = D->getSourceRange();
3691007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // FIXME: Multiple variables declared in a single declaration
3692007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // currently lack the information needed to correctly determine their
3693007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // ranges when accounting for the type-specifier.  We use context
3694007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
3695007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // and if so, whether it is the first decl.
3696007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
3697007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek      if (!cxcursor::isFirstInDeclGroup(C))
3698007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek        R.setBegin(VD->getLocation());
3699007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    }
3700007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    return R;
3701007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  }
37026653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  return SourceRange();
37036653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor}
37046653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
37056653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor/// \brief Retrieves the "raw" cursor extent, which is then extended to include
37066653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor/// the decl-specifier-seq for declarations.
37076653798ff5ce6deb58112777e21307ccc453133dDouglas Gregorstatic SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr) {
37086653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  if (C.kind >= CXCursor_FirstDecl && C.kind <= CXCursor_LastDecl) {
37096653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    Decl *D = cxcursor::getCursorDecl(C);
37106653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    SourceRange R = D->getSourceRange();
37112494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
37122494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // Adjust the start of the location for declarations preceded by
37132494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // declaration specifiers.
37142494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    SourceLocation StartLoc;
37156653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    if (const DeclaratorDecl *DD = dyn_cast<DeclaratorDecl>(D)) {
37162494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
37172494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
37182494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    } else if (TypedefDecl *Typedef = dyn_cast<TypedefDecl>(D)) {
37192494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
37202494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
37212494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    }
37226653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
37232494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (StartLoc.isValid() && R.getBegin().isValid() &&
37242494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        SrcMgr.isBeforeInTranslationUnit(StartLoc, R.getBegin()))
37252494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      R.setBegin(StartLoc);
37262494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
37272494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // FIXME: Multiple variables declared in a single declaration
37282494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // currently lack the information needed to correctly determine their
37292494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // ranges when accounting for the type-specifier.  We use context
37302494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
37312494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // and if so, whether it is the first decl.
37322494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
37332494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (!cxcursor::isFirstInDeclGroup(C))
37342494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        R.setBegin(VD->getLocation());
37356653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    }
37366653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
37376653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    return R;
37386653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  }
37396653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
37406653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  return getRawCursorExtent(C);
37416653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor}
3742a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
3743a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorextern "C" {
3744a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
3745a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas GregorCXSourceRange clang_getCursorExtent(CXCursor C) {
3746a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  SourceRange R = getRawCursorExtent(C);
3747a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (R.isInvalid())
37485352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
3749f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3750a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  return cxloc::translateSourceRange(getCursorContext(C), R);
3751a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor}
3752c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor
3753c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas GregorCXCursor clang_getCursorReferenced(CXCursor C) {
3754b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (clang_isInvalid(C.kind))
3755b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getNullCursor();
3756f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3757a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit tu = getCursorTU(C);
37581f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (clang_isDeclaration(C.kind)) {
37591f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Decl *D = getCursorDecl(C);
37601f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (UsingDecl *Using = dyn_cast<UsingDecl>(D))
3761a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Using, D->getLocation(), tu);
37621f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (ObjCClassDecl *Classes = dyn_cast<ObjCClassDecl>(D))
3763a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Classes, D->getLocation(), tu);
37641f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (ObjCForwardProtocolDecl *Protocols
37651f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                                        = dyn_cast<ObjCForwardProtocolDecl>(D))
3766a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Protocols, D->getLocation(), tu);
3767e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor    if (ObjCPropertyImplDecl *PropImpl =llvm::dyn_cast<ObjCPropertyImplDecl>(D))
3768e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor      if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
3769e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor        return MakeCXCursor(Property, tu);
3770e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
3771c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    return C;
37721f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  }
37731f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
377497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind)) {
37751f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Expr *E = getCursorExpr(C);
37761f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Decl *D = getDeclFromExpr(E);
377797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    if (D)
3778a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(D, tu);
37791f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
37801f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (OverloadExpr *Ovl = dyn_cast_or_null<OverloadExpr>(E))
3781a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Ovl, tu);
37821f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
378397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    return clang_getNullCursor();
378497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  }
378597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
378636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind)) {
378736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    Stmt *S = getCursorStmt(C);
378836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    if (GotoStmt *Goto = dyn_cast_or_null<GotoStmt>(S))
378937c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek      if (LabelDecl *label = Goto->getLabel())
379037c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek        if (LabelStmt *labelS = label->getStmt())
379137c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek        return MakeCXCursor(labelS, getCursorDecl(C), tu);
379236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
379336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return clang_getNullCursor();
379436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  }
379536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
3796bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor  if (C.kind == CXCursor_MacroInstantiation) {
3797bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor    if (MacroDefinition *Def = getCursorMacroInstantiation(C)->getDefinition())
3798a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeMacroDefinitionCursor(Def, tu);
3799bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor  }
3800bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor
3801c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  if (!clang_isReference(C.kind))
3802c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    return clang_getNullCursor();
3803f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3804c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  switch (C.kind) {
3805c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    case CXCursor_ObjCSuperClassRef:
3806a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorObjCSuperClassRef(C).first, tu);
3807f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3808f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCProtocolRef: {
3809a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorObjCProtocolRef(C).first, tu);
3810f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3811f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCClassRef:
3812a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorObjCClassRef(C).first, tu );
38137d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3814f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_TypeRef:
3815a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorTypeRef(C).first, tu );
38160b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
38170b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef:
3818a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorTemplateRef(C).first, tu );
38190b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
38206931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef:
3821a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorNamespaceRef(C).first, tu );
38226931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
3823a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef:
3824a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorMemberRef(C).first, tu );
3825a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
38263064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
38273064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      CXXBaseSpecifier *B = cxcursor::getCursorCXXBaseSpecifier(C);
38283064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      return clang_getTypeDeclaration(cxtype::MakeCXType(B->getType(),
3829a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                                         tu ));
38303064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
3831f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
383236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef:
383336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      // FIXME: We end up faking the "parent" declaration here because we
383436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      // don't want to make CXCursor larger.
383536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return MakeCXCursor(getCursorLabelRef(C).first,
3836a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek               static_cast<ASTUnit*>(tu->TUData)->getASTContext()
3837a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          .getTranslationUnitDecl(),
3838a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          tu);
383936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
38401f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
38411f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return C;
38421f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3843c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    default:
3844c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      // We would prefer to enumerate all non-reference cursor kinds here.
3845c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      llvm_unreachable("Unhandled reference cursor kind");
3846c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      break;
3847c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    }
3848c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  }
3849f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3850c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  return clang_getNullCursor();
3851c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor}
3852c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor
3853b699866820102a69d83d6ac6941985c5ef4e8c40Douglas GregorCXCursor clang_getCursorDefinition(CXCursor C) {
3854b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (clang_isInvalid(C.kind))
3855b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getNullCursor();
3856f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3857a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(C);
3858f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3859b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  bool WasReference = false;
386097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isReference(C.kind) || clang_isExpression(C.kind)) {
3861b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    C = clang_getCursorReferenced(C);
3862b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    WasReference = true;
3863b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
3864b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3865bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor  if (C.kind == CXCursor_MacroInstantiation)
3866bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor    return clang_getCursorReferenced(C);
3867bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor
3868b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!clang_isDeclaration(C.kind))
3869b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3870b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3871b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  Decl *D = getCursorDecl(C);
3872b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!D)
3873b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3874f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3875b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  switch (D->getKind()) {
3876b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds that don't really separate the notions of
3877b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // declaration and definition.
3878b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Namespace:
3879b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Typedef:
3880b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TemplateTypeParm:
3881b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::EnumConstant:
3882b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Field:
3883d98114647e16796a976b04af79975b4f0eacf22bBenjamin Kramer  case Decl::IndirectField:
3884b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCIvar:
3885b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCAtDefsField:
3886b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ImplicitParam:
3887b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ParmVar:
3888b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::NonTypeTemplateParm:
3889b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TemplateTemplateParm:
3890b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCategoryImpl:
3891b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCImplementation:
38926206d53f67613958ae1b023aba337ebb46f11a8bAbramo Bagnara  case Decl::AccessSpec:
3893b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::LinkageSpec:
3894b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCPropertyImpl:
3895b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FileScopeAsm:
3896b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::StaticAssert:
3897b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Block:
3898ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  case Decl::Label:  // FIXME: Is this right??
3899b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return C;
3900b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3901b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds that don't make any sense here, but are
3902b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // nonetheless harmless.
3903b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TranslationUnit:
3904b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    break;
3905b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3906b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds for which the definition is not resolvable.
3907b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UnresolvedUsingTypename:
3908b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UnresolvedUsingValue:
3909b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    break;
3910b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3911b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UsingDirective:
3912b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return MakeCXCursor(cast<UsingDirectiveDecl>(D)->getNominatedNamespace(),
3913a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                        TU);
3914b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3915b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::NamespaceAlias:
3916a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(cast<NamespaceAliasDecl>(D)->getNamespace(), TU);
3917b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3918b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Enum:
3919b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Record:
3920b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXRecord:
3921b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplateSpecialization:
3922b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplatePartialSpecialization:
3923952b017601f9c82b51119c3a1600f1312a833db9Douglas Gregor    if (TagDecl *Def = cast<TagDecl>(D)->getDefinition())
3924a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def, TU);
3925b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3926b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3927b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Function:
3928b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXMethod:
3929b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXConstructor:
3930b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXDestructor:
3931b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXConversion: {
3932b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    const FunctionDecl *Def = 0;
3933b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (cast<FunctionDecl>(D)->getBody(Def))
3934a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(const_cast<FunctionDecl *>(Def), TU);
3935b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3936b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
3937b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3938b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Var: {
393931310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    // Ask the variable if it has a definition.
394031310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    if (VarDecl *Def = cast<VarDecl>(D)->getDefinition())
3941a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def, TU);
394231310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    return clang_getNullCursor();
3943b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
3944f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3945b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FunctionTemplate: {
3946b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    const FunctionDecl *Def = 0;
3947b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (cast<FunctionTemplateDecl>(D)->getTemplatedDecl()->getBody(Def))
3948a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def->getDescribedFunctionTemplate(), TU);
3949b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3950b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
3951f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3952b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplate: {
3953b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (RecordDecl *Def = cast<ClassTemplateDecl>(D)->getTemplatedDecl()
3954952b017601f9c82b51119c3a1600f1312a833db9Douglas Gregor                                                            ->getDefinition())
39550b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return MakeCXCursor(cast<CXXRecordDecl>(Def)->getDescribedClassTemplate(),
3956a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          TU);
3957b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3958b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
3959b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
39601f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case Decl::Using:
39611f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return MakeCursorOverloadedDeclRef(cast<UsingDecl>(D),
3962a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                       D->getLocation(), TU);
3963b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3964b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UsingShadow:
3965b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getCursorDefinition(
3966f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                       MakeCXCursor(cast<UsingShadowDecl>(D)->getTargetDecl(),
3967a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                    TU));
3968b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3969b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCMethod: {
3970b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    ObjCMethodDecl *Method = cast<ObjCMethodDecl>(D);
3971b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (Method->isThisDeclarationADefinition())
3972b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      return C;
3973b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3974b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // Dig out the method definition in the associated
3975b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // @implementation, if we have it.
3976b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: The ASTs should make finding the definition easier.
3977b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCInterfaceDecl *Class
3978b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                       = dyn_cast<ObjCInterfaceDecl>(Method->getDeclContext()))
3979b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (ObjCImplementationDecl *ClassImpl = Class->getImplementation())
3980b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor        if (ObjCMethodDecl *Def = ClassImpl->getMethod(Method->getSelector(),
3981b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                                                  Method->isInstanceMethod()))
3982b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor          if (Def->isThisDeclarationADefinition())
3983a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek            return MakeCXCursor(Def, TU);
3984b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3985b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3986b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
3987b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3988b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCategory:
3989b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCCategoryImplDecl *Impl
3990b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                               = cast<ObjCCategoryDecl>(D)->getImplementation())
3991a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Impl, TU);
3992b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3993b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3994b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCProtocol:
3995b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (!cast<ObjCProtocolDecl>(D)->isForwardDecl())
3996b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      return C;
3997b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
3998b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
3999b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCInterface:
4000b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // There are two notions of a "definition" for an Objective-C
4001b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // class: the interface and its implementation. When we resolved a
4002b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // reference to an Objective-C class, produce the @interface as
4003b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // the definition; when we were provided with the interface,
4004b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // produce the @implementation as the definition.
4005b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (WasReference) {
4006b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (!cast<ObjCInterfaceDecl>(D)->isForwardDecl())
4007b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor        return C;
4008b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    } else if (ObjCImplementationDecl *Impl
4009b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                              = cast<ObjCInterfaceDecl>(D)->getImplementation())
4010a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Impl, TU);
4011b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4012f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4013b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCProperty:
4014b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: We don't really know where to find the
4015b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // ObjCPropertyImplDecls that implement this property.
4016b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4017b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4018b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCompatibleAlias:
4019b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCInterfaceDecl *Class
4020b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor          = cast<ObjCCompatibleAliasDecl>(D)->getClassInterface())
4021b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (!Class->isForwardDecl())
4022a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek        return MakeCXCursor(Class, TU);
4023f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4024b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4025b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
40261f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case Decl::ObjCForwardProtocol:
40271f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return MakeCursorOverloadedDeclRef(cast<ObjCForwardProtocolDecl>(D),
4028a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                       D->getLocation(), TU);
4029b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
40301f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case Decl::ObjCClass:
40319e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar    return MakeCursorOverloadedDeclRef(cast<ObjCClassDecl>(D), D->getLocation(),
4032a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                       TU);
4033b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4034b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Friend:
4035b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (NamedDecl *Friend = cast<FriendDecl>(D)->getFriendDecl())
4036a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
4037b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4038b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4039b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FriendTemplate:
4040b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (NamedDecl *Friend = cast<FriendTemplateDecl>(D)->getFriendDecl())
4041a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
4042b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4043b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4044b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4045b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  return clang_getNullCursor();
4046b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor}
4047b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4048b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregorunsigned clang_isCursorDefinition(CXCursor C) {
4049b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!clang_isDeclaration(C.kind))
4050b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return 0;
4051b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4052b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  return clang_getCursorDefinition(C) == C;
4053b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor}
4054b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
40551a9d0503b67a499797141af0fd6d315d5045f0eaDouglas GregorCXCursor clang_getCanonicalCursor(CXCursor C) {
40561a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  if (!clang_isDeclaration(C.kind))
40571a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor    return C;
40581a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
40591a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  if (Decl *D = getCursorDecl(C))
40601a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor    return MakeCXCursor(D->getCanonicalDecl(), getCursorTU(C));
40611a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
40621a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  return C;
40631a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor}
40641a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
40651f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregorunsigned clang_getNumOverloadedDecls(CXCursor C) {
40667c432dd959609a3689c2e4406450c092e6d76d6dDouglas Gregor  if (C.kind != CXCursor_OverloadedDeclRef)
40671f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return 0;
40681f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40691f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
40701f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
40711f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return E->getNumDecls();
40721f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40731f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadedTemplateStorage *S
40741f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                              = Storage.dyn_cast<OverloadedTemplateStorage*>())
40751f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return S->size();
40761f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40771f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  Decl *D = Storage.get<Decl*>();
40781f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (UsingDecl *Using = dyn_cast<UsingDecl>(D))
4079826faa22bae112e01293a58534a40711043cce65Argyrios Kyrtzidis    return Using->shadow_size();
40801f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (ObjCClassDecl *Classes = dyn_cast<ObjCClassDecl>(D))
40811f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return Classes->size();
40821f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (ObjCForwardProtocolDecl *Protocols =dyn_cast<ObjCForwardProtocolDecl>(D))
40831f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return Protocols->protocol_size();
40841f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40851f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  return 0;
40861f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor}
40871f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40881f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas GregorCXCursor clang_getOverloadedDecl(CXCursor cursor, unsigned index) {
40897c432dd959609a3689c2e4406450c092e6d76d6dDouglas Gregor  if (cursor.kind != CXCursor_OverloadedDeclRef)
40901f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return clang_getNullCursor();
40911f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40921f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (index >= clang_getNumOverloadedDecls(cursor))
40931f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return clang_getNullCursor();
40941f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
4095a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(cursor);
40961f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(cursor).first;
40971f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
4098a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(E->decls_begin()[index], TU);
40991f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
41001f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadedTemplateStorage *S
41011f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                              = Storage.dyn_cast<OverloadedTemplateStorage*>())
4102a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(S->begin()[index], TU);
41031f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
41041f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  Decl *D = Storage.get<Decl*>();
41051f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (UsingDecl *Using = dyn_cast<UsingDecl>(D)) {
41061f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    // FIXME: This is, unfortunately, linear time.
41071f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    UsingDecl::shadow_iterator Pos = Using->shadow_begin();
41081f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    std::advance(Pos, index);
4109a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(cast<UsingShadowDecl>(*Pos)->getTargetDecl(), TU);
41101f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  }
41111f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
41121f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (ObjCClassDecl *Classes = dyn_cast<ObjCClassDecl>(D))
4113a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(Classes->begin()[index].getInterface(), TU);
41141f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
41151f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (ObjCForwardProtocolDecl *Protocols = dyn_cast<ObjCForwardProtocolDecl>(D))
4116a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(Protocols->protocol_begin()[index], TU);
41171f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
41181f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  return clang_getNullCursor();
41191f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor}
41201f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
41210d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbarvoid clang_getDefinitionSpellingAndExtent(CXCursor C,
41224ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          const char **startBuf,
41234ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          const char **endBuf,
41244ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *startLine,
41254ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *startColumn,
41264ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *endLine,
41279ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          unsigned *endColumn) {
4128283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  assert(getCursorDecl(C) && "CXCursor has null decl");
4129283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  NamedDecl *ND = static_cast<NamedDecl *>(getCursorDecl(C));
41304ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  FunctionDecl *FD = dyn_cast<FunctionDecl>(ND);
41314ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  CompoundStmt *Body = dyn_cast<CompoundStmt>(FD->getBody());
4132f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
41334ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  SourceManager &SM = FD->getASTContext().getSourceManager();
41344ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startBuf = SM.getCharacterData(Body->getLBracLoc());
41354ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endBuf = SM.getCharacterData(Body->getRBracLoc());
41364ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startLine = SM.getSpellingLineNumber(Body->getLBracLoc());
41374ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startColumn = SM.getSpellingColumnNumber(Body->getLBracLoc());
41384ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endLine = SM.getSpellingLineNumber(Body->getRBracLoc());
41394ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endColumn = SM.getSpellingColumnNumber(Body->getRBracLoc());
41404ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff}
4141f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
41420a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregorvoid clang_enableStackTraces(void) {
41430a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  llvm::sys::PrintStackTraceOnErrorSignal();
41440a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor}
41450a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor
4146995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbarvoid clang_executeOnThread(void (*fn)(void*), void *user_data,
4147995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar                           unsigned stack_size) {
4148995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar  llvm::llvm_execute_on_thread(fn, user_data, stack_size);
4149995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar}
4150995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar
4151fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
4152fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
4153fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
4154fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor// Token-based Operations.
4155fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor//===----------------------------------------------------------------------===//
4156fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4157fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor/* CXToken layout:
4158fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[0]: a CXTokenKind
4159fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[1]: starting token location
4160fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[2]: token length
4161fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[3]: reserved
4162f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek *   ptr_data: for identifiers and keywords, an IdentifierInfo*.
4163fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   otherwise unused.
4164fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor */
4165fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregorextern "C" {
4166fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4167fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXTokenKind clang_getTokenKind(CXToken CXTok) {
4168fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  return static_cast<CXTokenKind>(CXTok.int_data[0]);
4169fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4170fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4171fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXString clang_getTokenSpelling(CXTranslationUnit TU, CXToken CXTok) {
4172fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  switch (clang_getTokenKind(CXTok)) {
4173fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Identifier:
4174fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Keyword:
4175fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // We know we have an IdentifierInfo*, so use that.
4176ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(static_cast<IdentifierInfo *>(CXTok.ptr_data)
4177ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek                            ->getNameStart());
4178fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4179fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Literal: {
4180fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // We have stashed the starting pointer in the ptr_data field. Use it.
4181fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    const char *Text = static_cast<const char *>(CXTok.ptr_data);
4182ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(llvm::StringRef(Text, CXTok.int_data[2]));
4183fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  }
4184f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4185fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Punctuation:
4186fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Comment:
4187fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    break;
4188fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  }
4189f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4190f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // We have to find the starting buffer pointer the hard way, by
4191fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // deconstructing the source location.
4192a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4193fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit)
4194ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
4195f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4196fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  SourceLocation Loc = SourceLocation::getFromRawEncoding(CXTok.int_data[1]);
4197fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> LocInfo
4198fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    = CXXUnit->getSourceManager().getDecomposedLoc(Loc);
4199f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  bool Invalid = false;
4200f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  llvm::StringRef Buffer
4201f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    = CXXUnit->getSourceManager().getBufferData(LocInfo.first, &Invalid);
4202f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  if (Invalid)
4203aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor    return createCXString("");
4204fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4205f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  return createCXString(Buffer.substr(LocInfo.second, CXTok.int_data[2]));
4206fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4207f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4208fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXSourceLocation clang_getTokenLocation(CXTranslationUnit TU, CXToken CXTok) {
4209a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4210fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit)
4211fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return clang_getNullLocation();
4212f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4213fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  return cxloc::translateSourceLocation(CXXUnit->getASTContext(),
4214fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                        SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
4215fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4216fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4217fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXSourceRange clang_getTokenExtent(CXTranslationUnit TU, CXToken CXTok) {
4218a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
42195352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  if (!CXXUnit)
42205352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
4221f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4222f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return cxloc::translateSourceRange(CXXUnit->getASTContext(),
4223fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                        SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
4224fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4225f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4226fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregorvoid clang_tokenize(CXTranslationUnit TU, CXSourceRange Range,
4227fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                    CXToken **Tokens, unsigned *NumTokens) {
4228fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (Tokens)
4229fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    *Tokens = 0;
4230fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (NumTokens)
4231fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    *NumTokens = 0;
4232f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4233a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4234fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit || !Tokens || !NumTokens)
4235fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4236f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4237bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
4238bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor
423985b988fdfa6adab6d43e16efd19ad4f3f7e2b49bDaniel Dunbar  SourceRange R = cxloc::translateCXSourceRange(Range);
4240fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (R.isInvalid())
4241fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4242f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4243fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  SourceManager &SourceMgr = CXXUnit->getSourceManager();
4244fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> BeginLocInfo
4245fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    = SourceMgr.getDecomposedLoc(R.getBegin());
4246fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> EndLocInfo
4247fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    = SourceMgr.getDecomposedLoc(R.getEnd());
4248f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4249fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // Cannot tokenize across files.
4250fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (BeginLocInfo.first != EndLocInfo.first)
4251fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4252f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4253f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // Create a lexer
4254f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  bool Invalid = false;
4255f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  llvm::StringRef Buffer
4256f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
425747a3fcd4afe122b23f9e7b6148f147bfa460cfe8Douglas Gregor  if (Invalid)
425847a3fcd4afe122b23f9e7b6148f147bfa460cfe8Douglas Gregor    return;
4259aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor
4260fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
4261fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor            CXXUnit->getASTContext().getLangOptions(),
4262f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer            Buffer.begin(), Buffer.data() + BeginLocInfo.second, Buffer.end());
4263fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Lex.SetCommentRetentionState(true);
4264f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4265fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // Lex tokens until we hit the end of the range.
4266f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  const char *EffectiveBufferEnd = Buffer.data() + EndLocInfo.second;
4267fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  llvm::SmallVector<CXToken, 32> CXTokens;
4268fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Token Tok;
4269096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall  bool previousWasAt = false;
4270fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  do {
4271fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // Lex the next token
4272fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    Lex.LexFromRawLexer(Tok);
4273fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    if (Tok.is(tok::eof))
4274fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      break;
4275f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4276fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // Initialize the CXToken.
4277fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXToken CXTok;
4278f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4279fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    //   - Common fields
4280fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[1] = Tok.getLocation().getRawEncoding();
4281fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[2] = Tok.getLength();
4282fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[3] = 0;
4283f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4284fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    //   - Kind-specific fields
4285fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    if (Tok.isLiteral()) {
4286fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Literal;
4287fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = (void *)Tok.getLiteralData();
4288c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara    } else if (Tok.is(tok::raw_identifier)) {
4289aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor      // Lookup the identifier to determine whether we have a keyword.
4290fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      IdentifierInfo *II
4291c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara        = CXXUnit->getPreprocessor().LookUpIdentifierInfo(Tok);
4292aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek
4293096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall      if ((II->getObjCKeywordID() != tok::objc_not_keyword) && previousWasAt) {
4294aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek        CXTok.int_data[0] = CXToken_Keyword;
4295aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      }
4296aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      else {
4297c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara        CXTok.int_data[0] = Tok.is(tok::identifier)
4298c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara          ? CXToken_Identifier
4299c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara          : CXToken_Keyword;
4300aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      }
4301fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = II;
4302fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else if (Tok.is(tok::comment)) {
4303fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Comment;
4304fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = 0;
4305fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else {
4306fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Punctuation;
4307fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = 0;
4308fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    }
4309fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTokens.push_back(CXTok);
4310096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall    previousWasAt = Tok.is(tok::at);
4311fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  } while (Lex.getBufferLocation() <= EffectiveBufferEnd);
4312f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4313fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (CXTokens.empty())
4314fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4315f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4316fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  *Tokens = (CXToken *)malloc(sizeof(CXToken) * CXTokens.size());
4317fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  memmove(*Tokens, CXTokens.data(), sizeof(CXToken) * CXTokens.size());
4318fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  *NumTokens = CXTokens.size();
4319fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
43200045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
43216db610934bedc6896393c1e1099525b35380acd6Ted Kremenekvoid clang_disposeTokens(CXTranslationUnit TU,
43226db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                         CXToken *Tokens, unsigned NumTokens) {
43236db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  free(Tokens);
43246db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
43256db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
43266db610934bedc6896393c1e1099525b35380acd6Ted Kremenek} // end: extern "C"
43276db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
43286db610934bedc6896393c1e1099525b35380acd6Ted Kremenek//===----------------------------------------------------------------------===//
43296db610934bedc6896393c1e1099525b35380acd6Ted Kremenek// Token annotation APIs.
43306db610934bedc6896393c1e1099525b35380acd6Ted Kremenek//===----------------------------------------------------------------------===//
43316db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
43320045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregortypedef llvm::DenseMap<unsigned, CXCursor> AnnotateTokensData;
4333fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekstatic enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
4334fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                     CXCursor parent,
4335fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                     CXClientData client_data);
43366db610934bedc6896393c1e1099525b35380acd6Ted Kremeneknamespace {
43376db610934bedc6896393c1e1099525b35380acd6Ted Kremenekclass AnnotateTokensWorker {
43386db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  AnnotateTokensData &Annotated;
433911949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  CXToken *Tokens;
434011949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  CXCursor *Cursors;
434111949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  unsigned NumTokens;
4342fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  unsigned TokIdx;
43434419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  unsigned PreprocessingTokIdx;
4344fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CursorVisitor AnnotateVis;
4345fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceManager &SrcMgr;
4346f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  bool HasContextSensitiveKeywords;
4347f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4348fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  bool MoreTokens() const { return TokIdx < NumTokens; }
4349fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  unsigned NextToken() const { return TokIdx; }
4350fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void AdvanceToken() { ++TokIdx; }
4351fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceLocation GetTokenLoc(unsigned tokI) {
4352fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[1]);
4353fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4354fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
43556db610934bedc6896393c1e1099525b35380acd6Ted Kremenekpublic:
435611949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  AnnotateTokensWorker(AnnotateTokensData &annotated,
4357fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                       CXToken *tokens, CXCursor *cursors, unsigned numTokens,
4358a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                       CXTranslationUnit tu, SourceRange RegionOfInterest)
435911949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek    : Annotated(annotated), Tokens(tokens), Cursors(cursors),
43604419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      NumTokens(numTokens), TokIdx(0), PreprocessingTokIdx(0),
4361a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      AnnotateVis(tu,
4362a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                  AnnotateTokensVisitor, this,
436304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                  Decl::MaxPCHLevel, true, RegionOfInterest),
4364f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      SrcMgr(static_cast<ASTUnit*>(tu->TUData)->getSourceManager()),
4365f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      HasContextSensitiveKeywords(false) { }
436611949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4367fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void VisitChildren(CXCursor C) { AnnotateVis.VisitChildren(C); }
43686db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  enum CXChildVisitResult Visit(CXCursor cursor, CXCursor parent);
4369fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void AnnotateTokens(CXCursor parent);
4370ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek  void AnnotateTokens() {
4371a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    AnnotateTokens(clang_getTranslationUnitCursor(AnnotateVis.getTU()));
4372ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek  }
4373f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4374f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  /// \brief Determine whether the annotator saw any cursors that have
4375f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  /// context-sensitive keywords.
4376f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  bool hasContextSensitiveKeywords() const {
4377f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    return HasContextSensitiveKeywords;
4378f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
43796db610934bedc6896393c1e1099525b35380acd6Ted Kremenek};
43806db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
43810045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
4382fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekvoid AnnotateTokensWorker::AnnotateTokens(CXCursor parent) {
4383fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Walk the AST within the region of interest, annotating tokens
4384fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // along the way.
4385fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  VisitChildren(parent);
4386fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4387fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = 0 ; I < TokIdx ; ++I) {
438811949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek    AnnotateTokensData::iterator Pos = Annotated.find(Tokens[I].int_data[1]);
43894419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    if (Pos != Annotated.end() &&
43904419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        (clang_isInvalid(Cursors[I].kind) ||
43914419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor         Pos->second.kind != CXCursor_PreprocessingDirective))
4392fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      Cursors[I] = Pos->second;
4393fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4394fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4395fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Finish up annotating any tokens left.
4396fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (!MoreTokens())
4397fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return;
439811949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4399fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const CXCursor &C = clang_getNullCursor();
4400fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = TokIdx ; I < NumTokens ; ++I) {
4401fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    AnnotateTokensData::iterator Pos = Annotated.find(Tokens[I].int_data[1]);
4402fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = (Pos == Annotated.end()) ? C : Pos->second;
440311949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  }
440411949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek}
440511949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
44066db610934bedc6896393c1e1099525b35380acd6Ted Kremenekenum CXChildVisitResult
44074419b675577d7c281a659fab1fec10e1bfbe04c5Douglas GregorAnnotateTokensWorker::Visit(CXCursor cursor, CXCursor parent) {
4408fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CXSourceLocation Loc = clang_getCursorLocation(cursor);
44094419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  SourceRange cursorRange = getRawCursorExtent(cursor);
441081d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor  if (cursorRange.isInvalid())
441181d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor    return CXChildVisit_Recurse;
4412f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4413f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  if (!HasContextSensitiveKeywords) {
4414f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // Objective-C properties can have context-sensitive keywords.
4415f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    if (cursor.kind == CXCursor_ObjCPropertyDecl) {
4416f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (ObjCPropertyDecl *Property
4417f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                  = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(cursor)))
4418f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        HasContextSensitiveKeywords = Property->getPropertyAttributesAsWritten() != 0;
4419f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4420f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // Objective-C methods can have context-sensitive keywords.
4421f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_ObjCInstanceMethodDecl ||
4422f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ObjCClassMethodDecl) {
4423f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (ObjCMethodDecl *Method
4424f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(cursor))) {
4425f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (Method->getObjCDeclQualifier())
4426f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4427f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        else {
4428f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
4429f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                                           PEnd = Method->param_end();
4430f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor               P != PEnd; ++P) {
4431f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            if ((*P)->getObjCDeclQualifier()) {
4432f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              HasContextSensitiveKeywords = true;
4433f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              break;
4434f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            }
4435f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          }
4436f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
4437f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4438f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4439f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // C++ methods can have context-sensitive keywords.
4440f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_CXXMethod) {
4441f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (CXXMethodDecl *Method
4442f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                  = dyn_cast_or_null<CXXMethodDecl>(getCursorDecl(cursor))) {
4443f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (Method->hasAttr<FinalAttr>() || Method->hasAttr<OverrideAttr>())
4444f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4445f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4446f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4447f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // C++ classes can have context-sensitive keywords.
4448f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_StructDecl ||
4449f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassDecl ||
4450f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassTemplate ||
4451f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassTemplatePartialSpecialization) {
4452f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Decl *D = getCursorDecl(cursor))
4453f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (D->hasAttr<FinalAttr>())
4454f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4455f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4456f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
4457f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
44584419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  if (clang_isPreprocessing(cursor.kind)) {
44594419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // For macro instantiations, just note where the beginning of the macro
44604419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // instantiation occurs.
44614419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    if (cursor.kind == CXCursor_MacroInstantiation) {
44624419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      Annotated[Loc.int_data] = cursor;
44634419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      return CXChildVisit_Recurse;
44644419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
44654419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
44664419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Items in the preprocessing record are kept separate from items in
44674419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // declarations, so we keep a separate token index.
44684419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    unsigned SavedTokIdx = TokIdx;
44694419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    TokIdx = PreprocessingTokIdx;
44704419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
44714419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Skip tokens up until we catch up to the beginning of the preprocessing
44724419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // entry.
44734419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    while (MoreTokens()) {
44744419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      const unsigned I = NextToken();
44754419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      SourceLocation TokLoc = GetTokenLoc(I);
44764419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
44774419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeBefore:
44784419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        AdvanceToken();
44794419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        continue;
44804419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeAfter:
44814419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeOverlap:
44824419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        break;
44834419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      }
44844419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      break;
44854419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
44864419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
44874419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Look at all of the tokens within this range.
44884419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    while (MoreTokens()) {
44894419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      const unsigned I = NextToken();
44904419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      SourceLocation TokLoc = GetTokenLoc(I);
44914419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
44924419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeBefore:
44934419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        assert(0 && "Infeasible");
44944419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeAfter:
44954419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        break;
44964419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeOverlap:
44974419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        Cursors[I] = cursor;
44984419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        AdvanceToken();
44994419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        continue;
45004419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      }
45014419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      break;
45024419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
45034419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
45044419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Save the preprocessing token index; restore the non-preprocessing
45054419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // token index.
45064419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    PreprocessingTokIdx = TokIdx;
45074419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    TokIdx = SavedTokIdx;
45080045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor    return CXChildVisit_Recurse;
45090045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  }
4510fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4511fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (cursorRange.isInvalid())
4512fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return CXChildVisit_Continue;
4513a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek
4514fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceLocation L = SourceLocation::getFromRawEncoding(Loc.int_data);
4515fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4516a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  // Adjust the annotated range based specific declarations.
4517a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  const enum CXCursorKind cursorK = clang_getCursorKind(cursor);
4518a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  if (cursorK >= CXCursor_FirstDecl && cursorK <= CXCursor_LastDecl) {
451923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    Decl *D = cxcursor::getCursorDecl(cursor);
452023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    // Don't visit synthesized ObjC methods, since they have no syntatic
452123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    // representation in the source.
452223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D)) {
452323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek      if (MD->isSynthesized())
452423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return CXChildVisit_Continue;
452523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    }
45262494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
45272494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    SourceLocation StartLoc;
452823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    if (const DeclaratorDecl *DD = dyn_cast<DeclaratorDecl>(D)) {
45292494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
45302494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
45312494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    } else if (TypedefDecl *Typedef = dyn_cast<TypedefDecl>(D)) {
45322494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
45332494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
4534a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek    }
45352494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
45362494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (StartLoc.isValid() && L.isValid() &&
45372494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        SrcMgr.isBeforeInTranslationUnit(StartLoc, L))
45382494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      cursorRange.setBegin(StartLoc);
4539a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  }
454081d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor
45413f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // If the location of the cursor occurs within a macro instantiation, record
45423f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // the spelling location of the cursor in our annotation map.  We can then
45433f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // paper over the token labelings during a post-processing step to try and
45443f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // get cursor mappings for tokens that are the *arguments* of a macro
45453f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // instantiation.
45463f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  if (L.isMacroID()) {
45473f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    unsigned rawEncoding = SrcMgr.getSpellingLoc(L).getRawEncoding();
45483f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // Only invalidate the old annotation if it isn't part of a preprocessing
45493f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // directive.  Here we assume that the default construction of CXCursor
45503f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // results in CXCursor.kind being an initialized value (i.e., 0).  If
45513f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // this isn't the case, we can fix by doing lookup + insertion.
45524419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
45533f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    CXCursor &oldC = Annotated[rawEncoding];
45543f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    if (!clang_isPreprocessing(oldC.kind))
45553f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek      oldC = cursor;
45563f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  }
45573f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek
4558fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const enum CXCursorKind K = clang_getCursorKind(parent);
4559fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const CXCursor updateC =
4560d8b0a84d586bc0a08695968acf2f169c9d01da69Ted Kremenek    (clang_isInvalid(K) || K == CXCursor_TranslationUnit)
4561d8b0a84d586bc0a08695968acf2f169c9d01da69Ted Kremenek     ? clang_getNullCursor() : parent;
4562fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4563fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  while (MoreTokens()) {
4564fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    const unsigned I = NextToken();
4565fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    SourceLocation TokLoc = GetTokenLoc(I);
4566fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
4567fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeBefore:
4568fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        Cursors[I] = updateC;
4569fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        AdvanceToken();
4570fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        continue;
4571fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeAfter:
4572fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeOverlap:
4573fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        break;
4574fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    }
4575fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    break;
4576fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4577fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4578fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Visit children to get their cursor information.
4579fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned BeforeChildren = NextToken();
4580fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  VisitChildren(cursor);
4581fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned AfterChildren = NextToken();
4582fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4583fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Adjust 'Last' to the last token within the extent of the cursor.
4584fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  while (MoreTokens()) {
4585fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    const unsigned I = NextToken();
4586fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    SourceLocation TokLoc = GetTokenLoc(I);
4587fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
4588fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeBefore:
4589fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        assert(0 && "Infeasible");
4590fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeAfter:
4591fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        break;
4592fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeOverlap:
4593fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        Cursors[I] = updateC;
4594fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        AdvanceToken();
4595fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        continue;
4596fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    }
4597fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    break;
4598fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4599fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned Last = NextToken();
46006db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
4601fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Scan the tokens that are at the beginning of the cursor, but are not
4602fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // capture by the child cursors.
4603fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4604fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // For AST elements within macros, rely on a post-annotate pass to
4605fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // to correctly annotate the tokens with cursors.  Otherwise we can
4606fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // get confusing results of having tokens that map to cursors that really
4607fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // are expanded by an instantiation.
4608fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (L.isMacroID())
4609fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    cursor = clang_getNullCursor();
4610fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4611fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = BeforeChildren; I != AfterChildren; ++I) {
4612fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    if (!clang_isInvalid(clang_getCursorKind(Cursors[I])))
4613fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      break;
46144419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
4615fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = cursor;
4616fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4617fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Scan the tokens that are at the end of the cursor, but are not captured
4618fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // but the child cursors.
4619fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = AfterChildren; I != Last; ++I)
4620fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = cursor;
4621fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4622fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  TokIdx = Last;
4623fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  return CXChildVisit_Continue;
46240045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor}
46250045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
46266db610934bedc6896393c1e1099525b35380acd6Ted Kremenekstatic enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
46276db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                                     CXCursor parent,
46286db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                                     CXClientData client_data) {
46296db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  return static_cast<AnnotateTokensWorker*>(client_data)->Visit(cursor, parent);
46306db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
46316db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
46326628a614c504263ae539462f049d523dd07ac1baTed Kremeneknamespace {
46336628a614c504263ae539462f049d523dd07ac1baTed Kremenek  struct clang_annotateTokens_Data {
46346628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXTranslationUnit TU;
46356628a614c504263ae539462f049d523dd07ac1baTed Kremenek    ASTUnit *CXXUnit;
46366628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXToken *Tokens;
46376628a614c504263ae539462f049d523dd07ac1baTed Kremenek    unsigned NumTokens;
46386628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXCursor *Cursors;
46396628a614c504263ae539462f049d523dd07ac1baTed Kremenek  };
4640ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek}
4641ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek
46426628a614c504263ae539462f049d523dd07ac1baTed Kremenek// This gets run a separate thread to avoid stack blowout.
46436628a614c504263ae539462f049d523dd07ac1baTed Kremenekstatic void clang_annotateTokensImpl(void *UserData) {
46446628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXTranslationUnit TU = ((clang_annotateTokens_Data*)UserData)->TU;
46456628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit *CXXUnit = ((clang_annotateTokens_Data*)UserData)->CXXUnit;
46466628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXToken *Tokens = ((clang_annotateTokens_Data*)UserData)->Tokens;
46476628a614c504263ae539462f049d523dd07ac1baTed Kremenek  const unsigned NumTokens = ((clang_annotateTokens_Data*)UserData)->NumTokens;
46486628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXCursor *Cursors = ((clang_annotateTokens_Data*)UserData)->Cursors;
4649fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
46500396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // Determine the region of interest, which contains all of the tokens.
46510045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  SourceRange RegionOfInterest;
46526628a614c504263ae539462f049d523dd07ac1baTed Kremenek  RegionOfInterest.setBegin(
46536628a614c504263ae539462f049d523dd07ac1baTed Kremenek    cxloc::translateSourceLocation(clang_getTokenLocation(TU, Tokens[0])));
46546628a614c504263ae539462f049d523dd07ac1baTed Kremenek  RegionOfInterest.setEnd(
46556628a614c504263ae539462f049d523dd07ac1baTed Kremenek    cxloc::translateSourceLocation(clang_getTokenLocation(TU,
46566628a614c504263ae539462f049d523dd07ac1baTed Kremenek                                                         Tokens[NumTokens-1])));
4657fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
46580396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // A mapping from the source locations found when re-lexing or traversing the
46590396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // region of interest to the corresponding cursors.
46600045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  AnnotateTokensData Annotated;
46616628a614c504263ae539462f049d523dd07ac1baTed Kremenek
4662fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Relex the tokens within the source range to look for preprocessing
46630396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // directives.
46649f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  SourceManager &SourceMgr = CXXUnit->getSourceManager();
46659f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  std::pair<FileID, unsigned> BeginLocInfo
46669f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    = SourceMgr.getDecomposedLoc(RegionOfInterest.getBegin());
46679f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  std::pair<FileID, unsigned> EndLocInfo
46689f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    = SourceMgr.getDecomposedLoc(RegionOfInterest.getEnd());
46696628a614c504263ae539462f049d523dd07ac1baTed Kremenek
46709f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  llvm::StringRef Buffer;
46710396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  bool Invalid = false;
46720396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  if (BeginLocInfo.first == EndLocInfo.first &&
46730396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor      ((Buffer = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid)),true) &&
46740396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor      !Invalid) {
46759f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
46769f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor              CXXUnit->getASTContext().getLangOptions(),
4677fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek              Buffer.begin(), Buffer.data() + BeginLocInfo.second,
46784ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor              Buffer.end());
46799f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    Lex.SetCommentRetentionState(true);
46806628a614c504263ae539462f049d523dd07ac1baTed Kremenek
4681fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    // Lex tokens in raw mode until we hit the end of the range, to avoid
46829f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    // entering #includes or expanding macros.
46834807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor    while (true) {
46849f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor      Token Tok;
46859f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor      Lex.LexFromRawLexer(Tok);
46866628a614c504263ae539462f049d523dd07ac1baTed Kremenek
46879f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    reprocess:
46889f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor      if (Tok.is(tok::hash) && Tok.isAtStartOfLine()) {
46899f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        // We have found a preprocessing directive. Gobble it up so that we
46909e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar        // don't see it while preprocessing these tokens later, but keep track
46919e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar        // of all of the token locations inside this preprocessing directive so
46929e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar        // that we can annotate them appropriately.
46939f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        //
46949f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        // FIXME: Some simple tests here could identify macro definitions and
46959f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        // #undefs, to provide specific cursor kinds for those.
46966628a614c504263ae539462f049d523dd07ac1baTed Kremenek        llvm::SmallVector<SourceLocation, 32> Locations;
46979f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        do {
46989f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor          Locations.push_back(Tok.getLocation());
4699fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek          Lex.LexFromRawLexer(Tok);
47009f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        } while (!Tok.isAtStartOfLine() && !Tok.is(tok::eof));
47016628a614c504263ae539462f049d523dd07ac1baTed Kremenek
47029f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        using namespace cxcursor;
47039f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        CXCursor Cursor
47046628a614c504263ae539462f049d523dd07ac1baTed Kremenek        = MakePreprocessingDirectiveCursor(SourceRange(Locations.front(),
47056628a614c504263ae539462f049d523dd07ac1baTed Kremenek                                                       Locations.back()),
4706a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                           TU);
47079f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        for (unsigned I = 0, N = Locations.size(); I != N; ++I) {
47089f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor          Annotated[Locations[I].getRawEncoding()] = Cursor;
47099f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        }
47106628a614c504263ae539462f049d523dd07ac1baTed Kremenek
47119f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        if (Tok.isAtStartOfLine())
47129f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor          goto reprocess;
47136628a614c504263ae539462f049d523dd07ac1baTed Kremenek
47149f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        continue;
47159f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor      }
47166628a614c504263ae539462f049d523dd07ac1baTed Kremenek
47174807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor      if (Tok.is(tok::eof))
47189f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        break;
47199f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    }
47204ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor  }
47216628a614c504263ae539462f049d523dd07ac1baTed Kremenek
47220396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // Annotate all of the source locations in the region of interest that map to
4723fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // a specific cursor.
4724fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  AnnotateTokensWorker W(Annotated, Tokens, Cursors, NumTokens,
4725a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                         TU, RegionOfInterest);
47266628a614c504263ae539462f049d523dd07ac1baTed Kremenek
47276c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // FIXME: We use a ridiculous stack size here because the data-recursion
47286c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // algorithm uses a large stack frame than the non-data recursive version,
47296c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // and AnnotationTokensWorker currently transforms the data-recursion
47306c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // algorithm back into a traditional recursion by explicitly calling
47316c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // VisitChildren().  We will need to remove this explicit recursive call.
47326628a614c504263ae539462f049d523dd07ac1baTed Kremenek  W.AnnotateTokens();
47336628a614c504263ae539462f049d523dd07ac1baTed Kremenek
4734f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  // If we ran into any entities that involve context-sensitive keywords,
4735f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  // take another pass through the tokens to mark them as such.
4736f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  if (W.hasContextSensitiveKeywords()) {
4737f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    for (unsigned I = 0; I != NumTokens; ++I) {
4738f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (clang_getTokenKind(Tokens[I]) != CXToken_Identifier)
4739f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
4740f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4741f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ObjCPropertyDecl) {
4742f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
4743f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (ObjCPropertyDecl *Property
47446628a614c504263ae539462f049d523dd07ac1baTed Kremenek            = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(Cursors[I]))) {
4745f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          if (Property->getPropertyAttributesAsWritten() != 0 &&
4746f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              llvm::StringSwitch<bool>(II->getName())
47476628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("readonly", true)
47486628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("assign", true)
47496628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("readwrite", true)
47506628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("retain", true)
47516628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("copy", true)
47526628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("nonatomic", true)
47536628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("atomic", true)
47546628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("getter", true)
47556628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("setter", true)
47566628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Default(false))
4757f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            Tokens[I].int_data[0] = CXToken_Keyword;
4758f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
4759f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
4760f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4761f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4762f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ObjCInstanceMethodDecl ||
4763f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Cursors[I].kind == CXCursor_ObjCClassMethodDecl) {
4764f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
4765f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (llvm::StringSwitch<bool>(II->getName())
47666628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("in", true)
47676628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("out", true)
47686628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("inout", true)
47696628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("oneway", true)
47706628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("bycopy", true)
47716628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("byref", true)
47726628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Default(false))
4773f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Tokens[I].int_data[0] = CXToken_Keyword;
4774f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
4775f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4776f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4777f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_CXXMethod) {
4778f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
4779f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (CXXMethodDecl *Method
47806628a614c504263ae539462f049d523dd07ac1baTed Kremenek            = dyn_cast_or_null<CXXMethodDecl>(getCursorDecl(Cursors[I]))) {
4781f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          if ((Method->hasAttr<FinalAttr>() ||
4782f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor               Method->hasAttr<OverrideAttr>()) &&
4783f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              Method->getLocation().getRawEncoding() != Tokens[I].int_data[1] &&
4784f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              llvm::StringSwitch<bool>(II->getName())
47856628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("final", true)
47866628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("override", true)
47876628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Default(false))
4788f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            Tokens[I].int_data[0] = CXToken_Keyword;
4789f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
4790f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
4791f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4792f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4793f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ClassDecl ||
4794f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Cursors[I].kind == CXCursor_StructDecl ||
4795f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Cursors[I].kind == CXCursor_ClassTemplate) {
4796f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
4797f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (II->getName() == "final") {
4798f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          // We have to be careful with 'final', since it could be the name
4799f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          // of a member class rather than the context-sensitive keyword.
4800f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          // So, check whether the cursor associated with this
4801f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Decl *D = getCursorDecl(Cursors[I]);
4802f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          if (CXXRecordDecl *Record = dyn_cast_or_null<CXXRecordDecl>(D)) {
4803f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            if ((Record->hasAttr<FinalAttr>()) &&
4804f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                Record->getIdentifier() != II)
4805f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              Tokens[I].int_data[0] = CXToken_Keyword;
4806f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          } else if (ClassTemplateDecl *ClassTemplate
48076628a614c504263ae539462f049d523dd07ac1baTed Kremenek                     = dyn_cast_or_null<ClassTemplateDecl>(D)) {
4808f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            CXXRecordDecl *Record = ClassTemplate->getTemplatedDecl();
4809f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            if ((Record->hasAttr<FinalAttr>()) &&
4810f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                Record->getIdentifier() != II)
48116628a614c504263ae539462f049d523dd07ac1baTed Kremenek              Tokens[I].int_data[0] = CXToken_Keyword;
4812f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          }
4813f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
48146628a614c504263ae539462f049d523dd07ac1baTed Kremenek        continue;
4815f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4816f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4817f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
4818fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
48196628a614c504263ae539462f049d523dd07ac1baTed Kremenek
48206628a614c504263ae539462f049d523dd07ac1baTed Kremenekextern "C" {
48216628a614c504263ae539462f049d523dd07ac1baTed Kremenek
48226628a614c504263ae539462f049d523dd07ac1baTed Kremenekvoid clang_annotateTokens(CXTranslationUnit TU,
48236628a614c504263ae539462f049d523dd07ac1baTed Kremenek                          CXToken *Tokens, unsigned NumTokens,
48246628a614c504263ae539462f049d523dd07ac1baTed Kremenek                          CXCursor *Cursors) {
48256628a614c504263ae539462f049d523dd07ac1baTed Kremenek
48266628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (NumTokens == 0 || !Tokens || !Cursors)
48276628a614c504263ae539462f049d523dd07ac1baTed Kremenek    return;
48286628a614c504263ae539462f049d523dd07ac1baTed Kremenek
48296628a614c504263ae539462f049d523dd07ac1baTed Kremenek  // Any token we don't specifically annotate will have a NULL cursor.
48306628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXCursor C = clang_getNullCursor();
48316628a614c504263ae539462f049d523dd07ac1baTed Kremenek  for (unsigned I = 0; I != NumTokens; ++I)
48326628a614c504263ae539462f049d523dd07ac1baTed Kremenek    Cursors[I] = C;
48336628a614c504263ae539462f049d523dd07ac1baTed Kremenek
48346628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
48356628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (!CXXUnit)
48366628a614c504263ae539462f049d523dd07ac1baTed Kremenek    return;
48376628a614c504263ae539462f049d523dd07ac1baTed Kremenek
48386628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
48396628a614c504263ae539462f049d523dd07ac1baTed Kremenek
48406628a614c504263ae539462f049d523dd07ac1baTed Kremenek  clang_annotateTokens_Data data = { TU, CXXUnit, Tokens, NumTokens, Cursors };
48416628a614c504263ae539462f049d523dd07ac1baTed Kremenek  llvm::CrashRecoveryContext CRC;
48426628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (!RunSafely(CRC, clang_annotateTokensImpl, &data,
48436628a614c504263ae539462f049d523dd07ac1baTed Kremenek                 GetSafetyThreadStackSize() * 2)) {
48446628a614c504263ae539462f049d523dd07ac1baTed Kremenek    fprintf(stderr, "libclang: crash detected while annotating tokens\n");
48456628a614c504263ae539462f049d523dd07ac1baTed Kremenek  }
48466628a614c504263ae539462f049d523dd07ac1baTed Kremenek}
48476628a614c504263ae539462f049d523dd07ac1baTed Kremenek
4848fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor} // end: extern "C"
4849fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4850fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor//===----------------------------------------------------------------------===//
485116b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek// Operations for querying linkage of a cursor.
485216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek//===----------------------------------------------------------------------===//
485316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
485416b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenekextern "C" {
485516b4259aecaa22b642d35d36fd89965ed700c1e0Ted KremenekCXLinkageKind clang_getCursorLinkage(CXCursor cursor) {
48560396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  if (!clang_isDeclaration(cursor.kind))
48570396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    return CXLinkage_Invalid;
48580396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor
485916b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  Decl *D = cxcursor::getCursorDecl(cursor);
486016b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  if (NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D))
486116b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek    switch (ND->getLinkage()) {
486216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case NoLinkage: return CXLinkage_NoLinkage;
486316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case InternalLinkage: return CXLinkage_Internal;
486416b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case UniqueExternalLinkage: return CXLinkage_UniqueExternal;
486516b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case ExternalLinkage: return CXLinkage_External;
486616b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek    };
486716b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
486816b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  return CXLinkage_Invalid;
486916b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek}
487016b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek} // end: extern "C"
487116b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
487216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek//===----------------------------------------------------------------------===//
487345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek// Operations for querying language of a cursor.
487445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek//===----------------------------------------------------------------------===//
487545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
487645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenekstatic CXLanguageKind getDeclLanguage(const Decl *D) {
487745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  switch (D->getKind()) {
487845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    default:
487945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      break;
488045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ImplicitParam:
488145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCAtDefsField:
488245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCategory:
488345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCategoryImpl:
488445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCClass:
488545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCompatibleAlias:
488645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCForwardProtocol:
488745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCImplementation:
488845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCInterface:
488945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCIvar:
489045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCMethod:
489145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCProperty:
489245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCPropertyImpl:
489345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCProtocol:
489445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      return CXLanguage_ObjC;
489545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXConstructor:
489645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXConversion:
489745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXDestructor:
489845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXMethod:
489945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXRecord:
490045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplate:
490145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplatePartialSpecialization:
490245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplateSpecialization:
490345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Friend:
490445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::FriendTemplate:
490545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::FunctionTemplate:
490645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::LinkageSpec:
490745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Namespace:
490845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::NamespaceAlias:
490945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::NonTypeTemplateParm:
491045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::StaticAssert:
491145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::TemplateTemplateParm:
491245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::TemplateTypeParm:
491345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UnresolvedUsingTypename:
491445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UnresolvedUsingValue:
491545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Using:
491645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UsingDirective:
491745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UsingShadow:
491845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      return CXLanguage_CPlusPlus;
491945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  }
492045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
492145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  return CXLanguage_C;
492245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek}
492345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
492445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenekextern "C" {
492558ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
492658ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregorenum CXAvailabilityKind clang_getCursorAvailability(CXCursor cursor) {
492758ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor  if (clang_isDeclaration(cursor.kind))
492858ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor    if (Decl *D = cxcursor::getCursorDecl(cursor)) {
49290a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      if (isa<FunctionDecl>(D) && cast<FunctionDecl>(D)->isDeleted())
493058ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor        return CXAvailability_Available;
493158ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
49320a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      switch (D->getAvailability()) {
49330a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Available:
49340a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_NotYetIntroduced:
49350a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor        return CXAvailability_Available;
49360a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
49370a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Deprecated:
493858ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor        return CXAvailability_Deprecated;
49390a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
49400a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Unavailable:
49410a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor        return CXAvailability_NotAvailable;
49420a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      }
494358ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor    }
49440a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
494558ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor  return CXAvailability_Available;
494658ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor}
494758ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
494845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted KremenekCXLanguageKind clang_getCursorLanguage(CXCursor cursor) {
494945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  if (clang_isDeclaration(cursor.kind))
495045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    return getDeclLanguage(cxcursor::getCursorDecl(cursor));
495145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
495245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  return CXLanguage_Invalid;
495345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek}
49543910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49553910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// \brief If the given cursor is the "templated" declaration
49563910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// descibing a class or function template, return the class or
49573910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// function template.
49583910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregorstatic Decl *maybeGetTemplateCursor(Decl *D) {
49593910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (!D)
49603910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    return 0;
49613910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49623910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
49633910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    if (FunctionTemplateDecl *FunTmpl = FD->getDescribedFunctionTemplate())
49643910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return FunTmpl;
49653910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49663910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(D))
49673910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    if (ClassTemplateDecl *ClassTmpl = RD->getDescribedClassTemplate())
49683910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return ClassTmpl;
49693910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49703910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  return D;
49713910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor}
49723910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49732be5bc9ad3981347a000742f81b91ab3080f1214Douglas GregorCXCursor clang_getCursorSemanticParent(CXCursor cursor) {
49742be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isDeclaration(cursor.kind)) {
49752be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor)) {
49762be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor      DeclContext *DC = D->getDeclContext();
49773910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      if (!DC)
49783910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor        return clang_getNullCursor();
49793910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
49803910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
49813910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor                          getCursorTU(cursor));
49822be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    }
49832be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
49842be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
49852be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isStatement(cursor.kind) || clang_isExpression(cursor.kind)) {
49862be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor))
4987a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(D, getCursorTU(cursor));
49882be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
49892be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
49902be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  return clang_getNullCursor();
49912be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor}
49922be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
49932be5bc9ad3981347a000742f81b91ab3080f1214Douglas GregorCXCursor clang_getCursorLexicalParent(CXCursor cursor) {
49942be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isDeclaration(cursor.kind)) {
49952be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor)) {
49962be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor      DeclContext *DC = D->getLexicalDeclContext();
49973910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      if (!DC)
49983910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor        return clang_getNullCursor();
49993910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
50003910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
50013910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor                          getCursorTU(cursor));
50022be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    }
50032be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
50042be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
50052be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  // FIXME: Note that we can't easily compute the lexical context of a
50062be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  // statement or expression, so we return nothing.
50072be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  return clang_getNullCursor();
50082be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor}
50092be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
50109f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregorstatic void CollectOverriddenMethods(DeclContext *Ctx,
50119f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                     ObjCMethodDecl *Method,
50129f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                            llvm::SmallVectorImpl<ObjCMethodDecl *> &Methods) {
50139f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!Ctx)
50149f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
50159f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50169f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  // If we have a class or category implementation, jump straight to the
50179f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  // interface.
50189f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (ObjCImplDecl *Impl = dyn_cast<ObjCImplDecl>(Ctx))
50199f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return CollectOverriddenMethods(Impl->getClassInterface(), Method, Methods);
50209f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50219f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  ObjCContainerDecl *Container = dyn_cast<ObjCContainerDecl>(Ctx);
50229f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!Container)
50239f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
50249f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50259f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  // Check whether we have a matching method at this level.
50269f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (ObjCMethodDecl *Overridden = Container->getMethod(Method->getSelector(),
50279f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                                    Method->isInstanceMethod()))
50289f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    if (Method != Overridden) {
50299f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      // We found an override at this level; there is no need to look
50309f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      // into other protocols or categories.
50319f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      Methods.push_back(Overridden);
50329f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      return;
50339f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    }
50349f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50359f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
50369f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    for (ObjCProtocolDecl::protocol_iterator P = Protocol->protocol_begin(),
50379f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                          PEnd = Protocol->protocol_end();
50389f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor         P != PEnd; ++P)
50399f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      CollectOverriddenMethods(*P, Method, Methods);
50409f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  }
50419f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50429f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(Container)) {
50439f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    for (ObjCCategoryDecl::protocol_iterator P = Category->protocol_begin(),
50449f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                          PEnd = Category->protocol_end();
50459f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor         P != PEnd; ++P)
50469f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      CollectOverriddenMethods(*P, Method, Methods);
50479f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  }
50489f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50499f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (ObjCInterfaceDecl *Interface = dyn_cast<ObjCInterfaceDecl>(Container)) {
50509f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    for (ObjCInterfaceDecl::protocol_iterator P = Interface->protocol_begin(),
50519f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                           PEnd = Interface->protocol_end();
50529f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor         P != PEnd; ++P)
50539f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      CollectOverriddenMethods(*P, Method, Methods);
50549f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50559f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    for (ObjCCategoryDecl *Category = Interface->getCategoryList();
50569f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor         Category; Category = Category->getNextClassCategory())
50579f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      CollectOverriddenMethods(Category, Method, Methods);
50589f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50599f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    // We only look into the superclass if we haven't found anything yet.
50609f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    if (Methods.empty())
50619f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      if (ObjCInterfaceDecl *Super = Interface->getSuperClass())
50629f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor        return CollectOverriddenMethods(Super, Method, Methods);
50639f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  }
50649f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor}
50659f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50669f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregorvoid clang_getOverriddenCursors(CXCursor cursor,
50679f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                CXCursor **overridden,
50689f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                unsigned *num_overridden) {
50699f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (overridden)
50709f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    *overridden = 0;
50719f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (num_overridden)
50729f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    *num_overridden = 0;
50739f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!overridden || !num_overridden)
50749f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
50759f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50769f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!clang_isDeclaration(cursor.kind))
50779f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
50789f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50799f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  Decl *D = getCursorDecl(cursor);
50809f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!D)
50819f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
50829f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50839f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  // Handle C++ member functions.
5084a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(cursor);
50859f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (CXXMethodDecl *CXXMethod = dyn_cast<CXXMethodDecl>(D)) {
50869f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    *num_overridden = CXXMethod->size_overridden_methods();
50879f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    if (!*num_overridden)
50889f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor      return;
50899f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
50909f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    *overridden = new CXCursor [*num_overridden];
50919f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    unsigned I = 0;
50929f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    for (CXXMethodDecl::method_iterator
50939f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor              M = CXXMethod->begin_overridden_methods(),
50949f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor           MEnd = CXXMethod->end_overridden_methods();
50959f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor         M != MEnd; (void)++M, ++I)
5096a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      (*overridden)[I] = MakeCXCursor(const_cast<CXXMethodDecl*>(*M), TU);
50979f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
50989f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  }
50999f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
51009f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(D);
51019f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!Method)
51029f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
51039f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
51049f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  // Handle Objective-C methods.
51059f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  llvm::SmallVector<ObjCMethodDecl *, 4> Methods;
51069f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  CollectOverriddenMethods(Method->getDeclContext(), Method, Methods);
51079f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
51089f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (Methods.empty())
51099f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
51109f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
51119f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  *num_overridden = Methods.size();
51129f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  *overridden = new CXCursor [Methods.size()];
51139f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  for (unsigned I = 0, N = Methods.size(); I != N; ++I)
5114a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    (*overridden)[I] = MakeCXCursor(Methods[I], TU);
51159f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor}
51169f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
51179f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregorvoid clang_disposeOverriddenCursors(CXCursor *overridden) {
51189f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  delete [] overridden;
51199f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor}
51209f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
5121ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas GregorCXFile clang_getIncludedFile(CXCursor cursor) {
5122ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (cursor.kind != CXCursor_InclusionDirective)
5123ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return 0;
5124ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
5125ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  InclusionDirective *ID = getCursorInclusionDirective(cursor);
5126ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  return (void *)ID->getFile();
5127ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor}
5128ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
512945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek} // end: extern "C"
513045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
51319ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
51329ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek//===----------------------------------------------------------------------===//
51339ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek// C++ AST instrospection.
51349ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek//===----------------------------------------------------------------------===//
51359ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
51369ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenekextern "C" {
51379ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenekunsigned clang_CXXMethod_isStatic(CXCursor C) {
51389ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek  if (!clang_isDeclaration(C.kind))
51399ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek    return 0;
514049f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor
514149f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  CXXMethodDecl *Method = 0;
514249f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  Decl *D = cxcursor::getCursorDecl(C);
514349f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast_or_null<FunctionTemplateDecl>(D))
514449f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor    Method = dyn_cast<CXXMethodDecl>(FunTmpl->getTemplatedDecl());
514549f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  else
514649f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor    Method = dyn_cast_or_null<CXXMethodDecl>(D);
514749f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  return (Method && Method->isStatic()) ? 1 : 0;
514840b492a43bac3ed0c465772aa6921d011cfc273fTed Kremenek}
5149b12903e1a4b8d1b611b8c7e4f910665d628e68cdTed Kremenek
51509ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek} // end: extern "C"
51519ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
515245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek//===----------------------------------------------------------------------===//
515395f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek// Attribute introspection.
515495f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek//===----------------------------------------------------------------------===//
515595f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
515695f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenekextern "C" {
515795f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted KremenekCXType clang_getIBOutletCollectionType(CXCursor C) {
515895f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek  if (C.kind != CXCursor_IBOutletCollectionAttr)
5159a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return cxtype::MakeCXType(QualType(), cxcursor::getCursorTU(C));
516095f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
516195f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek  IBOutletCollectionAttr *A =
516295f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek    cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(C));
516395f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
5164841b238087d6cdb21c2443b7429cb85bd1f9fce2Douglas Gregor  return cxtype::MakeCXType(A->getInterFace(), cxcursor::getCursorTU(C));
516595f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek}
516695f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek} // end: extern "C"
516795f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
516895f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek//===----------------------------------------------------------------------===//
516904bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek// Misc. utility functions.
517004bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek//===----------------------------------------------------------------------===//
5171f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
5172abdce7abc8a22dd2fe79a05c0b71864039bd8296Daniel Dunbar/// Default to using an 8 MB stack size on "safety" threads.
5173abdce7abc8a22dd2fe79a05c0b71864039bd8296Daniel Dunbarstatic unsigned SafetyStackThreadSize = 8 << 20;
5174bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5175bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarnamespace clang {
5176bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5177bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarbool RunSafely(llvm::CrashRecoveryContext &CRC,
51786c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek               void (*Fn)(void*), void *UserData,
51796c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek               unsigned Size) {
51806c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  if (!Size)
51816c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek    Size = GetSafetyThreadStackSize();
51826c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  if (Size)
5183bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar    return CRC.RunSafelyOnThread(Fn, UserData, Size);
5184bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  return CRC.RunSafely(Fn, UserData);
5185bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5186bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5187bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarunsigned GetSafetyThreadStackSize() {
5188bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  return SafetyStackThreadSize;
5189bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5190bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5191bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarvoid SetSafetyThreadStackSize(unsigned Value) {
5192bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  SafetyStackThreadSize = Value;
5193bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5194bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5195bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5196bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
519704bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenekextern "C" {
519804bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
5199a2a9d6e4e5b6001b86b7dfc5db1ea296ce29a3d3Ted KremenekCXString clang_getClangVersion() {
5200ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString(getClangFullVersion());
520104bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek}
520204bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
520304bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek} // end: extern "C"
5204