CIndex.cpp revision fbd84caf62a21afa4db5f730e2e2603ead17bb65
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"
17a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek#include "CXSourceLocation.h"
185352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor#include "CIndexDiagnostic.h"
19ab1889321f6f8f200f2b318ac26883ac18e49d03Ted Kremenek
2004bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek#include "clang/Basic/Version.h"
21936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor
2250398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff#include "clang/AST/DeclVisitor.h"
23fb5704295c6137685a7b90b92cd6b958028740c8Steve Naroff#include "clang/AST/StmtVisitor.h"
247d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor#include "clang/AST/TypeLocVisitor.h"
25b846debc1b22a37228efe4aa87b34482d15b6a3cBenjamin Kramer#include "clang/Basic/Diagnostic.h"
26b846debc1b22a37228efe4aa87b34482d15b6a3cBenjamin Kramer#include "clang/Frontend/ASTUnit.h"
27b846debc1b22a37228efe4aa87b34482d15b6a3cBenjamin Kramer#include "clang/Frontend/CompilerInstance.h"
28936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor#include "clang/Frontend/FrontendDiagnostic.h"
29d8210650ed948de65a08a8daf16d291b747717c4Ted Kremenek#include "clang/Lex/Lexer.h"
30b846debc1b22a37228efe4aa87b34482d15b6a3cBenjamin Kramer#include "clang/Lex/PreprocessingRecord.h"
3133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor#include "clang/Lex/Preprocessor.h"
3202465750c8c3fa96b1e7e596b02297e24361dc4fDouglas Gregor#include "llvm/Support/MemoryBuffer.h"
330829a839b05c687d3f11af0640d84c5d05f94124Benjamin Kramer#include "llvm/System/Program.h"
340a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor#include "llvm/System/Signals.h"
35fc0622155fa61349698a8fd0053773c37d9f7ac4Ted Kremenek
36c2a981614e324fea7f0a0533f8f1d103cbd17f6dBenjamin Kramer// Needed to define L_TMPNAM on some systems.
37c2a981614e324fea7f0a0533f8f1d103cbd17f6dBenjamin Kramer#include <cstdio>
38c2a981614e324fea7f0a0533f8f1d103cbd17f6dBenjamin Kramer
3950398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroffusing namespace clang;
4016c440a377b7ec8b722a2e2c7c864f75c95bd305Ted Kremenekusing namespace clang::cxcursor;
41ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenekusing namespace clang::cxstring;
4250398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff
438a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek//===----------------------------------------------------------------------===//
448a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek// Crash Reporting.
458a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek//===----------------------------------------------------------------------===//
468a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek
478a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek#ifdef __APPLE__
4829b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek#define USE_CRASHTRACER
498a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek#include "clang/Analysis/Support/SaveAndRestore.h"
508a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek// Integrate with crash reporter.
518a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenekextern "C" const char *__crashreporter_info__;
526b5699969d0dadaaee3f4eb51f18ce628fbfeca9Ted Kremenek#define NUM_CRASH_STRINGS 32
5329b7284c98253aed6e215aaa867c63084c80e944Ted Kremenekstatic unsigned crashtracer_counter = 0;
54254ba7cd3bc1b172c3f3c2b941aaae8335c25f85Ted Kremenekstatic unsigned crashtracer_counter_id[NUM_CRASH_STRINGS] = { 0 };
5529b7284c98253aed6e215aaa867c63084c80e944Ted Kremenekstatic const char *crashtracer_strings[NUM_CRASH_STRINGS] = { 0 };
5629b7284c98253aed6e215aaa867c63084c80e944Ted Kremenekstatic const char *agg_crashtracer_strings[NUM_CRASH_STRINGS] = { 0 };
5729b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek
5829b7284c98253aed6e215aaa867c63084c80e944Ted Kremenekstatic unsigned SetCrashTracerInfo(const char *str,
5929b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek                                   llvm::SmallString<1024> &AggStr) {
60f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
61254ba7cd3bc1b172c3f3c2b941aaae8335c25f85Ted Kremenek  unsigned slot = 0;
6229b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  while (crashtracer_strings[slot]) {
6329b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek    if (++slot == NUM_CRASH_STRINGS)
6429b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek      slot = 0;
6529b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  }
6629b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  crashtracer_strings[slot] = str;
67254ba7cd3bc1b172c3f3c2b941aaae8335c25f85Ted Kremenek  crashtracer_counter_id[slot] = ++crashtracer_counter;
6829b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek
6929b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  // We need to create an aggregate string because multiple threads
7029b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  // may be in this method at one time.  The crash reporter string
7129b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  // will attempt to overapproximate the set of in-flight invocations
7229b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  // of this function.  Race conditions can still cause this goal
7329b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  // to not be achieved.
7429b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  {
75f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    llvm::raw_svector_ostream Out(AggStr);
7629b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek    for (unsigned i = 0; i < NUM_CRASH_STRINGS; ++i)
7729b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek      if (crashtracer_strings[i]) Out << crashtracer_strings[i] << '\n';
7829b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  }
7929b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  __crashreporter_info__ = agg_crashtracer_strings[slot] =  AggStr.c_str();
8029b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  return slot;
8129b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek}
8229b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek
8329b7284c98253aed6e215aaa867c63084c80e944Ted Kremenekstatic void ResetCrashTracerInfo(unsigned slot) {
84254ba7cd3bc1b172c3f3c2b941aaae8335c25f85Ted Kremenek  unsigned max_slot = 0;
85254ba7cd3bc1b172c3f3c2b941aaae8335c25f85Ted Kremenek  unsigned max_value = 0;
86f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
87254ba7cd3bc1b172c3f3c2b941aaae8335c25f85Ted Kremenek  crashtracer_strings[slot] = agg_crashtracer_strings[slot] = 0;
88254ba7cd3bc1b172c3f3c2b941aaae8335c25f85Ted Kremenek
89254ba7cd3bc1b172c3f3c2b941aaae8335c25f85Ted Kremenek  for (unsigned i = 0 ; i < NUM_CRASH_STRINGS; ++i)
90254ba7cd3bc1b172c3f3c2b941aaae8335c25f85Ted Kremenek    if (agg_crashtracer_strings[i] &&
91254ba7cd3bc1b172c3f3c2b941aaae8335c25f85Ted Kremenek        crashtracer_counter_id[i] > max_value) {
92254ba7cd3bc1b172c3f3c2b941aaae8335c25f85Ted Kremenek      max_slot = i;
93254ba7cd3bc1b172c3f3c2b941aaae8335c25f85Ted Kremenek      max_value = crashtracer_counter_id[i];
9429b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek    }
95254ba7cd3bc1b172c3f3c2b941aaae8335c25f85Ted Kremenek
96254ba7cd3bc1b172c3f3c2b941aaae8335c25f85Ted Kremenek  __crashreporter_info__ = agg_crashtracer_strings[max_slot];
9729b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek}
9829b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek
9929b7284c98253aed6e215aaa867c63084c80e944Ted Kremeneknamespace {
10029b7284c98253aed6e215aaa867c63084c80e944Ted Kremenekclass ArgsCrashTracerInfo {
10129b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  llvm::SmallString<1024> CrashString;
10229b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  llvm::SmallString<1024> AggregateString;
10329b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  unsigned crashtracerSlot;
10429b7284c98253aed6e215aaa867c63084c80e944Ted Kremenekpublic:
10529b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  ArgsCrashTracerInfo(llvm::SmallVectorImpl<const char*> &Args)
10629b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek    : crashtracerSlot(0)
10729b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  {
10829b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek    {
10929b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek      llvm::raw_svector_ostream Out(CrashString);
1100baa95200b634cc55185489cea5d11b10edefb7dTed Kremenek      Out << "ClangCIndex [" << getClangFullVersion() << "]"
1110baa95200b634cc55185489cea5d11b10edefb7dTed Kremenek          << "[createTranslationUnitFromSourceFile]: clang";
11229b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek      for (llvm::SmallVectorImpl<const char*>::iterator I=Args.begin(),
11329b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek           E=Args.end(); I!=E; ++I)
11429b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek        Out << ' ' << *I;
11529b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek    }
11629b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek    crashtracerSlot = SetCrashTracerInfo(CrashString.c_str(),
11729b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek                                         AggregateString);
11829b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  }
119f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
12029b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  ~ArgsCrashTracerInfo() {
12129b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek    ResetCrashTracerInfo(crashtracerSlot);
12229b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek  }
12329b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek};
12429b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek}
12529b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek#endif
1268a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek
12733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// \brief The result of comparing two source ranges.
12833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregorenum RangeComparisonResult {
12933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief Either the ranges overlap or one of the ranges is invalid.
13033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  RangeOverlap,
131f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
13233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The first range ends before the second range starts.
13333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  RangeBefore,
134f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
13533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The first range starts after the second range ends.
13633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  RangeAfter
13733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor};
13833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
139f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek/// \brief Compare two source ranges to determine their relative position in
14033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// the translation unit.
141f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekstatic RangeComparisonResult RangeCompare(SourceManager &SM,
142f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                          SourceRange R1,
14333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor                                          SourceRange R2) {
14433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  assert(R1.isValid() && "First range is invalid?");
14533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  assert(R2.isValid() && "Second range is invalid?");
146d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar  if (R1.getEnd() == R2.getBegin() ||
147d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar      SM.isBeforeInTranslationUnit(R1.getEnd(), R2.getBegin()))
14833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    return RangeBefore;
149d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar  if (R2.getEnd() == R1.getBegin() ||
150d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar      SM.isBeforeInTranslationUnit(R2.getEnd(), R1.getBegin()))
15133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    return RangeAfter;
15233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  return RangeOverlap;
15333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
15433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
155fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek/// \brief Determine if a source location falls within, before, or after a
156fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek///   a given source range.
157fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekstatic RangeComparisonResult LocationCompare(SourceManager &SM,
158fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                             SourceLocation L, SourceRange R) {
159fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  assert(R.isValid() && "First range is invalid?");
160fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  assert(L.isValid() && "Second range is invalid?");
161fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (L == R.getBegin())
162fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeOverlap;
163fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (L == R.getEnd())
164fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeAfter;
165fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (SM.isBeforeInTranslationUnit(L, R.getBegin()))
166fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeBefore;
167fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (SM.isBeforeInTranslationUnit(R.getEnd(), L))
168fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeAfter;
169fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  return RangeOverlap;
170fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek}
171fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
17276dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// \brief Translate a Clang source range into a CIndex source range.
17376dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar///
17476dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// Clang internally represents ranges where the end location points to the
17576dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// start of the token at the end. However, for external clients it is more
17676dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// useful to have a CXSourceRange be a proper half-open interval. This routine
17776dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// does the appropriate translation.
178f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed KremenekCXSourceRange cxloc::translateSourceRange(const SourceManager &SM,
17976dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                                          const LangOptions &LangOpts,
18076dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                                          SourceRange R) {
18176dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  // We want the last character in this location, so we will adjust the
1826a5a23f8e7fb65e028c8092bc1d1a1d9dfe2e9bcDouglas Gregor  // location accordingly.
1836a5a23f8e7fb65e028c8092bc1d1a1d9dfe2e9bcDouglas Gregor  // FIXME: How do do this with a macro instantiation location?
18476dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  SourceLocation EndLoc = R.getEnd();
1856a5a23f8e7fb65e028c8092bc1d1a1d9dfe2e9bcDouglas Gregor  if (!EndLoc.isInvalid() && EndLoc.isFileID()) {
1866a5a23f8e7fb65e028c8092bc1d1a1d9dfe2e9bcDouglas Gregor    unsigned Length = Lexer::MeasureTokenLength(EndLoc, SM, LangOpts);
18776dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar    EndLoc = EndLoc.getFileLocWithOffset(Length);
18876dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  }
18976dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar
19076dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  CXSourceRange Result = { { (void *)&SM, (void *)&LangOpts },
19176dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                           R.getBegin().getRawEncoding(),
19276dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                           EndLoc.getRawEncoding() };
19376dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  return Result;
19476dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar}
1951db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
1968a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek//===----------------------------------------------------------------------===//
19733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor// Cursor visitor.
1988a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek//===----------------------------------------------------------------------===//
1998a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek
20089922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroffnamespace {
201f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
202b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor// Cursor visitor.
2037d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorclass CursorVisitor : public DeclVisitor<CursorVisitor, bool>,
204a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor                      public TypeLocVisitor<CursorVisitor, bool>,
205a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor                      public StmtVisitor<CursorVisitor, bool>
2067d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor{
20733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The translation unit we are traversing.
208b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  ASTUnit *TU;
209f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
21033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The parent cursor whose children we are traversing.
211b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  CXCursor Parent;
212f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
21333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The declaration that serves at the parent of any statement or
21433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// expression nodes.
215f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  Decl *StmtParent;
216f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
21733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The visitor function.
218b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  CXCursorVisitor Visitor;
219f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
22033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The opaque client data, to be passed along to the visitor.
221b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  CXClientData ClientData;
222f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2237d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  // MaxPCHLevel - the maximum PCH level of declarations that we will pass on
2247d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  // to the visitor. Declarations with a PCH level greater than this value will
2257d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  // be suppressed.
2267d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  unsigned MaxPCHLevel;
22733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
22833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief When valid, a source range to which the cursor should restrict
22933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// its search.
23033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  SourceRange RegionOfInterest;
231f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
232b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  using DeclVisitor<CursorVisitor, bool>::Visit;
2337d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  using TypeLocVisitor<CursorVisitor, bool>::Visit;
234a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  using StmtVisitor<CursorVisitor, bool>::Visit;
235f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
236f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  /// \brief Determine whether this particular source range comes before, comes
237f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  /// after, or overlaps the region of interest.
23833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  ///
239d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar  /// \param R a half-open source range retrieved from the abstract syntax tree.
240f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  RangeComparisonResult CompareRegionOfInterest(SourceRange R);
241f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
242b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorpublic:
243f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  CursorVisitor(ASTUnit *TU, CXCursorVisitor Visitor, CXClientData ClientData,
244f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                unsigned MaxPCHLevel,
24533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor                SourceRange RegionOfInterest = SourceRange())
246f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    : TU(TU), Visitor(Visitor), ClientData(ClientData),
24733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor      MaxPCHLevel(MaxPCHLevel), RegionOfInterest(RegionOfInterest)
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
25633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  bool Visit(CXCursor Cursor, bool CheckedRegionOfInterest = false);
257788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
258788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  std::pair<PreprocessingRecord::iterator, PreprocessingRecord::iterator>
259788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    getPreprocessedEntities();
260788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
261b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  bool VisitChildren(CXCursor Parent);
262f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2637d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  // Declaration visitors
26409dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek  bool VisitAttributes(Decl *D);
2651ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  bool VisitBlockDecl(BlockDecl *B);
266b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  bool VisitDeclContext(DeclContext *DC);
26779758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitTranslationUnitDecl(TranslationUnitDecl *D);
26879758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitTypedefDecl(TypedefDecl *D);
26979758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitTagDecl(TagDecl *D);
2704540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitEnumConstantDecl(EnumConstantDecl *D);
27179758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitDeclaratorDecl(DeclaratorDecl *DD);
2724540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitFunctionDecl(FunctionDecl *ND);
27379758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitFieldDecl(FieldDecl *D);
27479758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitVarDecl(VarDecl *);
27579758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCMethodDecl(ObjCMethodDecl *ND);
2764540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitObjCContainerDecl(ObjCContainerDecl *D);
27779758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCCategoryDecl(ObjCCategoryDecl *ND);
27879758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCProtocolDecl(ObjCProtocolDecl *PID);
27979758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCInterfaceDecl(ObjCInterfaceDecl *D);
2804540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitObjCImplDecl(ObjCImplDecl *D);
28179758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D);
2821ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  bool VisitObjCImplementationDecl(ObjCImplementationDecl *D);
2831ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // FIXME: ObjCPropertyDecl requires TypeSourceInfo, getter/setter locations,
2841ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // etc.
28579758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  // FIXME: ObjCCompatibleAliasDecl requires aliased-class locations.
28679758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCForwardProtocolDecl(ObjCForwardProtocolDecl *D);
28779758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCClassDecl(ObjCClassDecl *D);
2887d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
2897d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  // Type visitors
290f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // FIXME: QualifiedTypeLoc doesn't provide any location information
291f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitBuiltinTypeLoc(BuiltinTypeLoc TL);
2927d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  bool VisitTypedefTypeLoc(TypedefTypeLoc TL);
293f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitUnresolvedUsingTypeLoc(UnresolvedUsingTypeLoc TL);
294f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitTagTypeLoc(TagTypeLoc TL);
295f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // FIXME: TemplateTypeParmTypeLoc doesn't provide any location information
296f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL);
297f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitObjCObjectPointerTypeLoc(ObjCObjectPointerTypeLoc TL);
298f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitPointerTypeLoc(PointerTypeLoc TL);
299f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitBlockPointerTypeLoc(BlockPointerTypeLoc TL);
300f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitMemberPointerTypeLoc(MemberPointerTypeLoc TL);
301f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitLValueReferenceTypeLoc(LValueReferenceTypeLoc TL);
302f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitRValueReferenceTypeLoc(RValueReferenceTypeLoc TL);
303f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitFunctionTypeLoc(FunctionTypeLoc TL);
304f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitArrayTypeLoc(ArrayTypeLoc TL);
3052332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  // FIXME: Implement for TemplateSpecializationTypeLoc
3062332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  // FIXME: Implement visitors here when the unimplemented TypeLocs get
3072332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  // implemented
3082332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  bool VisitTypeOfExprTypeLoc(TypeOfExprTypeLoc TL);
3092332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  bool VisitTypeOfTypeLoc(TypeOfTypeLoc TL);
310f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
311a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  // Statement visitors
312a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  bool VisitStmt(Stmt *S);
313a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  bool VisitDeclStmt(DeclStmt *S);
314f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor  // FIXME: LabelStmt label?
315f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor  bool VisitIfStmt(IfStmt *S);
316f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor  bool VisitSwitchStmt(SwitchStmt *S);
317263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor  bool VisitWhileStmt(WhileStmt *S);
318263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor  bool VisitForStmt(ForStmt *S);
319f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
320336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor  // Expression visitors
3211ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  bool VisitBlockExpr(BlockExpr *B);
322336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor  bool VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
3231ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  bool VisitExplicitCastExpr(ExplicitCastExpr *E);
324c2350e553b853ad00914faf23fa731e5fc4a8a5cDouglas Gregor  bool VisitObjCMessageExpr(ObjCMessageExpr *E);
32581d3466d037dc5844234c7a93dab21a6ad986e7dDouglas Gregor  bool VisitObjCEncodeExpr(ObjCEncodeExpr *E);
3268ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  bool VisitOffsetOfExpr(OffsetOfExpr *E);
3271ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  bool VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E);
328b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor};
329f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
330b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor} // end anonymous namespace
3310d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
33233e9abd21083a0191a7676a04b497006d2da184dDouglas GregorRangeComparisonResult CursorVisitor::CompareRegionOfInterest(SourceRange R) {
33333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  return RangeCompare(TU->getSourceManager(), R, RegionOfInterest);
33433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
33533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
336b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \brief Visit the given cursor and, if requested by the visitor,
337b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// its children.
338b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor///
33933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// \param Cursor the cursor to visit.
34033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor///
34133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// \param CheckRegionOfInterest if true, then the caller already checked that
34233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// this cursor is within the region of interest.
34333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor///
344b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \returns true if the visitation should be aborted, false if it
345b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// should continue.
34633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregorbool CursorVisitor::Visit(CXCursor Cursor, bool CheckedRegionOfInterest) {
347b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isInvalid(Cursor.kind))
348b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return false;
349f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
350b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isDeclaration(Cursor.kind)) {
351b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Decl *D = getCursorDecl(Cursor);
352b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    assert(D && "Invalid declaration cursor");
353b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    if (D->getPCHLevel() > MaxPCHLevel)
354b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return false;
355b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor
356b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    if (D->isImplicit())
357b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return false;
358b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
3590d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
36033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  // If we have a range of interest, and this cursor doesn't intersect with it,
36133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  // we're done.
36233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (RegionOfInterest.isValid() && !CheckedRegionOfInterest) {
363f408f32aa9ae3d97bc656267dc5d78fa7d03499bDaniel Dunbar    SourceRange Range =
364f408f32aa9ae3d97bc656267dc5d78fa7d03499bDaniel Dunbar      cxloc::translateCXSourceRange(clang_getCursorExtent(Cursor));
365f408f32aa9ae3d97bc656267dc5d78fa7d03499bDaniel Dunbar    if (Range.isInvalid() || CompareRegionOfInterest(Range))
36633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor      return false;
36733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  }
368f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
369b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  switch (Visitor(Cursor, Parent, ClientData)) {
370b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Break:
371b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
3720d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
373b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Continue:
374b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return false;
3752e331b938b38057e333fab0ba841130ea8467794Douglas Gregor
376b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Recurse:
377b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return VisitChildren(Cursor);
378b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
3790d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
380fd64377225a6a140bddb3f997d52a036486f9360Douglas Gregor  return false;
381b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
3820d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
383788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregorstd::pair<PreprocessingRecord::iterator, PreprocessingRecord::iterator>
384788f5a1242c04762f91eaa7565c07b9865846d88Douglas GregorCursorVisitor::getPreprocessedEntities() {
385788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  PreprocessingRecord &PPRec
386788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    = *TU->getPreprocessor().getPreprocessingRecord();
387788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
388788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  bool OnlyLocalDecls
389788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    = !TU->isMainFileAST() && TU->getOnlyLocalDecls();
390788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
391788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  // There is no region of interest; we have to walk everything.
392788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  if (RegionOfInterest.isInvalid())
393788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    return std::make_pair(PPRec.begin(OnlyLocalDecls),
394788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor                          PPRec.end(OnlyLocalDecls));
395788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
396788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  // Find the file in which the region of interest lands.
397788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  SourceManager &SM = TU->getSourceManager();
398788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  std::pair<FileID, unsigned> Begin
399788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    = SM.getDecomposedInstantiationLoc(RegionOfInterest.getBegin());
400788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  std::pair<FileID, unsigned> End
401788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    = SM.getDecomposedInstantiationLoc(RegionOfInterest.getEnd());
402788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
403788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  // The region of interest spans files; we have to walk everything.
404788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  if (Begin.first != End.first)
405788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    return std::make_pair(PPRec.begin(OnlyLocalDecls),
406788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor                          PPRec.end(OnlyLocalDecls));
407788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
408788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  ASTUnit::PreprocessedEntitiesByFileMap &ByFileMap
409788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    = TU->getPreprocessedEntitiesByFile();
410788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  if (ByFileMap.empty()) {
411788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    // Build the mapping from files to sets of preprocessed entities.
412788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    for (PreprocessingRecord::iterator E = PPRec.begin(OnlyLocalDecls),
413788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor                                    EEnd = PPRec.end(OnlyLocalDecls);
414788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor         E != EEnd; ++E) {
415788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor      std::pair<FileID, unsigned> P
416788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor        = SM.getDecomposedInstantiationLoc((*E)->getSourceRange().getBegin());
417788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor      ByFileMap[P.first].push_back(*E);
418788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    }
419788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  }
420788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
421788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  return std::make_pair(ByFileMap[Begin.first].begin(),
422788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor                        ByFileMap[Begin.first].end());
423788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor}
424788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
425b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \brief Visit the children of the given cursor.
426b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor///
427b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \returns true if the visitation should be aborted, false if it
428b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// should continue.
429f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekbool CursorVisitor::VisitChildren(CXCursor Cursor) {
430a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  if (clang_isReference(Cursor.kind)) {
431a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    // By definition, references have no children.
432a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return false;
433a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  }
434f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
435f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // Set the Parent field to Cursor, then back to its old value once we're
436b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  // done.
437b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  class SetParentRAII {
438b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    CXCursor &Parent;
439f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    Decl *&StmtParent;
440b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    CXCursor OldParent;
441f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
442b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  public:
443f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    SetParentRAII(CXCursor &Parent, Decl *&StmtParent, CXCursor NewParent)
444f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek      : Parent(Parent), StmtParent(StmtParent), OldParent(Parent)
445b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    {
446b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      Parent = NewParent;
447f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor      if (clang_isDeclaration(Parent.kind))
448f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor        StmtParent = getCursorDecl(Parent);
449b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    }
450f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
451b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    ~SetParentRAII() {
452b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      Parent = OldParent;
453f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor      if (clang_isDeclaration(Parent.kind))
454f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor        StmtParent = getCursorDecl(Parent);
455b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    }
456f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  } SetParent(Parent, StmtParent, Cursor);
457f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
458b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isDeclaration(Cursor.kind)) {
459b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Decl *D = getCursorDecl(Cursor);
460b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    assert(D && "Invalid declaration cursor");
461539311e0221df256c70c1c3080c8af847cd29dffTed Kremenek    return VisitAttributes(D) || Visit(D);
462b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
463f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
464a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  if (clang_isStatement(Cursor.kind))
465a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return Visit(getCursorStmt(Cursor));
466a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  if (clang_isExpression(Cursor.kind))
467a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return Visit(getCursorExpr(Cursor));
468f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
469b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isTranslationUnit(Cursor.kind)) {
470b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    ASTUnit *CXXUnit = getCursorASTUnit(Cursor);
47133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    if (!CXXUnit->isMainFileAST() && CXXUnit->getOnlyLocalDecls() &&
47233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor        RegionOfInterest.isInvalid()) {
4737b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor      const std::vector<Decl*> &TLDs = CXXUnit->getTopLevelDecls();
4747b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor      for (std::vector<Decl*>::const_iterator it = TLDs.begin(),
4757b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor           ie = TLDs.end(); it != ie; ++it) {
47633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor        if (Visit(MakeCXCursor(*it, CXXUnit), true))
4777b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor          return true;
4787b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor      }
4790396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    } else if (VisitDeclContext(
4800396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor                            CXXUnit->getASTContext().getTranslationUnitDecl()))
4810396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor      return true;
4823178cb674ac8c3b59e1791e14d38d48619a1b621Bob Wilson
4830396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    // Walk the preprocessing record.
4848de30ffa2b60dbc672c0f3bcdaa07a8efcc3bcf8Daniel Dunbar    if (CXXUnit->getPreprocessor().getPreprocessingRecord()) {
4850396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor      // FIXME: Once we have the ability to deserialize a preprocessing record,
4860396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor      // do so.
487788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor      PreprocessingRecord::iterator E, EEnd;
488788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor      for (llvm::tie(E, EEnd) = getPreprocessedEntities(); E != EEnd; ++E) {
4890396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor        if (MacroInstantiation *MI = dyn_cast<MacroInstantiation>(*E)) {
4900396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor          if (Visit(MakeMacroInstantiationCursor(MI, CXXUnit)))
4910396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor            return true;
492788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
4930396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor          continue;
4940396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor        }
4950396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor
4960396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor        if (MacroDefinition *MD = dyn_cast<MacroDefinition>(*E)) {
4970396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor          if (Visit(MakeMacroDefinitionCursor(MD, CXXUnit)))
4980396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor            return true;
4990396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor
5000396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor          continue;
5010396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor        }
5020396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor      }
5030396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    }
5047b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor    return false;
505b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
506f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
507b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  // Nothing to visit at the moment.
508b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
509dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
510dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
5111ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenekbool CursorVisitor::VisitBlockDecl(BlockDecl *B) {
5121ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  for (BlockDecl::param_iterator I=B->param_begin(), E=B->param_end(); I!=E;++I)
5131ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek    if (Decl *D = *I)
5141ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek      if (Visit(D))
5151ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek        return true;
5161ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
5171ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  return Visit(MakeCXCursor(B->getBody(), StmtParent, TU));
5181ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek}
5191ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
520b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitDeclContext(DeclContext *DC) {
521b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  for (DeclContext::decl_iterator
522b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor       I = DC->decls_begin(), E = DC->decls_end(); I != E; ++I) {
52309dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
524d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar    CXCursor Cursor = MakeCXCursor(*I, TU);
525d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar
52633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    if (RegionOfInterest.isValid()) {
527d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar      SourceRange Range =
528d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar        cxloc::translateCXSourceRange(clang_getCursorExtent(Cursor));
529d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar      if (Range.isInvalid())
53033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor        continue;
531d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar
532d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar      switch (CompareRegionOfInterest(Range)) {
53333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor      case RangeBefore:
53433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor        // This declaration comes before the region of interest; skip it.
53533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor        continue;
536f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
53733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor      case RangeAfter:
53833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor        // This declaration comes after the region of interest; we're done.
53933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor        return false;
540f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
54133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor      case RangeOverlap:
54233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor        // This declaration overlaps the region of interest; visit it.
54333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor        break;
544f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek      }
54533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    }
546f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
547d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar    if (Visit(Cursor, true))
548b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
549b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
550f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
551b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
552dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
553dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
5541ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTranslationUnitDecl(TranslationUnitDecl *D) {
5551ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  llvm_unreachable("Translation units are visited directly by Visit()");
5561ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
5571ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
5581ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
5591ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTypedefDecl(TypedefDecl *D) {
5601ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
5611ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(TSInfo->getTypeLoc());
562f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
5631ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
5641ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
5651ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
5661ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTagDecl(TagDecl *D) {
5671ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitDeclContext(D);
5681ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
5691ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
5701ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitEnumConstantDecl(EnumConstantDecl *D) {
5711ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *Init = D->getInitExpr())
5721ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(MakeCXCursor(Init, StmtParent, TU));
5731ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
5741ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
5751ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
5767d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorbool CursorVisitor::VisitDeclaratorDecl(DeclaratorDecl *DD) {
5777d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  if (TypeSourceInfo *TSInfo = DD->getTypeSourceInfo())
5787d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    if (Visit(TSInfo->getTypeLoc()))
5797d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      return true;
5807d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
5817d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  return false;
5827d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor}
5837d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
584b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitFunctionDecl(FunctionDecl *ND) {
585f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (VisitDeclaratorDecl(ND))
586f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
587f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
588a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  if (ND->isThisDeclarationADefinition() &&
589a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor      Visit(MakeCXCursor(ND->getBody(), StmtParent, TU)))
590a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return true;
591f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
592b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
593b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
594dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
5951ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitFieldDecl(FieldDecl *D) {
5961ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (VisitDeclaratorDecl(D))
5971ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
598f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
5991ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *BitWidth = D->getBitWidth())
6001ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(MakeCXCursor(BitWidth, StmtParent, TU));
601f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
6021ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
6031ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6041ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6051ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitVarDecl(VarDecl *D) {
6061ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (VisitDeclaratorDecl(D))
6071ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
608f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
6091ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *Init = D->getInit())
6101ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(MakeCXCursor(Init, StmtParent, TU));
611f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
6121ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
6131ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6141ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6151ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCMethodDecl(ObjCMethodDecl *ND) {
6164bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor  if (TypeSourceInfo *TSInfo = ND->getResultTypeSourceInfo())
6174bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor    if (Visit(TSInfo->getTypeLoc()))
6184bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor      return true;
6194bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor
620f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  for (ObjCMethodDecl::param_iterator P = ND->param_begin(),
6211ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       PEnd = ND->param_end();
6221ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       P != PEnd; ++P) {
6231ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    if (Visit(MakeCXCursor(*P, TU)))
6241ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
6251ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  }
626f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
6271ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (ND->isThisDeclarationADefinition() &&
6281ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      Visit(MakeCXCursor(ND->getBody(), StmtParent, TU)))
6291ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
630f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
6311ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
6321ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6331ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
634a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregorbool CursorVisitor::VisitObjCContainerDecl(ObjCContainerDecl *D) {
635a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitDeclContext(D);
636a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor}
637a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor
638b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitObjCCategoryDecl(ObjCCategoryDecl *ND) {
639b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (Visit(MakeCursorObjCClassRef(ND->getClassInterface(), ND->getLocation(),
640b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                   TU)))
641b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
642f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
64378db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  ObjCCategoryDecl::protocol_loc_iterator PL = ND->protocol_loc_begin();
64478db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  for (ObjCCategoryDecl::protocol_iterator I = ND->protocol_begin(),
64578db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor         E = ND->protocol_end(); I != E; ++I, ++PL)
646b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
647b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
648f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
649a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitObjCContainerDecl(ND);
650dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
651dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
6521ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCProtocolDecl(ObjCProtocolDecl *PID) {
6531ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  ObjCProtocolDecl::protocol_loc_iterator PL = PID->protocol_loc_begin();
6541ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCProtocolDecl::protocol_iterator I = PID->protocol_begin(),
6551ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       E = PID->protocol_end(); I != E; ++I, ++PL)
6561ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
6571ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
658f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
6591ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCContainerDecl(PID);
6601ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6611ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
662b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitObjCInterfaceDecl(ObjCInterfaceDecl *D) {
663dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek  // Issue callbacks for super class.
664b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (D->getSuperClass() &&
665b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
666f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        D->getSuperClassLoc(),
667b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                        TU)))
668b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
669f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
67078db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  ObjCInterfaceDecl::protocol_loc_iterator PL = D->protocol_loc_begin();
67178db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  for (ObjCInterfaceDecl::protocol_iterator I = D->protocol_begin(),
67278db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor         E = D->protocol_end(); I != E; ++I, ++PL)
673b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
674b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
675f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
676a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitObjCContainerDecl(D);
677dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
678dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
6791ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCImplDecl(ObjCImplDecl *D) {
6801ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCContainerDecl(D);
6811ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6821ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6831ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D) {
684ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  // 'ID' could be null when dealing with invalid code.
685ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  if (ObjCInterfaceDecl *ID = D->getClassInterface())
686ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek    if (Visit(MakeCursorObjCClassRef(ID, D->getLocation(), TU)))
687ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek      return true;
688f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
6891ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCImplDecl(D);
6901ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6911ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6921ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCImplementationDecl(ObjCImplementationDecl *D) {
6931ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor#if 0
6941ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // Issue callbacks for super class.
6951ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // FIXME: No source location information!
6961ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (D->getSuperClass() &&
6971ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
698f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        D->getSuperClassLoc(),
6991ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor                                        TU)))
700a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return true;
7011ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor#endif
702f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
7031ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCImplDecl(D);
704dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
705dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
7061ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCForwardProtocolDecl(ObjCForwardProtocolDecl *D) {
7071ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  ObjCForwardProtocolDecl::protocol_loc_iterator PL = D->protocol_loc_begin();
7081ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCForwardProtocolDecl::protocol_iterator I = D->protocol_begin(),
7091ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor                                                  E = D->protocol_end();
7101ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       I != E; ++I, ++PL)
711b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
712b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
713f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
714f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return false;
715dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
716dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
7171ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCClassDecl(ObjCClassDecl *D) {
7181ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCClassDecl::iterator C = D->begin(), CEnd = D->end(); C != CEnd; ++C)
7191ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    if (Visit(MakeCursorObjCClassRef(C->getInterface(), C->getLocation(), TU)))
7201ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
721f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
7221ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
723dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
7245e4bc590b0ea010e38372d0b4a0aab578a746fe6Benjamin Kramer
725f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitBuiltinTypeLoc(BuiltinTypeLoc TL) {
726f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  ASTContext &Context = TU->getASTContext();
727f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
728f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // Some builtin types (such as Objective-C's "id", "sel", and
729f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // "Class") have associated declarations. Create cursors for those.
730f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  QualType VisitType;
731f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  switch (TL.getType()->getAs<BuiltinType>()->getKind()) {
7326b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Void:
733f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Bool:
7346b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Char_U:
7356b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UChar:
736f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Char16:
737f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Char32:
7386b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UShort:
7396b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UInt:
7406b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ULong:
7416b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ULongLong:
7426b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UInt128:
743f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Char_S:
7446b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::SChar:
7456b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::WChar:
7466b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Short:
747f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Int:
748f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Long:
749c4174cc4b9b657abb77d0825de473ea29cf48297Ted Kremenek  case BuiltinType::LongLong:
7506b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Int128:
7516b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Float:
7526b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Double:
7536b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::LongDouble:
754f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::NullPtr:
755f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Overload:
7566b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Dependent:
757f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
7586b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
7596b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UndeducedAuto: // FIXME: Deserves a cursor?
760f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
7616b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
762f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::ObjCId:
763f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    VisitType = Context.getObjCIdType();
764f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
7656b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
7666b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ObjCClass:
7676b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek    VisitType = Context.getObjCClassType();
7686b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek    break;
7696b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
770f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::ObjCSel:
771f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    VisitType = Context.getObjCSelType();
772f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
773f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
774f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
775f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (!VisitType.isNull()) {
776f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    if (const TypedefType *Typedef = VisitType->getAs<TypedefType>())
777f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek      return Visit(MakeCursorTypeRef(Typedef->getDecl(), TL.getBuiltinLoc(),
778f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                     TU));
779f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
780f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
781f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
782f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
783f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
7847d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorbool CursorVisitor::VisitTypedefTypeLoc(TypedefTypeLoc TL) {
7857d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getTypedefDecl(), TL.getNameLoc(), TU));
7867d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor}
7877d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
788f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitUnresolvedUsingTypeLoc(UnresolvedUsingTypeLoc TL) {
789f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
790f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
791f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
792f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitTagTypeLoc(TagTypeLoc TL) {
793f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
794f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
795f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
796f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL) {
797f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(MakeCursorObjCClassRef(TL.getIFaceDecl(), TL.getNameLoc(), TU)))
798f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
799f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
800f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  for (unsigned I = 0, N = TL.getNumProtocols(); I != N; ++I) {
801f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    if (Visit(MakeCursorObjCProtocolRef(TL.getProtocol(I), TL.getProtocolLoc(I),
802f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                        TU)))
803f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor      return true;
804f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
805f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
806f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
807f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
808f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
809f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitObjCObjectPointerTypeLoc(ObjCObjectPointerTypeLoc TL) {
810f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (TL.hasBaseTypeAsWritten() && Visit(TL.getBaseTypeLoc()))
811f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
812f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
813f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (TL.hasProtocolsAsWritten()) {
814f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    for (unsigned I = 0, N = TL.getNumProtocols(); I != N; ++I) {
815f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek      if (Visit(MakeCursorObjCProtocolRef(TL.getProtocol(I),
816f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                          TL.getProtocolLoc(I),
817f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                          TU)))
818f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor        return true;
819f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    }
820f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
821f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
822f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
823f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
824f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
825f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitPointerTypeLoc(PointerTypeLoc TL) {
826f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
827f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
828f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
829f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitBlockPointerTypeLoc(BlockPointerTypeLoc TL) {
830f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
831f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
832f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
833f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitMemberPointerTypeLoc(MemberPointerTypeLoc TL) {
834f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
835f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
836f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
837f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitLValueReferenceTypeLoc(LValueReferenceTypeLoc TL) {
838f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return Visit(TL.getPointeeLoc());
839f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
840f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
841f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitRValueReferenceTypeLoc(RValueReferenceTypeLoc TL) {
842f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return Visit(TL.getPointeeLoc());
843f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
844f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
845f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitFunctionTypeLoc(FunctionTypeLoc TL) {
846f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(TL.getResultLoc()))
847f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
848f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
849f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
8505dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek    if (Decl *D = TL.getArg(I))
8515dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek      if (Visit(MakeCXCursor(D, TU)))
8525dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek        return true;
853f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
854f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
855f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
856f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
857f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitArrayTypeLoc(ArrayTypeLoc TL) {
858f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(TL.getElementLoc()))
859f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
860f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
861f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Expr *Size = TL.getSizeExpr())
862f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return Visit(MakeCXCursor(Size, StmtParent, TU));
863f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
864f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
865f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
866f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
8672332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregorbool CursorVisitor::VisitTypeOfExprTypeLoc(TypeOfExprTypeLoc TL) {
8682332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  return Visit(MakeCXCursor(TL.getUnderlyingExpr(), StmtParent, TU));
8692332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor}
8702332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
8712332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregorbool CursorVisitor::VisitTypeOfTypeLoc(TypeOfTypeLoc TL) {
8722332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
8732332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor    return Visit(TSInfo->getTypeLoc());
8742332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
8752332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  return false;
8762332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor}
8772332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
878a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregorbool CursorVisitor::VisitStmt(Stmt *S) {
879a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  for (Stmt::child_iterator Child = S->child_begin(), ChildEnd = S->child_end();
880a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor       Child != ChildEnd; ++Child) {
88154d67caeee4fdce81f07163832f1163d5f2af5d2Daniel Dunbar    if (*Child && Visit(MakeCXCursor(*Child, StmtParent, TU)))
882a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor      return true;
883a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  }
884f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
885a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return false;
886a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor}
887a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor
888a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregorbool CursorVisitor::VisitDeclStmt(DeclStmt *S) {
889a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  for (DeclStmt::decl_iterator D = S->decl_begin(), DEnd = S->decl_end();
890a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor       D != DEnd; ++D) {
891263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor    if (*D && Visit(MakeCXCursor(*D, TU)))
892a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor      return true;
893a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  }
894f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
895a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return false;
896a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor}
897a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor
898f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregorbool CursorVisitor::VisitIfStmt(IfStmt *S) {
899f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor  if (VarDecl *Var = S->getConditionVariable()) {
900f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor    if (Visit(MakeCXCursor(Var, TU)))
901f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor      return true;
902f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  }
903f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
904263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor  if (S->getCond() && Visit(MakeCXCursor(S->getCond(), StmtParent, TU)))
905263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor    return true;
906f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor  if (S->getThen() && Visit(MakeCXCursor(S->getThen(), StmtParent, TU)))
907f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor    return true;
908f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor  if (S->getElse() && Visit(MakeCXCursor(S->getElse(), StmtParent, TU)))
909f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor    return true;
910f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor
911f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor  return false;
912f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor}
913f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor
914f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregorbool CursorVisitor::VisitSwitchStmt(SwitchStmt *S) {
915f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor  if (VarDecl *Var = S->getConditionVariable()) {
916f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor    if (Visit(MakeCXCursor(Var, TU)))
917f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor      return true;
918f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  }
919f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
920263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor  if (S->getCond() && Visit(MakeCXCursor(S->getCond(), StmtParent, TU)))
921f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor    return true;
922263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor  if (S->getBody() && Visit(MakeCXCursor(S->getBody(), StmtParent, TU)))
923263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor    return true;
924f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
925263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor  return false;
926263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor}
927f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor
928263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregorbool CursorVisitor::VisitWhileStmt(WhileStmt *S) {
929263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor  if (VarDecl *Var = S->getConditionVariable()) {
930263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor    if (Visit(MakeCXCursor(Var, TU)))
931263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor      return true;
932f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  }
933f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
934263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor  if (S->getCond() && Visit(MakeCXCursor(S->getCond(), StmtParent, TU)))
935263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor    return true;
936263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor  if (S->getBody() && Visit(MakeCXCursor(S->getBody(), StmtParent, TU)))
937263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor    return true;
938263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor
939263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor  return false;
940263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor}
941263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor
942263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregorbool CursorVisitor::VisitForStmt(ForStmt *S) {
943263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor  if (S->getInit() && Visit(MakeCXCursor(S->getInit(), StmtParent, TU)))
944263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor    return true;
945263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor  if (VarDecl *Var = S->getConditionVariable()) {
946263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor    if (Visit(MakeCXCursor(Var, TU)))
947263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor      return true;
948f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  }
949f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
950263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor  if (S->getCond() && Visit(MakeCXCursor(S->getCond(), StmtParent, TU)))
951263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor    return true;
952263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor  if (S->getInc() && Visit(MakeCXCursor(S->getInc(), StmtParent, TU)))
953263b47b6272bfd3ea1c0e61b7ac404e7b2e0f759Douglas Gregor    return true;
954f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor  if (S->getBody() && Visit(MakeCXCursor(S->getBody(), StmtParent, TU)))
955f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor    return true;
956f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
957f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor  return false;
958f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor}
959f5bab418aeba427a2f6b8f3e9ce77cbc4a7afa95Douglas Gregor
9601ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenekbool CursorVisitor::VisitBlockExpr(BlockExpr *B) {
9611ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  return Visit(B->getBlockDecl());
9621ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek}
9631ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
9648ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregorbool CursorVisitor::VisitOffsetOfExpr(OffsetOfExpr *E) {
9658ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  // FIXME: Visit fields as well?
9668ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  if (Visit(E->getTypeSourceInfo()->getTypeLoc()))
9678ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor    return true;
9688ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
9698ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor  return VisitExpr(E);
9708ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor}
9718ecdb65716cd7914ffb2eeee993fa9039fcd31e8Douglas Gregor
972336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregorbool CursorVisitor::VisitSizeOfAlignOfExpr(SizeOfAlignOfExpr *E) {
973336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor  if (E->isArgumentType()) {
974336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor    if (TypeSourceInfo *TSInfo = E->getArgumentTypeInfo())
975336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor      return Visit(TSInfo->getTypeLoc());
976f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
977336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor    return false;
978336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor  }
979f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
980336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor  return VisitExpr(E);
981336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor}
982336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor
983336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregorbool CursorVisitor::VisitExplicitCastExpr(ExplicitCastExpr *E) {
984336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor  if (TypeSourceInfo *TSInfo = E->getTypeInfoAsWritten())
985336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor    if (Visit(TSInfo->getTypeLoc()))
986336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor      return true;
987f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
988336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor  return VisitCastExpr(E);
989336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor}
990336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor
991336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregorbool CursorVisitor::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
992336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor  if (TypeSourceInfo *TSInfo = E->getTypeSourceInfo())
993336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor    if (Visit(TSInfo->getTypeLoc()))
994336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor      return true;
995f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
996336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor  return VisitExpr(E);
997336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor}
998336fd81e04c01ffbce2825b372aeb127c80d4d97Douglas Gregor
999c2350e553b853ad00914faf23fa731e5fc4a8a5cDouglas Gregorbool CursorVisitor::VisitObjCMessageExpr(ObjCMessageExpr *E) {
100004badcf84c8d504d8491c7c7e29b58f52cb16640Douglas Gregor  if (TypeSourceInfo *TSInfo = E->getClassReceiverTypeInfo())
100104badcf84c8d504d8491c7c7e29b58f52cb16640Douglas Gregor    if (Visit(TSInfo->getTypeLoc()))
100204badcf84c8d504d8491c7c7e29b58f52cb16640Douglas Gregor      return true;
1003c2350e553b853ad00914faf23fa731e5fc4a8a5cDouglas Gregor
1004c2350e553b853ad00914faf23fa731e5fc4a8a5cDouglas Gregor  return VisitExpr(E);
1005c2350e553b853ad00914faf23fa731e5fc4a8a5cDouglas Gregor}
1006c2350e553b853ad00914faf23fa731e5fc4a8a5cDouglas Gregor
100781d3466d037dc5844234c7a93dab21a6ad986e7dDouglas Gregorbool CursorVisitor::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
100881d3466d037dc5844234c7a93dab21a6ad986e7dDouglas Gregor  return Visit(E->getEncodedTypeSourceInfo()->getTypeLoc());
100981d3466d037dc5844234c7a93dab21a6ad986e7dDouglas Gregor}
101081d3466d037dc5844234c7a93dab21a6ad986e7dDouglas Gregor
101181d3466d037dc5844234c7a93dab21a6ad986e7dDouglas Gregor
101209dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenekbool CursorVisitor::VisitAttributes(Decl *D) {
101309dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek  for (const Attr *A = D->getAttrs(); A; A = A->getNext())
101409dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek    if (Visit(MakeCXCursor(A, D, TU)))
101509dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek        return true;
101609dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
101709dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek  return false;
101809dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek}
101909dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
10205e4bc590b0ea010e38372d0b4a0aab578a746fe6Benjamin Kramerextern "C" {
10210a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas GregorCXIndex clang_createIndex(int excludeDeclarationsFromPCH,
10220a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor                          int displayDiagnostics) {
1023a030b7cf5e6aad5889b1b662b6979840bc75f87fDouglas Gregor  CIndexer *CIdxr = new CIndexer();
1024e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  if (excludeDeclarationsFromPCH)
1025e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff    CIdxr->setOnlyLocalDecls();
10260a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  if (displayDiagnostics)
10270a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    CIdxr->setDisplayDiagnostics();
1028e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  return CIdxr;
1029600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
1030600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
10319ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarvoid clang_disposeIndex(CXIndex CIdx) {
10322b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (CIdx)
10332b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    delete static_cast<CIndexer *>(CIdx);
10342bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff}
10352bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff
10368506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbarvoid clang_setUseExternalASTGeneration(CXIndex CIdx, int value) {
10372b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (CIdx) {
10382b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
10392b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    CXXIdx->setUseExternalASTGeneration(value);
10402b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  }
10418506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar}
10428506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar
10439ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXTranslationUnit clang_createTranslationUnit(CXIndex CIdx,
1044a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                              const char *ast_filename) {
10452b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CIdx)
10462b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    return 0;
1047f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10487d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
10490d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
105028019772db70d4547be05a042eb950bc910f134fDouglas Gregor  llvm::IntrusiveRefCntPtr<Diagnostic> Diags;
105128019772db70d4547be05a042eb950bc910f134fDouglas Gregor  return ASTUnit::LoadFromPCHFile(ast_filename, Diags,
1052a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                  CXXIdx->getOnlyLocalDecls(),
1053a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                  0, 0, true);
1054600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
1055600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
10569ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXTranslationUnit
10579ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarclang_createTranslationUnitFromSourceFile(CXIndex CIdx,
10589ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          const char *source_filename,
10599ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          int num_command_line_args,
10604db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor                                          const char **command_line_args,
10614db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor                                          unsigned num_unsaved_files,
1062a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                          struct CXUnsavedFile *unsaved_files) {
10632b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CIdx)
10642b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    return 0;
1065f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1066e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
1067e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff
10685352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  // Configure the diagnostics.
10695352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  DiagnosticOptions DiagOpts;
107028019772db70d4547be05a042eb950bc910f134fDouglas Gregor  llvm::IntrusiveRefCntPtr<Diagnostic> Diags;
107128019772db70d4547be05a042eb950bc910f134fDouglas Gregor  Diags = CompilerInstance::createDiagnostics(DiagOpts, 0, 0);
1072f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10734db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  llvm::SmallVector<ASTUnit::RemappedFile, 4> RemappedFiles;
10744db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  for (unsigned I = 0; I != num_unsaved_files; ++I) {
1075a0a270c0f1c0a4e3482438bdc5f4a7bd3d25f0a6Chris Lattner    llvm::StringRef Data(unsaved_files[I].Contents, unsaved_files[I].Length);
1076f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    const llvm::MemoryBuffer *Buffer
1077a0a270c0f1c0a4e3482438bdc5f4a7bd3d25f0a6Chris Lattner      = llvm::MemoryBuffer::getMemBufferCopy(Data, unsaved_files[I].Filename);
10784db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor    RemappedFiles.push_back(std::make_pair(unsaved_files[I].Filename,
10794db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor                                           Buffer));
10804db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  }
1081f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10828506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar  if (!CXXIdx->getUseExternalASTGeneration()) {
10838506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar    llvm::SmallVector<const char *, 16> Args;
10848506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar
10858506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar    // The 'source_filename' argument is optional.  If the caller does not
10868506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar    // specify it then it is assumed that the source file is specified
10878506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar    // in the actual argument list.
10888506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar    if (source_filename)
10898506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar      Args.push_back(source_filename);
10908506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar    Args.insert(Args.end(), command_line_args,
10918506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar                command_line_args + num_command_line_args);
109294dc8f640ebea52241412512ed48601626edbc58Douglas Gregor    Args.push_back("-Xclang");
109394dc8f640ebea52241412512ed48601626edbc58Douglas Gregor    Args.push_back("-detailed-preprocessing-record");
10945352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    unsigned NumErrors = Diags->getNumErrors();
1095f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
109629b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek#ifdef USE_CRASHTRACER
109729b7284c98253aed6e215aaa867c63084c80e944Ted Kremenek    ArgsCrashTracerInfo ACTI(Args);
10988a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek#endif
1099f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1100942209729a70af6230af2c6f0436bb77d2f6891cDaniel Dunbar    llvm::OwningPtr<ASTUnit> Unit(
1101942209729a70af6230af2c6f0436bb77d2f6891cDaniel Dunbar      ASTUnit::LoadFromCommandLine(Args.data(), Args.data() + Args.size(),
11023687e9d3a5dbfa9963af02a49a2b139d91310813Douglas Gregor                                   Diags,
1103869824e87940f97b87064db2df2861e82e08a8c6Daniel Dunbar                                   CXXIdx->getClangResourcesPath(),
1104942209729a70af6230af2c6f0436bb77d2f6891cDaniel Dunbar                                   CXXIdx->getOnlyLocalDecls(),
11054db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor                                   RemappedFiles.data(),
1106a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                   RemappedFiles.size(),
110794dc8f640ebea52241412512ed48601626edbc58Douglas Gregor                                   /*CaptureDiagnostics=*/true));
1108f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1109942209729a70af6230af2c6f0436bb77d2f6891cDaniel Dunbar    // FIXME: Until we have broader testing, just drop the entire AST if we
1110942209729a70af6230af2c6f0436bb77d2f6891cDaniel Dunbar    // encountered an error.
11110a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    if (NumErrors != Diags->getNumErrors()) {
111234f6a322430e2cdcb38167781eb848b9b8e928bbTed Kremenek      // Make sure to check that 'Unit' is non-NULL.
111334f6a322430e2cdcb38167781eb848b9b8e928bbTed Kremenek      if (CXXIdx->getDisplayDiagnostics() && Unit.get()) {
1114405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor        for (ASTUnit::stored_diag_iterator D = Unit->stored_diag_begin(),
1115405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor                                        DEnd = Unit->stored_diag_end();
11160a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor             D != DEnd; ++D) {
11170a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor          CXStoredDiagnostic Diag(*D, Unit->getASTContext().getLangOptions());
1118274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor          CXString Msg = clang_formatDiagnostic(&Diag,
1119274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor                                      clang_defaultDiagnosticDisplayOptions());
1120274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor          fprintf(stderr, "%s\n", clang_getCString(Msg));
1121274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor          clang_disposeString(Msg);
11220a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor        }
1123274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor#ifdef LLVM_ON_WIN32
1124274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor        // On Windows, force a flush, since there may be multiple copies of
1125274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor        // stderr and stdout in the file system, all with different buffers
1126274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor        // but writing to the same device.
1127274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor        fflush(stderr);
112883c5184d3ae2ff94a993b3e770e96756193f15f0Ted Kremenek#endif
11290a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor      }
11300a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    }
1131942209729a70af6230af2c6f0436bb77d2f6891cDaniel Dunbar
1132942209729a70af6230af2c6f0436bb77d2f6891cDaniel Dunbar    return Unit.take();
11338506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar  }
11348506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar
1135139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek  // Build up the arguments for invoking 'clang'.
113674cd0694f9b7d8b9d939d6104b0cc319753c513cTed Kremenek  std::vector<const char *> argv;
11370d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
1138139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek  // First add the complete path to the 'clang' executable.
1139139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek  llvm::sys::Path ClangPath = static_cast<CIndexer *>(CIdx)->getClangPath();
11405e4bc590b0ea010e38372d0b4a0aab578a746fe6Benjamin Kramer  argv.push_back(ClangPath.c_str());
11410d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
1142139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek  // Add the '-emit-ast' option as our execution mode for 'clang'.
114374cd0694f9b7d8b9d939d6104b0cc319753c513cTed Kremenek  argv.push_back("-emit-ast");
11440d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
1145139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek  // The 'source_filename' argument is optional.  If the caller does not
1146139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek  // specify it then it is assumed that the source file is specified
1147139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek  // in the actual argument list.
11480d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar  if (source_filename)
11490d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar    argv.push_back(source_filename);
1150139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek
115137b5ac2909a7693fc6e6842c5a8825bdeaa636ddSteve Naroff  // Generate a temporary name for the AST file.
1152139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek  argv.push_back("-o");
1153c2a981614e324fea7f0a0533f8f1d103cbd17f6dBenjamin Kramer  char astTmpFile[L_tmpnam];
1154c2a981614e324fea7f0a0533f8f1d103cbd17f6dBenjamin Kramer  argv.push_back(tmpnam(astTmpFile));
1155139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek
11564db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  // Remap any unsaved files to temporary files.
11574db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  std::vector<llvm::sys::Path> TemporaryFiles;
11584db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  std::vector<std::string> RemapArgs;
11594db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  if (RemapFiles(num_unsaved_files, unsaved_files, RemapArgs, TemporaryFiles))
11604db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor    return 0;
1161f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
11624db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  // The pointers into the elements of RemapArgs are stable because we
11634db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  // won't be adding anything to RemapArgs after this point.
11644db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  for (unsigned i = 0, e = RemapArgs.size(); i != e; ++i)
11654db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor    argv.push_back(RemapArgs[i].c_str());
1166f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1167139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek  // Process the compiler options, stripping off '-o', '-c', '-fsyntax-only'.
1168139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek  for (int i = 0; i < num_command_line_args; ++i)
1169139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek    if (const char *arg = command_line_args[i]) {
1170139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek      if (strcmp(arg, "-o") == 0) {
1171139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek        ++i; // Also skip the matching argument.
1172139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek        continue;
1173139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek      }
1174139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek      if (strcmp(arg, "-emit-ast") == 0 ||
1175139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek          strcmp(arg, "-c") == 0 ||
1176139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek          strcmp(arg, "-fsyntax-only") == 0) {
1177139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek        continue;
1178139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek      }
1179139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek
1180139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek      // Keep the argument.
1181139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek      argv.push_back(arg);
1182e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff    }
11830d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
1184d93256e55673a17d18543397ec462416acb13792Douglas Gregor  // Generate a temporary name for the diagnostics file.
1185c2a981614e324fea7f0a0533f8f1d103cbd17f6dBenjamin Kramer  char tmpFileResults[L_tmpnam];
1186c2a981614e324fea7f0a0533f8f1d103cbd17f6dBenjamin Kramer  char *tmpResultsFileName = tmpnam(tmpFileResults);
1187c2a981614e324fea7f0a0533f8f1d103cbd17f6dBenjamin Kramer  llvm::sys::Path DiagnosticsFile(tmpResultsFileName);
1188d93256e55673a17d18543397ec462416acb13792Douglas Gregor  TemporaryFiles.push_back(DiagnosticsFile);
1189d93256e55673a17d18543397ec462416acb13792Douglas Gregor  argv.push_back("-fdiagnostics-binary");
1190d93256e55673a17d18543397ec462416acb13792Douglas Gregor
119194dc8f640ebea52241412512ed48601626edbc58Douglas Gregor  argv.push_back("-Xclang");
119294dc8f640ebea52241412512ed48601626edbc58Douglas Gregor  argv.push_back("-detailed-preprocessing-record");
119394dc8f640ebea52241412512ed48601626edbc58Douglas Gregor
1194139ba86a362593da2e350f881e5c7003917aa5a7Ted Kremenek  // Add the null terminator.
119574cd0694f9b7d8b9d939d6104b0cc319753c513cTed Kremenek  argv.push_back(NULL);
119674cd0694f9b7d8b9d939d6104b0cc319753c513cTed Kremenek
1197feb15e3f39de77ce5cdf8a5d9ee01ead5e679e9eTed Kremenek  // Invoke 'clang'.
1198feb15e3f39de77ce5cdf8a5d9ee01ead5e679e9eTed Kremenek  llvm::sys::Path DevNull; // leave empty, causes redirection to /dev/null
1199feb15e3f39de77ce5cdf8a5d9ee01ead5e679e9eTed Kremenek                           // on Unix or NUL (Windows).
1200379afec20ce8fd419551874daf3a1e683547575eTed Kremenek  std::string ErrMsg;
1201d93256e55673a17d18543397ec462416acb13792Douglas Gregor  const llvm::sys::Path *Redirects[] = { &DevNull, &DevNull, &DiagnosticsFile,
1202d93256e55673a17d18543397ec462416acb13792Douglas Gregor                                         NULL };
1203379afec20ce8fd419551874daf3a1e683547575eTed Kremenek  llvm::sys::Program::ExecuteAndWait(ClangPath, &argv[0], /* env */ NULL,
1204936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor      /* redirects */ &Redirects[0],
1205379afec20ce8fd419551874daf3a1e683547575eTed Kremenek      /* secondsToWait */ 0, /* memoryLimits */ 0, &ErrMsg);
12060d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
1207936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor  if (!ErrMsg.empty()) {
1208936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor    std::string AllArgs;
1209379afec20ce8fd419551874daf3a1e683547575eTed Kremenek    for (std::vector<const char*>::iterator I = argv.begin(), E = argv.end();
1210936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor         I != E; ++I) {
1211936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor      AllArgs += ' ';
1212779e5f487daaf1c96cbd22a145e988742395f8d4Ted Kremenek      if (*I)
1213936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor        AllArgs += *I;
1214779e5f487daaf1c96cbd22a145e988742395f8d4Ted Kremenek    }
1215f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
121632141c8f69a6a51d7cf95a18229cd335ecd59903Daniel Dunbar    Diags->Report(diag::err_fe_invoking) << AllArgs << ErrMsg;
1217379afec20ce8fd419551874daf3a1e683547575eTed Kremenek  }
12180829a839b05c687d3f11af0640d84c5d05f94124Benjamin Kramer
12193687e9d3a5dbfa9963af02a49a2b139d91310813Douglas Gregor  ASTUnit *ATU = ASTUnit::LoadFromPCHFile(astTmpFile, Diags,
12204db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor                                          CXXIdx->getOnlyLocalDecls(),
12214db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor                                          RemappedFiles.data(),
1222a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                          RemappedFiles.size(),
1223a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                          /*CaptureDiagnostics=*/true);
1224a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  if (ATU) {
1225a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor    LoadSerializedDiagnostics(DiagnosticsFile,
1226a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                              num_unsaved_files, unsaved_files,
1227a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                              ATU->getFileManager(),
1228a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                              ATU->getSourceManager(),
1229405634b215f19eec7183bd8005e34aa5a02f64a1Douglas Gregor                              ATU->getStoredDiagnostics());
12300a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  } else if (CXXIdx->getDisplayDiagnostics()) {
12310a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    // We failed to load the ASTUnit, but we can still deserialize the
12320a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    // diagnostics and emit them.
12330a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    FileManager FileMgr;
1234f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    Diagnostic Diag;
1235f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    SourceManager SourceMgr(Diag);
12360a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    // FIXME: Faked LangOpts!
12370a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    LangOptions LangOpts;
12380a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    llvm::SmallVector<StoredDiagnostic, 4> Diags;
12390a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    LoadSerializedDiagnostics(DiagnosticsFile,
12400a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor                              num_unsaved_files, unsaved_files,
12410a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor                              FileMgr, SourceMgr, Diags);
12420a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    for (llvm::SmallVector<StoredDiagnostic, 4>::iterator D = Diags.begin(),
12430a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor                                                       DEnd = Diags.end();
12440a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor         D != DEnd; ++D) {
12450a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor      CXStoredDiagnostic Diag(*D, LangOpts);
1246274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor      CXString Msg = clang_formatDiagnostic(&Diag,
1247274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor                                      clang_defaultDiagnosticDisplayOptions());
1248274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor      fprintf(stderr, "%s\n", clang_getCString(Msg));
1249274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor      clang_disposeString(Msg);
12500a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    }
1251274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor
1252274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor#ifdef LLVM_ON_WIN32
1253274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor    // On Windows, force a flush, since there may be multiple copies of
1254274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor    // stderr and stdout in the file system, all with different buffers
1255274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor    // but writing to the same device.
1256274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor    fflush(stderr);
1257274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor#endif
1258a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  }
1259d93256e55673a17d18543397ec462416acb13792Douglas Gregor
1260313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor  if (ATU) {
1261313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor    // Make the translation unit responsible for destroying all temporary files.
1262313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor    for (unsigned i = 0, e = TemporaryFiles.size(); i != e; ++i)
1263313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor      ATU->addTemporaryFile(TemporaryFiles[i]);
1264313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor    ATU->addTemporaryFile(llvm::sys::Path(ATU->getPCHFileName()));
1265313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor  } else {
1266313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor    // Destroy all of the temporary files now; they can't be referenced any
1267313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor    // longer.
1268313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor    llvm::sys::Path(astTmpFile).eraseFromDisk();
1269313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor    for (unsigned i = 0, e = TemporaryFiles.size(); i != e; ++i)
1270313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor      TemporaryFiles[i].eraseFromDisk();
1271313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor  }
1272313e26c4e81f0e467490a530548450f4c824a6c4Douglas Gregor
1273e19944c93961b7618f4f3f3185f698f46369ea54Steve Naroff  return ATU;
12745b7d8e254f6c2855b37b5521c0aee0a560dab237Steve Naroff}
12755b7d8e254f6c2855b37b5521c0aee0a560dab237Steve Naroff
12769ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarvoid clang_disposeTranslationUnit(CXTranslationUnit CTUnit) {
12772b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (CTUnit)
12782b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    delete static_cast<ASTUnit *>(CTUnit);
12792bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff}
12800d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
12819ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXString clang_getTranslationUnitSpelling(CXTranslationUnit CTUnit) {
12822b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CTUnit)
1283ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
1284f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
128577accc11f04ed4ff9afd4e27d430144d4714be56Steve Naroff  ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit);
1286ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString(CXXUnit->getOriginalSourceFileName(), true);
1287af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff}
12881eb79b58e56b99cf557d5d353586a10c5360364dDaniel Dunbar
12897eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas GregorCXCursor clang_getTranslationUnitCursor(CXTranslationUnit TU) {
1290b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  CXCursor Result = { CXCursor_TranslationUnit, { 0, 0, TU } };
12917eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  return Result;
12927eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor}
12937eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
1294fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
1295600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
1296fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
12971db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor// CXSourceLocation and CXSourceRange Operations.
12981db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor//===----------------------------------------------------------------------===//
12991db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
1300b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregorextern "C" {
1301b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXSourceLocation clang_getNullLocation() {
13025352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  CXSourceLocation Result = { { 0, 0 }, 0 };
1303b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  return Result;
1304b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
1305b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
1306b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregorunsigned clang_equalLocations(CXSourceLocation loc1, CXSourceLocation loc2) {
130790a6b9e1e4d4d1995ada044e319d6e722b07a6b4Daniel Dunbar  return (loc1.ptr_data[0] == loc2.ptr_data[0] &&
130890a6b9e1e4d4d1995ada044e319d6e722b07a6b4Daniel Dunbar          loc1.ptr_data[1] == loc2.ptr_data[1] &&
130990a6b9e1e4d4d1995ada044e319d6e722b07a6b4Daniel Dunbar          loc1.int_data == loc2.int_data);
1310b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
1311b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
1312b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXSourceLocation clang_getLocation(CXTranslationUnit tu,
1313b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor                                   CXFile file,
1314b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor                                   unsigned line,
1315b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor                                   unsigned column) {
131642748ec5cb2d75fe0dbb3a6db5aee6c11b5dc190Douglas Gregor  if (!tu || !file)
1317b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor    return clang_getNullLocation();
131842748ec5cb2d75fe0dbb3a6db5aee6c11b5dc190Douglas Gregor
1319b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu);
1320b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  SourceLocation SLoc
1321b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor    = CXXUnit->getSourceManager().getLocation(
1322f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        static_cast<const FileEntry *>(file),
1323b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor                                              line, column);
1324f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1325bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar  return cxloc::translateSourceLocation(CXXUnit->getASTContext(), SLoc);
1326b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
1327b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
13285352ac06d8f6194825bb2a99ffa009b61bafb503Douglas GregorCXSourceRange clang_getNullRange() {
13295352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  CXSourceRange Result = { { 0, 0 }, 0, 0 };
13305352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  return Result;
13315352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor}
1332d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar
1333b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXSourceRange clang_getRange(CXSourceLocation begin, CXSourceLocation end) {
13345352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  if (begin.ptr_data[0] != end.ptr_data[0] ||
13355352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor      begin.ptr_data[1] != end.ptr_data[1])
13365352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
1337f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1338f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  CXSourceRange Result = { { begin.ptr_data[0], begin.ptr_data[1] },
13395352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor                           begin.int_data, end.int_data };
1340b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  return Result;
1341b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
1342b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
134346766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregorvoid clang_getInstantiationLocation(CXSourceLocation location,
134446766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor                                    CXFile *file,
134546766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor                                    unsigned *line,
134646766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor                                    unsigned *column,
134746766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor                                    unsigned *offset) {
13481db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  SourceLocation Loc = SourceLocation::getFromRawEncoding(location.int_data);
13491db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
1350bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar  if (!location.ptr_data[0] || Loc.isInvalid()) {
135146766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor    if (file)
135246766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor      *file = 0;
135346766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor    if (line)
135446766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor      *line = 0;
135546766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor    if (column)
135646766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor      *column = 0;
135746766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor    if (offset)
135846766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor      *offset = 0;
135946766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor    return;
136046766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor  }
136146766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor
1362bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar  const SourceManager &SM =
1363bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar    *static_cast<const SourceManager*>(location.ptr_data[0]);
13641db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  SourceLocation InstLoc = SM.getInstantiationLoc(Loc);
13651db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
13661db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  if (file)
13671db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor    *file = (void *)SM.getFileEntryForID(SM.getFileID(InstLoc));
13681db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  if (line)
13691db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor    *line = SM.getInstantiationLineNumber(InstLoc);
13701db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  if (column)
13711db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor    *column = SM.getInstantiationColumnNumber(InstLoc);
1372e69517ce61638f12c9abe4605753a45275ac4e37Douglas Gregor  if (offset)
137346766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor    *offset = SM.getDecomposedLoc(InstLoc).second;
1374e69517ce61638f12c9abe4605753a45275ac4e37Douglas Gregor}
1375e69517ce61638f12c9abe4605753a45275ac4e37Douglas Gregor
13761db19dea8d221f27be46332d668d1e2decb7f1abDouglas GregorCXSourceLocation clang_getRangeStart(CXSourceRange range) {
1377f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  CXSourceLocation Result = { { range.ptr_data[0], range.ptr_data[1] },
13785352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor                              range.begin_int_data };
13791db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  return Result;
13801db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor}
13811db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
13821db19dea8d221f27be46332d668d1e2decb7f1abDouglas GregorCXSourceLocation clang_getRangeEnd(CXSourceRange range) {
1383bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar  CXSourceLocation Result = { { range.ptr_data[0], range.ptr_data[1] },
13845352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor                              range.end_int_data };
13851db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  return Result;
13861db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor}
13871db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
1388b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor} // end: extern "C"
1389b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
13901db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor//===----------------------------------------------------------------------===//
1391fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXFile Operations.
1392fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
1393fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
1394fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
139574844072411bae91d5dbb89955d200cbe1e0a1c8Ted KremenekCXString clang_getFileName(CXFile SFile) {
139698258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (!SFile)
139774844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek    return createCXString(NULL);
1398f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
139988145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  FileEntry *FEnt = static_cast<FileEntry *>(SFile);
140074844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek  return createCXString(FEnt->getName());
140188145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff}
140288145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff
140388145034694ed5267fa6fa5febc54fadc02bd479Steve Narofftime_t clang_getFileTime(CXFile SFile) {
140498258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (!SFile)
140598258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor    return 0;
1406f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
140788145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  FileEntry *FEnt = static_cast<FileEntry *>(SFile);
140888145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  return FEnt->getModificationTime();
1409ee9405e807d7c447c0143c2bd865b759192e97b3Steve Naroff}
1410f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1411b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXFile clang_getFile(CXTranslationUnit tu, const char *file_name) {
1412b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  if (!tu)
1413b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor    return 0;
1414f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1415b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu);
1416f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1417b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  FileManager &FMgr = CXXUnit->getFileManager();
1418b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  const FileEntry *File = FMgr.getFile(file_name, file_name+strlen(file_name));
1419b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  return const_cast<FileEntry *>(File);
1420b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
1421f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1422fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
1423fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
1424fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
1425fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXCursor Operations.
1426fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
1427fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
1428fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekstatic Decl *getDeclFromExpr(Stmt *E) {
1429fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (DeclRefExpr *RefExpr = dyn_cast<DeclRefExpr>(E))
1430fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return RefExpr->getDecl();
1431fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (MemberExpr *ME = dyn_cast<MemberExpr>(E))
1432fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return ME->getMemberDecl();
1433fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (ObjCIvarRefExpr *RE = dyn_cast<ObjCIvarRefExpr>(E))
1434fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return RE->getDecl();
1435f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1436fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (CallExpr *CE = dyn_cast<CallExpr>(E))
1437fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return getDeclFromExpr(CE->getCallee());
1438fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (CastExpr *CE = dyn_cast<CastExpr>(E))
1439fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return getDeclFromExpr(CE->getSubExpr());
1440fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (ObjCMessageExpr *OME = dyn_cast<ObjCMessageExpr>(E))
1441fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return OME->getMethodDecl();
1442f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1443fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  return 0;
1444fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek}
1445ee9405e807d7c447c0143c2bd865b759192e97b3Steve Naroff
1446c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbarstatic SourceLocation getLocationFromExpr(Expr *E) {
1447c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (ObjCMessageExpr *Msg = dyn_cast<ObjCMessageExpr>(E))
1448c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return /*FIXME:*/Msg->getLeftLoc();
1449c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
1450c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return DRE->getLocation();
1451c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (MemberExpr *Member = dyn_cast<MemberExpr>(E))
1452c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return Member->getMemberLoc();
1453c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (ObjCIvarRefExpr *Ivar = dyn_cast<ObjCIvarRefExpr>(E))
1454c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return Ivar->getLocation();
1455c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  return E->getLocStart();
1456c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar}
1457c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar
1458fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
1459f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1460f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekunsigned clang_visitChildren(CXCursor parent,
1461b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor                             CXCursorVisitor visitor,
1462b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor                             CXClientData client_data) {
1463b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  ASTUnit *CXXUnit = getCursorASTUnit(parent);
1464b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor
1465b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  unsigned PCHLevel = Decl::MaxPCHLevel;
1466f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1467b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  // Set the PCHLevel to filter out unwanted decls if requested.
1468b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (CXXUnit->getOnlyLocalDecls()) {
1469b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    PCHLevel = 0;
1470f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1471b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    // If the main input was an AST, bump the level.
1472b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    if (CXXUnit->isMainFileAST())
1473b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      ++PCHLevel;
1474b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
1475f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1476b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  CursorVisitor CursorVis(CXXUnit, visitor, client_data, PCHLevel);
1477b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return CursorVis.VisitChildren(parent);
1478b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
1479b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor
148078205d4bada39d95097e766af9eb30cdd0159461Douglas Gregorstatic CXString getDeclSpelling(Decl *D) {
148178205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D);
148278205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (!ND)
1483ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
1484f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
148578205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
1486ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(OMD->getSelector().getAsString());
1487f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
148878205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ObjCCategoryImplDecl *CIMP = dyn_cast<ObjCCategoryImplDecl>(ND))
148978205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // No, this isn't the same as the code below. getIdentifier() is non-virtual
149078205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // and returns different names. NamedDecl returns the class name and
149178205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // ObjCCategoryImplDecl returns the category name.
1492ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(CIMP->getIdentifier()->getNameStart());
1493f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
149478205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ND->getIdentifier())
1495ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(ND->getIdentifier()->getNameStart());
1496f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1497ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString("");
149878205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor}
1499f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
15009ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXString clang_getCursorSpelling(CXCursor C) {
15017eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  if (clang_isTranslationUnit(C.kind))
1502b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getTranslationUnitSpelling(C.data[2]);
15037eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
1504f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  if (clang_isReference(C.kind)) {
1505f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff    switch (C.kind) {
1506acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCSuperClassRef: {
15072e331b938b38057e333fab0ba841130ea8467794Douglas Gregor      ObjCInterfaceDecl *Super = getCursorObjCSuperClassRef(C).first;
1508ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(Super->getIdentifier()->getNameStart());
1509acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
1510acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCClassRef: {
15111adb082a709f7b588f03672999294e061234b2cfDouglas Gregor      ObjCInterfaceDecl *Class = getCursorObjCClassRef(C).first;
1512ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(Class->getIdentifier()->getNameStart());
1513acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
1514acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCProtocolRef: {
151578db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor      ObjCProtocolDecl *OID = getCursorObjCProtocolRef(C).first;
1516f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      assert(OID && "getCursorSpelling(): Missing protocol decl");
1517ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(OID->getIdentifier()->getNameStart());
1518acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
15197d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    case CXCursor_TypeRef: {
15207d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      TypeDecl *Type = getCursorTypeRef(C).first;
15217d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      assert(Type && "Missing type decl");
15227d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
1523ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(getCursorContext(C).getTypeDeclType(Type).
1524ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek                              getAsString());
15257d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    }
15267d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
1527acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    default:
1528ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString("<not implemented>");
1529f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff    }
1530f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  }
153197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
153297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind)) {
153397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    Decl *D = getDeclFromExpr(getCursorExpr(C));
153497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    if (D)
153578205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor      return getDeclSpelling(D);
1536ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
153797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  }
153897b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
15394ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor  if (C.kind == CXCursor_MacroInstantiation)
15404ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor    return createCXString(getCursorMacroInstantiation(C)->getName()
15414ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor                                                           ->getNameStart());
15424ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor
1543572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition)
1544572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return createCXString(getCursorMacroDefinition(C)->getName()
1545572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor                                                           ->getNameStart());
1546572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
154760cbfacd947590f83257a4191566dda92fbbde69Douglas Gregor  if (clang_isDeclaration(C.kind))
154860cbfacd947590f83257a4191566dda92fbbde69Douglas Gregor    return getDeclSpelling(getCursorDecl(C));
1549e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
1550ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString("");
1551f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff}
1552f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff
1553e68fff6fc083c6270d835216a3de0b82c6ef0310Ted KremenekCXString clang_getCursorKindSpelling(enum CXCursorKind Kind) {
155489922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  switch (Kind) {
1555e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_FunctionDecl:
1556e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("FunctionDecl");
1557e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TypedefDecl:
1558e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TypedefDecl");
1559e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_EnumDecl:
1560e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("EnumDecl");
1561e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_EnumConstantDecl:
1562e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("EnumConstantDecl");
1563e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_StructDecl:
1564e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("StructDecl");
1565e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnionDecl:
1566e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnionDecl");
1567e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ClassDecl:
1568e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ClassDecl");
1569e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_FieldDecl:
1570e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("FieldDecl");
1571e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_VarDecl:
1572e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("VarDecl");
1573e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ParmDecl:
1574e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ParmDecl");
1575e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCInterfaceDecl:
1576e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCInterfaceDecl");
1577e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCCategoryDecl:
1578e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCCategoryDecl");
1579e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCProtocolDecl:
1580e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCProtocolDecl");
1581e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCPropertyDecl:
1582e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCPropertyDecl");
1583e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCIvarDecl:
1584e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCIvarDecl");
1585e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCInstanceMethodDecl:
1586e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCInstanceMethodDecl");
1587e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCClassMethodDecl:
1588e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCClassMethodDecl");
1589e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCImplementationDecl:
1590e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCImplementationDecl");
1591e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCCategoryImplDecl:
1592e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCCategoryImplDecl");
15938bd5a69999cfd06b6b5a58fdd04e4f802b2df5a4Ted Kremenek  case CXCursor_CXXMethod:
15948bd5a69999cfd06b6b5a58fdd04e4f802b2df5a4Ted Kremenek      return createCXString("CXXMethod");
1595e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedDecl:
1596e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedDecl");
1597e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCSuperClassRef:
1598e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCSuperClassRef");
1599e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCProtocolRef:
1600e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCProtocolRef");
1601e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCClassRef:
1602e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCClassRef");
1603e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TypeRef:
1604e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TypeRef");
1605e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedExpr:
1606e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedExpr");
16071ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  case CXCursor_BlockExpr:
16081ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek      return createCXString("BlockExpr");
1609e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_DeclRefExpr:
1610e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("DeclRefExpr");
1611e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_MemberRefExpr:
1612e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("MemberRefExpr");
1613e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_CallExpr:
1614e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("CallExpr");
1615e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCMessageExpr:
1616e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCMessageExpr");
1617e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedStmt:
1618e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedStmt");
1619e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_InvalidFile:
1620e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("InvalidFile");
1621292db6401f040795db3ea4e00fc02622d6c3ba1dTed Kremenek  case CXCursor_InvalidCode:
1622292db6401f040795db3ea4e00fc02622d6c3ba1dTed Kremenek    return createCXString("InvalidCode");
1623e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_NoDeclFound:
1624e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("NoDeclFound");
1625e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_NotImplemented:
1626e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("NotImplemented");
1627e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TranslationUnit:
1628e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TranslationUnit");
1629e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek  case CXCursor_UnexposedAttr:
1630e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek      return createCXString("UnexposedAttr");
1631e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek  case CXCursor_IBActionAttr:
1632e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek      return createCXString("attribute(ibaction)");
16339f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  case CXCursor_IBOutletAttr:
16349f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor     return createCXString("attribute(iboutlet)");
16359f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  case CXCursor_PreprocessingDirective:
16369f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    return createCXString("preprocessing directive");
1637572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  case CXCursor_MacroDefinition:
1638572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return createCXString("macro definition");
16394807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  case CXCursor_MacroInstantiation:
16404807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor    return createCXString("macro instantiation");
164189922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  }
1642e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
1643deb06bd3566e18f677e76bc435d478b033fe328bTed Kremenek  llvm_unreachable("Unhandled CXCursorKind");
1644e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  return createCXString(NULL);
1645600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
164689922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff
1647e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenekenum CXChildVisitResult GetCursorVisitor(CXCursor cursor,
1648e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek                                         CXCursor parent,
164933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor                                         CXClientData client_data) {
165033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  CXCursor *BestCursor = static_cast<CXCursor *>(client_data);
165133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  *BestCursor = cursor;
165233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  return CXChildVisit_Recurse;
165333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
1654e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
1655b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXCursor clang_getCursor(CXTranslationUnit TU, CXSourceLocation Loc) {
1656b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  if (!TU)
1657f462989fe8d6f59ab2d7d0fe2b4b96292ce706eaTed Kremenek    return clang_getNullCursor();
1658e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
1659b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU);
1660b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
1661bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
1662bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor
1663a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek  SourceLocation SLoc = cxloc::translateSourceLocation(Loc);
166433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  CXCursor Result = MakeCXCursorInvalid(CXCursor_NoDeclFound);
166533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (SLoc.isValid()) {
1666d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar    SourceRange RegionOfInterest(SLoc, SLoc.getFileLocWithOffset(1));
1667e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
166833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    // FIXME: Would be great to have a "hint" cursor, then walk from that
166933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    // hint cursor upward until we find a cursor whose source range encloses
167033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    // the region of interest, rather than starting from the translation unit.
167133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    CXCursor Parent = clang_getTranslationUnitCursor(CXXUnit);
1672e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek    CursorVisitor CursorVis(CXXUnit, GetCursorVisitor, &Result,
167333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor                            Decl::MaxPCHLevel, RegionOfInterest);
167433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    CursorVis.VisitChildren(Parent);
167577128ddd3077fc045751a55bb3226802b15d5510Steve Naroff  }
1676e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  return Result;
167777128ddd3077fc045751a55bb3226802b15d5510Steve Naroff}
167877128ddd3077fc045751a55bb3226802b15d5510Steve Naroff
1679738855554394a6afcf39cc8345fd22c3756b8dd0Ted KremenekCXCursor clang_getNullCursor(void) {
16805bfb8c128c2ac8eb4032afc180cdc400a0f953caDouglas Gregor  return MakeCXCursorInvalid(CXCursor_InvalidFile);
1681738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek}
1682738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek
1683738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenekunsigned clang_equalCursors(CXCursor X, CXCursor Y) {
1684283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  return X == Y;
1685738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek}
16860d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
16879ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isInvalid(enum CXCursorKind K) {
168877128ddd3077fc045751a55bb3226802b15d5510Steve Naroff  return K >= CXCursor_FirstInvalid && K <= CXCursor_LastInvalid;
168977128ddd3077fc045751a55bb3226802b15d5510Steve Naroff}
169077128ddd3077fc045751a55bb3226802b15d5510Steve Naroff
16919ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isDeclaration(enum CXCursorKind K) {
169289922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  return K >= CXCursor_FirstDecl && K <= CXCursor_LastDecl;
169389922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff}
16942d4d629d8a0de5112c7ae9d05c03ddbf6dcd956aSteve Naroff
16959ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isReference(enum CXCursorKind K) {
1696f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  return K >= CXCursor_FirstRef && K <= CXCursor_LastRef;
1697f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff}
1698f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff
169997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregorunsigned clang_isExpression(enum CXCursorKind K) {
170097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  return K >= CXCursor_FirstExpr && K <= CXCursor_LastExpr;
170197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor}
170297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
170397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregorunsigned clang_isStatement(enum CXCursorKind K) {
170497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  return K >= CXCursor_FirstStmt && K <= CXCursor_LastStmt;
170597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor}
170697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
17077eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregorunsigned clang_isTranslationUnit(enum CXCursorKind K) {
17087eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  return K == CXCursor_TranslationUnit;
17097eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor}
17107eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
17119f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregorunsigned clang_isPreprocessing(enum CXCursorKind K) {
17129f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  return K >= CXCursor_FirstPreprocessing && K <= CXCursor_LastPreprocessing;
17139f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor}
17149f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor
1715ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenekunsigned clang_isUnexposed(enum CXCursorKind K) {
1716ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek  switch (K) {
1717ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedDecl:
1718ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedExpr:
1719ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedStmt:
1720ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedAttr:
1721ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek      return true;
1722ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    default:
1723ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek      return false;
1724ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek  }
1725ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek}
1726ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek
17279ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXCursorKind clang_getCursorKind(CXCursor C) {
17289efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff  return C.kind;
17299efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff}
17309efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff
173198258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas GregorCXSourceLocation clang_getCursorLocation(CXCursor C) {
173298258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (clang_isReference(C.kind)) {
1733f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    switch (C.kind) {
1734f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCSuperClassRef: {
1735f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCInterfaceDecl *, SourceLocation> P
1736f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCSuperClassRef(C);
1737a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
1738f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
1739f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor
1740f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCProtocolRef: {
1741f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCProtocolDecl *, SourceLocation> P
1742f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCProtocolRef(C);
1743a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
1744f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
1745f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor
1746f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCClassRef: {
1747f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCInterfaceDecl *, SourceLocation> P
1748f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCClassRef(C);
1749a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
1750f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
17517d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
1752f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_TypeRef: {
17537d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      std::pair<TypeDecl *, SourceLocation> P = getCursorTypeRef(C);
1754a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
17557d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    }
1756f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1757f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    default:
1758f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      // FIXME: Need a way to enumerate all non-reference cases.
1759f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      llvm_unreachable("Missed a reference kind");
1760f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
176198258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  }
176297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
176397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind))
1764f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    return cxloc::translateSourceLocation(getCursorContext(C),
176597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor                                   getLocationFromExpr(getCursorExpr(C)));
176697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
17679f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  if (C.kind == CXCursor_PreprocessingDirective) {
17689f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    SourceLocation L = cxcursor::getCursorPreprocessingDirective(C).getBegin();
17699f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
17709f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  }
17714807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor
17724807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  if (C.kind == CXCursor_MacroInstantiation) {
17734ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor    SourceLocation L
17744ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor      = cxcursor::getCursorMacroInstantiation(C)->getSourceRange().getBegin();
17754807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
17764807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  }
1777572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
1778572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition) {
1779572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    SourceLocation L = cxcursor::getCursorMacroDefinition(C)->getLocation();
1780572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
1781572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  }
17829f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor
17835352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  if (!getCursorDecl(C))
17845352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullLocation();
178598258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor
1786f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  Decl *D = getCursorDecl(C);
1787f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  SourceLocation Loc = D->getLocation();
1788f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(D))
1789f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    Loc = Class->getClassLoc();
17902ca54feee89d7277fb967e3247a64f40ef155a82Douglas Gregor  return cxloc::translateSourceLocation(getCursorContext(C), Loc);
179188145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff}
1792a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor
1793a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas GregorCXSourceRange clang_getCursorExtent(CXCursor C) {
1794a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  if (clang_isReference(C.kind)) {
1795a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor    switch (C.kind) {
1796f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek      case CXCursor_ObjCSuperClassRef: {
1797a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor        std::pair<ObjCInterfaceDecl *, SourceLocation> P
1798a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor          = getCursorObjCSuperClassRef(C);
1799a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek        return cxloc::translateSourceRange(P.first->getASTContext(), P.second);
1800a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor      }
1801f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1802f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek      case CXCursor_ObjCProtocolRef: {
1803a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor        std::pair<ObjCProtocolDecl *, SourceLocation> P
1804a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor          = getCursorObjCProtocolRef(C);
1805a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek        return cxloc::translateSourceRange(P.first->getASTContext(), P.second);
1806a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor      }
1807f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1808f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek      case CXCursor_ObjCClassRef: {
1809a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor        std::pair<ObjCInterfaceDecl *, SourceLocation> P
1810a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor          = getCursorObjCClassRef(C);
1811f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1812a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek        return cxloc::translateSourceRange(P.first->getASTContext(), P.second);
1813a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor      }
18147d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
1815f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek      case CXCursor_TypeRef: {
18167d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor        std::pair<TypeDecl *, SourceLocation> P = getCursorTypeRef(C);
1817a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek        return cxloc::translateSourceRange(P.first->getASTContext(), P.second);
18187d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      }
1819f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1820a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor      default:
1821a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor        // FIXME: Need a way to enumerate all non-reference cases.
1822a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor        llvm_unreachable("Missed a reference kind");
1823a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor    }
1824a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  }
182597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
182697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind))
1827f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    return cxloc::translateSourceRange(getCursorContext(C),
182897b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor                                getCursorExpr(C)->getSourceRange());
182933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
183033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (clang_isStatement(C.kind))
1831f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    return cxloc::translateSourceRange(getCursorContext(C),
183233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor                                getCursorStmt(C)->getSourceRange());
1833f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
18349f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  if (C.kind == CXCursor_PreprocessingDirective) {
18359f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    SourceRange R = cxcursor::getCursorPreprocessingDirective(C);
18369f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    return cxloc::translateSourceRange(getCursorContext(C), R);
18379f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  }
18384807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor
18394807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  if (C.kind == CXCursor_MacroInstantiation) {
18404ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor    SourceRange R = cxcursor::getCursorMacroInstantiation(C)->getSourceRange();
18414807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor    return cxloc::translateSourceRange(getCursorContext(C), R);
18424807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  }
1843572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
1844572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition) {
1845572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    SourceRange R = cxcursor::getCursorMacroDefinition(C)->getSourceRange();
1846572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return cxloc::translateSourceRange(getCursorContext(C), R);
1847572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  }
18489f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor
18495352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  if (!getCursorDecl(C))
18505352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
1851f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1852a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  Decl *D = getCursorDecl(C);
18532ca54feee89d7277fb967e3247a64f40ef155a82Douglas Gregor  return cxloc::translateSourceRange(getCursorContext(C), D->getSourceRange());
1854a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor}
1855c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor
1856c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas GregorCXCursor clang_getCursorReferenced(CXCursor C) {
1857b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (clang_isInvalid(C.kind))
1858b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getNullCursor();
1859f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1860b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  ASTUnit *CXXUnit = getCursorASTUnit(C);
1861b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (clang_isDeclaration(C.kind))
1862c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    return C;
1863f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
186497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind)) {
186597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    Decl *D = getDeclFromExpr(getCursorExpr(C));
186697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    if (D)
1867b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor      return MakeCXCursor(D, CXXUnit);
186897b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    return clang_getNullCursor();
186997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  }
187097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
1871bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor  if (C.kind == CXCursor_MacroInstantiation) {
1872bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor    if (MacroDefinition *Def = getCursorMacroInstantiation(C)->getDefinition())
1873bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor      return MakeMacroDefinitionCursor(Def, CXXUnit);
1874bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor  }
1875bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor
1876c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  if (!clang_isReference(C.kind))
1877c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    return clang_getNullCursor();
1878f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1879c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  switch (C.kind) {
1880c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    case CXCursor_ObjCSuperClassRef:
1881b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor      return MakeCXCursor(getCursorObjCSuperClassRef(C).first, CXXUnit);
1882f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1883f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCProtocolRef: {
1884b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor      return MakeCXCursor(getCursorObjCProtocolRef(C).first, CXXUnit);
1885f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1886f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCClassRef:
1887b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor      return MakeCXCursor(getCursorObjCClassRef(C).first, CXXUnit);
18887d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
1889f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_TypeRef:
18907d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      return MakeCXCursor(getCursorTypeRef(C).first, CXXUnit);
1891f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1892c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    default:
1893c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      // We would prefer to enumerate all non-reference cursor kinds here.
1894c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      llvm_unreachable("Unhandled reference cursor kind");
1895c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      break;
1896c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    }
1897c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  }
1898f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1899c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  return clang_getNullCursor();
1900c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor}
1901c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor
1902b699866820102a69d83d6ac6941985c5ef4e8c40Douglas GregorCXCursor clang_getCursorDefinition(CXCursor C) {
1903b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (clang_isInvalid(C.kind))
1904b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getNullCursor();
1905f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1906b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  ASTUnit *CXXUnit = getCursorASTUnit(C);
1907f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1908b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  bool WasReference = false;
190997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isReference(C.kind) || clang_isExpression(C.kind)) {
1910b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    C = clang_getCursorReferenced(C);
1911b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    WasReference = true;
1912b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
1913b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
1914bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor  if (C.kind == CXCursor_MacroInstantiation)
1915bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor    return clang_getCursorReferenced(C);
1916bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor
1917b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!clang_isDeclaration(C.kind))
1918b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
1919b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
1920b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  Decl *D = getCursorDecl(C);
1921b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!D)
1922b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
1923f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1924b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  switch (D->getKind()) {
1925b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds that don't really separate the notions of
1926b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // declaration and definition.
1927b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Namespace:
1928b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Typedef:
1929b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TemplateTypeParm:
1930b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::EnumConstant:
1931b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Field:
1932b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCIvar:
1933b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCAtDefsField:
1934b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ImplicitParam:
1935b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ParmVar:
1936b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::NonTypeTemplateParm:
1937b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TemplateTemplateParm:
1938b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCategoryImpl:
1939b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCImplementation:
1940b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::LinkageSpec:
1941b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCPropertyImpl:
1942b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FileScopeAsm:
1943b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::StaticAssert:
1944b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Block:
1945b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return C;
1946b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
1947b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds that don't make any sense here, but are
1948b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // nonetheless harmless.
1949b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TranslationUnit:
1950b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    break;
1951b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
1952b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds for which the definition is not resolvable.
1953b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UnresolvedUsingTypename:
1954b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UnresolvedUsingValue:
1955b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    break;
1956b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
1957b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UsingDirective:
1958b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return MakeCXCursor(cast<UsingDirectiveDecl>(D)->getNominatedNamespace(),
1959b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                        CXXUnit);
1960b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
1961b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::NamespaceAlias:
1962b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return MakeCXCursor(cast<NamespaceAliasDecl>(D)->getNamespace(), CXXUnit);
1963b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
1964b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Enum:
1965b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Record:
1966b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXRecord:
1967b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplateSpecialization:
1968b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplatePartialSpecialization:
1969952b017601f9c82b51119c3a1600f1312a833db9Douglas Gregor    if (TagDecl *Def = cast<TagDecl>(D)->getDefinition())
1970b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor      return MakeCXCursor(Def, CXXUnit);
1971b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
1972b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
1973b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Function:
1974b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXMethod:
1975b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXConstructor:
1976b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXDestructor:
1977b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXConversion: {
1978b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    const FunctionDecl *Def = 0;
1979b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (cast<FunctionDecl>(D)->getBody(Def))
1980b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor      return MakeCXCursor(const_cast<FunctionDecl *>(Def), CXXUnit);
1981b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
1982b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
1983b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
1984b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Var: {
198531310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    // Ask the variable if it has a definition.
198631310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    if (VarDecl *Def = cast<VarDecl>(D)->getDefinition())
198731310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl      return MakeCXCursor(Def, CXXUnit);
198831310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    return clang_getNullCursor();
1989b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
1990f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1991b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FunctionTemplate: {
1992b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    const FunctionDecl *Def = 0;
1993b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (cast<FunctionTemplateDecl>(D)->getTemplatedDecl()->getBody(Def))
1994b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor      return MakeCXCursor(Def->getDescribedFunctionTemplate(), CXXUnit);
1995b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
1996b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
1997f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1998b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplate: {
1999b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (RecordDecl *Def = cast<ClassTemplateDecl>(D)->getTemplatedDecl()
2000952b017601f9c82b51119c3a1600f1312a833db9Douglas Gregor                                                            ->getDefinition())
2001b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      return MakeCXCursor(
2002f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                         cast<CXXRecordDecl>(Def)->getDescribedClassTemplate(),
2003b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                          CXXUnit);
2004b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
2005b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
2006b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2007b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Using: {
2008b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    UsingDecl *Using = cast<UsingDecl>(D);
2009b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    CXCursor Def = clang_getNullCursor();
2010f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    for (UsingDecl::shadow_iterator S = Using->shadow_begin(),
2011f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                 SEnd = Using->shadow_end();
2012b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor         S != SEnd; ++S) {
2013b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (Def != clang_getNullCursor()) {
2014b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor        // FIXME: We have no way to return multiple results.
2015b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor        return clang_getNullCursor();
2016b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      }
2017b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2018f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek      Def = clang_getCursorDefinition(MakeCXCursor((*S)->getTargetDecl(),
2019b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                                   CXXUnit));
2020b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    }
2021b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2022b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return Def;
2023b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
2024b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2025b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UsingShadow:
2026b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getCursorDefinition(
2027f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                       MakeCXCursor(cast<UsingShadowDecl>(D)->getTargetDecl(),
2028b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                    CXXUnit));
2029b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2030b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCMethod: {
2031b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    ObjCMethodDecl *Method = cast<ObjCMethodDecl>(D);
2032b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (Method->isThisDeclarationADefinition())
2033b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      return C;
2034b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2035b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // Dig out the method definition in the associated
2036b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // @implementation, if we have it.
2037b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: The ASTs should make finding the definition easier.
2038b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCInterfaceDecl *Class
2039b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                       = dyn_cast<ObjCInterfaceDecl>(Method->getDeclContext()))
2040b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (ObjCImplementationDecl *ClassImpl = Class->getImplementation())
2041b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor        if (ObjCMethodDecl *Def = ClassImpl->getMethod(Method->getSelector(),
2042b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                                                  Method->isInstanceMethod()))
2043b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor          if (Def->isThisDeclarationADefinition())
2044b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor            return MakeCXCursor(Def, CXXUnit);
2045b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2046b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
2047b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
2048b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2049b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCategory:
2050b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCCategoryImplDecl *Impl
2051b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                               = cast<ObjCCategoryDecl>(D)->getImplementation())
2052b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor      return MakeCXCursor(Impl, CXXUnit);
2053b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
2054b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2055b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCProtocol:
2056b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (!cast<ObjCProtocolDecl>(D)->isForwardDecl())
2057b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      return C;
2058b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
2059b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2060b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCInterface:
2061b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // There are two notions of a "definition" for an Objective-C
2062b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // class: the interface and its implementation. When we resolved a
2063b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // reference to an Objective-C class, produce the @interface as
2064b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // the definition; when we were provided with the interface,
2065b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // produce the @implementation as the definition.
2066b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (WasReference) {
2067b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (!cast<ObjCInterfaceDecl>(D)->isForwardDecl())
2068b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor        return C;
2069b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    } else if (ObjCImplementationDecl *Impl
2070b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                              = cast<ObjCInterfaceDecl>(D)->getImplementation())
2071b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor      return MakeCXCursor(Impl, CXXUnit);
2072b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
2073f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2074b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCProperty:
2075b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: We don't really know where to find the
2076b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // ObjCPropertyImplDecls that implement this property.
2077b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
2078b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2079b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCompatibleAlias:
2080b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCInterfaceDecl *Class
2081b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor          = cast<ObjCCompatibleAliasDecl>(D)->getClassInterface())
2082b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (!Class->isForwardDecl())
2083b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor        return MakeCXCursor(Class, CXXUnit);
2084f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2085b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
2086b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2087b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCForwardProtocol: {
2088b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    ObjCForwardProtocolDecl *Forward = cast<ObjCForwardProtocolDecl>(D);
2089b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (Forward->protocol_size() == 1)
2090b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      return clang_getCursorDefinition(
2091f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                     MakeCXCursor(*Forward->protocol_begin(),
2092b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                                  CXXUnit));
2093b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2094b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: Cannot return multiple definitions.
2095b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
2096b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
2097b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2098b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCClass: {
2099b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    ObjCClassDecl *Class = cast<ObjCClassDecl>(D);
2100b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (Class->size() == 1) {
2101b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      ObjCInterfaceDecl *IFace = Class->begin()->getInterface();
2102b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (!IFace->isForwardDecl())
2103b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor        return MakeCXCursor(IFace, CXXUnit);
2104b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      return clang_getNullCursor();
2105b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    }
2106b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2107b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: Cannot return multiple definitions.
2108b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
2109b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
2110b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2111b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Friend:
2112b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (NamedDecl *Friend = cast<FriendDecl>(D)->getFriendDecl())
2113b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor      return clang_getCursorDefinition(MakeCXCursor(Friend, CXXUnit));
2114b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
2115b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2116b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FriendTemplate:
2117b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (NamedDecl *Friend = cast<FriendTemplateDecl>(D)->getFriendDecl())
2118b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor      return clang_getCursorDefinition(MakeCXCursor(Friend, CXXUnit));
2119b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
2120b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
2121b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2122b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  return clang_getNullCursor();
2123b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor}
2124b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2125b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregorunsigned clang_isCursorDefinition(CXCursor C) {
2126b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!clang_isDeclaration(C.kind))
2127b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return 0;
2128b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
2129b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  return clang_getCursorDefinition(C) == C;
2130b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor}
2131b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
21320d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbarvoid clang_getDefinitionSpellingAndExtent(CXCursor C,
21334ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          const char **startBuf,
21344ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          const char **endBuf,
21354ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *startLine,
21364ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *startColumn,
21374ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *endLine,
21389ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          unsigned *endColumn) {
2139283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  assert(getCursorDecl(C) && "CXCursor has null decl");
2140283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  NamedDecl *ND = static_cast<NamedDecl *>(getCursorDecl(C));
21414ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  FunctionDecl *FD = dyn_cast<FunctionDecl>(ND);
21424ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  CompoundStmt *Body = dyn_cast<CompoundStmt>(FD->getBody());
2143f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
21444ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  SourceManager &SM = FD->getASTContext().getSourceManager();
21454ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startBuf = SM.getCharacterData(Body->getLBracLoc());
21464ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endBuf = SM.getCharacterData(Body->getRBracLoc());
21474ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startLine = SM.getSpellingLineNumber(Body->getLBracLoc());
21484ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startColumn = SM.getSpellingColumnNumber(Body->getLBracLoc());
21494ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endLine = SM.getSpellingLineNumber(Body->getRBracLoc());
21504ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endColumn = SM.getSpellingColumnNumber(Body->getRBracLoc());
21514ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff}
2152f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
21530a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregorvoid clang_enableStackTraces(void) {
21540a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  llvm::sys::PrintStackTraceOnErrorSignal();
21550a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor}
21560a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor
2157fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
2158fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2159fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2160fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor// Token-based Operations.
2161fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor//===----------------------------------------------------------------------===//
2162fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
2163fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor/* CXToken layout:
2164fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[0]: a CXTokenKind
2165fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[1]: starting token location
2166fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[2]: token length
2167fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[3]: reserved
2168f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek *   ptr_data: for identifiers and keywords, an IdentifierInfo*.
2169fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   otherwise unused.
2170fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor */
2171fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregorextern "C" {
2172fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
2173fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXTokenKind clang_getTokenKind(CXToken CXTok) {
2174fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  return static_cast<CXTokenKind>(CXTok.int_data[0]);
2175fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
2176fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
2177fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXString clang_getTokenSpelling(CXTranslationUnit TU, CXToken CXTok) {
2178fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  switch (clang_getTokenKind(CXTok)) {
2179fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Identifier:
2180fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Keyword:
2181fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // We know we have an IdentifierInfo*, so use that.
2182ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(static_cast<IdentifierInfo *>(CXTok.ptr_data)
2183ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek                            ->getNameStart());
2184fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
2185fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Literal: {
2186fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // We have stashed the starting pointer in the ptr_data field. Use it.
2187fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    const char *Text = static_cast<const char *>(CXTok.ptr_data);
2188ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(llvm::StringRef(Text, CXTok.int_data[2]));
2189fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  }
2190f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2191fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Punctuation:
2192fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Comment:
2193fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    break;
2194fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  }
2195f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2196f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // We have to find the starting buffer pointer the hard way, by
2197fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // deconstructing the source location.
2198fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU);
2199fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit)
2200ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
2201f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2202fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  SourceLocation Loc = SourceLocation::getFromRawEncoding(CXTok.int_data[1]);
2203fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> LocInfo
2204fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    = CXXUnit->getSourceManager().getDecomposedLoc(Loc);
2205f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  bool Invalid = false;
2206f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  llvm::StringRef Buffer
2207f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    = CXXUnit->getSourceManager().getBufferData(LocInfo.first, &Invalid);
2208f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  if (Invalid)
2209aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor    return createCXString("");
2210fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
2211f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  return createCXString(Buffer.substr(LocInfo.second, CXTok.int_data[2]));
2212fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
2213f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2214fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXSourceLocation clang_getTokenLocation(CXTranslationUnit TU, CXToken CXTok) {
2215fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU);
2216fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit)
2217fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return clang_getNullLocation();
2218f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2219fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  return cxloc::translateSourceLocation(CXXUnit->getASTContext(),
2220fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                        SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
2221fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
2222fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
2223fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXSourceRange clang_getTokenExtent(CXTranslationUnit TU, CXToken CXTok) {
2224fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU);
22255352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  if (!CXXUnit)
22265352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
2227f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2228f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return cxloc::translateSourceRange(CXXUnit->getASTContext(),
2229fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                        SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
2230fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
2231f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2232fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregorvoid clang_tokenize(CXTranslationUnit TU, CXSourceRange Range,
2233fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                    CXToken **Tokens, unsigned *NumTokens) {
2234fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (Tokens)
2235fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    *Tokens = 0;
2236fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (NumTokens)
2237fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    *NumTokens = 0;
2238f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2239fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU);
2240fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit || !Tokens || !NumTokens)
2241fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
2242f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2243bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
2244bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor
224585b988fdfa6adab6d43e16efd19ad4f3f7e2b49bDaniel Dunbar  SourceRange R = cxloc::translateCXSourceRange(Range);
2246fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (R.isInvalid())
2247fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
2248f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2249fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  SourceManager &SourceMgr = CXXUnit->getSourceManager();
2250fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> BeginLocInfo
2251fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    = SourceMgr.getDecomposedLoc(R.getBegin());
2252fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> EndLocInfo
2253fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    = SourceMgr.getDecomposedLoc(R.getEnd());
2254f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2255fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // Cannot tokenize across files.
2256fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (BeginLocInfo.first != EndLocInfo.first)
2257fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
2258f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2259f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // Create a lexer
2260f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  bool Invalid = false;
2261f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  llvm::StringRef Buffer
2262f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
226347a3fcd4afe122b23f9e7b6148f147bfa460cfe8Douglas Gregor  if (Invalid)
226447a3fcd4afe122b23f9e7b6148f147bfa460cfe8Douglas Gregor    return;
2265aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor
2266fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
2267fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor            CXXUnit->getASTContext().getLangOptions(),
2268f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer            Buffer.begin(), Buffer.data() + BeginLocInfo.second, Buffer.end());
2269fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Lex.SetCommentRetentionState(true);
2270f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2271fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // Lex tokens until we hit the end of the range.
2272f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  const char *EffectiveBufferEnd = Buffer.data() + EndLocInfo.second;
2273fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  llvm::SmallVector<CXToken, 32> CXTokens;
2274fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Token Tok;
2275fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  do {
2276fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // Lex the next token
2277fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    Lex.LexFromRawLexer(Tok);
2278fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    if (Tok.is(tok::eof))
2279fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      break;
2280f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2281fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // Initialize the CXToken.
2282fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXToken CXTok;
2283f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2284fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    //   - Common fields
2285fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[1] = Tok.getLocation().getRawEncoding();
2286fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[2] = Tok.getLength();
2287fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[3] = 0;
2288f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2289fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    //   - Kind-specific fields
2290fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    if (Tok.isLiteral()) {
2291fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Literal;
2292fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = (void *)Tok.getLiteralData();
2293fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else if (Tok.is(tok::identifier)) {
2294aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor      // Lookup the identifier to determine whether we have a keyword.
2295fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      std::pair<FileID, unsigned> LocInfo
2296fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor        = SourceMgr.getDecomposedLoc(Tok.getLocation());
2297f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor      bool Invalid = false;
2298f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer      llvm::StringRef Buf
2299f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor        = CXXUnit->getSourceManager().getBufferData(LocInfo.first, &Invalid);
2300f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor      if (Invalid)
2301aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor        return;
2302aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor
2303f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer      const char *StartPos = Buf.data() + LocInfo.second;
2304fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      IdentifierInfo *II
2305fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor        = CXXUnit->getPreprocessor().LookUpIdentifierInfo(Tok, StartPos);
2306aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek
2307aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      if (II->getObjCKeywordID() != tok::objc_not_keyword) {
2308aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek        CXTok.int_data[0] = CXToken_Keyword;
2309aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      }
2310aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      else {
2311aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek        CXTok.int_data[0] = II->getTokenID() == tok::identifier?
2312aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek                                CXToken_Identifier
2313aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek                              : CXToken_Keyword;
2314aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      }
2315fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = II;
2316fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else if (Tok.is(tok::comment)) {
2317fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Comment;
2318fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = 0;
2319fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else {
2320fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Punctuation;
2321fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = 0;
2322fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    }
2323fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTokens.push_back(CXTok);
2324fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  } while (Lex.getBufferLocation() <= EffectiveBufferEnd);
2325f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2326fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (CXTokens.empty())
2327fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
2328f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2329fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  *Tokens = (CXToken *)malloc(sizeof(CXToken) * CXTokens.size());
2330fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  memmove(*Tokens, CXTokens.data(), sizeof(CXToken) * CXTokens.size());
2331fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  *NumTokens = CXTokens.size();
2332fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
23330045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
23346db610934bedc6896393c1e1099525b35380acd6Ted Kremenekvoid clang_disposeTokens(CXTranslationUnit TU,
23356db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                         CXToken *Tokens, unsigned NumTokens) {
23366db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  free(Tokens);
23376db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
23386db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
23396db610934bedc6896393c1e1099525b35380acd6Ted Kremenek} // end: extern "C"
23406db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
23416db610934bedc6896393c1e1099525b35380acd6Ted Kremenek//===----------------------------------------------------------------------===//
23426db610934bedc6896393c1e1099525b35380acd6Ted Kremenek// Token annotation APIs.
23436db610934bedc6896393c1e1099525b35380acd6Ted Kremenek//===----------------------------------------------------------------------===//
23446db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
23450045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregortypedef llvm::DenseMap<unsigned, CXCursor> AnnotateTokensData;
2346fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekstatic enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
2347fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                     CXCursor parent,
2348fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                     CXClientData client_data);
23496db610934bedc6896393c1e1099525b35380acd6Ted Kremeneknamespace {
23506db610934bedc6896393c1e1099525b35380acd6Ted Kremenekclass AnnotateTokensWorker {
23516db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  AnnotateTokensData &Annotated;
235211949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  CXToken *Tokens;
235311949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  CXCursor *Cursors;
235411949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  unsigned NumTokens;
2355fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  unsigned TokIdx;
2356fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CursorVisitor AnnotateVis;
2357fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceManager &SrcMgr;
2358fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
2359fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  bool MoreTokens() const { return TokIdx < NumTokens; }
2360fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  unsigned NextToken() const { return TokIdx; }
2361fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void AdvanceToken() { ++TokIdx; }
2362fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceLocation GetTokenLoc(unsigned tokI) {
2363fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[1]);
2364fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
2365fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
23666db610934bedc6896393c1e1099525b35380acd6Ted Kremenekpublic:
236711949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  AnnotateTokensWorker(AnnotateTokensData &annotated,
2368fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                       CXToken *tokens, CXCursor *cursors, unsigned numTokens,
2369fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                       ASTUnit *CXXUnit, SourceRange RegionOfInterest)
237011949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek    : Annotated(annotated), Tokens(tokens), Cursors(cursors),
2371fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      NumTokens(numTokens), TokIdx(0),
2372fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      AnnotateVis(CXXUnit, AnnotateTokensVisitor, this,
2373fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                  Decl::MaxPCHLevel, RegionOfInterest),
2374fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      SrcMgr(CXXUnit->getSourceManager()) {}
237511949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
2376fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void VisitChildren(CXCursor C) { AnnotateVis.VisitChildren(C); }
23776db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  enum CXChildVisitResult Visit(CXCursor cursor, CXCursor parent);
2378fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void AnnotateTokens(CXCursor parent);
23796db610934bedc6896393c1e1099525b35380acd6Ted Kremenek};
23806db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
23810045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
2382fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekvoid AnnotateTokensWorker::AnnotateTokens(CXCursor parent) {
2383fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Walk the AST within the region of interest, annotating tokens
2384fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // along the way.
2385fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  VisitChildren(parent);
2386fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
2387fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = 0 ; I < TokIdx ; ++I) {
238811949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek    AnnotateTokensData::iterator Pos = Annotated.find(Tokens[I].int_data[1]);
2389fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    if (Pos != Annotated.end())
2390fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      Cursors[I] = Pos->second;
2391fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
2392fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
2393fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Finish up annotating any tokens left.
2394fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (!MoreTokens())
2395fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return;
239611949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
2397fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const CXCursor &C = clang_getNullCursor();
2398fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = TokIdx ; I < NumTokens ; ++I) {
2399fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    AnnotateTokensData::iterator Pos = Annotated.find(Tokens[I].int_data[1]);
2400fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = (Pos == Annotated.end()) ? C : Pos->second;
240111949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  }
240211949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek}
240311949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
24046db610934bedc6896393c1e1099525b35380acd6Ted Kremenekenum CXChildVisitResult
24056db610934bedc6896393c1e1099525b35380acd6Ted KremenekAnnotateTokensWorker::Visit(CXCursor cursor, CXCursor parent) {
2406fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CXSourceLocation Loc = clang_getCursorLocation(cursor);
2407fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // We can always annotate a preprocessing directive/macro instantiation.
2408fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (clang_isPreprocessing(cursor.kind)) {
2409fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Annotated[Loc.int_data] = cursor;
24100045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor    return CXChildVisit_Recurse;
24110045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  }
2412fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
2413fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CXSourceRange cursorExtent = clang_getCursorExtent(cursor);
2414fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceRange cursorRange = cxloc::translateCXSourceRange(cursorExtent);
2415fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
2416fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (cursorRange.isInvalid())
2417fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return CXChildVisit_Continue;
2418fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
2419fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceLocation L = SourceLocation::getFromRawEncoding(Loc.int_data);
2420fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
2421fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const enum CXCursorKind K = clang_getCursorKind(parent);
2422fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const CXCursor updateC =
2423fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    (clang_isInvalid(K) || K == CXCursor_TranslationUnit ||
2424fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek     L.isMacroID())
2425fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    ? clang_getNullCursor() : parent;
2426fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
2427fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  while (MoreTokens()) {
2428fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    const unsigned I = NextToken();
2429fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    SourceLocation TokLoc = GetTokenLoc(I);
2430fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
2431fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeBefore:
2432fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        Cursors[I] = updateC;
2433fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        AdvanceToken();
2434fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        continue;
2435fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeAfter:
2436fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        return CXChildVisit_Continue;
2437fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeOverlap:
2438fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        break;
2439fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    }
2440fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    break;
2441fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
2442fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
2443fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Visit children to get their cursor information.
2444fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned BeforeChildren = NextToken();
2445fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  VisitChildren(cursor);
2446fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned AfterChildren = NextToken();
2447fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
2448fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Adjust 'Last' to the last token within the extent of the cursor.
2449fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  while (MoreTokens()) {
2450fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    const unsigned I = NextToken();
2451fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    SourceLocation TokLoc = GetTokenLoc(I);
2452fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
2453fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeBefore:
2454fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        assert(0 && "Infeasible");
2455fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeAfter:
2456fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        break;
2457fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      case RangeOverlap:
2458fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        Cursors[I] = updateC;
2459fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        AdvanceToken();
2460fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek        continue;
2461fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    }
2462fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    break;
2463fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
2464fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned Last = NextToken();
24656db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
2466fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Scan the tokens that are at the beginning of the cursor, but are not
2467fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // capture by the child cursors.
2468fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
2469fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // For AST elements within macros, rely on a post-annotate pass to
2470fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // to correctly annotate the tokens with cursors.  Otherwise we can
2471fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // get confusing results of having tokens that map to cursors that really
2472fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // are expanded by an instantiation.
2473fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (L.isMacroID())
2474fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    cursor = clang_getNullCursor();
2475fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
2476fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = BeforeChildren; I != AfterChildren; ++I) {
2477fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    if (!clang_isInvalid(clang_getCursorKind(Cursors[I])))
2478fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      break;
2479fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = cursor;
2480fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
2481fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Scan the tokens that are at the end of the cursor, but are not captured
2482fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // but the child cursors.
2483fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = AfterChildren; I != Last; ++I)
2484fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = cursor;
2485fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
2486fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  TokIdx = Last;
2487fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  return CXChildVisit_Continue;
24880045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor}
24890045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
24906db610934bedc6896393c1e1099525b35380acd6Ted Kremenekstatic enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
24916db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                                     CXCursor parent,
24926db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                                     CXClientData client_data) {
24936db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  return static_cast<AnnotateTokensWorker*>(client_data)->Visit(cursor, parent);
24946db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
24956db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
24966db610934bedc6896393c1e1099525b35380acd6Ted Kremenekextern "C" {
24976db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
2498fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregorvoid clang_annotateTokens(CXTranslationUnit TU,
2499fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                          CXToken *Tokens, unsigned NumTokens,
2500fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                          CXCursor *Cursors) {
2501fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
2502fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (NumTokens == 0 || !Tokens || !Cursors)
25030045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor    return;
2504fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
25050045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU);
2506fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (!CXXUnit) {
2507fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    // Any token we don't specifically annotate will have a NULL cursor.
2508fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    const CXCursor &C = clang_getNullCursor();
2509fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    for (unsigned I = 0; I != NumTokens; ++I)
2510fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      Cursors[I] = C;
25110045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor    return;
2512fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
2513fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
2514bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
2515fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
25160396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // Determine the region of interest, which contains all of the tokens.
25170045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  SourceRange RegionOfInterest;
2518fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  RegionOfInterest.setBegin(cxloc::translateSourceLocation(
2519fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                        clang_getTokenLocation(TU, Tokens[0])));
2520fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
25210045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  SourceLocation End
2522fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    = cxloc::translateSourceLocation(clang_getTokenLocation(TU,
2523fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                        Tokens[NumTokens - 1]));
2524d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar  RegionOfInterest.setEnd(CXXUnit->getPreprocessor().getLocForEndOfToken(End));
2525fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
25260396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // A mapping from the source locations found when re-lexing or traversing the
25270396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // region of interest to the corresponding cursors.
25280045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  AnnotateTokensData Annotated;
2529fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
2530fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Relex the tokens within the source range to look for preprocessing
25310396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // directives.
25329f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  SourceManager &SourceMgr = CXXUnit->getSourceManager();
25339f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  std::pair<FileID, unsigned> BeginLocInfo
25349f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    = SourceMgr.getDecomposedLoc(RegionOfInterest.getBegin());
25359f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  std::pair<FileID, unsigned> EndLocInfo
25369f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    = SourceMgr.getDecomposedLoc(RegionOfInterest.getEnd());
2537fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
25389f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  llvm::StringRef Buffer;
25390396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  bool Invalid = false;
25400396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  if (BeginLocInfo.first == EndLocInfo.first &&
25410396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor      ((Buffer = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid)),true) &&
25420396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor      !Invalid) {
25439f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
25449f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor              CXXUnit->getASTContext().getLangOptions(),
2545fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek              Buffer.begin(), Buffer.data() + BeginLocInfo.second,
25464ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor              Buffer.end());
25479f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    Lex.SetCommentRetentionState(true);
2548fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
2549fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    // Lex tokens in raw mode until we hit the end of the range, to avoid
25509f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    // entering #includes or expanding macros.
25514807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor    while (true) {
25529f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor      Token Tok;
25539f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor      Lex.LexFromRawLexer(Tok);
2554fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
25559f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    reprocess:
25569f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor      if (Tok.is(tok::hash) && Tok.isAtStartOfLine()) {
25579f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        // We have found a preprocessing directive. Gobble it up so that we
25589f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        // don't see it while preprocessing these tokens later, but keep track of
25599f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        // all of the token locations inside this preprocessing directive so that
25609f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        // we can annotate them appropriately.
25619f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        //
25629f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        // FIXME: Some simple tests here could identify macro definitions and
25639f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        // #undefs, to provide specific cursor kinds for those.
25649f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        std::vector<SourceLocation> Locations;
25659f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        do {
25669f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor          Locations.push_back(Tok.getLocation());
2567fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek          Lex.LexFromRawLexer(Tok);
25689f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        } while (!Tok.isAtStartOfLine() && !Tok.is(tok::eof));
2569fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
25709f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        using namespace cxcursor;
25719f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        CXCursor Cursor
2572fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek          = MakePreprocessingDirectiveCursor(SourceRange(Locations.front(),
2573fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                         Locations.back()),
25746db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                           CXXUnit);
25759f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        for (unsigned I = 0, N = Locations.size(); I != N; ++I) {
25769f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor          Annotated[Locations[I].getRawEncoding()] = Cursor;
25779f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        }
2578fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
25799f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        if (Tok.isAtStartOfLine())
25809f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor          goto reprocess;
2581fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
25829f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        continue;
25839f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor      }
2584fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
25854807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor      if (Tok.is(tok::eof))
25869f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor        break;
25879f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    }
25884ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor  }
2589fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
25900396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // Annotate all of the source locations in the region of interest that map to
2591fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // a specific cursor.
2592fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  AnnotateTokensWorker W(Annotated, Tokens, Cursors, NumTokens,
2593fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                         CXXUnit, RegionOfInterest);
2594fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  W.AnnotateTokens(clang_getTranslationUnitCursor(CXXUnit));
2595fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
2596fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor} // end: extern "C"
2597fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
2598fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor//===----------------------------------------------------------------------===//
259916b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek// Operations for querying linkage of a cursor.
260016b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek//===----------------------------------------------------------------------===//
260116b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
260216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenekextern "C" {
260316b4259aecaa22b642d35d36fd89965ed700c1e0Ted KremenekCXLinkageKind clang_getCursorLinkage(CXCursor cursor) {
26040396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  if (!clang_isDeclaration(cursor.kind))
26050396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    return CXLinkage_Invalid;
26060396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor
260716b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  Decl *D = cxcursor::getCursorDecl(cursor);
260816b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  if (NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D))
260916b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek    switch (ND->getLinkage()) {
261016b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case NoLinkage: return CXLinkage_NoLinkage;
261116b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case InternalLinkage: return CXLinkage_Internal;
261216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case UniqueExternalLinkage: return CXLinkage_UniqueExternal;
261316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case ExternalLinkage: return CXLinkage_External;
261416b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek    };
261516b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
261616b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  return CXLinkage_Invalid;
261716b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek}
261816b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek} // end: extern "C"
261916b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
262016b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek//===----------------------------------------------------------------------===//
262145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek// Operations for querying language of a cursor.
262245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek//===----------------------------------------------------------------------===//
262345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
262445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenekstatic CXLanguageKind getDeclLanguage(const Decl *D) {
262545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  switch (D->getKind()) {
262645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    default:
262745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      break;
262845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ImplicitParam:
262945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCAtDefsField:
263045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCategory:
263145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCategoryImpl:
263245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCClass:
263345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCompatibleAlias:
263445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCForwardProtocol:
263545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCImplementation:
263645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCInterface:
263745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCIvar:
263845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCMethod:
263945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCProperty:
264045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCPropertyImpl:
264145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCProtocol:
264245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      return CXLanguage_ObjC;
264345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXConstructor:
264445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXConversion:
264545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXDestructor:
264645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXMethod:
264745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXRecord:
264845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplate:
264945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplatePartialSpecialization:
265045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplateSpecialization:
265145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Friend:
265245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::FriendTemplate:
265345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::FunctionTemplate:
265445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::LinkageSpec:
265545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Namespace:
265645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::NamespaceAlias:
265745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::NonTypeTemplateParm:
265845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::StaticAssert:
265945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::TemplateTemplateParm:
266045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::TemplateTypeParm:
266145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UnresolvedUsingTypename:
266245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UnresolvedUsingValue:
266345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Using:
266445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UsingDirective:
266545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UsingShadow:
266645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      return CXLanguage_CPlusPlus;
266745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  }
266845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
266945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  return CXLanguage_C;
267045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek}
267145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
267245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenekextern "C" {
267345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted KremenekCXLanguageKind clang_getCursorLanguage(CXCursor cursor) {
267445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  if (clang_isDeclaration(cursor.kind))
267545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    return getDeclLanguage(cxcursor::getCursorDecl(cursor));
267645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
267745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  return CXLanguage_Invalid;
267845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek}
267945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek} // end: extern "C"
268045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
268145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek//===----------------------------------------------------------------------===//
2682fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXString Operations.
2683fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2684fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2685fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
2686fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekconst char *clang_getCString(CXString string) {
2687fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  return string.Spelling;
2688fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek}
26894ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff
2690fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekvoid clang_disposeString(CXString string) {
2691fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (string.MustFreeString && string.Spelling)
2692fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    free((void*)string.Spelling);
2693fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek}
269404bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
2695fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
269604bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
2697ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremeneknamespace clang { namespace cxstring {
2698ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted KremenekCXString createCXString(const char *String, bool DupString){
2699ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  CXString Str;
2700ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  if (DupString) {
2701ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    Str.Spelling = strdup(String);
2702ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    Str.MustFreeString = 1;
2703ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  } else {
2704ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    Str.Spelling = String;
2705ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    Str.MustFreeString = 0;
2706ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  }
2707ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return Str;
2708ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek}
2709ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek
2710ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted KremenekCXString createCXString(llvm::StringRef String, bool DupString) {
2711ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  CXString Result;
2712ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  if (DupString || (!String.empty() && String.data()[String.size()] != 0)) {
2713ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    char *Spelling = (char *)malloc(String.size() + 1);
2714ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    memmove(Spelling, String.data(), String.size());
2715ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    Spelling[String.size()] = 0;
2716ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    Result.Spelling = Spelling;
2717ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    Result.MustFreeString = 1;
2718ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  } else {
2719ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    Result.Spelling = String.data();
2720ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    Result.MustFreeString = 0;
2721ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  }
2722ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return Result;
2723ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek}
2724ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek}}
2725ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek
272604bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek//===----------------------------------------------------------------------===//
272704bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek// Misc. utility functions.
272804bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek//===----------------------------------------------------------------------===//
2729f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
273004bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenekextern "C" {
273104bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
2732a2a9d6e4e5b6001b86b7dfc5db1ea296ce29a3d3Ted KremenekCXString clang_getClangVersion() {
2733ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString(getClangFullVersion());
273404bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek}
273504bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
273604bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek} // end: extern "C"
2737