blob: 994386ea4ac89f890f0f0ba6304a9175d84c6ff6 [file] [log] [blame]
Guy Benyei11169dd2012-12-18 14:30:41 +00001//===- CIndex.cpp - Clang-C Source Indexing Library -----------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the main API hooks in the Clang-C Source Indexing
11// library.
12//
13//===----------------------------------------------------------------------===//
14
15#include "CIndexer.h"
16#include "CIndexDiagnostic.h"
Chandler Carruth4b417452013-01-19 08:09:44 +000017#include "CLog.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000018#include "CXCursor.h"
19#include "CXSourceLocation.h"
20#include "CXString.h"
21#include "CXTranslationUnit.h"
22#include "CXType.h"
23#include "CursorVisitor.h"
David Blaikie0a4e61f2013-09-13 18:32:52 +000024#include "clang/AST/Attr.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000025#include "clang/AST/StmtVisitor.h"
26#include "clang/Basic/Diagnostic.h"
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +000027#include "clang/Basic/DiagnosticCategories.h"
28#include "clang/Basic/DiagnosticIDs.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000029#include "clang/Basic/Version.h"
30#include "clang/Frontend/ASTUnit.h"
31#include "clang/Frontend/CompilerInstance.h"
32#include "clang/Frontend/FrontendDiagnostic.h"
Argyrios Kyrtzidisca741ce2016-02-14 22:30:14 +000033#include "clang/Index/CodegenNameGenerator.h"
Dmitri Gribenko9e605112013-11-13 22:16:51 +000034#include "clang/Index/CommentToXML.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000035#include "clang/Lex/HeaderSearch.h"
36#include "clang/Lex/Lexer.h"
37#include "clang/Lex/PreprocessingRecord.h"
38#include "clang/Lex/Preprocessor.h"
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +000039#include "clang/Serialization/SerializationDiagnostic.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000040#include "llvm/ADT/Optional.h"
41#include "llvm/ADT/STLExtras.h"
42#include "llvm/ADT/StringSwitch.h"
Alp Toker1d257e12014-06-04 03:28:55 +000043#include "llvm/Config/llvm-config.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000044#include "llvm/Support/Compiler.h"
45#include "llvm/Support/CrashRecoveryContext.h"
Chandler Carruth4b417452013-01-19 08:09:44 +000046#include "llvm/Support/Format.h"
Chandler Carruth37ad2582014-06-27 15:14:39 +000047#include "llvm/Support/ManagedStatic.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000048#include "llvm/Support/MemoryBuffer.h"
49#include "llvm/Support/Mutex.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000050#include "llvm/Support/Program.h"
51#include "llvm/Support/SaveAndRestore.h"
52#include "llvm/Support/Signals.h"
Adrian Prantlbc068582015-07-08 01:00:30 +000053#include "llvm/Support/TargetSelect.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000054#include "llvm/Support/Threading.h"
55#include "llvm/Support/Timer.h"
56#include "llvm/Support/raw_ostream.h"
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +000057
Alp Toker1a86ad22014-07-06 06:24:00 +000058#if LLVM_ENABLE_THREADS != 0 && defined(__APPLE__)
59#define USE_DARWIN_THREADS
60#endif
61
62#ifdef USE_DARWIN_THREADS
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +000063#include <pthread.h>
64#endif
Guy Benyei11169dd2012-12-18 14:30:41 +000065
66using namespace clang;
67using namespace clang::cxcursor;
Guy Benyei11169dd2012-12-18 14:30:41 +000068using namespace clang::cxtu;
69using namespace clang::cxindex;
70
Dmitri Gribenkod36209e2013-01-26 21:32:42 +000071CXTranslationUnit cxtu::MakeCXTranslationUnit(CIndexer *CIdx, ASTUnit *AU) {
72 if (!AU)
Craig Topper69186e72014-06-08 08:38:04 +000073 return nullptr;
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +000074 assert(CIdx);
Guy Benyei11169dd2012-12-18 14:30:41 +000075 CXTranslationUnit D = new CXTranslationUnitImpl();
76 D->CIdx = CIdx;
Dmitri Gribenkod36209e2013-01-26 21:32:42 +000077 D->TheASTUnit = AU;
Dmitri Gribenko74895212013-02-03 13:52:47 +000078 D->StringPool = new cxstring::CXStringPool();
Craig Topper69186e72014-06-08 08:38:04 +000079 D->Diagnostics = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +000080 D->OverridenCursorsPool = createOverridenCXCursorsPool();
Craig Topper69186e72014-06-08 08:38:04 +000081 D->CommentToXML = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +000082 return D;
83}
84
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +000085bool cxtu::isASTReadError(ASTUnit *AU) {
86 for (ASTUnit::stored_diag_iterator D = AU->stored_diag_begin(),
87 DEnd = AU->stored_diag_end();
88 D != DEnd; ++D) {
89 if (D->getLevel() >= DiagnosticsEngine::Error &&
90 DiagnosticIDs::getCategoryNumberForDiag(D->getID()) ==
91 diag::DiagCat_AST_Deserialization_Issue)
92 return true;
93 }
94 return false;
95}
96
Guy Benyei11169dd2012-12-18 14:30:41 +000097cxtu::CXTUOwner::~CXTUOwner() {
98 if (TU)
99 clang_disposeTranslationUnit(TU);
100}
101
102/// \brief Compare two source ranges to determine their relative position in
103/// the translation unit.
104static RangeComparisonResult RangeCompare(SourceManager &SM,
105 SourceRange R1,
106 SourceRange R2) {
107 assert(R1.isValid() && "First range is invalid?");
108 assert(R2.isValid() && "Second range is invalid?");
109 if (R1.getEnd() != R2.getBegin() &&
110 SM.isBeforeInTranslationUnit(R1.getEnd(), R2.getBegin()))
111 return RangeBefore;
112 if (R2.getEnd() != R1.getBegin() &&
113 SM.isBeforeInTranslationUnit(R2.getEnd(), R1.getBegin()))
114 return RangeAfter;
115 return RangeOverlap;
116}
117
118/// \brief Determine if a source location falls within, before, or after a
119/// a given source range.
120static RangeComparisonResult LocationCompare(SourceManager &SM,
121 SourceLocation L, SourceRange R) {
122 assert(R.isValid() && "First range is invalid?");
123 assert(L.isValid() && "Second range is invalid?");
124 if (L == R.getBegin() || L == R.getEnd())
125 return RangeOverlap;
126 if (SM.isBeforeInTranslationUnit(L, R.getBegin()))
127 return RangeBefore;
128 if (SM.isBeforeInTranslationUnit(R.getEnd(), L))
129 return RangeAfter;
130 return RangeOverlap;
131}
132
133/// \brief Translate a Clang source range into a CIndex source range.
134///
135/// Clang internally represents ranges where the end location points to the
136/// start of the token at the end. However, for external clients it is more
137/// useful to have a CXSourceRange be a proper half-open interval. This routine
138/// does the appropriate translation.
139CXSourceRange cxloc::translateSourceRange(const SourceManager &SM,
140 const LangOptions &LangOpts,
141 const CharSourceRange &R) {
142 // We want the last character in this location, so we will adjust the
143 // location accordingly.
144 SourceLocation EndLoc = R.getEnd();
145 if (EndLoc.isValid() && EndLoc.isMacroID() && !SM.isMacroArgExpansion(EndLoc))
146 EndLoc = SM.getExpansionRange(EndLoc).second;
Yaron Keren8b563662015-10-03 10:46:20 +0000147 if (R.isTokenRange() && EndLoc.isValid()) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000148 unsigned Length = Lexer::MeasureTokenLength(SM.getSpellingLoc(EndLoc),
149 SM, LangOpts);
150 EndLoc = EndLoc.getLocWithOffset(Length);
151 }
152
Bill Wendlingeade3622013-01-23 08:25:41 +0000153 CXSourceRange Result = {
Dmitri Gribenkof9304482013-01-23 15:56:07 +0000154 { &SM, &LangOpts },
Bill Wendlingeade3622013-01-23 08:25:41 +0000155 R.getBegin().getRawEncoding(),
156 EndLoc.getRawEncoding()
157 };
Guy Benyei11169dd2012-12-18 14:30:41 +0000158 return Result;
159}
160
161//===----------------------------------------------------------------------===//
162// Cursor visitor.
163//===----------------------------------------------------------------------===//
164
165static SourceRange getRawCursorExtent(CXCursor C);
166static SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr);
167
168
169RangeComparisonResult CursorVisitor::CompareRegionOfInterest(SourceRange R) {
170 return RangeCompare(AU->getSourceManager(), R, RegionOfInterest);
171}
172
173/// \brief Visit the given cursor and, if requested by the visitor,
174/// its children.
175///
176/// \param Cursor the cursor to visit.
177///
178/// \param CheckedRegionOfInterest if true, then the caller already checked
179/// that this cursor is within the region of interest.
180///
181/// \returns true if the visitation should be aborted, false if it
182/// should continue.
183bool CursorVisitor::Visit(CXCursor Cursor, bool CheckedRegionOfInterest) {
184 if (clang_isInvalid(Cursor.kind))
185 return false;
186
187 if (clang_isDeclaration(Cursor.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +0000188 const Decl *D = getCursorDecl(Cursor);
Guy Benyei11169dd2012-12-18 14:30:41 +0000189 if (!D) {
190 assert(0 && "Invalid declaration cursor");
191 return true; // abort.
192 }
193
194 // Ignore implicit declarations, unless it's an objc method because
195 // currently we should report implicit methods for properties when indexing.
196 if (D->isImplicit() && !isa<ObjCMethodDecl>(D))
197 return false;
198 }
199
200 // If we have a range of interest, and this cursor doesn't intersect with it,
201 // we're done.
202 if (RegionOfInterest.isValid() && !CheckedRegionOfInterest) {
203 SourceRange Range = getRawCursorExtent(Cursor);
204 if (Range.isInvalid() || CompareRegionOfInterest(Range))
205 return false;
206 }
207
208 switch (Visitor(Cursor, Parent, ClientData)) {
209 case CXChildVisit_Break:
210 return true;
211
212 case CXChildVisit_Continue:
213 return false;
214
215 case CXChildVisit_Recurse: {
216 bool ret = VisitChildren(Cursor);
217 if (PostChildrenVisitor)
218 if (PostChildrenVisitor(Cursor, ClientData))
219 return true;
220 return ret;
221 }
222 }
223
224 llvm_unreachable("Invalid CXChildVisitResult!");
225}
226
227static bool visitPreprocessedEntitiesInRange(SourceRange R,
228 PreprocessingRecord &PPRec,
229 CursorVisitor &Visitor) {
230 SourceManager &SM = Visitor.getASTUnit()->getSourceManager();
231 FileID FID;
232
233 if (!Visitor.shouldVisitIncludedEntities()) {
234 // If the begin/end of the range lie in the same FileID, do the optimization
235 // where we skip preprocessed entities that do not come from the same FileID.
236 FID = SM.getFileID(SM.getFileLoc(R.getBegin()));
237 if (FID != SM.getFileID(SM.getFileLoc(R.getEnd())))
238 FID = FileID();
239 }
240
Benjamin Kramerb4ef6682015-02-06 17:25:10 +0000241 const auto &Entities = PPRec.getPreprocessedEntitiesInRange(R);
242 return Visitor.visitPreprocessedEntities(Entities.begin(), Entities.end(),
Guy Benyei11169dd2012-12-18 14:30:41 +0000243 PPRec, FID);
244}
245
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000246bool CursorVisitor::visitFileRegion() {
Guy Benyei11169dd2012-12-18 14:30:41 +0000247 if (RegionOfInterest.isInvalid())
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000248 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000249
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +0000250 ASTUnit *Unit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +0000251 SourceManager &SM = Unit->getSourceManager();
252
253 std::pair<FileID, unsigned>
254 Begin = SM.getDecomposedLoc(SM.getFileLoc(RegionOfInterest.getBegin())),
255 End = SM.getDecomposedLoc(SM.getFileLoc(RegionOfInterest.getEnd()));
256
257 if (End.first != Begin.first) {
258 // If the end does not reside in the same file, try to recover by
259 // picking the end of the file of begin location.
260 End.first = Begin.first;
261 End.second = SM.getFileIDSize(Begin.first);
262 }
263
264 assert(Begin.first == End.first);
265 if (Begin.second > End.second)
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000266 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000267
268 FileID File = Begin.first;
269 unsigned Offset = Begin.second;
270 unsigned Length = End.second - Begin.second;
271
272 if (!VisitDeclsOnly && !VisitPreprocessorLast)
273 if (visitPreprocessedEntitiesInRegion())
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000274 return true; // visitation break.
Guy Benyei11169dd2012-12-18 14:30:41 +0000275
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000276 if (visitDeclsFromFileRegion(File, Offset, Length))
277 return true; // visitation break.
Guy Benyei11169dd2012-12-18 14:30:41 +0000278
279 if (!VisitDeclsOnly && VisitPreprocessorLast)
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000280 return visitPreprocessedEntitiesInRegion();
281
282 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000283}
284
285static bool isInLexicalContext(Decl *D, DeclContext *DC) {
286 if (!DC)
287 return false;
288
289 for (DeclContext *DeclDC = D->getLexicalDeclContext();
290 DeclDC; DeclDC = DeclDC->getLexicalParent()) {
291 if (DeclDC == DC)
292 return true;
293 }
294 return false;
295}
296
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000297bool CursorVisitor::visitDeclsFromFileRegion(FileID File,
Guy Benyei11169dd2012-12-18 14:30:41 +0000298 unsigned Offset, unsigned Length) {
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +0000299 ASTUnit *Unit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +0000300 SourceManager &SM = Unit->getSourceManager();
301 SourceRange Range = RegionOfInterest;
302
303 SmallVector<Decl *, 16> Decls;
304 Unit->findFileRegionDecls(File, Offset, Length, Decls);
305
306 // If we didn't find any file level decls for the file, try looking at the
307 // file that it was included from.
308 while (Decls.empty() || Decls.front()->isTopLevelDeclInObjCContainer()) {
309 bool Invalid = false;
310 const SrcMgr::SLocEntry &SLEntry = SM.getSLocEntry(File, &Invalid);
311 if (Invalid)
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000312 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000313
314 SourceLocation Outer;
315 if (SLEntry.isFile())
316 Outer = SLEntry.getFile().getIncludeLoc();
317 else
318 Outer = SLEntry.getExpansion().getExpansionLocStart();
319 if (Outer.isInvalid())
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000320 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000321
Benjamin Kramer867ea1d2014-03-02 13:01:17 +0000322 std::tie(File, Offset) = SM.getDecomposedExpansionLoc(Outer);
Guy Benyei11169dd2012-12-18 14:30:41 +0000323 Length = 0;
324 Unit->findFileRegionDecls(File, Offset, Length, Decls);
325 }
326
327 assert(!Decls.empty());
328
329 bool VisitedAtLeastOnce = false;
Craig Topper69186e72014-06-08 08:38:04 +0000330 DeclContext *CurDC = nullptr;
Craig Topper2341c0d2013-07-04 03:08:24 +0000331 SmallVectorImpl<Decl *>::iterator DIt = Decls.begin();
332 for (SmallVectorImpl<Decl *>::iterator DE = Decls.end(); DIt != DE; ++DIt) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000333 Decl *D = *DIt;
334 if (D->getSourceRange().isInvalid())
335 continue;
336
337 if (isInLexicalContext(D, CurDC))
338 continue;
339
340 CurDC = dyn_cast<DeclContext>(D);
341
342 if (TagDecl *TD = dyn_cast<TagDecl>(D))
343 if (!TD->isFreeStanding())
344 continue;
345
346 RangeComparisonResult CompRes = RangeCompare(SM, D->getSourceRange(),Range);
347 if (CompRes == RangeBefore)
348 continue;
349 if (CompRes == RangeAfter)
350 break;
351
352 assert(CompRes == RangeOverlap);
353 VisitedAtLeastOnce = true;
354
355 if (isa<ObjCContainerDecl>(D)) {
356 FileDI_current = &DIt;
357 FileDE_current = DE;
358 } else {
Craig Topper69186e72014-06-08 08:38:04 +0000359 FileDI_current = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +0000360 }
361
362 if (Visit(MakeCXCursor(D, TU, Range), /*CheckedRegionOfInterest=*/true))
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000363 return true; // visitation break.
Guy Benyei11169dd2012-12-18 14:30:41 +0000364 }
365
366 if (VisitedAtLeastOnce)
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000367 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000368
369 // No Decls overlapped with the range. Move up the lexical context until there
370 // is a context that contains the range or we reach the translation unit
371 // level.
372 DeclContext *DC = DIt == Decls.begin() ? (*DIt)->getLexicalDeclContext()
373 : (*(DIt-1))->getLexicalDeclContext();
374
375 while (DC && !DC->isTranslationUnit()) {
376 Decl *D = cast<Decl>(DC);
377 SourceRange CurDeclRange = D->getSourceRange();
378 if (CurDeclRange.isInvalid())
379 break;
380
381 if (RangeCompare(SM, CurDeclRange, Range) == RangeOverlap) {
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000382 if (Visit(MakeCXCursor(D, TU, Range), /*CheckedRegionOfInterest=*/true))
383 return true; // visitation break.
Guy Benyei11169dd2012-12-18 14:30:41 +0000384 }
385
386 DC = D->getLexicalDeclContext();
387 }
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000388
389 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000390}
391
392bool CursorVisitor::visitPreprocessedEntitiesInRegion() {
393 if (!AU->getPreprocessor().getPreprocessingRecord())
394 return false;
395
396 PreprocessingRecord &PPRec
397 = *AU->getPreprocessor().getPreprocessingRecord();
398 SourceManager &SM = AU->getSourceManager();
399
400 if (RegionOfInterest.isValid()) {
401 SourceRange MappedRange = AU->mapRangeToPreamble(RegionOfInterest);
402 SourceLocation B = MappedRange.getBegin();
403 SourceLocation E = MappedRange.getEnd();
404
405 if (AU->isInPreambleFileID(B)) {
406 if (SM.isLoadedSourceLocation(E))
407 return visitPreprocessedEntitiesInRange(SourceRange(B, E),
408 PPRec, *this);
409
410 // Beginning of range lies in the preamble but it also extends beyond
411 // it into the main file. Split the range into 2 parts, one covering
412 // the preamble and another covering the main file. This allows subsequent
413 // calls to visitPreprocessedEntitiesInRange to accept a source range that
414 // lies in the same FileID, allowing it to skip preprocessed entities that
415 // do not come from the same FileID.
416 bool breaked =
417 visitPreprocessedEntitiesInRange(
418 SourceRange(B, AU->getEndOfPreambleFileID()),
419 PPRec, *this);
420 if (breaked) return true;
421 return visitPreprocessedEntitiesInRange(
422 SourceRange(AU->getStartOfMainFileID(), E),
423 PPRec, *this);
424 }
425
426 return visitPreprocessedEntitiesInRange(SourceRange(B, E), PPRec, *this);
427 }
428
429 bool OnlyLocalDecls
430 = !AU->isMainFileAST() && AU->getOnlyLocalDecls();
431
432 if (OnlyLocalDecls)
433 return visitPreprocessedEntities(PPRec.local_begin(), PPRec.local_end(),
434 PPRec);
435
436 return visitPreprocessedEntities(PPRec.begin(), PPRec.end(), PPRec);
437}
438
439template<typename InputIterator>
440bool CursorVisitor::visitPreprocessedEntities(InputIterator First,
441 InputIterator Last,
442 PreprocessingRecord &PPRec,
443 FileID FID) {
444 for (; First != Last; ++First) {
445 if (!FID.isInvalid() && !PPRec.isEntityInFileID(First, FID))
446 continue;
447
448 PreprocessedEntity *PPE = *First;
Argyrios Kyrtzidis1030f262013-05-07 20:37:17 +0000449 if (!PPE)
450 continue;
451
Guy Benyei11169dd2012-12-18 14:30:41 +0000452 if (MacroExpansion *ME = dyn_cast<MacroExpansion>(PPE)) {
453 if (Visit(MakeMacroExpansionCursor(ME, TU)))
454 return true;
Richard Smith66a81862015-05-04 02:25:31 +0000455
Guy Benyei11169dd2012-12-18 14:30:41 +0000456 continue;
457 }
Richard Smith66a81862015-05-04 02:25:31 +0000458
459 if (MacroDefinitionRecord *MD = dyn_cast<MacroDefinitionRecord>(PPE)) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000460 if (Visit(MakeMacroDefinitionCursor(MD, TU)))
461 return true;
Richard Smith66a81862015-05-04 02:25:31 +0000462
Guy Benyei11169dd2012-12-18 14:30:41 +0000463 continue;
464 }
465
466 if (InclusionDirective *ID = dyn_cast<InclusionDirective>(PPE)) {
467 if (Visit(MakeInclusionDirectiveCursor(ID, TU)))
468 return true;
469
470 continue;
471 }
472 }
473
474 return false;
475}
476
477/// \brief Visit the children of the given cursor.
478///
479/// \returns true if the visitation should be aborted, false if it
480/// should continue.
481bool CursorVisitor::VisitChildren(CXCursor Cursor) {
482 if (clang_isReference(Cursor.kind) &&
483 Cursor.kind != CXCursor_CXXBaseSpecifier) {
484 // By definition, references have no children.
485 return false;
486 }
487
488 // Set the Parent field to Cursor, then back to its old value once we're
489 // done.
490 SetParentRAII SetParent(Parent, StmtParent, Cursor);
491
492 if (clang_isDeclaration(Cursor.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +0000493 Decl *D = const_cast<Decl *>(getCursorDecl(Cursor));
Guy Benyei11169dd2012-12-18 14:30:41 +0000494 if (!D)
495 return false;
496
497 return VisitAttributes(D) || Visit(D);
498 }
499
500 if (clang_isStatement(Cursor.kind)) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +0000501 if (const Stmt *S = getCursorStmt(Cursor))
Guy Benyei11169dd2012-12-18 14:30:41 +0000502 return Visit(S);
503
504 return false;
505 }
506
507 if (clang_isExpression(Cursor.kind)) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +0000508 if (const Expr *E = getCursorExpr(Cursor))
Guy Benyei11169dd2012-12-18 14:30:41 +0000509 return Visit(E);
510
511 return false;
512 }
513
514 if (clang_isTranslationUnit(Cursor.kind)) {
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +0000515 CXTranslationUnit TU = getCursorTU(Cursor);
516 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +0000517
518 int VisitOrder[2] = { VisitPreprocessorLast, !VisitPreprocessorLast };
519 for (unsigned I = 0; I != 2; ++I) {
520 if (VisitOrder[I]) {
521 if (!CXXUnit->isMainFileAST() && CXXUnit->getOnlyLocalDecls() &&
522 RegionOfInterest.isInvalid()) {
523 for (ASTUnit::top_level_iterator TL = CXXUnit->top_level_begin(),
524 TLEnd = CXXUnit->top_level_end();
525 TL != TLEnd; ++TL) {
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +0000526 if (Visit(MakeCXCursor(*TL, TU, RegionOfInterest), true))
Guy Benyei11169dd2012-12-18 14:30:41 +0000527 return true;
528 }
529 } else if (VisitDeclContext(
530 CXXUnit->getASTContext().getTranslationUnitDecl()))
531 return true;
532 continue;
533 }
534
535 // Walk the preprocessing record.
536 if (CXXUnit->getPreprocessor().getPreprocessingRecord())
537 visitPreprocessedEntitiesInRegion();
538 }
539
540 return false;
541 }
542
543 if (Cursor.kind == CXCursor_CXXBaseSpecifier) {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +0000544 if (const CXXBaseSpecifier *Base = getCursorCXXBaseSpecifier(Cursor)) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000545 if (TypeSourceInfo *BaseTSInfo = Base->getTypeSourceInfo()) {
546 return Visit(BaseTSInfo->getTypeLoc());
547 }
548 }
549 }
550
551 if (Cursor.kind == CXCursor_IBOutletCollectionAttr) {
Dmitri Gribenkoe4baea62013-01-26 18:08:08 +0000552 const IBOutletCollectionAttr *A =
Guy Benyei11169dd2012-12-18 14:30:41 +0000553 cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(Cursor));
Richard Smithb1f9a282013-10-31 01:56:18 +0000554 if (const ObjCObjectType *ObjT = A->getInterface()->getAs<ObjCObjectType>())
Richard Smithb87c4652013-10-31 21:23:20 +0000555 return Visit(cxcursor::MakeCursorObjCClassRef(
556 ObjT->getInterface(),
557 A->getInterfaceLoc()->getTypeLoc().getLocStart(), TU));
Guy Benyei11169dd2012-12-18 14:30:41 +0000558 }
559
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +0000560 // If pointing inside a macro definition, check if the token is an identifier
561 // that was ever defined as a macro. In such a case, create a "pseudo" macro
562 // expansion cursor for that token.
563 SourceLocation BeginLoc = RegionOfInterest.getBegin();
564 if (Cursor.kind == CXCursor_MacroDefinition &&
565 BeginLoc == RegionOfInterest.getEnd()) {
566 SourceLocation Loc = AU->mapLocationToPreamble(BeginLoc);
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +0000567 const MacroInfo *MI =
568 getMacroInfo(cxcursor::getCursorMacroDefinition(Cursor), TU);
Richard Smith66a81862015-05-04 02:25:31 +0000569 if (MacroDefinitionRecord *MacroDef =
570 checkForMacroInMacroDefinition(MI, Loc, TU))
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +0000571 return Visit(cxcursor::MakeMacroExpansionCursor(MacroDef, BeginLoc, TU));
572 }
573
Guy Benyei11169dd2012-12-18 14:30:41 +0000574 // Nothing to visit at the moment.
575 return false;
576}
577
578bool CursorVisitor::VisitBlockDecl(BlockDecl *B) {
579 if (TypeSourceInfo *TSInfo = B->getSignatureAsWritten())
580 if (Visit(TSInfo->getTypeLoc()))
581 return true;
582
583 if (Stmt *Body = B->getBody())
584 return Visit(MakeCXCursor(Body, StmtParent, TU, RegionOfInterest));
585
586 return false;
587}
588
Ted Kremenek03325582013-02-21 01:29:01 +0000589Optional<bool> CursorVisitor::shouldVisitCursor(CXCursor Cursor) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000590 if (RegionOfInterest.isValid()) {
591 SourceRange Range = getFullCursorExtent(Cursor, AU->getSourceManager());
592 if (Range.isInvalid())
David Blaikie7a30dc52013-02-21 01:47:18 +0000593 return None;
Guy Benyei11169dd2012-12-18 14:30:41 +0000594
595 switch (CompareRegionOfInterest(Range)) {
596 case RangeBefore:
597 // This declaration comes before the region of interest; skip it.
David Blaikie7a30dc52013-02-21 01:47:18 +0000598 return None;
Guy Benyei11169dd2012-12-18 14:30:41 +0000599
600 case RangeAfter:
601 // This declaration comes after the region of interest; we're done.
602 return false;
603
604 case RangeOverlap:
605 // This declaration overlaps the region of interest; visit it.
606 break;
607 }
608 }
609 return true;
610}
611
612bool CursorVisitor::VisitDeclContext(DeclContext *DC) {
613 DeclContext::decl_iterator I = DC->decls_begin(), E = DC->decls_end();
614
615 // FIXME: Eventually remove. This part of a hack to support proper
616 // iteration over all Decls contained lexically within an ObjC container.
617 SaveAndRestore<DeclContext::decl_iterator*> DI_saved(DI_current, &I);
618 SaveAndRestore<DeclContext::decl_iterator> DE_saved(DE_current, E);
619
620 for ( ; I != E; ++I) {
621 Decl *D = *I;
622 if (D->getLexicalDeclContext() != DC)
623 continue;
624 CXCursor Cursor = MakeCXCursor(D, TU, RegionOfInterest);
625
626 // Ignore synthesized ivars here, otherwise if we have something like:
627 // @synthesize prop = _prop;
628 // and '_prop' is not declared, we will encounter a '_prop' ivar before
629 // encountering the 'prop' synthesize declaration and we will think that
630 // we passed the region-of-interest.
631 if (ObjCIvarDecl *ivarD = dyn_cast<ObjCIvarDecl>(D)) {
632 if (ivarD->getSynthesize())
633 continue;
634 }
635
636 // FIXME: ObjCClassRef/ObjCProtocolRef for forward class/protocol
637 // declarations is a mismatch with the compiler semantics.
638 if (Cursor.kind == CXCursor_ObjCInterfaceDecl) {
639 ObjCInterfaceDecl *ID = cast<ObjCInterfaceDecl>(D);
640 if (!ID->isThisDeclarationADefinition())
641 Cursor = MakeCursorObjCClassRef(ID, ID->getLocation(), TU);
642
643 } else if (Cursor.kind == CXCursor_ObjCProtocolDecl) {
644 ObjCProtocolDecl *PD = cast<ObjCProtocolDecl>(D);
645 if (!PD->isThisDeclarationADefinition())
646 Cursor = MakeCursorObjCProtocolRef(PD, PD->getLocation(), TU);
647 }
648
Ted Kremenek03325582013-02-21 01:29:01 +0000649 const Optional<bool> &V = shouldVisitCursor(Cursor);
Guy Benyei11169dd2012-12-18 14:30:41 +0000650 if (!V.hasValue())
651 continue;
652 if (!V.getValue())
653 return false;
654 if (Visit(Cursor, true))
655 return true;
656 }
657 return false;
658}
659
660bool CursorVisitor::VisitTranslationUnitDecl(TranslationUnitDecl *D) {
661 llvm_unreachable("Translation units are visited directly by Visit()");
662}
663
Sergey Kalinichev8f3b1872015-11-15 13:48:32 +0000664bool CursorVisitor::VisitTypeAliasTemplateDecl(TypeAliasTemplateDecl *D) {
665 if (VisitTemplateParameters(D->getTemplateParameters()))
666 return true;
667
668 return Visit(MakeCXCursor(D->getTemplatedDecl(), TU, RegionOfInterest));
669}
670
Guy Benyei11169dd2012-12-18 14:30:41 +0000671bool CursorVisitor::VisitTypeAliasDecl(TypeAliasDecl *D) {
672 if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
673 return Visit(TSInfo->getTypeLoc());
674
675 return false;
676}
677
678bool CursorVisitor::VisitTypedefDecl(TypedefDecl *D) {
679 if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
680 return Visit(TSInfo->getTypeLoc());
681
682 return false;
683}
684
685bool CursorVisitor::VisitTagDecl(TagDecl *D) {
686 return VisitDeclContext(D);
687}
688
689bool CursorVisitor::VisitClassTemplateSpecializationDecl(
690 ClassTemplateSpecializationDecl *D) {
691 bool ShouldVisitBody = false;
692 switch (D->getSpecializationKind()) {
693 case TSK_Undeclared:
694 case TSK_ImplicitInstantiation:
695 // Nothing to visit
696 return false;
697
698 case TSK_ExplicitInstantiationDeclaration:
699 case TSK_ExplicitInstantiationDefinition:
700 break;
701
702 case TSK_ExplicitSpecialization:
703 ShouldVisitBody = true;
704 break;
705 }
706
707 // Visit the template arguments used in the specialization.
708 if (TypeSourceInfo *SpecType = D->getTypeAsWritten()) {
709 TypeLoc TL = SpecType->getTypeLoc();
David Blaikie6adc78e2013-02-18 22:06:02 +0000710 if (TemplateSpecializationTypeLoc TSTLoc =
711 TL.getAs<TemplateSpecializationTypeLoc>()) {
712 for (unsigned I = 0, N = TSTLoc.getNumArgs(); I != N; ++I)
713 if (VisitTemplateArgumentLoc(TSTLoc.getArgLoc(I)))
Guy Benyei11169dd2012-12-18 14:30:41 +0000714 return true;
715 }
716 }
Alexander Kornienko1a9f1842015-12-28 15:24:08 +0000717
718 return ShouldVisitBody && VisitCXXRecordDecl(D);
Guy Benyei11169dd2012-12-18 14:30:41 +0000719}
720
721bool CursorVisitor::VisitClassTemplatePartialSpecializationDecl(
722 ClassTemplatePartialSpecializationDecl *D) {
723 // FIXME: Visit the "outer" template parameter lists on the TagDecl
724 // before visiting these template parameters.
725 if (VisitTemplateParameters(D->getTemplateParameters()))
726 return true;
727
728 // Visit the partial specialization arguments.
Enea Zaffanella6dbe1872013-08-10 07:24:53 +0000729 const ASTTemplateArgumentListInfo *Info = D->getTemplateArgsAsWritten();
730 const TemplateArgumentLoc *TemplateArgs = Info->getTemplateArgs();
731 for (unsigned I = 0, N = Info->NumTemplateArgs; I != N; ++I)
Guy Benyei11169dd2012-12-18 14:30:41 +0000732 if (VisitTemplateArgumentLoc(TemplateArgs[I]))
733 return true;
734
735 return VisitCXXRecordDecl(D);
736}
737
738bool CursorVisitor::VisitTemplateTypeParmDecl(TemplateTypeParmDecl *D) {
739 // Visit the default argument.
740 if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
741 if (TypeSourceInfo *DefArg = D->getDefaultArgumentInfo())
742 if (Visit(DefArg->getTypeLoc()))
743 return true;
744
745 return false;
746}
747
748bool CursorVisitor::VisitEnumConstantDecl(EnumConstantDecl *D) {
749 if (Expr *Init = D->getInitExpr())
750 return Visit(MakeCXCursor(Init, StmtParent, TU, RegionOfInterest));
751 return false;
752}
753
754bool CursorVisitor::VisitDeclaratorDecl(DeclaratorDecl *DD) {
Argyrios Kyrtzidis2ec76742013-04-05 21:04:10 +0000755 unsigned NumParamList = DD->getNumTemplateParameterLists();
756 for (unsigned i = 0; i < NumParamList; i++) {
757 TemplateParameterList* Params = DD->getTemplateParameterList(i);
758 if (VisitTemplateParameters(Params))
759 return true;
760 }
761
Guy Benyei11169dd2012-12-18 14:30:41 +0000762 if (TypeSourceInfo *TSInfo = DD->getTypeSourceInfo())
763 if (Visit(TSInfo->getTypeLoc()))
764 return true;
765
766 // Visit the nested-name-specifier, if present.
767 if (NestedNameSpecifierLoc QualifierLoc = DD->getQualifierLoc())
768 if (VisitNestedNameSpecifierLoc(QualifierLoc))
769 return true;
770
771 return false;
772}
773
Benjamin Kramer4cadf292014-03-07 21:51:58 +0000774/// \brief Compare two base or member initializers based on their source order.
775static int CompareCXXCtorInitializers(CXXCtorInitializer *const *X,
776 CXXCtorInitializer *const *Y) {
777 return (*X)->getSourceOrder() - (*Y)->getSourceOrder();
778}
779
Guy Benyei11169dd2012-12-18 14:30:41 +0000780bool CursorVisitor::VisitFunctionDecl(FunctionDecl *ND) {
Argyrios Kyrtzidis2ec76742013-04-05 21:04:10 +0000781 unsigned NumParamList = ND->getNumTemplateParameterLists();
782 for (unsigned i = 0; i < NumParamList; i++) {
783 TemplateParameterList* Params = ND->getTemplateParameterList(i);
784 if (VisitTemplateParameters(Params))
785 return true;
786 }
787
Guy Benyei11169dd2012-12-18 14:30:41 +0000788 if (TypeSourceInfo *TSInfo = ND->getTypeSourceInfo()) {
789 // Visit the function declaration's syntactic components in the order
790 // written. This requires a bit of work.
791 TypeLoc TL = TSInfo->getTypeLoc().IgnoreParens();
David Blaikie6adc78e2013-02-18 22:06:02 +0000792 FunctionTypeLoc FTL = TL.getAs<FunctionTypeLoc>();
Guy Benyei11169dd2012-12-18 14:30:41 +0000793
794 // If we have a function declared directly (without the use of a typedef),
795 // visit just the return type. Otherwise, just visit the function's type
796 // now.
Alp Toker42a16a62014-01-25 23:51:36 +0000797 if ((FTL && !isa<CXXConversionDecl>(ND) && Visit(FTL.getReturnLoc())) ||
Guy Benyei11169dd2012-12-18 14:30:41 +0000798 (!FTL && Visit(TL)))
799 return true;
800
801 // Visit the nested-name-specifier, if present.
802 if (NestedNameSpecifierLoc QualifierLoc = ND->getQualifierLoc())
803 if (VisitNestedNameSpecifierLoc(QualifierLoc))
804 return true;
805
806 // Visit the declaration name.
Argyrios Kyrtzidis4a4d2b42014-02-09 08:13:47 +0000807 if (!isa<CXXDestructorDecl>(ND))
808 if (VisitDeclarationNameInfo(ND->getNameInfo()))
809 return true;
Guy Benyei11169dd2012-12-18 14:30:41 +0000810
811 // FIXME: Visit explicitly-specified template arguments!
812
813 // Visit the function parameters, if we have a function type.
David Blaikie6adc78e2013-02-18 22:06:02 +0000814 if (FTL && VisitFunctionTypeLoc(FTL, true))
Guy Benyei11169dd2012-12-18 14:30:41 +0000815 return true;
816
Bill Wendling44426052012-12-20 19:22:21 +0000817 // FIXME: Attributes?
Guy Benyei11169dd2012-12-18 14:30:41 +0000818 }
819
820 if (ND->doesThisDeclarationHaveABody() && !ND->isLateTemplateParsed()) {
821 if (CXXConstructorDecl *Constructor = dyn_cast<CXXConstructorDecl>(ND)) {
822 // Find the initializers that were written in the source.
823 SmallVector<CXXCtorInitializer *, 4> WrittenInits;
Aaron Ballman0ad78302014-03-13 17:34:31 +0000824 for (auto *I : Constructor->inits()) {
825 if (!I->isWritten())
Guy Benyei11169dd2012-12-18 14:30:41 +0000826 continue;
827
Aaron Ballman0ad78302014-03-13 17:34:31 +0000828 WrittenInits.push_back(I);
Guy Benyei11169dd2012-12-18 14:30:41 +0000829 }
830
831 // Sort the initializers in source order
Benjamin Kramer4cadf292014-03-07 21:51:58 +0000832 llvm::array_pod_sort(WrittenInits.begin(), WrittenInits.end(),
833 &CompareCXXCtorInitializers);
834
Guy Benyei11169dd2012-12-18 14:30:41 +0000835 // Visit the initializers in source order
836 for (unsigned I = 0, N = WrittenInits.size(); I != N; ++I) {
837 CXXCtorInitializer *Init = WrittenInits[I];
838 if (Init->isAnyMemberInitializer()) {
839 if (Visit(MakeCursorMemberRef(Init->getAnyMember(),
840 Init->getMemberLocation(), TU)))
841 return true;
842 } else if (TypeSourceInfo *TInfo = Init->getTypeSourceInfo()) {
843 if (Visit(TInfo->getTypeLoc()))
844 return true;
845 }
846
847 // Visit the initializer value.
848 if (Expr *Initializer = Init->getInit())
849 if (Visit(MakeCXCursor(Initializer, ND, TU, RegionOfInterest)))
850 return true;
851 }
852 }
853
854 if (Visit(MakeCXCursor(ND->getBody(), StmtParent, TU, RegionOfInterest)))
855 return true;
856 }
857
858 return false;
859}
860
861bool CursorVisitor::VisitFieldDecl(FieldDecl *D) {
862 if (VisitDeclaratorDecl(D))
863 return true;
864
865 if (Expr *BitWidth = D->getBitWidth())
866 return Visit(MakeCXCursor(BitWidth, StmtParent, TU, RegionOfInterest));
867
868 return false;
869}
870
871bool CursorVisitor::VisitVarDecl(VarDecl *D) {
872 if (VisitDeclaratorDecl(D))
873 return true;
874
875 if (Expr *Init = D->getInit())
876 return Visit(MakeCXCursor(Init, StmtParent, TU, RegionOfInterest));
877
878 return false;
879}
880
881bool CursorVisitor::VisitNonTypeTemplateParmDecl(NonTypeTemplateParmDecl *D) {
882 if (VisitDeclaratorDecl(D))
883 return true;
884
885 if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
886 if (Expr *DefArg = D->getDefaultArgument())
887 return Visit(MakeCXCursor(DefArg, StmtParent, TU, RegionOfInterest));
888
889 return false;
890}
891
892bool CursorVisitor::VisitFunctionTemplateDecl(FunctionTemplateDecl *D) {
893 // FIXME: Visit the "outer" template parameter lists on the FunctionDecl
894 // before visiting these template parameters.
895 if (VisitTemplateParameters(D->getTemplateParameters()))
896 return true;
897
898 return VisitFunctionDecl(D->getTemplatedDecl());
899}
900
901bool CursorVisitor::VisitClassTemplateDecl(ClassTemplateDecl *D) {
902 // FIXME: Visit the "outer" template parameter lists on the TagDecl
903 // before visiting these template parameters.
904 if (VisitTemplateParameters(D->getTemplateParameters()))
905 return true;
906
907 return VisitCXXRecordDecl(D->getTemplatedDecl());
908}
909
910bool CursorVisitor::VisitTemplateTemplateParmDecl(TemplateTemplateParmDecl *D) {
911 if (VisitTemplateParameters(D->getTemplateParameters()))
912 return true;
913
914 if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited() &&
915 VisitTemplateArgumentLoc(D->getDefaultArgument()))
916 return true;
917
918 return false;
919}
920
Douglas Gregor9bda6cf2015-07-07 03:58:14 +0000921bool CursorVisitor::VisitObjCTypeParamDecl(ObjCTypeParamDecl *D) {
922 // Visit the bound, if it's explicit.
923 if (D->hasExplicitBound()) {
924 if (auto TInfo = D->getTypeSourceInfo()) {
925 if (Visit(TInfo->getTypeLoc()))
926 return true;
927 }
928 }
929
930 return false;
931}
932
Guy Benyei11169dd2012-12-18 14:30:41 +0000933bool CursorVisitor::VisitObjCMethodDecl(ObjCMethodDecl *ND) {
Alp Toker314cc812014-01-25 16:55:45 +0000934 if (TypeSourceInfo *TSInfo = ND->getReturnTypeSourceInfo())
Guy Benyei11169dd2012-12-18 14:30:41 +0000935 if (Visit(TSInfo->getTypeLoc()))
936 return true;
937
Aaron Ballman43b68be2014-03-07 17:50:17 +0000938 for (const auto *P : ND->params()) {
939 if (Visit(MakeCXCursor(P, TU, RegionOfInterest)))
Guy Benyei11169dd2012-12-18 14:30:41 +0000940 return true;
941 }
942
Alexander Kornienko1a9f1842015-12-28 15:24:08 +0000943 return ND->isThisDeclarationADefinition() &&
944 Visit(MakeCXCursor(ND->getBody(), StmtParent, TU, RegionOfInterest));
Guy Benyei11169dd2012-12-18 14:30:41 +0000945}
946
947template <typename DeclIt>
948static void addRangedDeclsInContainer(DeclIt *DI_current, DeclIt DE_current,
949 SourceManager &SM, SourceLocation EndLoc,
950 SmallVectorImpl<Decl *> &Decls) {
951 DeclIt next = *DI_current;
952 while (++next != DE_current) {
953 Decl *D_next = *next;
954 if (!D_next)
955 break;
956 SourceLocation L = D_next->getLocStart();
957 if (!L.isValid())
958 break;
959 if (SM.isBeforeInTranslationUnit(L, EndLoc)) {
960 *DI_current = next;
961 Decls.push_back(D_next);
962 continue;
963 }
964 break;
965 }
966}
967
Guy Benyei11169dd2012-12-18 14:30:41 +0000968bool CursorVisitor::VisitObjCContainerDecl(ObjCContainerDecl *D) {
969 // FIXME: Eventually convert back to just 'VisitDeclContext()'. Essentially
970 // an @implementation can lexically contain Decls that are not properly
971 // nested in the AST. When we identify such cases, we need to retrofit
972 // this nesting here.
973 if (!DI_current && !FileDI_current)
974 return VisitDeclContext(D);
975
976 // Scan the Decls that immediately come after the container
977 // in the current DeclContext. If any fall within the
978 // container's lexical region, stash them into a vector
979 // for later processing.
980 SmallVector<Decl *, 24> DeclsInContainer;
981 SourceLocation EndLoc = D->getSourceRange().getEnd();
982 SourceManager &SM = AU->getSourceManager();
983 if (EndLoc.isValid()) {
984 if (DI_current) {
985 addRangedDeclsInContainer(DI_current, DE_current, SM, EndLoc,
986 DeclsInContainer);
987 } else {
988 addRangedDeclsInContainer(FileDI_current, FileDE_current, SM, EndLoc,
989 DeclsInContainer);
990 }
991 }
992
993 // The common case.
994 if (DeclsInContainer.empty())
995 return VisitDeclContext(D);
996
997 // Get all the Decls in the DeclContext, and sort them with the
998 // additional ones we've collected. Then visit them.
Aaron Ballman629afae2014-03-07 19:56:05 +0000999 for (auto *SubDecl : D->decls()) {
1000 if (!SubDecl || SubDecl->getLexicalDeclContext() != D ||
1001 SubDecl->getLocStart().isInvalid())
Guy Benyei11169dd2012-12-18 14:30:41 +00001002 continue;
Aaron Ballman629afae2014-03-07 19:56:05 +00001003 DeclsInContainer.push_back(SubDecl);
Guy Benyei11169dd2012-12-18 14:30:41 +00001004 }
1005
1006 // Now sort the Decls so that they appear in lexical order.
1007 std::sort(DeclsInContainer.begin(), DeclsInContainer.end(),
Benjamin Kramerbbdd7642014-03-01 14:48:57 +00001008 [&SM](Decl *A, Decl *B) {
1009 SourceLocation L_A = A->getLocStart();
1010 SourceLocation L_B = B->getLocStart();
1011 assert(L_A.isValid() && L_B.isValid());
1012 return SM.isBeforeInTranslationUnit(L_A, L_B);
1013 });
Guy Benyei11169dd2012-12-18 14:30:41 +00001014
1015 // Now visit the decls.
1016 for (SmallVectorImpl<Decl*>::iterator I = DeclsInContainer.begin(),
1017 E = DeclsInContainer.end(); I != E; ++I) {
1018 CXCursor Cursor = MakeCXCursor(*I, TU, RegionOfInterest);
Ted Kremenek03325582013-02-21 01:29:01 +00001019 const Optional<bool> &V = shouldVisitCursor(Cursor);
Guy Benyei11169dd2012-12-18 14:30:41 +00001020 if (!V.hasValue())
1021 continue;
1022 if (!V.getValue())
1023 return false;
1024 if (Visit(Cursor, true))
1025 return true;
1026 }
1027 return false;
1028}
1029
1030bool CursorVisitor::VisitObjCCategoryDecl(ObjCCategoryDecl *ND) {
1031 if (Visit(MakeCursorObjCClassRef(ND->getClassInterface(), ND->getLocation(),
1032 TU)))
1033 return true;
1034
Douglas Gregore9d95f12015-07-07 03:57:35 +00001035 if (VisitObjCTypeParamList(ND->getTypeParamList()))
1036 return true;
1037
Guy Benyei11169dd2012-12-18 14:30:41 +00001038 ObjCCategoryDecl::protocol_loc_iterator PL = ND->protocol_loc_begin();
1039 for (ObjCCategoryDecl::protocol_iterator I = ND->protocol_begin(),
1040 E = ND->protocol_end(); I != E; ++I, ++PL)
1041 if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1042 return true;
1043
1044 return VisitObjCContainerDecl(ND);
1045}
1046
1047bool CursorVisitor::VisitObjCProtocolDecl(ObjCProtocolDecl *PID) {
1048 if (!PID->isThisDeclarationADefinition())
1049 return Visit(MakeCursorObjCProtocolRef(PID, PID->getLocation(), TU));
1050
1051 ObjCProtocolDecl::protocol_loc_iterator PL = PID->protocol_loc_begin();
1052 for (ObjCProtocolDecl::protocol_iterator I = PID->protocol_begin(),
1053 E = PID->protocol_end(); I != E; ++I, ++PL)
1054 if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1055 return true;
1056
1057 return VisitObjCContainerDecl(PID);
1058}
1059
1060bool CursorVisitor::VisitObjCPropertyDecl(ObjCPropertyDecl *PD) {
1061 if (PD->getTypeSourceInfo() && Visit(PD->getTypeSourceInfo()->getTypeLoc()))
1062 return true;
1063
1064 // FIXME: This implements a workaround with @property declarations also being
1065 // installed in the DeclContext for the @interface. Eventually this code
1066 // should be removed.
1067 ObjCCategoryDecl *CDecl = dyn_cast<ObjCCategoryDecl>(PD->getDeclContext());
1068 if (!CDecl || !CDecl->IsClassExtension())
1069 return false;
1070
1071 ObjCInterfaceDecl *ID = CDecl->getClassInterface();
1072 if (!ID)
1073 return false;
1074
1075 IdentifierInfo *PropertyId = PD->getIdentifier();
1076 ObjCPropertyDecl *prevDecl =
Manman Ren5b786402016-01-28 18:49:28 +00001077 ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(ID), PropertyId,
1078 PD->getQueryKind());
Guy Benyei11169dd2012-12-18 14:30:41 +00001079
1080 if (!prevDecl)
1081 return false;
1082
1083 // Visit synthesized methods since they will be skipped when visiting
1084 // the @interface.
1085 if (ObjCMethodDecl *MD = prevDecl->getGetterMethodDecl())
1086 if (MD->isPropertyAccessor() && MD->getLexicalDeclContext() == CDecl)
1087 if (Visit(MakeCXCursor(MD, TU, RegionOfInterest)))
1088 return true;
1089
1090 if (ObjCMethodDecl *MD = prevDecl->getSetterMethodDecl())
1091 if (MD->isPropertyAccessor() && MD->getLexicalDeclContext() == CDecl)
1092 if (Visit(MakeCXCursor(MD, TU, RegionOfInterest)))
1093 return true;
1094
1095 return false;
1096}
1097
Douglas Gregore9d95f12015-07-07 03:57:35 +00001098bool CursorVisitor::VisitObjCTypeParamList(ObjCTypeParamList *typeParamList) {
1099 if (!typeParamList)
1100 return false;
1101
1102 for (auto *typeParam : *typeParamList) {
1103 // Visit the type parameter.
1104 if (Visit(MakeCXCursor(typeParam, TU, RegionOfInterest)))
1105 return true;
Douglas Gregore9d95f12015-07-07 03:57:35 +00001106 }
1107
1108 return false;
1109}
1110
Guy Benyei11169dd2012-12-18 14:30:41 +00001111bool CursorVisitor::VisitObjCInterfaceDecl(ObjCInterfaceDecl *D) {
1112 if (!D->isThisDeclarationADefinition()) {
1113 // Forward declaration is treated like a reference.
1114 return Visit(MakeCursorObjCClassRef(D, D->getLocation(), TU));
1115 }
1116
Douglas Gregore9d95f12015-07-07 03:57:35 +00001117 // Objective-C type parameters.
1118 if (VisitObjCTypeParamList(D->getTypeParamListAsWritten()))
1119 return true;
1120
Guy Benyei11169dd2012-12-18 14:30:41 +00001121 // Issue callbacks for super class.
1122 if (D->getSuperClass() &&
1123 Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1124 D->getSuperClassLoc(),
1125 TU)))
1126 return true;
1127
Douglas Gregore9d95f12015-07-07 03:57:35 +00001128 if (TypeSourceInfo *SuperClassTInfo = D->getSuperClassTInfo())
1129 if (Visit(SuperClassTInfo->getTypeLoc()))
1130 return true;
1131
Guy Benyei11169dd2012-12-18 14:30:41 +00001132 ObjCInterfaceDecl::protocol_loc_iterator PL = D->protocol_loc_begin();
1133 for (ObjCInterfaceDecl::protocol_iterator I = D->protocol_begin(),
1134 E = D->protocol_end(); I != E; ++I, ++PL)
1135 if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1136 return true;
1137
1138 return VisitObjCContainerDecl(D);
1139}
1140
1141bool CursorVisitor::VisitObjCImplDecl(ObjCImplDecl *D) {
1142 return VisitObjCContainerDecl(D);
1143}
1144
1145bool CursorVisitor::VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D) {
1146 // 'ID' could be null when dealing with invalid code.
1147 if (ObjCInterfaceDecl *ID = D->getClassInterface())
1148 if (Visit(MakeCursorObjCClassRef(ID, D->getLocation(), TU)))
1149 return true;
1150
1151 return VisitObjCImplDecl(D);
1152}
1153
1154bool CursorVisitor::VisitObjCImplementationDecl(ObjCImplementationDecl *D) {
1155#if 0
1156 // Issue callbacks for super class.
1157 // FIXME: No source location information!
1158 if (D->getSuperClass() &&
1159 Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1160 D->getSuperClassLoc(),
1161 TU)))
1162 return true;
1163#endif
1164
1165 return VisitObjCImplDecl(D);
1166}
1167
1168bool CursorVisitor::VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *PD) {
1169 if (ObjCIvarDecl *Ivar = PD->getPropertyIvarDecl())
1170 if (PD->isIvarNameSpecified())
1171 return Visit(MakeCursorMemberRef(Ivar, PD->getPropertyIvarDeclLoc(), TU));
1172
1173 return false;
1174}
1175
1176bool CursorVisitor::VisitNamespaceDecl(NamespaceDecl *D) {
1177 return VisitDeclContext(D);
1178}
1179
1180bool CursorVisitor::VisitNamespaceAliasDecl(NamespaceAliasDecl *D) {
1181 // Visit nested-name-specifier.
1182 if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1183 if (VisitNestedNameSpecifierLoc(QualifierLoc))
1184 return true;
1185
1186 return Visit(MakeCursorNamespaceRef(D->getAliasedNamespace(),
1187 D->getTargetNameLoc(), TU));
1188}
1189
1190bool CursorVisitor::VisitUsingDecl(UsingDecl *D) {
1191 // Visit nested-name-specifier.
1192 if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1193 if (VisitNestedNameSpecifierLoc(QualifierLoc))
1194 return true;
1195 }
1196
1197 if (Visit(MakeCursorOverloadedDeclRef(D, D->getLocation(), TU)))
1198 return true;
1199
1200 return VisitDeclarationNameInfo(D->getNameInfo());
1201}
1202
1203bool CursorVisitor::VisitUsingDirectiveDecl(UsingDirectiveDecl *D) {
1204 // Visit nested-name-specifier.
1205 if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1206 if (VisitNestedNameSpecifierLoc(QualifierLoc))
1207 return true;
1208
1209 return Visit(MakeCursorNamespaceRef(D->getNominatedNamespaceAsWritten(),
1210 D->getIdentLocation(), TU));
1211}
1212
1213bool CursorVisitor::VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D) {
1214 // Visit nested-name-specifier.
1215 if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1216 if (VisitNestedNameSpecifierLoc(QualifierLoc))
1217 return true;
1218 }
1219
1220 return VisitDeclarationNameInfo(D->getNameInfo());
1221}
1222
1223bool CursorVisitor::VisitUnresolvedUsingTypenameDecl(
1224 UnresolvedUsingTypenameDecl *D) {
1225 // Visit nested-name-specifier.
1226 if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1227 if (VisitNestedNameSpecifierLoc(QualifierLoc))
1228 return true;
1229
1230 return false;
1231}
1232
1233bool CursorVisitor::VisitDeclarationNameInfo(DeclarationNameInfo Name) {
1234 switch (Name.getName().getNameKind()) {
1235 case clang::DeclarationName::Identifier:
1236 case clang::DeclarationName::CXXLiteralOperatorName:
1237 case clang::DeclarationName::CXXOperatorName:
1238 case clang::DeclarationName::CXXUsingDirective:
1239 return false;
1240
1241 case clang::DeclarationName::CXXConstructorName:
1242 case clang::DeclarationName::CXXDestructorName:
1243 case clang::DeclarationName::CXXConversionFunctionName:
1244 if (TypeSourceInfo *TSInfo = Name.getNamedTypeInfo())
1245 return Visit(TSInfo->getTypeLoc());
1246 return false;
1247
1248 case clang::DeclarationName::ObjCZeroArgSelector:
1249 case clang::DeclarationName::ObjCOneArgSelector:
1250 case clang::DeclarationName::ObjCMultiArgSelector:
1251 // FIXME: Per-identifier location info?
1252 return false;
1253 }
1254
1255 llvm_unreachable("Invalid DeclarationName::Kind!");
1256}
1257
1258bool CursorVisitor::VisitNestedNameSpecifier(NestedNameSpecifier *NNS,
1259 SourceRange Range) {
1260 // FIXME: This whole routine is a hack to work around the lack of proper
1261 // source information in nested-name-specifiers (PR5791). Since we do have
1262 // a beginning source location, we can visit the first component of the
1263 // nested-name-specifier, if it's a single-token component.
1264 if (!NNS)
1265 return false;
1266
1267 // Get the first component in the nested-name-specifier.
1268 while (NestedNameSpecifier *Prefix = NNS->getPrefix())
1269 NNS = Prefix;
1270
1271 switch (NNS->getKind()) {
1272 case NestedNameSpecifier::Namespace:
1273 return Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(), Range.getBegin(),
1274 TU));
1275
1276 case NestedNameSpecifier::NamespaceAlias:
1277 return Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
1278 Range.getBegin(), TU));
1279
1280 case NestedNameSpecifier::TypeSpec: {
1281 // If the type has a form where we know that the beginning of the source
1282 // range matches up with a reference cursor. Visit the appropriate reference
1283 // cursor.
1284 const Type *T = NNS->getAsType();
1285 if (const TypedefType *Typedef = dyn_cast<TypedefType>(T))
1286 return Visit(MakeCursorTypeRef(Typedef->getDecl(), Range.getBegin(), TU));
1287 if (const TagType *Tag = dyn_cast<TagType>(T))
1288 return Visit(MakeCursorTypeRef(Tag->getDecl(), Range.getBegin(), TU));
1289 if (const TemplateSpecializationType *TST
1290 = dyn_cast<TemplateSpecializationType>(T))
1291 return VisitTemplateName(TST->getTemplateName(), Range.getBegin());
1292 break;
1293 }
1294
1295 case NestedNameSpecifier::TypeSpecWithTemplate:
1296 case NestedNameSpecifier::Global:
1297 case NestedNameSpecifier::Identifier:
Nikola Smiljanic67860242014-09-26 00:28:20 +00001298 case NestedNameSpecifier::Super:
Guy Benyei11169dd2012-12-18 14:30:41 +00001299 break;
1300 }
1301
1302 return false;
1303}
1304
1305bool
1306CursorVisitor::VisitNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
1307 SmallVector<NestedNameSpecifierLoc, 4> Qualifiers;
1308 for (; Qualifier; Qualifier = Qualifier.getPrefix())
1309 Qualifiers.push_back(Qualifier);
1310
1311 while (!Qualifiers.empty()) {
1312 NestedNameSpecifierLoc Q = Qualifiers.pop_back_val();
1313 NestedNameSpecifier *NNS = Q.getNestedNameSpecifier();
1314 switch (NNS->getKind()) {
1315 case NestedNameSpecifier::Namespace:
1316 if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(),
1317 Q.getLocalBeginLoc(),
1318 TU)))
1319 return true;
1320
1321 break;
1322
1323 case NestedNameSpecifier::NamespaceAlias:
1324 if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
1325 Q.getLocalBeginLoc(),
1326 TU)))
1327 return true;
1328
1329 break;
1330
1331 case NestedNameSpecifier::TypeSpec:
1332 case NestedNameSpecifier::TypeSpecWithTemplate:
1333 if (Visit(Q.getTypeLoc()))
1334 return true;
1335
1336 break;
1337
1338 case NestedNameSpecifier::Global:
1339 case NestedNameSpecifier::Identifier:
Nikola Smiljanic67860242014-09-26 00:28:20 +00001340 case NestedNameSpecifier::Super:
Guy Benyei11169dd2012-12-18 14:30:41 +00001341 break;
1342 }
1343 }
1344
1345 return false;
1346}
1347
1348bool CursorVisitor::VisitTemplateParameters(
1349 const TemplateParameterList *Params) {
1350 if (!Params)
1351 return false;
1352
1353 for (TemplateParameterList::const_iterator P = Params->begin(),
1354 PEnd = Params->end();
1355 P != PEnd; ++P) {
1356 if (Visit(MakeCXCursor(*P, TU, RegionOfInterest)))
1357 return true;
1358 }
1359
1360 return false;
1361}
1362
1363bool CursorVisitor::VisitTemplateName(TemplateName Name, SourceLocation Loc) {
1364 switch (Name.getKind()) {
1365 case TemplateName::Template:
1366 return Visit(MakeCursorTemplateRef(Name.getAsTemplateDecl(), Loc, TU));
1367
1368 case TemplateName::OverloadedTemplate:
1369 // Visit the overloaded template set.
1370 if (Visit(MakeCursorOverloadedDeclRef(Name, Loc, TU)))
1371 return true;
1372
1373 return false;
1374
1375 case TemplateName::DependentTemplate:
1376 // FIXME: Visit nested-name-specifier.
1377 return false;
1378
1379 case TemplateName::QualifiedTemplate:
1380 // FIXME: Visit nested-name-specifier.
1381 return Visit(MakeCursorTemplateRef(
1382 Name.getAsQualifiedTemplateName()->getDecl(),
1383 Loc, TU));
1384
1385 case TemplateName::SubstTemplateTemplateParm:
1386 return Visit(MakeCursorTemplateRef(
1387 Name.getAsSubstTemplateTemplateParm()->getParameter(),
1388 Loc, TU));
1389
1390 case TemplateName::SubstTemplateTemplateParmPack:
1391 return Visit(MakeCursorTemplateRef(
1392 Name.getAsSubstTemplateTemplateParmPack()->getParameterPack(),
1393 Loc, TU));
1394 }
1395
1396 llvm_unreachable("Invalid TemplateName::Kind!");
1397}
1398
1399bool CursorVisitor::VisitTemplateArgumentLoc(const TemplateArgumentLoc &TAL) {
1400 switch (TAL.getArgument().getKind()) {
1401 case TemplateArgument::Null:
1402 case TemplateArgument::Integral:
1403 case TemplateArgument::Pack:
1404 return false;
1405
1406 case TemplateArgument::Type:
1407 if (TypeSourceInfo *TSInfo = TAL.getTypeSourceInfo())
1408 return Visit(TSInfo->getTypeLoc());
1409 return false;
1410
1411 case TemplateArgument::Declaration:
1412 if (Expr *E = TAL.getSourceDeclExpression())
1413 return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1414 return false;
1415
1416 case TemplateArgument::NullPtr:
1417 if (Expr *E = TAL.getSourceNullPtrExpression())
1418 return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1419 return false;
1420
1421 case TemplateArgument::Expression:
1422 if (Expr *E = TAL.getSourceExpression())
1423 return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1424 return false;
1425
1426 case TemplateArgument::Template:
1427 case TemplateArgument::TemplateExpansion:
1428 if (VisitNestedNameSpecifierLoc(TAL.getTemplateQualifierLoc()))
1429 return true;
1430
1431 return VisitTemplateName(TAL.getArgument().getAsTemplateOrTemplatePattern(),
1432 TAL.getTemplateNameLoc());
1433 }
1434
1435 llvm_unreachable("Invalid TemplateArgument::Kind!");
1436}
1437
1438bool CursorVisitor::VisitLinkageSpecDecl(LinkageSpecDecl *D) {
1439 return VisitDeclContext(D);
1440}
1441
1442bool CursorVisitor::VisitQualifiedTypeLoc(QualifiedTypeLoc TL) {
1443 return Visit(TL.getUnqualifiedLoc());
1444}
1445
1446bool CursorVisitor::VisitBuiltinTypeLoc(BuiltinTypeLoc TL) {
1447 ASTContext &Context = AU->getASTContext();
1448
1449 // Some builtin types (such as Objective-C's "id", "sel", and
1450 // "Class") have associated declarations. Create cursors for those.
1451 QualType VisitType;
1452 switch (TL.getTypePtr()->getKind()) {
1453
1454 case BuiltinType::Void:
1455 case BuiltinType::NullPtr:
1456 case BuiltinType::Dependent:
Alexey Bader954ba212016-04-08 13:40:33 +00001457#define IMAGE_TYPE(ImgType, Id, SingletonId, Access, Suffix) \
1458 case BuiltinType::Id:
Alexey Baderb62f1442016-04-13 08:33:41 +00001459#include "clang/Basic/OpenCLImageTypes.def"
NAKAMURA Takumi288c42e2013-02-07 12:47:42 +00001460 case BuiltinType::OCLSampler:
Guy Benyei1b4fb3e2013-01-20 12:31:11 +00001461 case BuiltinType::OCLEvent:
Alexey Bader9c8453f2015-09-15 11:18:52 +00001462 case BuiltinType::OCLClkEvent:
1463 case BuiltinType::OCLQueue:
1464 case BuiltinType::OCLNDRange:
1465 case BuiltinType::OCLReserveID:
Guy Benyei11169dd2012-12-18 14:30:41 +00001466#define BUILTIN_TYPE(Id, SingletonId)
1467#define SIGNED_TYPE(Id, SingletonId) case BuiltinType::Id:
1468#define UNSIGNED_TYPE(Id, SingletonId) case BuiltinType::Id:
1469#define FLOATING_TYPE(Id, SingletonId) case BuiltinType::Id:
1470#define PLACEHOLDER_TYPE(Id, SingletonId) case BuiltinType::Id:
1471#include "clang/AST/BuiltinTypes.def"
1472 break;
1473
1474 case BuiltinType::ObjCId:
1475 VisitType = Context.getObjCIdType();
1476 break;
1477
1478 case BuiltinType::ObjCClass:
1479 VisitType = Context.getObjCClassType();
1480 break;
1481
1482 case BuiltinType::ObjCSel:
1483 VisitType = Context.getObjCSelType();
1484 break;
1485 }
1486
1487 if (!VisitType.isNull()) {
1488 if (const TypedefType *Typedef = VisitType->getAs<TypedefType>())
1489 return Visit(MakeCursorTypeRef(Typedef->getDecl(), TL.getBuiltinLoc(),
1490 TU));
1491 }
1492
1493 return false;
1494}
1495
1496bool CursorVisitor::VisitTypedefTypeLoc(TypedefTypeLoc TL) {
1497 return Visit(MakeCursorTypeRef(TL.getTypedefNameDecl(), TL.getNameLoc(), TU));
1498}
1499
1500bool CursorVisitor::VisitUnresolvedUsingTypeLoc(UnresolvedUsingTypeLoc TL) {
1501 return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1502}
1503
1504bool CursorVisitor::VisitTagTypeLoc(TagTypeLoc TL) {
1505 if (TL.isDefinition())
1506 return Visit(MakeCXCursor(TL.getDecl(), TU, RegionOfInterest));
1507
1508 return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1509}
1510
1511bool CursorVisitor::VisitTemplateTypeParmTypeLoc(TemplateTypeParmTypeLoc TL) {
1512 return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1513}
1514
1515bool CursorVisitor::VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL) {
Hans Wennborg59dbe862015-09-29 20:56:43 +00001516 return Visit(MakeCursorObjCClassRef(TL.getIFaceDecl(), TL.getNameLoc(), TU));
Guy Benyei11169dd2012-12-18 14:30:41 +00001517}
1518
1519bool CursorVisitor::VisitObjCObjectTypeLoc(ObjCObjectTypeLoc TL) {
1520 if (TL.hasBaseTypeAsWritten() && Visit(TL.getBaseLoc()))
1521 return true;
1522
Douglas Gregore9d95f12015-07-07 03:57:35 +00001523 for (unsigned I = 0, N = TL.getNumTypeArgs(); I != N; ++I) {
1524 if (Visit(TL.getTypeArgTInfo(I)->getTypeLoc()))
1525 return true;
1526 }
1527
Guy Benyei11169dd2012-12-18 14:30:41 +00001528 for (unsigned I = 0, N = TL.getNumProtocols(); I != N; ++I) {
1529 if (Visit(MakeCursorObjCProtocolRef(TL.getProtocol(I), TL.getProtocolLoc(I),
1530 TU)))
1531 return true;
1532 }
1533
1534 return false;
1535}
1536
1537bool CursorVisitor::VisitObjCObjectPointerTypeLoc(ObjCObjectPointerTypeLoc TL) {
1538 return Visit(TL.getPointeeLoc());
1539}
1540
1541bool CursorVisitor::VisitParenTypeLoc(ParenTypeLoc TL) {
1542 return Visit(TL.getInnerLoc());
1543}
1544
1545bool CursorVisitor::VisitPointerTypeLoc(PointerTypeLoc TL) {
1546 return Visit(TL.getPointeeLoc());
1547}
1548
1549bool CursorVisitor::VisitBlockPointerTypeLoc(BlockPointerTypeLoc TL) {
1550 return Visit(TL.getPointeeLoc());
1551}
1552
1553bool CursorVisitor::VisitMemberPointerTypeLoc(MemberPointerTypeLoc TL) {
1554 return Visit(TL.getPointeeLoc());
1555}
1556
1557bool CursorVisitor::VisitLValueReferenceTypeLoc(LValueReferenceTypeLoc TL) {
1558 return Visit(TL.getPointeeLoc());
1559}
1560
1561bool CursorVisitor::VisitRValueReferenceTypeLoc(RValueReferenceTypeLoc TL) {
1562 return Visit(TL.getPointeeLoc());
1563}
1564
1565bool CursorVisitor::VisitAttributedTypeLoc(AttributedTypeLoc TL) {
1566 return Visit(TL.getModifiedLoc());
1567}
1568
1569bool CursorVisitor::VisitFunctionTypeLoc(FunctionTypeLoc TL,
1570 bool SkipResultType) {
Alp Toker42a16a62014-01-25 23:51:36 +00001571 if (!SkipResultType && Visit(TL.getReturnLoc()))
Guy Benyei11169dd2012-12-18 14:30:41 +00001572 return true;
1573
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00001574 for (unsigned I = 0, N = TL.getNumParams(); I != N; ++I)
1575 if (Decl *D = TL.getParam(I))
Guy Benyei11169dd2012-12-18 14:30:41 +00001576 if (Visit(MakeCXCursor(D, TU, RegionOfInterest)))
1577 return true;
1578
1579 return false;
1580}
1581
1582bool CursorVisitor::VisitArrayTypeLoc(ArrayTypeLoc TL) {
1583 if (Visit(TL.getElementLoc()))
1584 return true;
1585
1586 if (Expr *Size = TL.getSizeExpr())
1587 return Visit(MakeCXCursor(Size, StmtParent, TU, RegionOfInterest));
1588
1589 return false;
1590}
1591
Reid Kleckner8a365022013-06-24 17:51:48 +00001592bool CursorVisitor::VisitDecayedTypeLoc(DecayedTypeLoc TL) {
1593 return Visit(TL.getOriginalLoc());
1594}
1595
Reid Kleckner0503a872013-12-05 01:23:43 +00001596bool CursorVisitor::VisitAdjustedTypeLoc(AdjustedTypeLoc TL) {
1597 return Visit(TL.getOriginalLoc());
1598}
1599
Guy Benyei11169dd2012-12-18 14:30:41 +00001600bool CursorVisitor::VisitTemplateSpecializationTypeLoc(
1601 TemplateSpecializationTypeLoc TL) {
1602 // Visit the template name.
1603 if (VisitTemplateName(TL.getTypePtr()->getTemplateName(),
1604 TL.getTemplateNameLoc()))
1605 return true;
1606
1607 // Visit the template arguments.
1608 for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
1609 if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
1610 return true;
1611
1612 return false;
1613}
1614
1615bool CursorVisitor::VisitTypeOfExprTypeLoc(TypeOfExprTypeLoc TL) {
1616 return Visit(MakeCXCursor(TL.getUnderlyingExpr(), StmtParent, TU));
1617}
1618
1619bool CursorVisitor::VisitTypeOfTypeLoc(TypeOfTypeLoc TL) {
1620 if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
1621 return Visit(TSInfo->getTypeLoc());
1622
1623 return false;
1624}
1625
1626bool CursorVisitor::VisitUnaryTransformTypeLoc(UnaryTransformTypeLoc TL) {
1627 if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
1628 return Visit(TSInfo->getTypeLoc());
1629
1630 return false;
1631}
1632
1633bool CursorVisitor::VisitDependentNameTypeLoc(DependentNameTypeLoc TL) {
Hans Wennborg59dbe862015-09-29 20:56:43 +00001634 return VisitNestedNameSpecifierLoc(TL.getQualifierLoc());
Guy Benyei11169dd2012-12-18 14:30:41 +00001635}
1636
1637bool CursorVisitor::VisitDependentTemplateSpecializationTypeLoc(
1638 DependentTemplateSpecializationTypeLoc TL) {
1639 // Visit the nested-name-specifier, if there is one.
1640 if (TL.getQualifierLoc() &&
1641 VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
1642 return true;
1643
1644 // Visit the template arguments.
1645 for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
1646 if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
1647 return true;
1648
1649 return false;
1650}
1651
1652bool CursorVisitor::VisitElaboratedTypeLoc(ElaboratedTypeLoc TL) {
1653 if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
1654 return true;
1655
1656 return Visit(TL.getNamedTypeLoc());
1657}
1658
1659bool CursorVisitor::VisitPackExpansionTypeLoc(PackExpansionTypeLoc TL) {
1660 return Visit(TL.getPatternLoc());
1661}
1662
1663bool CursorVisitor::VisitDecltypeTypeLoc(DecltypeTypeLoc TL) {
1664 if (Expr *E = TL.getUnderlyingExpr())
1665 return Visit(MakeCXCursor(E, StmtParent, TU));
1666
1667 return false;
1668}
1669
1670bool CursorVisitor::VisitInjectedClassNameTypeLoc(InjectedClassNameTypeLoc TL) {
1671 return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1672}
1673
1674bool CursorVisitor::VisitAtomicTypeLoc(AtomicTypeLoc TL) {
1675 return Visit(TL.getValueLoc());
1676}
1677
Xiuli Pan9c14e282016-01-09 12:53:17 +00001678bool CursorVisitor::VisitPipeTypeLoc(PipeTypeLoc TL) {
1679 return Visit(TL.getValueLoc());
1680}
1681
Guy Benyei11169dd2012-12-18 14:30:41 +00001682#define DEFAULT_TYPELOC_IMPL(CLASS, PARENT) \
1683bool CursorVisitor::Visit##CLASS##TypeLoc(CLASS##TypeLoc TL) { \
1684 return Visit##PARENT##Loc(TL); \
1685}
1686
1687DEFAULT_TYPELOC_IMPL(Complex, Type)
1688DEFAULT_TYPELOC_IMPL(ConstantArray, ArrayType)
1689DEFAULT_TYPELOC_IMPL(IncompleteArray, ArrayType)
1690DEFAULT_TYPELOC_IMPL(VariableArray, ArrayType)
1691DEFAULT_TYPELOC_IMPL(DependentSizedArray, ArrayType)
1692DEFAULT_TYPELOC_IMPL(DependentSizedExtVector, Type)
1693DEFAULT_TYPELOC_IMPL(Vector, Type)
1694DEFAULT_TYPELOC_IMPL(ExtVector, VectorType)
1695DEFAULT_TYPELOC_IMPL(FunctionProto, FunctionType)
1696DEFAULT_TYPELOC_IMPL(FunctionNoProto, FunctionType)
1697DEFAULT_TYPELOC_IMPL(Record, TagType)
1698DEFAULT_TYPELOC_IMPL(Enum, TagType)
1699DEFAULT_TYPELOC_IMPL(SubstTemplateTypeParm, Type)
1700DEFAULT_TYPELOC_IMPL(SubstTemplateTypeParmPack, Type)
1701DEFAULT_TYPELOC_IMPL(Auto, Type)
1702
1703bool CursorVisitor::VisitCXXRecordDecl(CXXRecordDecl *D) {
1704 // Visit the nested-name-specifier, if present.
1705 if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1706 if (VisitNestedNameSpecifierLoc(QualifierLoc))
1707 return true;
1708
1709 if (D->isCompleteDefinition()) {
Aaron Ballman574705e2014-03-13 15:41:46 +00001710 for (const auto &I : D->bases()) {
1711 if (Visit(cxcursor::MakeCursorCXXBaseSpecifier(&I, TU)))
Guy Benyei11169dd2012-12-18 14:30:41 +00001712 return true;
1713 }
1714 }
1715
1716 return VisitTagDecl(D);
1717}
1718
1719bool CursorVisitor::VisitAttributes(Decl *D) {
Aaron Ballmanb97112e2014-03-08 22:19:01 +00001720 for (const auto *I : D->attrs())
1721 if (Visit(MakeCXCursor(I, D, TU)))
Guy Benyei11169dd2012-12-18 14:30:41 +00001722 return true;
1723
1724 return false;
1725}
1726
1727//===----------------------------------------------------------------------===//
1728// Data-recursive visitor methods.
1729//===----------------------------------------------------------------------===//
1730
1731namespace {
1732#define DEF_JOB(NAME, DATA, KIND)\
1733class NAME : public VisitorJob {\
1734public:\
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001735 NAME(const DATA *d, CXCursor parent) : \
1736 VisitorJob(parent, VisitorJob::KIND, d) {} \
Guy Benyei11169dd2012-12-18 14:30:41 +00001737 static bool classof(const VisitorJob *VJ) { return VJ->getKind() == KIND; }\
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001738 const DATA *get() const { return static_cast<const DATA*>(data[0]); }\
Guy Benyei11169dd2012-12-18 14:30:41 +00001739};
1740
1741DEF_JOB(StmtVisit, Stmt, StmtVisitKind)
1742DEF_JOB(MemberExprParts, MemberExpr, MemberExprPartsKind)
1743DEF_JOB(DeclRefExprParts, DeclRefExpr, DeclRefExprPartsKind)
1744DEF_JOB(OverloadExprParts, OverloadExpr, OverloadExprPartsKind)
Guy Benyei11169dd2012-12-18 14:30:41 +00001745DEF_JOB(SizeOfPackExprParts, SizeOfPackExpr, SizeOfPackExprPartsKind)
1746DEF_JOB(LambdaExprParts, LambdaExpr, LambdaExprPartsKind)
1747DEF_JOB(PostChildrenVisit, void, PostChildrenVisitKind)
1748#undef DEF_JOB
1749
James Y Knight04ec5bf2015-12-24 02:59:37 +00001750class ExplicitTemplateArgsVisit : public VisitorJob {
1751public:
1752 ExplicitTemplateArgsVisit(const TemplateArgumentLoc *Begin,
1753 const TemplateArgumentLoc *End, CXCursor parent)
1754 : VisitorJob(parent, VisitorJob::ExplicitTemplateArgsVisitKind, Begin,
1755 End) {}
1756 static bool classof(const VisitorJob *VJ) {
1757 return VJ->getKind() == ExplicitTemplateArgsVisitKind;
1758 }
1759 const TemplateArgumentLoc *begin() const {
1760 return static_cast<const TemplateArgumentLoc *>(data[0]);
1761 }
1762 const TemplateArgumentLoc *end() {
1763 return static_cast<const TemplateArgumentLoc *>(data[1]);
1764 }
1765};
Guy Benyei11169dd2012-12-18 14:30:41 +00001766class DeclVisit : public VisitorJob {
1767public:
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001768 DeclVisit(const Decl *D, CXCursor parent, bool isFirst) :
Guy Benyei11169dd2012-12-18 14:30:41 +00001769 VisitorJob(parent, VisitorJob::DeclVisitKind,
Craig Topper69186e72014-06-08 08:38:04 +00001770 D, isFirst ? (void*) 1 : (void*) nullptr) {}
Guy Benyei11169dd2012-12-18 14:30:41 +00001771 static bool classof(const VisitorJob *VJ) {
1772 return VJ->getKind() == DeclVisitKind;
1773 }
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001774 const Decl *get() const { return static_cast<const Decl *>(data[0]); }
Dmitri Gribenkoe5423a72015-03-23 19:23:50 +00001775 bool isFirst() const { return data[1] != nullptr; }
Guy Benyei11169dd2012-12-18 14:30:41 +00001776};
1777class TypeLocVisit : public VisitorJob {
1778public:
1779 TypeLocVisit(TypeLoc tl, CXCursor parent) :
1780 VisitorJob(parent, VisitorJob::TypeLocVisitKind,
1781 tl.getType().getAsOpaquePtr(), tl.getOpaqueData()) {}
1782
1783 static bool classof(const VisitorJob *VJ) {
1784 return VJ->getKind() == TypeLocVisitKind;
1785 }
1786
1787 TypeLoc get() const {
1788 QualType T = QualType::getFromOpaquePtr(data[0]);
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001789 return TypeLoc(T, const_cast<void *>(data[1]));
Guy Benyei11169dd2012-12-18 14:30:41 +00001790 }
1791};
1792
1793class LabelRefVisit : public VisitorJob {
1794public:
1795 LabelRefVisit(LabelDecl *LD, SourceLocation labelLoc, CXCursor parent)
1796 : VisitorJob(parent, VisitorJob::LabelRefVisitKind, LD,
1797 labelLoc.getPtrEncoding()) {}
1798
1799 static bool classof(const VisitorJob *VJ) {
1800 return VJ->getKind() == VisitorJob::LabelRefVisitKind;
1801 }
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001802 const LabelDecl *get() const {
1803 return static_cast<const LabelDecl *>(data[0]);
1804 }
Guy Benyei11169dd2012-12-18 14:30:41 +00001805 SourceLocation getLoc() const {
1806 return SourceLocation::getFromPtrEncoding(data[1]); }
1807};
1808
1809class NestedNameSpecifierLocVisit : public VisitorJob {
1810public:
1811 NestedNameSpecifierLocVisit(NestedNameSpecifierLoc Qualifier, CXCursor parent)
1812 : VisitorJob(parent, VisitorJob::NestedNameSpecifierLocVisitKind,
1813 Qualifier.getNestedNameSpecifier(),
1814 Qualifier.getOpaqueData()) { }
1815
1816 static bool classof(const VisitorJob *VJ) {
1817 return VJ->getKind() == VisitorJob::NestedNameSpecifierLocVisitKind;
1818 }
1819
1820 NestedNameSpecifierLoc get() const {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001821 return NestedNameSpecifierLoc(
1822 const_cast<NestedNameSpecifier *>(
1823 static_cast<const NestedNameSpecifier *>(data[0])),
1824 const_cast<void *>(data[1]));
Guy Benyei11169dd2012-12-18 14:30:41 +00001825 }
1826};
1827
1828class DeclarationNameInfoVisit : public VisitorJob {
1829public:
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001830 DeclarationNameInfoVisit(const Stmt *S, CXCursor parent)
Dmitri Gribenkodd7dacf2013-02-03 13:19:54 +00001831 : VisitorJob(parent, VisitorJob::DeclarationNameInfoVisitKind, S) {}
Guy Benyei11169dd2012-12-18 14:30:41 +00001832 static bool classof(const VisitorJob *VJ) {
1833 return VJ->getKind() == VisitorJob::DeclarationNameInfoVisitKind;
1834 }
1835 DeclarationNameInfo get() const {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001836 const Stmt *S = static_cast<const Stmt *>(data[0]);
Guy Benyei11169dd2012-12-18 14:30:41 +00001837 switch (S->getStmtClass()) {
1838 default:
1839 llvm_unreachable("Unhandled Stmt");
1840 case clang::Stmt::MSDependentExistsStmtClass:
1841 return cast<MSDependentExistsStmt>(S)->getNameInfo();
1842 case Stmt::CXXDependentScopeMemberExprClass:
1843 return cast<CXXDependentScopeMemberExpr>(S)->getMemberNameInfo();
1844 case Stmt::DependentScopeDeclRefExprClass:
1845 return cast<DependentScopeDeclRefExpr>(S)->getNameInfo();
Alexander Musmand9ed09f2014-07-21 09:42:05 +00001846 case Stmt::OMPCriticalDirectiveClass:
1847 return cast<OMPCriticalDirective>(S)->getDirectiveName();
Guy Benyei11169dd2012-12-18 14:30:41 +00001848 }
1849 }
1850};
1851class MemberRefVisit : public VisitorJob {
1852public:
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001853 MemberRefVisit(const FieldDecl *D, SourceLocation L, CXCursor parent)
Guy Benyei11169dd2012-12-18 14:30:41 +00001854 : VisitorJob(parent, VisitorJob::MemberRefVisitKind, D,
1855 L.getPtrEncoding()) {}
1856 static bool classof(const VisitorJob *VJ) {
1857 return VJ->getKind() == VisitorJob::MemberRefVisitKind;
1858 }
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001859 const FieldDecl *get() const {
1860 return static_cast<const FieldDecl *>(data[0]);
Guy Benyei11169dd2012-12-18 14:30:41 +00001861 }
1862 SourceLocation getLoc() const {
1863 return SourceLocation::getFromRawEncoding((unsigned)(uintptr_t) data[1]);
1864 }
1865};
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001866class EnqueueVisitor : public ConstStmtVisitor<EnqueueVisitor, void> {
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001867 friend class OMPClauseEnqueue;
Guy Benyei11169dd2012-12-18 14:30:41 +00001868 VisitorWorkList &WL;
1869 CXCursor Parent;
1870public:
1871 EnqueueVisitor(VisitorWorkList &wl, CXCursor parent)
1872 : WL(wl), Parent(parent) {}
1873
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001874 void VisitAddrLabelExpr(const AddrLabelExpr *E);
1875 void VisitBlockExpr(const BlockExpr *B);
1876 void VisitCompoundLiteralExpr(const CompoundLiteralExpr *E);
1877 void VisitCompoundStmt(const CompoundStmt *S);
1878 void VisitCXXDefaultArgExpr(const CXXDefaultArgExpr *E) { /* Do nothing. */ }
1879 void VisitMSDependentExistsStmt(const MSDependentExistsStmt *S);
1880 void VisitCXXDependentScopeMemberExpr(const CXXDependentScopeMemberExpr *E);
1881 void VisitCXXNewExpr(const CXXNewExpr *E);
1882 void VisitCXXScalarValueInitExpr(const CXXScalarValueInitExpr *E);
1883 void VisitCXXOperatorCallExpr(const CXXOperatorCallExpr *E);
1884 void VisitCXXPseudoDestructorExpr(const CXXPseudoDestructorExpr *E);
1885 void VisitCXXTemporaryObjectExpr(const CXXTemporaryObjectExpr *E);
1886 void VisitCXXTypeidExpr(const CXXTypeidExpr *E);
1887 void VisitCXXUnresolvedConstructExpr(const CXXUnresolvedConstructExpr *E);
1888 void VisitCXXUuidofExpr(const CXXUuidofExpr *E);
1889 void VisitCXXCatchStmt(const CXXCatchStmt *S);
Argyrios Kyrtzidis99891242014-11-13 09:03:21 +00001890 void VisitCXXForRangeStmt(const CXXForRangeStmt *S);
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001891 void VisitDeclRefExpr(const DeclRefExpr *D);
1892 void VisitDeclStmt(const DeclStmt *S);
1893 void VisitDependentScopeDeclRefExpr(const DependentScopeDeclRefExpr *E);
1894 void VisitDesignatedInitExpr(const DesignatedInitExpr *E);
1895 void VisitExplicitCastExpr(const ExplicitCastExpr *E);
1896 void VisitForStmt(const ForStmt *FS);
1897 void VisitGotoStmt(const GotoStmt *GS);
1898 void VisitIfStmt(const IfStmt *If);
1899 void VisitInitListExpr(const InitListExpr *IE);
1900 void VisitMemberExpr(const MemberExpr *M);
1901 void VisitOffsetOfExpr(const OffsetOfExpr *E);
1902 void VisitObjCEncodeExpr(const ObjCEncodeExpr *E);
1903 void VisitObjCMessageExpr(const ObjCMessageExpr *M);
1904 void VisitOverloadExpr(const OverloadExpr *E);
1905 void VisitUnaryExprOrTypeTraitExpr(const UnaryExprOrTypeTraitExpr *E);
1906 void VisitStmt(const Stmt *S);
1907 void VisitSwitchStmt(const SwitchStmt *S);
1908 void VisitWhileStmt(const WhileStmt *W);
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001909 void VisitTypeTraitExpr(const TypeTraitExpr *E);
1910 void VisitArrayTypeTraitExpr(const ArrayTypeTraitExpr *E);
1911 void VisitExpressionTraitExpr(const ExpressionTraitExpr *E);
1912 void VisitUnresolvedMemberExpr(const UnresolvedMemberExpr *U);
1913 void VisitVAArgExpr(const VAArgExpr *E);
1914 void VisitSizeOfPackExpr(const SizeOfPackExpr *E);
1915 void VisitPseudoObjectExpr(const PseudoObjectExpr *E);
1916 void VisitOpaqueValueExpr(const OpaqueValueExpr *E);
1917 void VisitLambdaExpr(const LambdaExpr *E);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001918 void VisitOMPExecutableDirective(const OMPExecutableDirective *D);
Alexander Musman3aaab662014-08-19 11:27:13 +00001919 void VisitOMPLoopDirective(const OMPLoopDirective *D);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001920 void VisitOMPParallelDirective(const OMPParallelDirective *D);
Alexey Bataev1b59ab52014-02-27 08:29:12 +00001921 void VisitOMPSimdDirective(const OMPSimdDirective *D);
Alexey Bataevf29276e2014-06-18 04:14:57 +00001922 void VisitOMPForDirective(const OMPForDirective *D);
Alexander Musmanf82886e2014-09-18 05:12:34 +00001923 void VisitOMPForSimdDirective(const OMPForSimdDirective *D);
Alexey Bataevd3f8dd22014-06-25 11:44:49 +00001924 void VisitOMPSectionsDirective(const OMPSectionsDirective *D);
Alexey Bataev1e0498a2014-06-26 08:21:58 +00001925 void VisitOMPSectionDirective(const OMPSectionDirective *D);
Alexey Bataevd1e40fb2014-06-26 12:05:45 +00001926 void VisitOMPSingleDirective(const OMPSingleDirective *D);
Alexander Musman80c22892014-07-17 08:54:58 +00001927 void VisitOMPMasterDirective(const OMPMasterDirective *D);
Alexander Musmand9ed09f2014-07-21 09:42:05 +00001928 void VisitOMPCriticalDirective(const OMPCriticalDirective *D);
Alexey Bataev4acb8592014-07-07 13:01:15 +00001929 void VisitOMPParallelForDirective(const OMPParallelForDirective *D);
Alexander Musmane4e893b2014-09-23 09:33:00 +00001930 void VisitOMPParallelForSimdDirective(const OMPParallelForSimdDirective *D);
Alexey Bataev84d0b3e2014-07-08 08:12:03 +00001931 void VisitOMPParallelSectionsDirective(const OMPParallelSectionsDirective *D);
Alexey Bataev9c2e8ee2014-07-11 11:25:16 +00001932 void VisitOMPTaskDirective(const OMPTaskDirective *D);
Alexey Bataev68446b72014-07-18 07:47:19 +00001933 void VisitOMPTaskyieldDirective(const OMPTaskyieldDirective *D);
Alexey Bataev4d1dfea2014-07-18 09:11:51 +00001934 void VisitOMPBarrierDirective(const OMPBarrierDirective *D);
Alexey Bataev2df347a2014-07-18 10:17:07 +00001935 void VisitOMPTaskwaitDirective(const OMPTaskwaitDirective *D);
Alexey Bataevc30dd2d2015-06-18 12:14:09 +00001936 void VisitOMPTaskgroupDirective(const OMPTaskgroupDirective *D);
Alexey Bataev6d4ed052015-07-01 06:57:41 +00001937 void
1938 VisitOMPCancellationPointDirective(const OMPCancellationPointDirective *D);
Alexey Bataev80909872015-07-02 11:25:17 +00001939 void VisitOMPCancelDirective(const OMPCancelDirective *D);
Alexey Bataev6125da92014-07-21 11:26:11 +00001940 void VisitOMPFlushDirective(const OMPFlushDirective *D);
Alexey Bataev9fb6e642014-07-22 06:45:04 +00001941 void VisitOMPOrderedDirective(const OMPOrderedDirective *D);
Alexey Bataev0162e452014-07-22 10:10:35 +00001942 void VisitOMPAtomicDirective(const OMPAtomicDirective *D);
Alexey Bataev0bd520b2014-09-19 08:19:49 +00001943 void VisitOMPTargetDirective(const OMPTargetDirective *D);
Michael Wong65f367f2015-07-21 13:44:28 +00001944 void VisitOMPTargetDataDirective(const OMPTargetDataDirective *D);
Samuel Antaodf67fc42016-01-19 19:15:56 +00001945 void VisitOMPTargetEnterDataDirective(const OMPTargetEnterDataDirective *D);
Samuel Antao72590762016-01-19 20:04:50 +00001946 void VisitOMPTargetExitDataDirective(const OMPTargetExitDataDirective *D);
Arpith Chacko Jacobe955b3d2016-01-26 18:48:41 +00001947 void VisitOMPTargetParallelDirective(const OMPTargetParallelDirective *D);
Arpith Chacko Jacob05bebb52016-02-03 15:46:42 +00001948 void
1949 VisitOMPTargetParallelForDirective(const OMPTargetParallelForDirective *D);
Alexey Bataev13314bf2014-10-09 04:18:56 +00001950 void VisitOMPTeamsDirective(const OMPTeamsDirective *D);
Alexey Bataev49f6e782015-12-01 04:18:41 +00001951 void VisitOMPTaskLoopDirective(const OMPTaskLoopDirective *D);
Alexey Bataev0a6ed842015-12-03 09:40:15 +00001952 void VisitOMPTaskLoopSimdDirective(const OMPTaskLoopSimdDirective *D);
Carlo Bertolli6200a3d2015-12-14 14:51:25 +00001953 void VisitOMPDistributeDirective(const OMPDistributeDirective *D);
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001954
Guy Benyei11169dd2012-12-18 14:30:41 +00001955private:
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001956 void AddDeclarationNameInfo(const Stmt *S);
Guy Benyei11169dd2012-12-18 14:30:41 +00001957 void AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier);
James Y Knight04ec5bf2015-12-24 02:59:37 +00001958 void AddExplicitTemplateArgs(const TemplateArgumentLoc *A,
1959 unsigned NumTemplateArgs);
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001960 void AddMemberRef(const FieldDecl *D, SourceLocation L);
1961 void AddStmt(const Stmt *S);
1962 void AddDecl(const Decl *D, bool isFirst = true);
Guy Benyei11169dd2012-12-18 14:30:41 +00001963 void AddTypeLoc(TypeSourceInfo *TI);
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001964 void EnqueueChildren(const Stmt *S);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001965 void EnqueueChildren(const OMPClause *S);
Guy Benyei11169dd2012-12-18 14:30:41 +00001966};
Alexander Kornienkoab9db512015-06-22 23:07:51 +00001967} // end anonyous namespace
Guy Benyei11169dd2012-12-18 14:30:41 +00001968
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001969void EnqueueVisitor::AddDeclarationNameInfo(const Stmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001970 // 'S' should always be non-null, since it comes from the
1971 // statement we are visiting.
1972 WL.push_back(DeclarationNameInfoVisit(S, Parent));
1973}
1974
1975void
1976EnqueueVisitor::AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
1977 if (Qualifier)
1978 WL.push_back(NestedNameSpecifierLocVisit(Qualifier, Parent));
1979}
1980
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001981void EnqueueVisitor::AddStmt(const Stmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001982 if (S)
1983 WL.push_back(StmtVisit(S, Parent));
1984}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001985void EnqueueVisitor::AddDecl(const Decl *D, bool isFirst) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001986 if (D)
1987 WL.push_back(DeclVisit(D, Parent, isFirst));
1988}
James Y Knight04ec5bf2015-12-24 02:59:37 +00001989void EnqueueVisitor::AddExplicitTemplateArgs(const TemplateArgumentLoc *A,
1990 unsigned NumTemplateArgs) {
1991 WL.push_back(ExplicitTemplateArgsVisit(A, A + NumTemplateArgs, Parent));
Guy Benyei11169dd2012-12-18 14:30:41 +00001992}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001993void EnqueueVisitor::AddMemberRef(const FieldDecl *D, SourceLocation L) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001994 if (D)
1995 WL.push_back(MemberRefVisit(D, L, Parent));
1996}
1997void EnqueueVisitor::AddTypeLoc(TypeSourceInfo *TI) {
1998 if (TI)
1999 WL.push_back(TypeLocVisit(TI->getTypeLoc(), Parent));
2000 }
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002001void EnqueueVisitor::EnqueueChildren(const Stmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002002 unsigned size = WL.size();
Benjamin Kramer642f1732015-07-02 21:03:14 +00002003 for (const Stmt *SubStmt : S->children()) {
2004 AddStmt(SubStmt);
Guy Benyei11169dd2012-12-18 14:30:41 +00002005 }
2006 if (size == WL.size())
2007 return;
2008 // Now reverse the entries we just added. This will match the DFS
2009 // ordering performed by the worklist.
2010 VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
2011 std::reverse(I, E);
2012}
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002013namespace {
2014class OMPClauseEnqueue : public ConstOMPClauseVisitor<OMPClauseEnqueue> {
2015 EnqueueVisitor *Visitor;
Alexey Bataev756c1962013-09-24 03:17:45 +00002016 /// \brief Process clauses with list of variables.
2017 template <typename T>
2018 void VisitOMPClauseList(T *Node);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002019public:
2020 OMPClauseEnqueue(EnqueueVisitor *Visitor) : Visitor(Visitor) { }
2021#define OPENMP_CLAUSE(Name, Class) \
2022 void Visit##Class(const Class *C);
2023#include "clang/Basic/OpenMPKinds.def"
Alexey Bataev3392d762016-02-16 11:18:12 +00002024 void VisitOMPClauseWithPreInit(const OMPClauseWithPreInit *C);
Alexey Bataev005248a2016-02-25 05:25:57 +00002025 void VisitOMPClauseWithPostUpdate(const OMPClauseWithPostUpdate *C);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002026};
2027
Alexey Bataev3392d762016-02-16 11:18:12 +00002028void OMPClauseEnqueue::VisitOMPClauseWithPreInit(
2029 const OMPClauseWithPreInit *C) {
2030 Visitor->AddStmt(C->getPreInitStmt());
2031}
2032
Alexey Bataev005248a2016-02-25 05:25:57 +00002033void OMPClauseEnqueue::VisitOMPClauseWithPostUpdate(
2034 const OMPClauseWithPostUpdate *C) {
Alexey Bataev37e594c2016-03-04 07:21:16 +00002035 VisitOMPClauseWithPreInit(C);
Alexey Bataev005248a2016-02-25 05:25:57 +00002036 Visitor->AddStmt(C->getPostUpdateExpr());
2037}
2038
Alexey Bataevaadd52e2014-02-13 05:29:23 +00002039void OMPClauseEnqueue::VisitOMPIfClause(const OMPIfClause *C) {
2040 Visitor->AddStmt(C->getCondition());
2041}
2042
Alexey Bataev3778b602014-07-17 07:32:53 +00002043void OMPClauseEnqueue::VisitOMPFinalClause(const OMPFinalClause *C) {
2044 Visitor->AddStmt(C->getCondition());
2045}
2046
Alexey Bataev568a8332014-03-06 06:15:19 +00002047void OMPClauseEnqueue::VisitOMPNumThreadsClause(const OMPNumThreadsClause *C) {
2048 Visitor->AddStmt(C->getNumThreads());
2049}
2050
Alexey Bataev62c87d22014-03-21 04:51:18 +00002051void OMPClauseEnqueue::VisitOMPSafelenClause(const OMPSafelenClause *C) {
2052 Visitor->AddStmt(C->getSafelen());
2053}
2054
Alexey Bataev66b15b52015-08-21 11:14:16 +00002055void OMPClauseEnqueue::VisitOMPSimdlenClause(const OMPSimdlenClause *C) {
2056 Visitor->AddStmt(C->getSimdlen());
2057}
2058
Alexander Musman8bd31e62014-05-27 15:12:19 +00002059void OMPClauseEnqueue::VisitOMPCollapseClause(const OMPCollapseClause *C) {
2060 Visitor->AddStmt(C->getNumForLoops());
2061}
2062
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002063void OMPClauseEnqueue::VisitOMPDefaultClause(const OMPDefaultClause *C) { }
Alexey Bataev756c1962013-09-24 03:17:45 +00002064
Alexey Bataevbcbadb62014-05-06 06:04:14 +00002065void OMPClauseEnqueue::VisitOMPProcBindClause(const OMPProcBindClause *C) { }
2066
Alexey Bataev56dafe82014-06-20 07:16:17 +00002067void OMPClauseEnqueue::VisitOMPScheduleClause(const OMPScheduleClause *C) {
Alexey Bataev3392d762016-02-16 11:18:12 +00002068 VisitOMPClauseWithPreInit(C);
Alexey Bataev56dafe82014-06-20 07:16:17 +00002069 Visitor->AddStmt(C->getChunkSize());
2070}
2071
Alexey Bataev10e775f2015-07-30 11:36:16 +00002072void OMPClauseEnqueue::VisitOMPOrderedClause(const OMPOrderedClause *C) {
2073 Visitor->AddStmt(C->getNumForLoops());
2074}
Alexey Bataev142e1fc2014-06-20 09:44:06 +00002075
Alexey Bataev236070f2014-06-20 11:19:47 +00002076void OMPClauseEnqueue::VisitOMPNowaitClause(const OMPNowaitClause *) {}
2077
Alexey Bataev7aea99a2014-07-17 12:19:31 +00002078void OMPClauseEnqueue::VisitOMPUntiedClause(const OMPUntiedClause *) {}
2079
Alexey Bataev74ba3a52014-07-17 12:47:03 +00002080void OMPClauseEnqueue::VisitOMPMergeableClause(const OMPMergeableClause *) {}
2081
Alexey Bataevf98b00c2014-07-23 02:27:21 +00002082void OMPClauseEnqueue::VisitOMPReadClause(const OMPReadClause *) {}
2083
Alexey Bataevdea47612014-07-23 07:46:59 +00002084void OMPClauseEnqueue::VisitOMPWriteClause(const OMPWriteClause *) {}
2085
Alexey Bataev67a4f222014-07-23 10:25:33 +00002086void OMPClauseEnqueue::VisitOMPUpdateClause(const OMPUpdateClause *) {}
2087
Alexey Bataev459dec02014-07-24 06:46:57 +00002088void OMPClauseEnqueue::VisitOMPCaptureClause(const OMPCaptureClause *) {}
2089
Alexey Bataev82bad8b2014-07-24 08:55:34 +00002090void OMPClauseEnqueue::VisitOMPSeqCstClause(const OMPSeqCstClause *) {}
2091
Alexey Bataev346265e2015-09-25 10:37:12 +00002092void OMPClauseEnqueue::VisitOMPThreadsClause(const OMPThreadsClause *) {}
2093
Alexey Bataevd14d1e62015-09-28 06:39:35 +00002094void OMPClauseEnqueue::VisitOMPSIMDClause(const OMPSIMDClause *) {}
2095
Alexey Bataevb825de12015-12-07 10:51:44 +00002096void OMPClauseEnqueue::VisitOMPNogroupClause(const OMPNogroupClause *) {}
2097
Michael Wonge710d542015-08-07 16:16:36 +00002098void OMPClauseEnqueue::VisitOMPDeviceClause(const OMPDeviceClause *C) {
2099 Visitor->AddStmt(C->getDevice());
2100}
2101
Kelvin Li099bb8c2015-11-24 20:50:12 +00002102void OMPClauseEnqueue::VisitOMPNumTeamsClause(const OMPNumTeamsClause *C) {
2103 Visitor->AddStmt(C->getNumTeams());
2104}
2105
Kelvin Lia15fb1a2015-11-27 18:47:36 +00002106void OMPClauseEnqueue::VisitOMPThreadLimitClause(const OMPThreadLimitClause *C) {
2107 Visitor->AddStmt(C->getThreadLimit());
2108}
2109
Alexey Bataeva0569352015-12-01 10:17:31 +00002110void OMPClauseEnqueue::VisitOMPPriorityClause(const OMPPriorityClause *C) {
2111 Visitor->AddStmt(C->getPriority());
2112}
2113
Alexey Bataev1fd4aed2015-12-07 12:52:51 +00002114void OMPClauseEnqueue::VisitOMPGrainsizeClause(const OMPGrainsizeClause *C) {
2115 Visitor->AddStmt(C->getGrainsize());
2116}
2117
Alexey Bataev382967a2015-12-08 12:06:20 +00002118void OMPClauseEnqueue::VisitOMPNumTasksClause(const OMPNumTasksClause *C) {
2119 Visitor->AddStmt(C->getNumTasks());
2120}
2121
Alexey Bataev28c75412015-12-15 08:19:24 +00002122void OMPClauseEnqueue::VisitOMPHintClause(const OMPHintClause *C) {
2123 Visitor->AddStmt(C->getHint());
2124}
2125
Alexey Bataev756c1962013-09-24 03:17:45 +00002126template<typename T>
2127void OMPClauseEnqueue::VisitOMPClauseList(T *Node) {
Alexey Bataev03b340a2014-10-21 03:16:40 +00002128 for (const auto *I : Node->varlists()) {
Aaron Ballman2205d2a2014-03-14 15:55:35 +00002129 Visitor->AddStmt(I);
Alexey Bataev03b340a2014-10-21 03:16:40 +00002130 }
Alexey Bataev756c1962013-09-24 03:17:45 +00002131}
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002132
2133void OMPClauseEnqueue::VisitOMPPrivateClause(const OMPPrivateClause *C) {
Alexey Bataev756c1962013-09-24 03:17:45 +00002134 VisitOMPClauseList(C);
Alexey Bataev03b340a2014-10-21 03:16:40 +00002135 for (const auto *E : C->private_copies()) {
2136 Visitor->AddStmt(E);
2137 }
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002138}
Alexey Bataevd5af8e42013-10-01 05:32:34 +00002139void OMPClauseEnqueue::VisitOMPFirstprivateClause(
2140 const OMPFirstprivateClause *C) {
2141 VisitOMPClauseList(C);
Alexey Bataev417089f2016-02-17 13:19:37 +00002142 VisitOMPClauseWithPreInit(C);
2143 for (const auto *E : C->private_copies()) {
2144 Visitor->AddStmt(E);
2145 }
2146 for (const auto *E : C->inits()) {
2147 Visitor->AddStmt(E);
2148 }
Alexey Bataevd5af8e42013-10-01 05:32:34 +00002149}
Alexander Musman1bb328c2014-06-04 13:06:39 +00002150void OMPClauseEnqueue::VisitOMPLastprivateClause(
2151 const OMPLastprivateClause *C) {
2152 VisitOMPClauseList(C);
Alexey Bataev005248a2016-02-25 05:25:57 +00002153 VisitOMPClauseWithPostUpdate(C);
Alexey Bataev38e89532015-04-16 04:54:05 +00002154 for (auto *E : C->private_copies()) {
2155 Visitor->AddStmt(E);
2156 }
2157 for (auto *E : C->source_exprs()) {
2158 Visitor->AddStmt(E);
2159 }
2160 for (auto *E : C->destination_exprs()) {
2161 Visitor->AddStmt(E);
2162 }
2163 for (auto *E : C->assignment_ops()) {
2164 Visitor->AddStmt(E);
2165 }
Alexander Musman1bb328c2014-06-04 13:06:39 +00002166}
Alexey Bataev758e55e2013-09-06 18:03:48 +00002167void OMPClauseEnqueue::VisitOMPSharedClause(const OMPSharedClause *C) {
Alexey Bataev756c1962013-09-24 03:17:45 +00002168 VisitOMPClauseList(C);
Alexey Bataev758e55e2013-09-06 18:03:48 +00002169}
Alexey Bataevc5e02582014-06-16 07:08:35 +00002170void OMPClauseEnqueue::VisitOMPReductionClause(const OMPReductionClause *C) {
2171 VisitOMPClauseList(C);
Alexey Bataev61205072016-03-02 04:57:40 +00002172 VisitOMPClauseWithPostUpdate(C);
Alexey Bataevf24e7b12015-10-08 09:10:53 +00002173 for (auto *E : C->privates()) {
2174 Visitor->AddStmt(E);
2175 }
Alexey Bataev794ba0d2015-04-10 10:43:45 +00002176 for (auto *E : C->lhs_exprs()) {
2177 Visitor->AddStmt(E);
2178 }
2179 for (auto *E : C->rhs_exprs()) {
2180 Visitor->AddStmt(E);
2181 }
2182 for (auto *E : C->reduction_ops()) {
2183 Visitor->AddStmt(E);
2184 }
Alexey Bataevc5e02582014-06-16 07:08:35 +00002185}
Alexander Musman8dba6642014-04-22 13:09:42 +00002186void OMPClauseEnqueue::VisitOMPLinearClause(const OMPLinearClause *C) {
2187 VisitOMPClauseList(C);
Alexey Bataev78849fb2016-03-09 09:49:00 +00002188 VisitOMPClauseWithPostUpdate(C);
Alexey Bataevbd9fec12015-08-18 06:47:21 +00002189 for (const auto *E : C->privates()) {
2190 Visitor->AddStmt(E);
2191 }
Alexander Musman3276a272015-03-21 10:12:56 +00002192 for (const auto *E : C->inits()) {
2193 Visitor->AddStmt(E);
2194 }
2195 for (const auto *E : C->updates()) {
2196 Visitor->AddStmt(E);
2197 }
2198 for (const auto *E : C->finals()) {
2199 Visitor->AddStmt(E);
2200 }
Alexander Musman8dba6642014-04-22 13:09:42 +00002201 Visitor->AddStmt(C->getStep());
Alexander Musman3276a272015-03-21 10:12:56 +00002202 Visitor->AddStmt(C->getCalcStep());
Alexander Musman8dba6642014-04-22 13:09:42 +00002203}
Alexander Musmanf0d76e72014-05-29 14:36:25 +00002204void OMPClauseEnqueue::VisitOMPAlignedClause(const OMPAlignedClause *C) {
2205 VisitOMPClauseList(C);
2206 Visitor->AddStmt(C->getAlignment());
2207}
Alexey Bataevd48bcd82014-03-31 03:36:38 +00002208void OMPClauseEnqueue::VisitOMPCopyinClause(const OMPCopyinClause *C) {
2209 VisitOMPClauseList(C);
Alexey Bataevf56f98c2015-04-16 05:39:01 +00002210 for (auto *E : C->source_exprs()) {
2211 Visitor->AddStmt(E);
2212 }
2213 for (auto *E : C->destination_exprs()) {
2214 Visitor->AddStmt(E);
2215 }
2216 for (auto *E : C->assignment_ops()) {
2217 Visitor->AddStmt(E);
2218 }
Alexey Bataevd48bcd82014-03-31 03:36:38 +00002219}
Alexey Bataevbae9a792014-06-27 10:37:06 +00002220void
2221OMPClauseEnqueue::VisitOMPCopyprivateClause(const OMPCopyprivateClause *C) {
2222 VisitOMPClauseList(C);
Alexey Bataeva63048e2015-03-23 06:18:07 +00002223 for (auto *E : C->source_exprs()) {
2224 Visitor->AddStmt(E);
2225 }
2226 for (auto *E : C->destination_exprs()) {
2227 Visitor->AddStmt(E);
2228 }
2229 for (auto *E : C->assignment_ops()) {
2230 Visitor->AddStmt(E);
2231 }
Alexey Bataevbae9a792014-06-27 10:37:06 +00002232}
Alexey Bataev6125da92014-07-21 11:26:11 +00002233void OMPClauseEnqueue::VisitOMPFlushClause(const OMPFlushClause *C) {
2234 VisitOMPClauseList(C);
2235}
Alexey Bataev1c2cfbc2015-06-23 14:25:19 +00002236void OMPClauseEnqueue::VisitOMPDependClause(const OMPDependClause *C) {
2237 VisitOMPClauseList(C);
2238}
Kelvin Li0bff7af2015-11-23 05:32:03 +00002239void OMPClauseEnqueue::VisitOMPMapClause(const OMPMapClause *C) {
2240 VisitOMPClauseList(C);
2241}
Carlo Bertollib4adf552016-01-15 18:50:31 +00002242void OMPClauseEnqueue::VisitOMPDistScheduleClause(
2243 const OMPDistScheduleClause *C) {
Alexey Bataev3392d762016-02-16 11:18:12 +00002244 VisitOMPClauseWithPreInit(C);
Carlo Bertollib4adf552016-01-15 18:50:31 +00002245 Visitor->AddStmt(C->getChunkSize());
Carlo Bertollib4adf552016-01-15 18:50:31 +00002246}
Alexey Bataev3392d762016-02-16 11:18:12 +00002247void OMPClauseEnqueue::VisitOMPDefaultmapClause(
2248 const OMPDefaultmapClause * /*C*/) {}
Alexander Kornienkoab9db512015-06-22 23:07:51 +00002249}
Alexey Bataev756c1962013-09-24 03:17:45 +00002250
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002251void EnqueueVisitor::EnqueueChildren(const OMPClause *S) {
2252 unsigned size = WL.size();
2253 OMPClauseEnqueue Visitor(this);
2254 Visitor.Visit(S);
2255 if (size == WL.size())
2256 return;
2257 // Now reverse the entries we just added. This will match the DFS
2258 // ordering performed by the worklist.
2259 VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
2260 std::reverse(I, E);
2261}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002262void EnqueueVisitor::VisitAddrLabelExpr(const AddrLabelExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002263 WL.push_back(LabelRefVisit(E->getLabel(), E->getLabelLoc(), Parent));
2264}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002265void EnqueueVisitor::VisitBlockExpr(const BlockExpr *B) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002266 AddDecl(B->getBlockDecl());
2267}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002268void EnqueueVisitor::VisitCompoundLiteralExpr(const CompoundLiteralExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002269 EnqueueChildren(E);
2270 AddTypeLoc(E->getTypeSourceInfo());
2271}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002272void EnqueueVisitor::VisitCompoundStmt(const CompoundStmt *S) {
Pete Cooper57d3f142015-07-30 17:22:52 +00002273 for (auto &I : llvm::reverse(S->body()))
2274 AddStmt(I);
Guy Benyei11169dd2012-12-18 14:30:41 +00002275}
2276void EnqueueVisitor::
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002277VisitMSDependentExistsStmt(const MSDependentExistsStmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002278 AddStmt(S->getSubStmt());
2279 AddDeclarationNameInfo(S);
2280 if (NestedNameSpecifierLoc QualifierLoc = S->getQualifierLoc())
2281 AddNestedNameSpecifierLoc(QualifierLoc);
2282}
2283
2284void EnqueueVisitor::
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002285VisitCXXDependentScopeMemberExpr(const CXXDependentScopeMemberExpr *E) {
James Y Knight04ec5bf2015-12-24 02:59:37 +00002286 if (E->hasExplicitTemplateArgs())
2287 AddExplicitTemplateArgs(E->getTemplateArgs(), E->getNumTemplateArgs());
Guy Benyei11169dd2012-12-18 14:30:41 +00002288 AddDeclarationNameInfo(E);
2289 if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
2290 AddNestedNameSpecifierLoc(QualifierLoc);
2291 if (!E->isImplicitAccess())
2292 AddStmt(E->getBase());
2293}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002294void EnqueueVisitor::VisitCXXNewExpr(const CXXNewExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002295 // Enqueue the initializer , if any.
2296 AddStmt(E->getInitializer());
2297 // Enqueue the array size, if any.
2298 AddStmt(E->getArraySize());
2299 // Enqueue the allocated type.
2300 AddTypeLoc(E->getAllocatedTypeSourceInfo());
2301 // Enqueue the placement arguments.
2302 for (unsigned I = E->getNumPlacementArgs(); I > 0; --I)
2303 AddStmt(E->getPlacementArg(I-1));
2304}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002305void EnqueueVisitor::VisitCXXOperatorCallExpr(const CXXOperatorCallExpr *CE) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002306 for (unsigned I = CE->getNumArgs(); I > 1 /* Yes, this is 1 */; --I)
2307 AddStmt(CE->getArg(I-1));
2308 AddStmt(CE->getCallee());
2309 AddStmt(CE->getArg(0));
2310}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002311void EnqueueVisitor::VisitCXXPseudoDestructorExpr(
2312 const CXXPseudoDestructorExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002313 // Visit the name of the type being destroyed.
2314 AddTypeLoc(E->getDestroyedTypeInfo());
2315 // Visit the scope type that looks disturbingly like the nested-name-specifier
2316 // but isn't.
2317 AddTypeLoc(E->getScopeTypeInfo());
2318 // Visit the nested-name-specifier.
2319 if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
2320 AddNestedNameSpecifierLoc(QualifierLoc);
2321 // Visit base expression.
2322 AddStmt(E->getBase());
2323}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002324void EnqueueVisitor::VisitCXXScalarValueInitExpr(
2325 const CXXScalarValueInitExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002326 AddTypeLoc(E->getTypeSourceInfo());
2327}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002328void EnqueueVisitor::VisitCXXTemporaryObjectExpr(
2329 const CXXTemporaryObjectExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002330 EnqueueChildren(E);
2331 AddTypeLoc(E->getTypeSourceInfo());
2332}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002333void EnqueueVisitor::VisitCXXTypeidExpr(const CXXTypeidExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002334 EnqueueChildren(E);
2335 if (E->isTypeOperand())
2336 AddTypeLoc(E->getTypeOperandSourceInfo());
2337}
2338
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002339void EnqueueVisitor::VisitCXXUnresolvedConstructExpr(
2340 const CXXUnresolvedConstructExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002341 EnqueueChildren(E);
2342 AddTypeLoc(E->getTypeSourceInfo());
2343}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002344void EnqueueVisitor::VisitCXXUuidofExpr(const CXXUuidofExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002345 EnqueueChildren(E);
2346 if (E->isTypeOperand())
2347 AddTypeLoc(E->getTypeOperandSourceInfo());
2348}
2349
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002350void EnqueueVisitor::VisitCXXCatchStmt(const CXXCatchStmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002351 EnqueueChildren(S);
2352 AddDecl(S->getExceptionDecl());
2353}
2354
Argyrios Kyrtzidis99891242014-11-13 09:03:21 +00002355void EnqueueVisitor::VisitCXXForRangeStmt(const CXXForRangeStmt *S) {
Argyrios Kyrtzidiscde70692014-11-13 09:50:19 +00002356 AddStmt(S->getBody());
Argyrios Kyrtzidis99891242014-11-13 09:03:21 +00002357 AddStmt(S->getRangeInit());
Argyrios Kyrtzidiscde70692014-11-13 09:50:19 +00002358 AddDecl(S->getLoopVariable());
Argyrios Kyrtzidis99891242014-11-13 09:03:21 +00002359}
2360
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002361void EnqueueVisitor::VisitDeclRefExpr(const DeclRefExpr *DR) {
James Y Knight04ec5bf2015-12-24 02:59:37 +00002362 if (DR->hasExplicitTemplateArgs())
2363 AddExplicitTemplateArgs(DR->getTemplateArgs(), DR->getNumTemplateArgs());
Guy Benyei11169dd2012-12-18 14:30:41 +00002364 WL.push_back(DeclRefExprParts(DR, Parent));
2365}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002366void EnqueueVisitor::VisitDependentScopeDeclRefExpr(
2367 const DependentScopeDeclRefExpr *E) {
James Y Knight04ec5bf2015-12-24 02:59:37 +00002368 if (E->hasExplicitTemplateArgs())
2369 AddExplicitTemplateArgs(E->getTemplateArgs(), E->getNumTemplateArgs());
Guy Benyei11169dd2012-12-18 14:30:41 +00002370 AddDeclarationNameInfo(E);
2371 AddNestedNameSpecifierLoc(E->getQualifierLoc());
2372}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002373void EnqueueVisitor::VisitDeclStmt(const DeclStmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002374 unsigned size = WL.size();
2375 bool isFirst = true;
Aaron Ballman535bbcc2014-03-14 17:01:24 +00002376 for (const auto *D : S->decls()) {
2377 AddDecl(D, isFirst);
Guy Benyei11169dd2012-12-18 14:30:41 +00002378 isFirst = false;
2379 }
2380 if (size == WL.size())
2381 return;
2382 // Now reverse the entries we just added. This will match the DFS
2383 // ordering performed by the worklist.
2384 VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
2385 std::reverse(I, E);
2386}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002387void EnqueueVisitor::VisitDesignatedInitExpr(const DesignatedInitExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002388 AddStmt(E->getInit());
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002389 for (DesignatedInitExpr::const_reverse_designators_iterator
Guy Benyei11169dd2012-12-18 14:30:41 +00002390 D = E->designators_rbegin(), DEnd = E->designators_rend();
2391 D != DEnd; ++D) {
2392 if (D->isFieldDesignator()) {
2393 if (FieldDecl *Field = D->getField())
2394 AddMemberRef(Field, D->getFieldLoc());
2395 continue;
2396 }
2397 if (D->isArrayDesignator()) {
2398 AddStmt(E->getArrayIndex(*D));
2399 continue;
2400 }
2401 assert(D->isArrayRangeDesignator() && "Unknown designator kind");
2402 AddStmt(E->getArrayRangeEnd(*D));
2403 AddStmt(E->getArrayRangeStart(*D));
2404 }
2405}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002406void EnqueueVisitor::VisitExplicitCastExpr(const ExplicitCastExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002407 EnqueueChildren(E);
2408 AddTypeLoc(E->getTypeInfoAsWritten());
2409}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002410void EnqueueVisitor::VisitForStmt(const ForStmt *FS) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002411 AddStmt(FS->getBody());
2412 AddStmt(FS->getInc());
2413 AddStmt(FS->getCond());
2414 AddDecl(FS->getConditionVariable());
2415 AddStmt(FS->getInit());
2416}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002417void EnqueueVisitor::VisitGotoStmt(const GotoStmt *GS) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002418 WL.push_back(LabelRefVisit(GS->getLabel(), GS->getLabelLoc(), Parent));
2419}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002420void EnqueueVisitor::VisitIfStmt(const IfStmt *If) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002421 AddStmt(If->getElse());
2422 AddStmt(If->getThen());
2423 AddStmt(If->getCond());
2424 AddDecl(If->getConditionVariable());
2425}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002426void EnqueueVisitor::VisitInitListExpr(const InitListExpr *IE) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002427 // We care about the syntactic form of the initializer list, only.
2428 if (InitListExpr *Syntactic = IE->getSyntacticForm())
2429 IE = Syntactic;
2430 EnqueueChildren(IE);
2431}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002432void EnqueueVisitor::VisitMemberExpr(const MemberExpr *M) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002433 WL.push_back(MemberExprParts(M, Parent));
2434
2435 // If the base of the member access expression is an implicit 'this', don't
2436 // visit it.
2437 // FIXME: If we ever want to show these implicit accesses, this will be
2438 // unfortunate. However, clang_getCursor() relies on this behavior.
Argyrios Kyrtzidis58d0e7a2015-03-13 04:40:07 +00002439 if (M->isImplicitAccess())
2440 return;
2441
2442 // Ignore base anonymous struct/union fields, otherwise they will shadow the
2443 // real field that that we are interested in.
2444 if (auto *SubME = dyn_cast<MemberExpr>(M->getBase())) {
2445 if (auto *FD = dyn_cast_or_null<FieldDecl>(SubME->getMemberDecl())) {
2446 if (FD->isAnonymousStructOrUnion()) {
2447 AddStmt(SubME->getBase());
2448 return;
2449 }
2450 }
2451 }
2452
2453 AddStmt(M->getBase());
Guy Benyei11169dd2012-12-18 14:30:41 +00002454}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002455void EnqueueVisitor::VisitObjCEncodeExpr(const ObjCEncodeExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002456 AddTypeLoc(E->getEncodedTypeSourceInfo());
2457}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002458void EnqueueVisitor::VisitObjCMessageExpr(const ObjCMessageExpr *M) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002459 EnqueueChildren(M);
2460 AddTypeLoc(M->getClassReceiverTypeInfo());
2461}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002462void EnqueueVisitor::VisitOffsetOfExpr(const OffsetOfExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002463 // Visit the components of the offsetof expression.
2464 for (unsigned N = E->getNumComponents(), I = N; I > 0; --I) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002465 const OffsetOfNode &Node = E->getComponent(I-1);
2466 switch (Node.getKind()) {
2467 case OffsetOfNode::Array:
2468 AddStmt(E->getIndexExpr(Node.getArrayExprIndex()));
2469 break;
2470 case OffsetOfNode::Field:
2471 AddMemberRef(Node.getField(), Node.getSourceRange().getEnd());
2472 break;
2473 case OffsetOfNode::Identifier:
2474 case OffsetOfNode::Base:
2475 continue;
2476 }
2477 }
2478 // Visit the type into which we're computing the offset.
2479 AddTypeLoc(E->getTypeSourceInfo());
2480}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002481void EnqueueVisitor::VisitOverloadExpr(const OverloadExpr *E) {
James Y Knight04ec5bf2015-12-24 02:59:37 +00002482 if (E->hasExplicitTemplateArgs())
2483 AddExplicitTemplateArgs(E->getTemplateArgs(), E->getNumTemplateArgs());
Guy Benyei11169dd2012-12-18 14:30:41 +00002484 WL.push_back(OverloadExprParts(E, Parent));
2485}
2486void EnqueueVisitor::VisitUnaryExprOrTypeTraitExpr(
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002487 const UnaryExprOrTypeTraitExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002488 EnqueueChildren(E);
2489 if (E->isArgumentType())
2490 AddTypeLoc(E->getArgumentTypeInfo());
2491}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002492void EnqueueVisitor::VisitStmt(const Stmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002493 EnqueueChildren(S);
2494}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002495void EnqueueVisitor::VisitSwitchStmt(const SwitchStmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002496 AddStmt(S->getBody());
2497 AddStmt(S->getCond());
2498 AddDecl(S->getConditionVariable());
2499}
2500
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002501void EnqueueVisitor::VisitWhileStmt(const WhileStmt *W) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002502 AddStmt(W->getBody());
2503 AddStmt(W->getCond());
2504 AddDecl(W->getConditionVariable());
2505}
2506
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002507void EnqueueVisitor::VisitTypeTraitExpr(const TypeTraitExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002508 for (unsigned I = E->getNumArgs(); I > 0; --I)
2509 AddTypeLoc(E->getArg(I-1));
2510}
2511
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002512void EnqueueVisitor::VisitArrayTypeTraitExpr(const ArrayTypeTraitExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002513 AddTypeLoc(E->getQueriedTypeSourceInfo());
2514}
2515
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002516void EnqueueVisitor::VisitExpressionTraitExpr(const ExpressionTraitExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002517 EnqueueChildren(E);
2518}
2519
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002520void EnqueueVisitor::VisitUnresolvedMemberExpr(const UnresolvedMemberExpr *U) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002521 VisitOverloadExpr(U);
2522 if (!U->isImplicitAccess())
2523 AddStmt(U->getBase());
2524}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002525void EnqueueVisitor::VisitVAArgExpr(const VAArgExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002526 AddStmt(E->getSubExpr());
2527 AddTypeLoc(E->getWrittenTypeInfo());
2528}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002529void EnqueueVisitor::VisitSizeOfPackExpr(const SizeOfPackExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002530 WL.push_back(SizeOfPackExprParts(E, Parent));
2531}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002532void EnqueueVisitor::VisitOpaqueValueExpr(const OpaqueValueExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002533 // If the opaque value has a source expression, just transparently
2534 // visit that. This is useful for (e.g.) pseudo-object expressions.
2535 if (Expr *SourceExpr = E->getSourceExpr())
2536 return Visit(SourceExpr);
2537}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002538void EnqueueVisitor::VisitLambdaExpr(const LambdaExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002539 AddStmt(E->getBody());
2540 WL.push_back(LambdaExprParts(E, Parent));
2541}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002542void EnqueueVisitor::VisitPseudoObjectExpr(const PseudoObjectExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002543 // Treat the expression like its syntactic form.
2544 Visit(E->getSyntacticForm());
2545}
2546
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002547void EnqueueVisitor::VisitOMPExecutableDirective(
2548 const OMPExecutableDirective *D) {
2549 EnqueueChildren(D);
2550 for (ArrayRef<OMPClause *>::iterator I = D->clauses().begin(),
2551 E = D->clauses().end();
2552 I != E; ++I)
2553 EnqueueChildren(*I);
2554}
2555
Alexander Musman3aaab662014-08-19 11:27:13 +00002556void EnqueueVisitor::VisitOMPLoopDirective(const OMPLoopDirective *D) {
2557 VisitOMPExecutableDirective(D);
2558}
2559
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002560void EnqueueVisitor::VisitOMPParallelDirective(const OMPParallelDirective *D) {
2561 VisitOMPExecutableDirective(D);
2562}
2563
Alexey Bataev1b59ab52014-02-27 08:29:12 +00002564void EnqueueVisitor::VisitOMPSimdDirective(const OMPSimdDirective *D) {
Alexander Musman3aaab662014-08-19 11:27:13 +00002565 VisitOMPLoopDirective(D);
Alexey Bataev1b59ab52014-02-27 08:29:12 +00002566}
2567
Alexey Bataevf29276e2014-06-18 04:14:57 +00002568void EnqueueVisitor::VisitOMPForDirective(const OMPForDirective *D) {
Alexander Musman3aaab662014-08-19 11:27:13 +00002569 VisitOMPLoopDirective(D);
Alexey Bataevf29276e2014-06-18 04:14:57 +00002570}
2571
Alexander Musmanf82886e2014-09-18 05:12:34 +00002572void EnqueueVisitor::VisitOMPForSimdDirective(const OMPForSimdDirective *D) {
2573 VisitOMPLoopDirective(D);
2574}
2575
Alexey Bataevd3f8dd22014-06-25 11:44:49 +00002576void EnqueueVisitor::VisitOMPSectionsDirective(const OMPSectionsDirective *D) {
2577 VisitOMPExecutableDirective(D);
2578}
2579
Alexey Bataev1e0498a2014-06-26 08:21:58 +00002580void EnqueueVisitor::VisitOMPSectionDirective(const OMPSectionDirective *D) {
2581 VisitOMPExecutableDirective(D);
2582}
2583
Alexey Bataevd1e40fb2014-06-26 12:05:45 +00002584void EnqueueVisitor::VisitOMPSingleDirective(const OMPSingleDirective *D) {
2585 VisitOMPExecutableDirective(D);
2586}
2587
Alexander Musman80c22892014-07-17 08:54:58 +00002588void EnqueueVisitor::VisitOMPMasterDirective(const OMPMasterDirective *D) {
2589 VisitOMPExecutableDirective(D);
2590}
2591
Alexander Musmand9ed09f2014-07-21 09:42:05 +00002592void EnqueueVisitor::VisitOMPCriticalDirective(const OMPCriticalDirective *D) {
2593 VisitOMPExecutableDirective(D);
2594 AddDeclarationNameInfo(D);
2595}
2596
Alexey Bataev4acb8592014-07-07 13:01:15 +00002597void
2598EnqueueVisitor::VisitOMPParallelForDirective(const OMPParallelForDirective *D) {
Alexander Musman3aaab662014-08-19 11:27:13 +00002599 VisitOMPLoopDirective(D);
Alexey Bataev4acb8592014-07-07 13:01:15 +00002600}
2601
Alexander Musmane4e893b2014-09-23 09:33:00 +00002602void EnqueueVisitor::VisitOMPParallelForSimdDirective(
2603 const OMPParallelForSimdDirective *D) {
2604 VisitOMPLoopDirective(D);
2605}
2606
Alexey Bataev84d0b3e2014-07-08 08:12:03 +00002607void EnqueueVisitor::VisitOMPParallelSectionsDirective(
2608 const OMPParallelSectionsDirective *D) {
2609 VisitOMPExecutableDirective(D);
2610}
2611
Alexey Bataev9c2e8ee2014-07-11 11:25:16 +00002612void EnqueueVisitor::VisitOMPTaskDirective(const OMPTaskDirective *D) {
2613 VisitOMPExecutableDirective(D);
2614}
2615
Alexey Bataev68446b72014-07-18 07:47:19 +00002616void
2617EnqueueVisitor::VisitOMPTaskyieldDirective(const OMPTaskyieldDirective *D) {
2618 VisitOMPExecutableDirective(D);
2619}
2620
Alexey Bataev4d1dfea2014-07-18 09:11:51 +00002621void EnqueueVisitor::VisitOMPBarrierDirective(const OMPBarrierDirective *D) {
2622 VisitOMPExecutableDirective(D);
2623}
2624
Alexey Bataev2df347a2014-07-18 10:17:07 +00002625void EnqueueVisitor::VisitOMPTaskwaitDirective(const OMPTaskwaitDirective *D) {
2626 VisitOMPExecutableDirective(D);
2627}
2628
Alexey Bataevc30dd2d2015-06-18 12:14:09 +00002629void EnqueueVisitor::VisitOMPTaskgroupDirective(
2630 const OMPTaskgroupDirective *D) {
2631 VisitOMPExecutableDirective(D);
2632}
2633
Alexey Bataev6125da92014-07-21 11:26:11 +00002634void EnqueueVisitor::VisitOMPFlushDirective(const OMPFlushDirective *D) {
2635 VisitOMPExecutableDirective(D);
2636}
2637
Alexey Bataev9fb6e642014-07-22 06:45:04 +00002638void EnqueueVisitor::VisitOMPOrderedDirective(const OMPOrderedDirective *D) {
2639 VisitOMPExecutableDirective(D);
2640}
2641
Alexey Bataev0162e452014-07-22 10:10:35 +00002642void EnqueueVisitor::VisitOMPAtomicDirective(const OMPAtomicDirective *D) {
2643 VisitOMPExecutableDirective(D);
2644}
2645
Alexey Bataev0bd520b2014-09-19 08:19:49 +00002646void EnqueueVisitor::VisitOMPTargetDirective(const OMPTargetDirective *D) {
2647 VisitOMPExecutableDirective(D);
2648}
2649
Michael Wong65f367f2015-07-21 13:44:28 +00002650void EnqueueVisitor::VisitOMPTargetDataDirective(const
2651 OMPTargetDataDirective *D) {
2652 VisitOMPExecutableDirective(D);
2653}
2654
Samuel Antaodf67fc42016-01-19 19:15:56 +00002655void EnqueueVisitor::VisitOMPTargetEnterDataDirective(
2656 const OMPTargetEnterDataDirective *D) {
2657 VisitOMPExecutableDirective(D);
2658}
2659
Samuel Antao72590762016-01-19 20:04:50 +00002660void EnqueueVisitor::VisitOMPTargetExitDataDirective(
2661 const OMPTargetExitDataDirective *D) {
2662 VisitOMPExecutableDirective(D);
2663}
2664
Arpith Chacko Jacobe955b3d2016-01-26 18:48:41 +00002665void EnqueueVisitor::VisitOMPTargetParallelDirective(
2666 const OMPTargetParallelDirective *D) {
2667 VisitOMPExecutableDirective(D);
2668}
2669
Arpith Chacko Jacob05bebb52016-02-03 15:46:42 +00002670void EnqueueVisitor::VisitOMPTargetParallelForDirective(
2671 const OMPTargetParallelForDirective *D) {
2672 VisitOMPLoopDirective(D);
2673}
2674
Alexey Bataev13314bf2014-10-09 04:18:56 +00002675void EnqueueVisitor::VisitOMPTeamsDirective(const OMPTeamsDirective *D) {
2676 VisitOMPExecutableDirective(D);
2677}
2678
Alexey Bataev6d4ed052015-07-01 06:57:41 +00002679void EnqueueVisitor::VisitOMPCancellationPointDirective(
2680 const OMPCancellationPointDirective *D) {
2681 VisitOMPExecutableDirective(D);
2682}
2683
Alexey Bataev80909872015-07-02 11:25:17 +00002684void EnqueueVisitor::VisitOMPCancelDirective(const OMPCancelDirective *D) {
2685 VisitOMPExecutableDirective(D);
2686}
2687
Alexey Bataev49f6e782015-12-01 04:18:41 +00002688void EnqueueVisitor::VisitOMPTaskLoopDirective(const OMPTaskLoopDirective *D) {
2689 VisitOMPLoopDirective(D);
2690}
2691
Alexey Bataev0a6ed842015-12-03 09:40:15 +00002692void EnqueueVisitor::VisitOMPTaskLoopSimdDirective(
2693 const OMPTaskLoopSimdDirective *D) {
2694 VisitOMPLoopDirective(D);
2695}
2696
Carlo Bertolli6200a3d2015-12-14 14:51:25 +00002697void EnqueueVisitor::VisitOMPDistributeDirective(
2698 const OMPDistributeDirective *D) {
2699 VisitOMPLoopDirective(D);
2700}
2701
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002702void CursorVisitor::EnqueueWorkList(VisitorWorkList &WL, const Stmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002703 EnqueueVisitor(WL, MakeCXCursor(S, StmtParent, TU,RegionOfInterest)).Visit(S);
2704}
2705
2706bool CursorVisitor::IsInRegionOfInterest(CXCursor C) {
2707 if (RegionOfInterest.isValid()) {
2708 SourceRange Range = getRawCursorExtent(C);
2709 if (Range.isInvalid() || CompareRegionOfInterest(Range))
2710 return false;
2711 }
2712 return true;
2713}
2714
2715bool CursorVisitor::RunVisitorWorkList(VisitorWorkList &WL) {
2716 while (!WL.empty()) {
2717 // Dequeue the worklist item.
Robert Wilhelm25284cc2013-08-23 16:11:15 +00002718 VisitorJob LI = WL.pop_back_val();
Guy Benyei11169dd2012-12-18 14:30:41 +00002719
2720 // Set the Parent field, then back to its old value once we're done.
2721 SetParentRAII SetParent(Parent, StmtParent, LI.getParent());
2722
2723 switch (LI.getKind()) {
2724 case VisitorJob::DeclVisitKind: {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002725 const Decl *D = cast<DeclVisit>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002726 if (!D)
2727 continue;
2728
2729 // For now, perform default visitation for Decls.
2730 if (Visit(MakeCXCursor(D, TU, RegionOfInterest,
2731 cast<DeclVisit>(&LI)->isFirst())))
2732 return true;
2733
2734 continue;
2735 }
2736 case VisitorJob::ExplicitTemplateArgsVisitKind: {
James Y Knight04ec5bf2015-12-24 02:59:37 +00002737 for (const TemplateArgumentLoc &Arg :
2738 *cast<ExplicitTemplateArgsVisit>(&LI)) {
2739 if (VisitTemplateArgumentLoc(Arg))
Guy Benyei11169dd2012-12-18 14:30:41 +00002740 return true;
2741 }
2742 continue;
2743 }
2744 case VisitorJob::TypeLocVisitKind: {
2745 // Perform default visitation for TypeLocs.
2746 if (Visit(cast<TypeLocVisit>(&LI)->get()))
2747 return true;
2748 continue;
2749 }
2750 case VisitorJob::LabelRefVisitKind: {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002751 const LabelDecl *LS = cast<LabelRefVisit>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002752 if (LabelStmt *stmt = LS->getStmt()) {
2753 if (Visit(MakeCursorLabelRef(stmt, cast<LabelRefVisit>(&LI)->getLoc(),
2754 TU))) {
2755 return true;
2756 }
2757 }
2758 continue;
2759 }
2760
2761 case VisitorJob::NestedNameSpecifierLocVisitKind: {
2762 NestedNameSpecifierLocVisit *V = cast<NestedNameSpecifierLocVisit>(&LI);
2763 if (VisitNestedNameSpecifierLoc(V->get()))
2764 return true;
2765 continue;
2766 }
2767
2768 case VisitorJob::DeclarationNameInfoVisitKind: {
2769 if (VisitDeclarationNameInfo(cast<DeclarationNameInfoVisit>(&LI)
2770 ->get()))
2771 return true;
2772 continue;
2773 }
2774 case VisitorJob::MemberRefVisitKind: {
2775 MemberRefVisit *V = cast<MemberRefVisit>(&LI);
2776 if (Visit(MakeCursorMemberRef(V->get(), V->getLoc(), TU)))
2777 return true;
2778 continue;
2779 }
2780 case VisitorJob::StmtVisitKind: {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002781 const Stmt *S = cast<StmtVisit>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002782 if (!S)
2783 continue;
2784
2785 // Update the current cursor.
2786 CXCursor Cursor = MakeCXCursor(S, StmtParent, TU, RegionOfInterest);
2787 if (!IsInRegionOfInterest(Cursor))
2788 continue;
2789 switch (Visitor(Cursor, Parent, ClientData)) {
2790 case CXChildVisit_Break: return true;
2791 case CXChildVisit_Continue: break;
2792 case CXChildVisit_Recurse:
2793 if (PostChildrenVisitor)
Craig Topper69186e72014-06-08 08:38:04 +00002794 WL.push_back(PostChildrenVisit(nullptr, Cursor));
Guy Benyei11169dd2012-12-18 14:30:41 +00002795 EnqueueWorkList(WL, S);
2796 break;
2797 }
2798 continue;
2799 }
2800 case VisitorJob::MemberExprPartsKind: {
2801 // Handle the other pieces in the MemberExpr besides the base.
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002802 const MemberExpr *M = cast<MemberExprParts>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002803
2804 // Visit the nested-name-specifier
2805 if (NestedNameSpecifierLoc QualifierLoc = M->getQualifierLoc())
2806 if (VisitNestedNameSpecifierLoc(QualifierLoc))
2807 return true;
2808
2809 // Visit the declaration name.
2810 if (VisitDeclarationNameInfo(M->getMemberNameInfo()))
2811 return true;
2812
2813 // Visit the explicitly-specified template arguments, if any.
2814 if (M->hasExplicitTemplateArgs()) {
2815 for (const TemplateArgumentLoc *Arg = M->getTemplateArgs(),
2816 *ArgEnd = Arg + M->getNumTemplateArgs();
2817 Arg != ArgEnd; ++Arg) {
2818 if (VisitTemplateArgumentLoc(*Arg))
2819 return true;
2820 }
2821 }
2822 continue;
2823 }
2824 case VisitorJob::DeclRefExprPartsKind: {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002825 const DeclRefExpr *DR = cast<DeclRefExprParts>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002826 // Visit nested-name-specifier, if present.
2827 if (NestedNameSpecifierLoc QualifierLoc = DR->getQualifierLoc())
2828 if (VisitNestedNameSpecifierLoc(QualifierLoc))
2829 return true;
2830 // Visit declaration name.
2831 if (VisitDeclarationNameInfo(DR->getNameInfo()))
2832 return true;
2833 continue;
2834 }
2835 case VisitorJob::OverloadExprPartsKind: {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002836 const OverloadExpr *O = cast<OverloadExprParts>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002837 // Visit the nested-name-specifier.
2838 if (NestedNameSpecifierLoc QualifierLoc = O->getQualifierLoc())
2839 if (VisitNestedNameSpecifierLoc(QualifierLoc))
2840 return true;
2841 // Visit the declaration name.
2842 if (VisitDeclarationNameInfo(O->getNameInfo()))
2843 return true;
2844 // Visit the overloaded declaration reference.
2845 if (Visit(MakeCursorOverloadedDeclRef(O, TU)))
2846 return true;
2847 continue;
2848 }
2849 case VisitorJob::SizeOfPackExprPartsKind: {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002850 const SizeOfPackExpr *E = cast<SizeOfPackExprParts>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002851 NamedDecl *Pack = E->getPack();
2852 if (isa<TemplateTypeParmDecl>(Pack)) {
2853 if (Visit(MakeCursorTypeRef(cast<TemplateTypeParmDecl>(Pack),
2854 E->getPackLoc(), TU)))
2855 return true;
2856
2857 continue;
2858 }
2859
2860 if (isa<TemplateTemplateParmDecl>(Pack)) {
2861 if (Visit(MakeCursorTemplateRef(cast<TemplateTemplateParmDecl>(Pack),
2862 E->getPackLoc(), TU)))
2863 return true;
2864
2865 continue;
2866 }
2867
2868 // Non-type template parameter packs and function parameter packs are
2869 // treated like DeclRefExpr cursors.
2870 continue;
2871 }
2872
2873 case VisitorJob::LambdaExprPartsKind: {
2874 // Visit captures.
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002875 const LambdaExpr *E = cast<LambdaExprParts>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002876 for (LambdaExpr::capture_iterator C = E->explicit_capture_begin(),
2877 CEnd = E->explicit_capture_end();
2878 C != CEnd; ++C) {
Richard Smithba71c082013-05-16 06:20:58 +00002879 // FIXME: Lambda init-captures.
2880 if (!C->capturesVariable())
Guy Benyei11169dd2012-12-18 14:30:41 +00002881 continue;
Richard Smithba71c082013-05-16 06:20:58 +00002882
Guy Benyei11169dd2012-12-18 14:30:41 +00002883 if (Visit(MakeCursorVariableRef(C->getCapturedVar(),
2884 C->getLocation(),
2885 TU)))
2886 return true;
2887 }
2888
2889 // Visit parameters and return type, if present.
2890 if (E->hasExplicitParameters() || E->hasExplicitResultType()) {
2891 TypeLoc TL = E->getCallOperator()->getTypeSourceInfo()->getTypeLoc();
2892 if (E->hasExplicitParameters() && E->hasExplicitResultType()) {
2893 // Visit the whole type.
2894 if (Visit(TL))
2895 return true;
David Blaikie6adc78e2013-02-18 22:06:02 +00002896 } else if (FunctionProtoTypeLoc Proto =
2897 TL.getAs<FunctionProtoTypeLoc>()) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002898 if (E->hasExplicitParameters()) {
2899 // Visit parameters.
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00002900 for (unsigned I = 0, N = Proto.getNumParams(); I != N; ++I)
2901 if (Visit(MakeCXCursor(Proto.getParam(I), TU)))
Guy Benyei11169dd2012-12-18 14:30:41 +00002902 return true;
2903 } else {
2904 // Visit result type.
Alp Toker42a16a62014-01-25 23:51:36 +00002905 if (Visit(Proto.getReturnLoc()))
Guy Benyei11169dd2012-12-18 14:30:41 +00002906 return true;
2907 }
2908 }
2909 }
2910 break;
2911 }
2912
2913 case VisitorJob::PostChildrenVisitKind:
2914 if (PostChildrenVisitor(Parent, ClientData))
2915 return true;
2916 break;
2917 }
2918 }
2919 return false;
2920}
2921
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002922bool CursorVisitor::Visit(const Stmt *S) {
Craig Topper69186e72014-06-08 08:38:04 +00002923 VisitorWorkList *WL = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00002924 if (!WorkListFreeList.empty()) {
2925 WL = WorkListFreeList.back();
2926 WL->clear();
2927 WorkListFreeList.pop_back();
2928 }
2929 else {
2930 WL = new VisitorWorkList();
2931 WorkListCache.push_back(WL);
2932 }
2933 EnqueueWorkList(*WL, S);
2934 bool result = RunVisitorWorkList(*WL);
2935 WorkListFreeList.push_back(WL);
2936 return result;
2937}
2938
2939namespace {
Dmitri Gribenkof8579502013-01-12 19:30:44 +00002940typedef SmallVector<SourceRange, 4> RefNamePieces;
James Y Knight04ec5bf2015-12-24 02:59:37 +00002941RefNamePieces buildPieces(unsigned NameFlags, bool IsMemberRefExpr,
2942 const DeclarationNameInfo &NI, SourceRange QLoc,
2943 const SourceRange *TemplateArgsLoc = nullptr) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002944 const bool WantQualifier = NameFlags & CXNameRange_WantQualifier;
2945 const bool WantTemplateArgs = NameFlags & CXNameRange_WantTemplateArgs;
2946 const bool WantSinglePiece = NameFlags & CXNameRange_WantSinglePiece;
2947
2948 const DeclarationName::NameKind Kind = NI.getName().getNameKind();
2949
2950 RefNamePieces Pieces;
2951
2952 if (WantQualifier && QLoc.isValid())
2953 Pieces.push_back(QLoc);
2954
2955 if (Kind != DeclarationName::CXXOperatorName || IsMemberRefExpr)
2956 Pieces.push_back(NI.getLoc());
James Y Knight04ec5bf2015-12-24 02:59:37 +00002957
2958 if (WantTemplateArgs && TemplateArgsLoc && TemplateArgsLoc->isValid())
2959 Pieces.push_back(*TemplateArgsLoc);
2960
Guy Benyei11169dd2012-12-18 14:30:41 +00002961 if (Kind == DeclarationName::CXXOperatorName) {
2962 Pieces.push_back(SourceLocation::getFromRawEncoding(
2963 NI.getInfo().CXXOperatorName.BeginOpNameLoc));
2964 Pieces.push_back(SourceLocation::getFromRawEncoding(
2965 NI.getInfo().CXXOperatorName.EndOpNameLoc));
2966 }
2967
2968 if (WantSinglePiece) {
2969 SourceRange R(Pieces.front().getBegin(), Pieces.back().getEnd());
2970 Pieces.clear();
2971 Pieces.push_back(R);
2972 }
2973
2974 return Pieces;
2975}
Alexander Kornienkoab9db512015-06-22 23:07:51 +00002976}
Guy Benyei11169dd2012-12-18 14:30:41 +00002977
2978//===----------------------------------------------------------------------===//
2979// Misc. API hooks.
2980//===----------------------------------------------------------------------===//
2981
Chad Rosier05c71aa2013-03-27 18:28:23 +00002982static void fatal_error_handler(void *user_data, const std::string& reason,
2983 bool gen_crash_diag) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002984 // Write the result out to stderr avoiding errs() because raw_ostreams can
2985 // call report_fatal_error.
2986 fprintf(stderr, "LIBCLANG FATAL ERROR: %s\n", reason.c_str());
2987 ::abort();
2988}
2989
Chandler Carruth66660742014-06-27 16:37:27 +00002990namespace {
2991struct RegisterFatalErrorHandler {
2992 RegisterFatalErrorHandler() {
2993 llvm::install_fatal_error_handler(fatal_error_handler, nullptr);
2994 }
2995};
2996}
2997
2998static llvm::ManagedStatic<RegisterFatalErrorHandler> RegisterFatalErrorHandlerOnce;
2999
Guy Benyei11169dd2012-12-18 14:30:41 +00003000extern "C" {
3001CXIndex clang_createIndex(int excludeDeclarationsFromPCH,
3002 int displayDiagnostics) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003003 // We use crash recovery to make some of our APIs more reliable, implicitly
3004 // enable it.
Argyrios Kyrtzidis3701f542013-11-27 08:58:09 +00003005 if (!getenv("LIBCLANG_DISABLE_CRASH_RECOVERY"))
3006 llvm::CrashRecoveryContext::Enable();
Guy Benyei11169dd2012-12-18 14:30:41 +00003007
Chandler Carruth66660742014-06-27 16:37:27 +00003008 // Look through the managed static to trigger construction of the managed
3009 // static which registers our fatal error handler. This ensures it is only
3010 // registered once.
3011 (void)*RegisterFatalErrorHandlerOnce;
Guy Benyei11169dd2012-12-18 14:30:41 +00003012
Adrian Prantlbc068582015-07-08 01:00:30 +00003013 // Initialize targets for clang module support.
3014 llvm::InitializeAllTargets();
3015 llvm::InitializeAllTargetMCs();
3016 llvm::InitializeAllAsmPrinters();
3017 llvm::InitializeAllAsmParsers();
3018
Adrian Prantlfb2398d2015-07-17 01:19:54 +00003019 CIndexer *CIdxr = new CIndexer();
3020
Guy Benyei11169dd2012-12-18 14:30:41 +00003021 if (excludeDeclarationsFromPCH)
3022 CIdxr->setOnlyLocalDecls();
3023 if (displayDiagnostics)
3024 CIdxr->setDisplayDiagnostics();
3025
3026 if (getenv("LIBCLANG_BGPRIO_INDEX"))
3027 CIdxr->setCXGlobalOptFlags(CIdxr->getCXGlobalOptFlags() |
3028 CXGlobalOpt_ThreadBackgroundPriorityForIndexing);
3029 if (getenv("LIBCLANG_BGPRIO_EDIT"))
3030 CIdxr->setCXGlobalOptFlags(CIdxr->getCXGlobalOptFlags() |
3031 CXGlobalOpt_ThreadBackgroundPriorityForEditing);
3032
3033 return CIdxr;
3034}
3035
3036void clang_disposeIndex(CXIndex CIdx) {
3037 if (CIdx)
3038 delete static_cast<CIndexer *>(CIdx);
3039}
3040
3041void clang_CXIndex_setGlobalOptions(CXIndex CIdx, unsigned options) {
3042 if (CIdx)
3043 static_cast<CIndexer *>(CIdx)->setCXGlobalOptFlags(options);
3044}
3045
3046unsigned clang_CXIndex_getGlobalOptions(CXIndex CIdx) {
3047 if (CIdx)
3048 return static_cast<CIndexer *>(CIdx)->getCXGlobalOptFlags();
3049 return 0;
3050}
3051
3052void clang_toggleCrashRecovery(unsigned isEnabled) {
3053 if (isEnabled)
3054 llvm::CrashRecoveryContext::Enable();
3055 else
3056 llvm::CrashRecoveryContext::Disable();
3057}
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003058
Guy Benyei11169dd2012-12-18 14:30:41 +00003059CXTranslationUnit clang_createTranslationUnit(CXIndex CIdx,
3060 const char *ast_filename) {
Dmitri Gribenko8850cda2014-02-19 10:24:00 +00003061 CXTranslationUnit TU;
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003062 enum CXErrorCode Result =
3063 clang_createTranslationUnit2(CIdx, ast_filename, &TU);
Reid Klecknerfd48fc62014-02-12 23:56:20 +00003064 (void)Result;
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003065 assert((TU && Result == CXError_Success) ||
3066 (!TU && Result != CXError_Success));
3067 return TU;
3068}
3069
3070enum CXErrorCode clang_createTranslationUnit2(CXIndex CIdx,
3071 const char *ast_filename,
3072 CXTranslationUnit *out_TU) {
Dmitri Gribenko8850cda2014-02-19 10:24:00 +00003073 if (out_TU)
Craig Topper69186e72014-06-08 08:38:04 +00003074 *out_TU = nullptr;
Dmitri Gribenko8850cda2014-02-19 10:24:00 +00003075
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003076 if (!CIdx || !ast_filename || !out_TU)
3077 return CXError_InvalidArguments;
Guy Benyei11169dd2012-12-18 14:30:41 +00003078
Argyrios Kyrtzidis27021012013-05-24 22:24:07 +00003079 LOG_FUNC_SECTION {
3080 *Log << ast_filename;
3081 }
3082
Guy Benyei11169dd2012-12-18 14:30:41 +00003083 CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
3084 FileSystemOptions FileSystemOpts;
3085
Justin Bognerd512c1e2014-10-15 00:33:06 +00003086 IntrusiveRefCntPtr<DiagnosticsEngine> Diags =
3087 CompilerInstance::createDiagnostics(new DiagnosticOptions());
David Blaikie6f7382d2014-08-10 19:08:04 +00003088 std::unique_ptr<ASTUnit> AU = ASTUnit::LoadFromASTFile(
Adrian Prantlfb2398d2015-07-17 01:19:54 +00003089 ast_filename, CXXIdx->getPCHContainerOperations()->getRawReader(), Diags,
Adrian Prantl6b21ab22015-08-27 19:46:20 +00003090 FileSystemOpts, /*UseDebugInfo=*/false,
3091 CXXIdx->getOnlyLocalDecls(), None,
David Blaikie6f7382d2014-08-10 19:08:04 +00003092 /*CaptureDiagnostics=*/true,
3093 /*AllowPCHWithCompilerErrors=*/true,
3094 /*UserFilesAreVolatile=*/true);
3095 *out_TU = MakeCXTranslationUnit(CXXIdx, AU.release());
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003096 return *out_TU ? CXError_Success : CXError_Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00003097}
3098
3099unsigned clang_defaultEditingTranslationUnitOptions() {
3100 return CXTranslationUnit_PrecompiledPreamble |
3101 CXTranslationUnit_CacheCompletionResults;
3102}
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003103
Guy Benyei11169dd2012-12-18 14:30:41 +00003104CXTranslationUnit
3105clang_createTranslationUnitFromSourceFile(CXIndex CIdx,
3106 const char *source_filename,
3107 int num_command_line_args,
3108 const char * const *command_line_args,
3109 unsigned num_unsaved_files,
3110 struct CXUnsavedFile *unsaved_files) {
3111 unsigned Options = CXTranslationUnit_DetailedPreprocessingRecord;
3112 return clang_parseTranslationUnit(CIdx, source_filename,
3113 command_line_args, num_command_line_args,
3114 unsaved_files, num_unsaved_files,
3115 Options);
3116}
3117
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003118static CXErrorCode
3119clang_parseTranslationUnit_Impl(CXIndex CIdx, const char *source_filename,
3120 const char *const *command_line_args,
3121 int num_command_line_args,
3122 ArrayRef<CXUnsavedFile> unsaved_files,
3123 unsigned options, CXTranslationUnit *out_TU) {
Dmitri Gribenko1bf8d912014-02-18 15:20:02 +00003124 // Set up the initial return values.
3125 if (out_TU)
Craig Topper69186e72014-06-08 08:38:04 +00003126 *out_TU = nullptr;
Dmitri Gribenko1bf8d912014-02-18 15:20:02 +00003127
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003128 // Check arguments.
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003129 if (!CIdx || !out_TU)
3130 return CXError_InvalidArguments;
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003131
Guy Benyei11169dd2012-12-18 14:30:41 +00003132 CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
3133
3134 if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForIndexing))
3135 setThreadBackgroundPriority();
3136
3137 bool PrecompilePreamble = options & CXTranslationUnit_PrecompiledPreamble;
Benjamin Kramer5c248d82015-12-15 09:30:31 +00003138 bool CreatePreambleOnFirstParse =
3139 options & CXTranslationUnit_CreatePreambleOnFirstParse;
Guy Benyei11169dd2012-12-18 14:30:41 +00003140 // FIXME: Add a flag for modules.
3141 TranslationUnitKind TUKind
3142 = (options & CXTranslationUnit_Incomplete)? TU_Prefix : TU_Complete;
Alp Toker8c8a8752013-12-03 06:53:35 +00003143 bool CacheCodeCompletionResults
Guy Benyei11169dd2012-12-18 14:30:41 +00003144 = options & CXTranslationUnit_CacheCompletionResults;
3145 bool IncludeBriefCommentsInCodeCompletion
3146 = options & CXTranslationUnit_IncludeBriefCommentsInCodeCompletion;
3147 bool SkipFunctionBodies = options & CXTranslationUnit_SkipFunctionBodies;
3148 bool ForSerialization = options & CXTranslationUnit_ForSerialization;
3149
3150 // Configure the diagnostics.
3151 IntrusiveRefCntPtr<DiagnosticsEngine>
Sean Silvaf1b49e22013-01-20 01:58:28 +00003152 Diags(CompilerInstance::createDiagnostics(new DiagnosticOptions));
Guy Benyei11169dd2012-12-18 14:30:41 +00003153
Manuel Klimek016c0242016-03-01 10:56:19 +00003154 if (options & CXTranslationUnit_KeepGoing)
3155 Diags->setFatalsAsError(true);
3156
Guy Benyei11169dd2012-12-18 14:30:41 +00003157 // Recover resources if we crash before exiting this function.
3158 llvm::CrashRecoveryContextCleanupRegistrar<DiagnosticsEngine,
3159 llvm::CrashRecoveryContextReleaseRefCleanup<DiagnosticsEngine> >
Alp Tokerf994cef2014-07-05 03:08:06 +00003160 DiagCleanup(Diags.get());
Guy Benyei11169dd2012-12-18 14:30:41 +00003161
Ahmed Charlesb8984322014-03-07 20:03:18 +00003162 std::unique_ptr<std::vector<ASTUnit::RemappedFile>> RemappedFiles(
3163 new std::vector<ASTUnit::RemappedFile>());
Guy Benyei11169dd2012-12-18 14:30:41 +00003164
3165 // Recover resources if we crash before exiting this function.
3166 llvm::CrashRecoveryContextCleanupRegistrar<
3167 std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
3168
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003169 for (auto &UF : unsaved_files) {
Rafael Espindolad87f8d72014-08-27 20:03:29 +00003170 std::unique_ptr<llvm::MemoryBuffer> MB =
Alp Toker9d85b182014-07-07 01:23:14 +00003171 llvm::MemoryBuffer::getMemBufferCopy(getContents(UF), UF.Filename);
Rafael Espindolad87f8d72014-08-27 20:03:29 +00003172 RemappedFiles->push_back(std::make_pair(UF.Filename, MB.release()));
Guy Benyei11169dd2012-12-18 14:30:41 +00003173 }
3174
Ahmed Charlesb8984322014-03-07 20:03:18 +00003175 std::unique_ptr<std::vector<const char *>> Args(
3176 new std::vector<const char *>());
Guy Benyei11169dd2012-12-18 14:30:41 +00003177
3178 // Recover resources if we crash before exiting this method.
3179 llvm::CrashRecoveryContextCleanupRegistrar<std::vector<const char*> >
3180 ArgsCleanup(Args.get());
3181
3182 // Since the Clang C library is primarily used by batch tools dealing with
3183 // (often very broken) source code, where spell-checking can have a
3184 // significant negative impact on performance (particularly when
3185 // precompiled headers are involved), we disable it by default.
3186 // Only do this if we haven't found a spell-checking-related argument.
3187 bool FoundSpellCheckingArgument = false;
3188 for (int I = 0; I != num_command_line_args; ++I) {
3189 if (strcmp(command_line_args[I], "-fno-spell-checking") == 0 ||
3190 strcmp(command_line_args[I], "-fspell-checking") == 0) {
3191 FoundSpellCheckingArgument = true;
3192 break;
3193 }
3194 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003195 Args->insert(Args->end(), command_line_args,
3196 command_line_args + num_command_line_args);
3197
Benjamin Kramerc02670e2015-11-18 16:14:27 +00003198 if (!FoundSpellCheckingArgument)
3199 Args->insert(Args->begin() + 1, "-fno-spell-checking");
3200
Guy Benyei11169dd2012-12-18 14:30:41 +00003201 // The 'source_filename' argument is optional. If the caller does not
3202 // specify it then it is assumed that the source file is specified
3203 // in the actual argument list.
3204 // Put the source file after command_line_args otherwise if '-x' flag is
3205 // present it will be unused.
3206 if (source_filename)
3207 Args->push_back(source_filename);
3208
3209 // Do we need the detailed preprocessing record?
3210 if (options & CXTranslationUnit_DetailedPreprocessingRecord) {
3211 Args->push_back("-Xclang");
3212 Args->push_back("-detailed-preprocessing-record");
3213 }
3214
3215 unsigned NumErrors = Diags->getClient()->getNumErrors();
Ahmed Charlesb8984322014-03-07 20:03:18 +00003216 std::unique_ptr<ASTUnit> ErrUnit;
Benjamin Kramer5c248d82015-12-15 09:30:31 +00003217 // Unless the user specified that they want the preamble on the first parse
3218 // set it up to be created on the first reparse. This makes the first parse
3219 // faster, trading for a slower (first) reparse.
3220 unsigned PrecompilePreambleAfterNParses =
3221 !PrecompilePreamble ? 0 : 2 - CreatePreambleOnFirstParse;
Ahmed Charlesb8984322014-03-07 20:03:18 +00003222 std::unique_ptr<ASTUnit> Unit(ASTUnit::LoadFromCommandLine(
Adrian Prantlbb165fb2015-06-20 18:53:08 +00003223 Args->data(), Args->data() + Args->size(),
3224 CXXIdx->getPCHContainerOperations(), Diags,
Ahmed Charlesb8984322014-03-07 20:03:18 +00003225 CXXIdx->getClangResourcesPath(), CXXIdx->getOnlyLocalDecls(),
3226 /*CaptureDiagnostics=*/true, *RemappedFiles.get(),
Benjamin Kramer5c248d82015-12-15 09:30:31 +00003227 /*RemappedFilesKeepOriginalName=*/true, PrecompilePreambleAfterNParses,
3228 TUKind, CacheCodeCompletionResults, IncludeBriefCommentsInCodeCompletion,
Ahmed Charlesb8984322014-03-07 20:03:18 +00003229 /*AllowPCHWithCompilerErrors=*/true, SkipFunctionBodies,
Argyrios Kyrtzidisa3e2ff12015-11-20 03:36:21 +00003230 /*UserFilesAreVolatile=*/true, ForSerialization,
3231 CXXIdx->getPCHContainerOperations()->getRawReader().getFormat(),
3232 &ErrUnit));
Guy Benyei11169dd2012-12-18 14:30:41 +00003233
Artem Belevich0ff05cd2015-07-13 23:27:56 +00003234 // Early failures in LoadFromCommandLine may return with ErrUnit unset.
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003235 if (!Unit && !ErrUnit)
3236 return CXError_ASTReadError;
Artem Belevich0ff05cd2015-07-13 23:27:56 +00003237
Guy Benyei11169dd2012-12-18 14:30:41 +00003238 if (NumErrors != Diags->getClient()->getNumErrors()) {
3239 // Make sure to check that 'Unit' is non-NULL.
3240 if (CXXIdx->getDisplayDiagnostics())
3241 printDiagsToStderr(Unit ? Unit.get() : ErrUnit.get());
3242 }
3243
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003244 if (isASTReadError(Unit ? Unit.get() : ErrUnit.get()))
3245 return CXError_ASTReadError;
3246
3247 *out_TU = MakeCXTranslationUnit(CXXIdx, Unit.release());
3248 return *out_TU ? CXError_Success : CXError_Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00003249}
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003250
3251CXTranslationUnit
3252clang_parseTranslationUnit(CXIndex CIdx,
3253 const char *source_filename,
3254 const char *const *command_line_args,
3255 int num_command_line_args,
3256 struct CXUnsavedFile *unsaved_files,
3257 unsigned num_unsaved_files,
3258 unsigned options) {
3259 CXTranslationUnit TU;
3260 enum CXErrorCode Result = clang_parseTranslationUnit2(
3261 CIdx, source_filename, command_line_args, num_command_line_args,
3262 unsaved_files, num_unsaved_files, options, &TU);
Reid Kleckner6eaf05a2014-02-13 01:19:59 +00003263 (void)Result;
Dmitri Gribenko1bf8d912014-02-18 15:20:02 +00003264 assert((TU && Result == CXError_Success) ||
3265 (!TU && Result != CXError_Success));
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003266 return TU;
3267}
3268
3269enum CXErrorCode clang_parseTranslationUnit2(
Benjamin Kramerc02670e2015-11-18 16:14:27 +00003270 CXIndex CIdx, const char *source_filename,
3271 const char *const *command_line_args, int num_command_line_args,
3272 struct CXUnsavedFile *unsaved_files, unsigned num_unsaved_files,
3273 unsigned options, CXTranslationUnit *out_TU) {
3274 SmallVector<const char *, 4> Args;
3275 Args.push_back("clang");
3276 Args.append(command_line_args, command_line_args + num_command_line_args);
3277 return clang_parseTranslationUnit2FullArgv(
3278 CIdx, source_filename, Args.data(), Args.size(), unsaved_files,
3279 num_unsaved_files, options, out_TU);
3280}
3281
3282enum CXErrorCode clang_parseTranslationUnit2FullArgv(
3283 CXIndex CIdx, const char *source_filename,
3284 const char *const *command_line_args, int num_command_line_args,
3285 struct CXUnsavedFile *unsaved_files, unsigned num_unsaved_files,
3286 unsigned options, CXTranslationUnit *out_TU) {
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00003287 LOG_FUNC_SECTION {
3288 *Log << source_filename << ": ";
3289 for (int i = 0; i != num_command_line_args; ++i)
3290 *Log << command_line_args[i] << " ";
3291 }
3292
Alp Toker9d85b182014-07-07 01:23:14 +00003293 if (num_unsaved_files && !unsaved_files)
3294 return CXError_InvalidArguments;
3295
Alp Toker5c532982014-07-07 22:42:03 +00003296 CXErrorCode result = CXError_Failure;
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003297 auto ParseTranslationUnitImpl = [=, &result] {
3298 result = clang_parseTranslationUnit_Impl(
3299 CIdx, source_filename, command_line_args, num_command_line_args,
3300 llvm::makeArrayRef(unsaved_files, num_unsaved_files), options, out_TU);
3301 };
Guy Benyei11169dd2012-12-18 14:30:41 +00003302 llvm::CrashRecoveryContext CRC;
3303
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003304 if (!RunSafely(CRC, ParseTranslationUnitImpl)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003305 fprintf(stderr, "libclang: crash detected during parsing: {\n");
3306 fprintf(stderr, " 'source_filename' : '%s'\n", source_filename);
3307 fprintf(stderr, " 'command_line_args' : [");
3308 for (int i = 0; i != num_command_line_args; ++i) {
3309 if (i)
3310 fprintf(stderr, ", ");
3311 fprintf(stderr, "'%s'", command_line_args[i]);
3312 }
3313 fprintf(stderr, "],\n");
3314 fprintf(stderr, " 'unsaved_files' : [");
3315 for (unsigned i = 0; i != num_unsaved_files; ++i) {
3316 if (i)
3317 fprintf(stderr, ", ");
3318 fprintf(stderr, "('%s', '...', %ld)", unsaved_files[i].Filename,
3319 unsaved_files[i].Length);
3320 }
3321 fprintf(stderr, "],\n");
3322 fprintf(stderr, " 'options' : %d,\n", options);
3323 fprintf(stderr, "}\n");
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003324
3325 return CXError_Crashed;
Guy Benyei11169dd2012-12-18 14:30:41 +00003326 } else if (getenv("LIBCLANG_RESOURCE_USAGE")) {
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003327 if (CXTranslationUnit *TU = out_TU)
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003328 PrintLibclangResourceUsage(*TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00003329 }
Alp Toker5c532982014-07-07 22:42:03 +00003330
3331 return result;
Guy Benyei11169dd2012-12-18 14:30:41 +00003332}
3333
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003334CXString clang_Type_getObjCEncoding(CXType CT) {
3335 CXTranslationUnit tu = static_cast<CXTranslationUnit>(CT.data[1]);
3336 ASTContext &Ctx = getASTUnit(tu)->getASTContext();
3337 std::string encoding;
3338 Ctx.getObjCEncodingForType(QualType::getFromOpaquePtr(CT.data[0]),
3339 encoding);
3340
3341 return cxstring::createDup(encoding);
3342}
3343
3344static const IdentifierInfo *getMacroIdentifier(CXCursor C) {
3345 if (C.kind == CXCursor_MacroDefinition) {
3346 if (const MacroDefinitionRecord *MDR = getCursorMacroDefinition(C))
3347 return MDR->getName();
3348 } else if (C.kind == CXCursor_MacroExpansion) {
3349 MacroExpansionCursor ME = getCursorMacroExpansion(C);
3350 return ME.getName();
3351 }
3352 return nullptr;
3353}
3354
3355unsigned clang_Cursor_isMacroFunctionLike(CXCursor C) {
3356 const IdentifierInfo *II = getMacroIdentifier(C);
3357 if (!II) {
3358 return false;
3359 }
3360 ASTUnit *ASTU = getCursorASTUnit(C);
3361 Preprocessor &PP = ASTU->getPreprocessor();
3362 if (const MacroInfo *MI = PP.getMacroInfo(II))
3363 return MI->isFunctionLike();
3364 return false;
3365}
3366
3367unsigned clang_Cursor_isMacroBuiltin(CXCursor C) {
3368 const IdentifierInfo *II = getMacroIdentifier(C);
3369 if (!II) {
3370 return false;
3371 }
3372 ASTUnit *ASTU = getCursorASTUnit(C);
3373 Preprocessor &PP = ASTU->getPreprocessor();
3374 if (const MacroInfo *MI = PP.getMacroInfo(II))
3375 return MI->isBuiltinMacro();
3376 return false;
3377}
3378
3379unsigned clang_Cursor_isFunctionInlined(CXCursor C) {
3380 const Decl *D = getCursorDecl(C);
3381 const FunctionDecl *FD = dyn_cast_or_null<FunctionDecl>(D);
3382 if (!FD) {
3383 return false;
3384 }
3385 return FD->isInlined();
3386}
3387
3388static StringLiteral* getCFSTR_value(CallExpr *callExpr) {
3389 if (callExpr->getNumArgs() != 1) {
3390 return nullptr;
3391 }
3392
3393 StringLiteral *S = nullptr;
3394 auto *arg = callExpr->getArg(0);
3395 if (arg->getStmtClass() == Stmt::ImplicitCastExprClass) {
3396 ImplicitCastExpr *I = static_cast<ImplicitCastExpr *>(arg);
3397 auto *subExpr = I->getSubExprAsWritten();
3398
3399 if(subExpr->getStmtClass() != Stmt::StringLiteralClass){
3400 return nullptr;
3401 }
3402
3403 S = static_cast<StringLiteral *>(I->getSubExprAsWritten());
3404 } else if (arg->getStmtClass() == Stmt::StringLiteralClass) {
3405 S = static_cast<StringLiteral *>(callExpr->getArg(0));
3406 } else {
3407 return nullptr;
3408 }
3409 return S;
3410}
3411
3412typedef struct {
3413 CXEvalResultKind EvalType;
3414 union {
3415 int intVal;
3416 double floatVal;
3417 char *stringVal;
3418 } EvalData;
3419} ExprEvalResult;
3420
3421void clang_EvalResult_dispose(CXEvalResult E) {
3422 ExprEvalResult *ER = (ExprEvalResult *)E;
3423 if (ER) {
3424 CXEvalResultKind evalType = ER->EvalType;
3425
3426 if (evalType != CXEval_UnExposed && evalType != CXEval_Float &&
3427 evalType != CXEval_Int && ER->EvalData.stringVal) {
3428 free((void *) ER->EvalData.stringVal);
3429 }
3430 free((void *)ER);
3431 }
3432}
3433
3434CXEvalResultKind clang_EvalResult_getKind(CXEvalResult E) {
3435 if (!E) {
3436 return CXEval_UnExposed;
3437 }
3438 return ((ExprEvalResult *)E)->EvalType;
3439}
3440
3441int clang_EvalResult_getAsInt(CXEvalResult E) {
3442 if (!E) {
3443 return 0;
3444 }
3445 return ((ExprEvalResult *)E)->EvalData.intVal;
3446}
3447
3448double clang_EvalResult_getAsDouble(CXEvalResult E) {
3449 if (!E) {
3450 return 0;
3451 }
3452 return ((ExprEvalResult *)E)->EvalData.floatVal;
3453}
3454
3455const char* clang_EvalResult_getAsStr(CXEvalResult E) {
3456 if (!E) {
3457 return nullptr;
3458 }
3459 return ((ExprEvalResult *)E)->EvalData.stringVal;
3460}
3461
3462static const ExprEvalResult* evaluateExpr(Expr *expr, CXCursor C) {
3463 Expr::EvalResult ER;
3464 ASTContext &ctx = getCursorContext(C);
3465 if (!expr) {
3466 return nullptr;
3467 }
3468 expr = expr->IgnoreParens();
3469 bool res = expr->EvaluateAsRValue(ER, ctx);
3470 QualType rettype;
3471 CallExpr *callExpr;
3472 ExprEvalResult *result = (ExprEvalResult *) malloc(sizeof(ExprEvalResult));
3473 if (!result) {
3474 return nullptr;
3475 }
3476 result->EvalType = CXEval_UnExposed;
3477
3478 if (res) {
3479
3480 if (ER.Val.isInt()) {
3481 result->EvalType = CXEval_Int;
3482 result->EvalData.intVal = ER.Val.getInt().getExtValue();
3483 return result;
3484 } else if (ER.Val.isFloat()) {
3485
3486 llvm::SmallVector<char, 100> Buffer;
3487 ER.Val.getFloat().toString(Buffer);
3488 std::string floatStr(Buffer.data(), Buffer.size());
3489 result->EvalType = CXEval_Float;
3490 bool ignored;
3491 llvm::APFloat apFloat = ER.Val.getFloat();
3492 apFloat.convert(llvm::APFloat::IEEEdouble,
3493 llvm::APFloat::rmNearestTiesToEven, &ignored);
3494 result->EvalData.floatVal = apFloat.convertToDouble();
3495 return result;
3496
3497 } else if (expr->getStmtClass() == Stmt::ImplicitCastExprClass) {
3498
3499 const ImplicitCastExpr *I = dyn_cast<ImplicitCastExpr>(expr);
3500 auto *subExpr = I->getSubExprAsWritten();
3501 if (subExpr->getStmtClass() == Stmt::StringLiteralClass ||
3502 subExpr->getStmtClass() == Stmt::ObjCStringLiteralClass) {
3503
3504 const StringLiteral *StrE = nullptr;
3505 const ObjCStringLiteral *ObjCExpr;
3506 ObjCExpr = dyn_cast<ObjCStringLiteral>(subExpr);
3507
3508 if (ObjCExpr) {
3509 StrE = ObjCExpr->getString();
3510 result->EvalType = CXEval_ObjCStrLiteral;
3511 } else {
3512 StrE = cast<StringLiteral>(I->getSubExprAsWritten());
3513 result->EvalType = CXEval_StrLiteral;
3514 }
3515
3516 std::string strRef(StrE->getString().str());
3517 result->EvalData.stringVal = (char *)malloc(strRef.size()+1);
3518 strncpy((char*)result->EvalData.stringVal, strRef.c_str(),
3519 strRef.size());
3520 result->EvalData.stringVal[strRef.size()] = '\0';
3521 return result;
3522 }
3523
3524 } else if (expr->getStmtClass() == Stmt::ObjCStringLiteralClass ||
3525 expr->getStmtClass() == Stmt::StringLiteralClass) {
3526
3527 const StringLiteral *StrE = nullptr;
3528 const ObjCStringLiteral *ObjCExpr;
3529 ObjCExpr = dyn_cast<ObjCStringLiteral>(expr);
3530
3531 if (ObjCExpr) {
3532 StrE = ObjCExpr->getString();
3533 result->EvalType = CXEval_ObjCStrLiteral;
3534 } else {
3535 StrE = cast<StringLiteral>(expr);
3536 result->EvalType = CXEval_StrLiteral;
3537 }
3538
3539 std::string strRef(StrE->getString().str());
3540 result->EvalData.stringVal = (char *)malloc(strRef.size()+1);
3541 strncpy((char*)result->EvalData.stringVal, strRef.c_str(),
3542 strRef.size());
3543 result->EvalData.stringVal[strRef.size()] = '\0';
3544 return result;
3545
3546 } else if (expr->getStmtClass() == Stmt::CStyleCastExprClass) {
3547
3548 CStyleCastExpr *CC = static_cast<CStyleCastExpr *>(expr);
3549
3550 rettype = CC->getType();
3551 if (rettype.getAsString() == "CFStringRef" &&
3552 CC->getSubExpr()->getStmtClass() == Stmt::CallExprClass) {
3553
3554 callExpr = static_cast<CallExpr *>(CC->getSubExpr());
3555 StringLiteral* S = getCFSTR_value(callExpr);
3556 if (S) {
3557 std::string strLiteral(S->getString().str());
3558 result->EvalType = CXEval_CFStr;
3559
3560 result->EvalData.stringVal = (char *)malloc(strLiteral.size()+1);
3561 strncpy((char*)result->EvalData.stringVal, strLiteral.c_str(),
3562 strLiteral.size());
3563 result->EvalData.stringVal[strLiteral.size()] = '\0';
3564 return result;
3565 }
3566 }
3567
3568 } else if (expr->getStmtClass() == Stmt::CallExprClass) {
3569
3570 callExpr = static_cast<CallExpr *>(expr);
3571 rettype = callExpr->getCallReturnType(ctx);
3572
3573 if (rettype->isVectorType() || callExpr->getNumArgs() > 1) {
Hans Wennborg6917b4a2016-04-11 20:53:59 +00003574 clang_EvalResult_dispose((CXEvalResult *)result);
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003575 return nullptr;
3576 }
3577 if (rettype->isIntegralType(ctx) || rettype->isRealFloatingType()) {
3578 if(callExpr->getNumArgs() == 1 &&
Hans Wennborg6917b4a2016-04-11 20:53:59 +00003579 !callExpr->getArg(0)->getType()->isIntegralType(ctx)) {
3580 clang_EvalResult_dispose((CXEvalResult *)result);
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003581 return nullptr;
3582 }
3583 } else if(rettype.getAsString() == "CFStringRef") {
3584
3585 StringLiteral* S = getCFSTR_value(callExpr);
3586 if (S) {
3587 std::string strLiteral(S->getString().str());
3588 result->EvalType = CXEval_CFStr;
3589 result->EvalData.stringVal = (char *)malloc(strLiteral.size()+1);
3590 strncpy((char*)result->EvalData.stringVal, strLiteral.c_str(),
3591 strLiteral.size());
3592 result->EvalData.stringVal[strLiteral.size()] = '\0';
3593 return result;
3594 }
3595 }
3596
3597 } else if (expr->getStmtClass() == Stmt::DeclRefExprClass) {
3598
3599 DeclRefExpr *D = static_cast<DeclRefExpr *>(expr);
3600 ValueDecl *V = D->getDecl();
3601 if (V->getKind() == Decl::Function) {
3602 std::string strName(V->getNameAsString());
3603 result->EvalType = CXEval_Other;
3604 result->EvalData.stringVal = (char *)malloc(strName.size()+1);
3605 strncpy((char*)result->EvalData.stringVal, strName.c_str(),
3606 strName.size());
3607 result->EvalData.stringVal[strName.size()] = '\0';
3608 return result;
3609 }
3610 }
3611
3612 }
3613
3614 clang_EvalResult_dispose((CXEvalResult *)result);
3615 return nullptr;
3616}
3617
3618CXEvalResult clang_Cursor_Evaluate(CXCursor C) {
3619 const Decl *D = getCursorDecl(C);
3620 if (D) {
3621 const Expr *expr = nullptr;
3622 if (auto *Var = dyn_cast<VarDecl>(D)) {
3623 expr = Var->getInit();
3624 } else if (auto *Field = dyn_cast<FieldDecl>(D)) {
3625 expr = Field->getInClassInitializer();
3626 }
3627 if (expr)
Aaron Ballman01dc1572016-01-20 15:25:30 +00003628 return const_cast<CXEvalResult>(reinterpret_cast<const void *>(
3629 evaluateExpr(const_cast<Expr *>(expr), C)));
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003630 return nullptr;
3631 }
3632
3633 const CompoundStmt *compoundStmt = dyn_cast_or_null<CompoundStmt>(getCursorStmt(C));
3634 if (compoundStmt) {
3635 Expr *expr = nullptr;
3636 for (auto *bodyIterator : compoundStmt->body()) {
3637 if ((expr = dyn_cast<Expr>(bodyIterator))) {
3638 break;
3639 }
3640 }
3641 if (expr)
Aaron Ballman01dc1572016-01-20 15:25:30 +00003642 return const_cast<CXEvalResult>(
3643 reinterpret_cast<const void *>(evaluateExpr(expr, C)));
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003644 }
3645 return nullptr;
3646}
3647
3648unsigned clang_Cursor_hasAttrs(CXCursor C) {
3649 const Decl *D = getCursorDecl(C);
3650 if (!D) {
3651 return 0;
3652 }
3653
3654 if (D->hasAttrs()) {
3655 return 1;
3656 }
3657
3658 return 0;
3659}
Guy Benyei11169dd2012-12-18 14:30:41 +00003660unsigned clang_defaultSaveOptions(CXTranslationUnit TU) {
3661 return CXSaveTranslationUnit_None;
3662}
3663
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003664static CXSaveError clang_saveTranslationUnit_Impl(CXTranslationUnit TU,
3665 const char *FileName,
3666 unsigned options) {
3667 CIndexer *CXXIdx = TU->CIdx;
Guy Benyei11169dd2012-12-18 14:30:41 +00003668 if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForIndexing))
3669 setThreadBackgroundPriority();
3670
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003671 bool hadError = cxtu::getASTUnit(TU)->Save(FileName);
3672 return hadError ? CXSaveError_Unknown : CXSaveError_None;
Guy Benyei11169dd2012-12-18 14:30:41 +00003673}
3674
3675int clang_saveTranslationUnit(CXTranslationUnit TU, const char *FileName,
3676 unsigned options) {
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00003677 LOG_FUNC_SECTION {
3678 *Log << TU << ' ' << FileName;
3679 }
3680
Dmitri Gribenko852d6222014-02-11 15:02:48 +00003681 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003682 LOG_BAD_TU(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00003683 return CXSaveError_InvalidTU;
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003684 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003685
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003686 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00003687 ASTUnit::ConcurrencyCheck Check(*CXXUnit);
3688 if (!CXXUnit->hasSema())
3689 return CXSaveError_InvalidTU;
3690
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003691 CXSaveError result;
3692 auto SaveTranslationUnitImpl = [=, &result]() {
3693 result = clang_saveTranslationUnit_Impl(TU, FileName, options);
3694 };
Guy Benyei11169dd2012-12-18 14:30:41 +00003695
3696 if (!CXXUnit->getDiagnostics().hasUnrecoverableErrorOccurred() ||
3697 getenv("LIBCLANG_NOTHREADS")) {
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003698 SaveTranslationUnitImpl();
Guy Benyei11169dd2012-12-18 14:30:41 +00003699
3700 if (getenv("LIBCLANG_RESOURCE_USAGE"))
3701 PrintLibclangResourceUsage(TU);
3702
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003703 return result;
Guy Benyei11169dd2012-12-18 14:30:41 +00003704 }
3705
3706 // We have an AST that has invalid nodes due to compiler errors.
3707 // Use a crash recovery thread for protection.
3708
3709 llvm::CrashRecoveryContext CRC;
3710
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003711 if (!RunSafely(CRC, SaveTranslationUnitImpl)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003712 fprintf(stderr, "libclang: crash detected during AST saving: {\n");
3713 fprintf(stderr, " 'filename' : '%s'\n", FileName);
3714 fprintf(stderr, " 'options' : %d,\n", options);
3715 fprintf(stderr, "}\n");
3716
3717 return CXSaveError_Unknown;
3718
3719 } else if (getenv("LIBCLANG_RESOURCE_USAGE")) {
3720 PrintLibclangResourceUsage(TU);
3721 }
3722
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003723 return result;
Guy Benyei11169dd2012-12-18 14:30:41 +00003724}
3725
3726void clang_disposeTranslationUnit(CXTranslationUnit CTUnit) {
3727 if (CTUnit) {
3728 // If the translation unit has been marked as unsafe to free, just discard
3729 // it.
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003730 ASTUnit *Unit = cxtu::getASTUnit(CTUnit);
3731 if (Unit && Unit->isUnsafeToFree())
Guy Benyei11169dd2012-12-18 14:30:41 +00003732 return;
3733
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003734 delete cxtu::getASTUnit(CTUnit);
Dmitri Gribenkob95b3f12013-01-26 22:44:19 +00003735 delete CTUnit->StringPool;
Guy Benyei11169dd2012-12-18 14:30:41 +00003736 delete static_cast<CXDiagnosticSetImpl *>(CTUnit->Diagnostics);
3737 disposeOverridenCXCursorsPool(CTUnit->OverridenCursorsPool);
Dmitri Gribenko9e605112013-11-13 22:16:51 +00003738 delete CTUnit->CommentToXML;
Guy Benyei11169dd2012-12-18 14:30:41 +00003739 delete CTUnit;
3740 }
3741}
3742
3743unsigned clang_defaultReparseOptions(CXTranslationUnit TU) {
3744 return CXReparse_None;
3745}
3746
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003747static CXErrorCode
3748clang_reparseTranslationUnit_Impl(CXTranslationUnit TU,
3749 ArrayRef<CXUnsavedFile> unsaved_files,
3750 unsigned options) {
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003751 // Check arguments.
Dmitri Gribenko852d6222014-02-11 15:02:48 +00003752 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003753 LOG_BAD_TU(TU);
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003754 return CXError_InvalidArguments;
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003755 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003756
3757 // Reset the associated diagnostics.
3758 delete static_cast<CXDiagnosticSetImpl*>(TU->Diagnostics);
Craig Topper69186e72014-06-08 08:38:04 +00003759 TU->Diagnostics = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00003760
Dmitri Gribenko183436e2013-01-26 21:49:50 +00003761 CIndexer *CXXIdx = TU->CIdx;
Guy Benyei11169dd2012-12-18 14:30:41 +00003762 if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForEditing))
3763 setThreadBackgroundPriority();
3764
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003765 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00003766 ASTUnit::ConcurrencyCheck Check(*CXXUnit);
Ahmed Charlesb8984322014-03-07 20:03:18 +00003767
3768 std::unique_ptr<std::vector<ASTUnit::RemappedFile>> RemappedFiles(
3769 new std::vector<ASTUnit::RemappedFile>());
3770
Guy Benyei11169dd2012-12-18 14:30:41 +00003771 // Recover resources if we crash before exiting this function.
3772 llvm::CrashRecoveryContextCleanupRegistrar<
3773 std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
Alp Toker9d85b182014-07-07 01:23:14 +00003774
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003775 for (auto &UF : unsaved_files) {
Rafael Espindolad87f8d72014-08-27 20:03:29 +00003776 std::unique_ptr<llvm::MemoryBuffer> MB =
Alp Toker9d85b182014-07-07 01:23:14 +00003777 llvm::MemoryBuffer::getMemBufferCopy(getContents(UF), UF.Filename);
Rafael Espindolad87f8d72014-08-27 20:03:29 +00003778 RemappedFiles->push_back(std::make_pair(UF.Filename, MB.release()));
Guy Benyei11169dd2012-12-18 14:30:41 +00003779 }
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003780
Adrian Prantlbb165fb2015-06-20 18:53:08 +00003781 if (!CXXUnit->Reparse(CXXIdx->getPCHContainerOperations(),
3782 *RemappedFiles.get()))
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003783 return CXError_Success;
3784 if (isASTReadError(CXXUnit))
3785 return CXError_ASTReadError;
3786 return CXError_Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00003787}
3788
3789int clang_reparseTranslationUnit(CXTranslationUnit TU,
3790 unsigned num_unsaved_files,
3791 struct CXUnsavedFile *unsaved_files,
3792 unsigned options) {
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00003793 LOG_FUNC_SECTION {
3794 *Log << TU;
3795 }
3796
Alp Toker9d85b182014-07-07 01:23:14 +00003797 if (num_unsaved_files && !unsaved_files)
3798 return CXError_InvalidArguments;
3799
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003800 CXErrorCode result;
3801 auto ReparseTranslationUnitImpl = [=, &result]() {
3802 result = clang_reparseTranslationUnit_Impl(
3803 TU, llvm::makeArrayRef(unsaved_files, num_unsaved_files), options);
3804 };
Guy Benyei11169dd2012-12-18 14:30:41 +00003805
3806 if (getenv("LIBCLANG_NOTHREADS")) {
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003807 ReparseTranslationUnitImpl();
Alp Toker5c532982014-07-07 22:42:03 +00003808 return result;
Guy Benyei11169dd2012-12-18 14:30:41 +00003809 }
3810
3811 llvm::CrashRecoveryContext CRC;
3812
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003813 if (!RunSafely(CRC, ReparseTranslationUnitImpl)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003814 fprintf(stderr, "libclang: crash detected during reparsing\n");
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003815 cxtu::getASTUnit(TU)->setUnsafeToFree(true);
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003816 return CXError_Crashed;
Guy Benyei11169dd2012-12-18 14:30:41 +00003817 } else if (getenv("LIBCLANG_RESOURCE_USAGE"))
3818 PrintLibclangResourceUsage(TU);
3819
Alp Toker5c532982014-07-07 22:42:03 +00003820 return result;
Guy Benyei11169dd2012-12-18 14:30:41 +00003821}
3822
3823
3824CXString clang_getTranslationUnitSpelling(CXTranslationUnit CTUnit) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00003825 if (isNotUsableTU(CTUnit)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003826 LOG_BAD_TU(CTUnit);
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00003827 return cxstring::createEmpty();
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003828 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003829
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003830 ASTUnit *CXXUnit = cxtu::getASTUnit(CTUnit);
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00003831 return cxstring::createDup(CXXUnit->getOriginalSourceFileName());
Guy Benyei11169dd2012-12-18 14:30:41 +00003832}
3833
3834CXCursor clang_getTranslationUnitCursor(CXTranslationUnit TU) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00003835 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003836 LOG_BAD_TU(TU);
Argyrios Kyrtzidis0e95fca2013-04-04 22:40:59 +00003837 return clang_getNullCursor();
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003838 }
Argyrios Kyrtzidis0e95fca2013-04-04 22:40:59 +00003839
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003840 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00003841 return MakeCXCursor(CXXUnit->getASTContext().getTranslationUnitDecl(), TU);
3842}
3843
3844} // end: extern "C"
3845
3846//===----------------------------------------------------------------------===//
3847// CXFile Operations.
3848//===----------------------------------------------------------------------===//
3849
3850extern "C" {
3851CXString clang_getFileName(CXFile SFile) {
3852 if (!SFile)
Dmitri Gribenkof98dfba2013-02-01 14:13:32 +00003853 return cxstring::createNull();
Guy Benyei11169dd2012-12-18 14:30:41 +00003854
3855 FileEntry *FEnt = static_cast<FileEntry *>(SFile);
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00003856 return cxstring::createRef(FEnt->getName());
Guy Benyei11169dd2012-12-18 14:30:41 +00003857}
3858
3859time_t clang_getFileTime(CXFile SFile) {
3860 if (!SFile)
3861 return 0;
3862
3863 FileEntry *FEnt = static_cast<FileEntry *>(SFile);
3864 return FEnt->getModificationTime();
3865}
3866
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003867CXFile clang_getFile(CXTranslationUnit TU, const char *file_name) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00003868 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003869 LOG_BAD_TU(TU);
Craig Topper69186e72014-06-08 08:38:04 +00003870 return nullptr;
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003871 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003872
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003873 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00003874
3875 FileManager &FMgr = CXXUnit->getFileManager();
3876 return const_cast<FileEntry *>(FMgr.getFile(file_name));
3877}
3878
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003879unsigned clang_isFileMultipleIncludeGuarded(CXTranslationUnit TU,
3880 CXFile file) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00003881 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003882 LOG_BAD_TU(TU);
3883 return 0;
3884 }
3885
3886 if (!file)
Guy Benyei11169dd2012-12-18 14:30:41 +00003887 return 0;
3888
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003889 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00003890 FileEntry *FEnt = static_cast<FileEntry *>(file);
3891 return CXXUnit->getPreprocessor().getHeaderSearchInfo()
3892 .isFileMultipleIncludeGuarded(FEnt);
3893}
3894
Argyrios Kyrtzidisac08b262013-01-26 04:52:52 +00003895int clang_getFileUniqueID(CXFile file, CXFileUniqueID *outID) {
3896 if (!file || !outID)
3897 return 1;
3898
Argyrios Kyrtzidisac08b262013-01-26 04:52:52 +00003899 FileEntry *FEnt = static_cast<FileEntry *>(file);
Rafael Espindolaf8f91b82013-08-01 21:42:11 +00003900 const llvm::sys::fs::UniqueID &ID = FEnt->getUniqueID();
3901 outID->data[0] = ID.getDevice();
3902 outID->data[1] = ID.getFile();
Argyrios Kyrtzidisac08b262013-01-26 04:52:52 +00003903 outID->data[2] = FEnt->getModificationTime();
3904 return 0;
Argyrios Kyrtzidisac08b262013-01-26 04:52:52 +00003905}
3906
Argyrios Kyrtzidisac3997e2014-08-16 00:26:19 +00003907int clang_File_isEqual(CXFile file1, CXFile file2) {
3908 if (file1 == file2)
3909 return true;
3910
3911 if (!file1 || !file2)
3912 return false;
3913
3914 FileEntry *FEnt1 = static_cast<FileEntry *>(file1);
3915 FileEntry *FEnt2 = static_cast<FileEntry *>(file2);
3916 return FEnt1->getUniqueID() == FEnt2->getUniqueID();
3917}
3918
Guy Benyei11169dd2012-12-18 14:30:41 +00003919} // end: extern "C"
3920
3921//===----------------------------------------------------------------------===//
3922// CXCursor Operations.
3923//===----------------------------------------------------------------------===//
3924
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003925static const Decl *getDeclFromExpr(const Stmt *E) {
3926 if (const ImplicitCastExpr *CE = dyn_cast<ImplicitCastExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003927 return getDeclFromExpr(CE->getSubExpr());
3928
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003929 if (const DeclRefExpr *RefExpr = dyn_cast<DeclRefExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003930 return RefExpr->getDecl();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003931 if (const MemberExpr *ME = dyn_cast<MemberExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003932 return ME->getMemberDecl();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003933 if (const ObjCIvarRefExpr *RE = dyn_cast<ObjCIvarRefExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003934 return RE->getDecl();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003935 if (const ObjCPropertyRefExpr *PRE = dyn_cast<ObjCPropertyRefExpr>(E)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003936 if (PRE->isExplicitProperty())
3937 return PRE->getExplicitProperty();
3938 // It could be messaging both getter and setter as in:
3939 // ++myobj.myprop;
3940 // in which case prefer to associate the setter since it is less obvious
3941 // from inspecting the source that the setter is going to get called.
3942 if (PRE->isMessagingSetter())
3943 return PRE->getImplicitPropertySetter();
3944 return PRE->getImplicitPropertyGetter();
3945 }
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003946 if (const PseudoObjectExpr *POE = dyn_cast<PseudoObjectExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003947 return getDeclFromExpr(POE->getSyntacticForm());
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003948 if (const OpaqueValueExpr *OVE = dyn_cast<OpaqueValueExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003949 if (Expr *Src = OVE->getSourceExpr())
3950 return getDeclFromExpr(Src);
3951
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003952 if (const CallExpr *CE = dyn_cast<CallExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003953 return getDeclFromExpr(CE->getCallee());
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003954 if (const CXXConstructExpr *CE = dyn_cast<CXXConstructExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003955 if (!CE->isElidable())
3956 return CE->getConstructor();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003957 if (const ObjCMessageExpr *OME = dyn_cast<ObjCMessageExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003958 return OME->getMethodDecl();
3959
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003960 if (const ObjCProtocolExpr *PE = dyn_cast<ObjCProtocolExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003961 return PE->getProtocol();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003962 if (const SubstNonTypeTemplateParmPackExpr *NTTP
Guy Benyei11169dd2012-12-18 14:30:41 +00003963 = dyn_cast<SubstNonTypeTemplateParmPackExpr>(E))
3964 return NTTP->getParameterPack();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003965 if (const SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003966 if (isa<NonTypeTemplateParmDecl>(SizeOfPack->getPack()) ||
3967 isa<ParmVarDecl>(SizeOfPack->getPack()))
3968 return SizeOfPack->getPack();
Craig Topper69186e72014-06-08 08:38:04 +00003969
3970 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00003971}
3972
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003973static SourceLocation getLocationFromExpr(const Expr *E) {
3974 if (const ImplicitCastExpr *CE = dyn_cast<ImplicitCastExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003975 return getLocationFromExpr(CE->getSubExpr());
3976
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003977 if (const ObjCMessageExpr *Msg = dyn_cast<ObjCMessageExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003978 return /*FIXME:*/Msg->getLeftLoc();
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003979 if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003980 return DRE->getLocation();
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003981 if (const MemberExpr *Member = dyn_cast<MemberExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003982 return Member->getMemberLoc();
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003983 if (const ObjCIvarRefExpr *Ivar = dyn_cast<ObjCIvarRefExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003984 return Ivar->getLocation();
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003985 if (const SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003986 return SizeOfPack->getPackLoc();
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003987 if (const ObjCPropertyRefExpr *PropRef = dyn_cast<ObjCPropertyRefExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003988 return PropRef->getLocation();
3989
3990 return E->getLocStart();
3991}
3992
3993extern "C" {
3994
3995unsigned clang_visitChildren(CXCursor parent,
3996 CXCursorVisitor visitor,
3997 CXClientData client_data) {
3998 CursorVisitor CursorVis(getCursorTU(parent), visitor, client_data,
3999 /*VisitPreprocessorLast=*/false);
4000 return CursorVis.VisitChildren(parent);
4001}
4002
4003#ifndef __has_feature
4004#define __has_feature(x) 0
4005#endif
4006#if __has_feature(blocks)
4007typedef enum CXChildVisitResult
4008 (^CXCursorVisitorBlock)(CXCursor cursor, CXCursor parent);
4009
4010static enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
4011 CXClientData client_data) {
4012 CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
4013 return block(cursor, parent);
4014}
4015#else
4016// If we are compiled with a compiler that doesn't have native blocks support,
4017// define and call the block manually, so the
4018typedef struct _CXChildVisitResult
4019{
4020 void *isa;
4021 int flags;
4022 int reserved;
4023 enum CXChildVisitResult(*invoke)(struct _CXChildVisitResult*, CXCursor,
4024 CXCursor);
4025} *CXCursorVisitorBlock;
4026
4027static enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
4028 CXClientData client_data) {
4029 CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
4030 return block->invoke(block, cursor, parent);
4031}
4032#endif
4033
4034
4035unsigned clang_visitChildrenWithBlock(CXCursor parent,
4036 CXCursorVisitorBlock block) {
4037 return clang_visitChildren(parent, visitWithBlock, block);
4038}
4039
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004040static CXString getDeclSpelling(const Decl *D) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004041 if (!D)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004042 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004043
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004044 const NamedDecl *ND = dyn_cast<NamedDecl>(D);
Guy Benyei11169dd2012-12-18 14:30:41 +00004045 if (!ND) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004046 if (const ObjCPropertyImplDecl *PropImpl =
4047 dyn_cast<ObjCPropertyImplDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00004048 if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004049 return cxstring::createDup(Property->getIdentifier()->getName());
Guy Benyei11169dd2012-12-18 14:30:41 +00004050
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004051 if (const ImportDecl *ImportD = dyn_cast<ImportDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00004052 if (Module *Mod = ImportD->getImportedModule())
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004053 return cxstring::createDup(Mod->getFullModuleName());
Guy Benyei11169dd2012-12-18 14:30:41 +00004054
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004055 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004056 }
4057
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004058 if (const ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004059 return cxstring::createDup(OMD->getSelector().getAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004060
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004061 if (const ObjCCategoryImplDecl *CIMP = dyn_cast<ObjCCategoryImplDecl>(ND))
Guy Benyei11169dd2012-12-18 14:30:41 +00004062 // No, this isn't the same as the code below. getIdentifier() is non-virtual
4063 // and returns different names. NamedDecl returns the class name and
4064 // ObjCCategoryImplDecl returns the category name.
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004065 return cxstring::createRef(CIMP->getIdentifier()->getNameStart());
Guy Benyei11169dd2012-12-18 14:30:41 +00004066
4067 if (isa<UsingDirectiveDecl>(D))
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004068 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004069
4070 SmallString<1024> S;
4071 llvm::raw_svector_ostream os(S);
4072 ND->printName(os);
4073
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004074 return cxstring::createDup(os.str());
Guy Benyei11169dd2012-12-18 14:30:41 +00004075}
4076
4077CXString clang_getCursorSpelling(CXCursor C) {
4078 if (clang_isTranslationUnit(C.kind))
Dmitri Gribenko2c173b42013-01-11 19:28:44 +00004079 return clang_getTranslationUnitSpelling(getCursorTU(C));
Guy Benyei11169dd2012-12-18 14:30:41 +00004080
4081 if (clang_isReference(C.kind)) {
4082 switch (C.kind) {
4083 case CXCursor_ObjCSuperClassRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004084 const ObjCInterfaceDecl *Super = getCursorObjCSuperClassRef(C).first;
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004085 return cxstring::createRef(Super->getIdentifier()->getNameStart());
Guy Benyei11169dd2012-12-18 14:30:41 +00004086 }
4087 case CXCursor_ObjCClassRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004088 const ObjCInterfaceDecl *Class = getCursorObjCClassRef(C).first;
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004089 return cxstring::createRef(Class->getIdentifier()->getNameStart());
Guy Benyei11169dd2012-12-18 14:30:41 +00004090 }
4091 case CXCursor_ObjCProtocolRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004092 const ObjCProtocolDecl *OID = getCursorObjCProtocolRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004093 assert(OID && "getCursorSpelling(): Missing protocol decl");
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004094 return cxstring::createRef(OID->getIdentifier()->getNameStart());
Guy Benyei11169dd2012-12-18 14:30:41 +00004095 }
4096 case CXCursor_CXXBaseSpecifier: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004097 const CXXBaseSpecifier *B = getCursorCXXBaseSpecifier(C);
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004098 return cxstring::createDup(B->getType().getAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004099 }
4100 case CXCursor_TypeRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004101 const TypeDecl *Type = getCursorTypeRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004102 assert(Type && "Missing type decl");
4103
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004104 return cxstring::createDup(getCursorContext(C).getTypeDeclType(Type).
Guy Benyei11169dd2012-12-18 14:30:41 +00004105 getAsString());
4106 }
4107 case CXCursor_TemplateRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004108 const TemplateDecl *Template = getCursorTemplateRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004109 assert(Template && "Missing template decl");
4110
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004111 return cxstring::createDup(Template->getNameAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004112 }
4113
4114 case CXCursor_NamespaceRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004115 const NamedDecl *NS = getCursorNamespaceRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004116 assert(NS && "Missing namespace decl");
4117
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004118 return cxstring::createDup(NS->getNameAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004119 }
4120
4121 case CXCursor_MemberRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004122 const FieldDecl *Field = getCursorMemberRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004123 assert(Field && "Missing member decl");
4124
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004125 return cxstring::createDup(Field->getNameAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004126 }
4127
4128 case CXCursor_LabelRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004129 const LabelStmt *Label = getCursorLabelRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004130 assert(Label && "Missing label");
4131
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004132 return cxstring::createRef(Label->getName());
Guy Benyei11169dd2012-12-18 14:30:41 +00004133 }
4134
4135 case CXCursor_OverloadedDeclRef: {
4136 OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004137 if (const Decl *D = Storage.dyn_cast<const Decl *>()) {
4138 if (const NamedDecl *ND = dyn_cast<NamedDecl>(D))
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004139 return cxstring::createDup(ND->getNameAsString());
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004140 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004141 }
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004142 if (const OverloadExpr *E = Storage.dyn_cast<const OverloadExpr *>())
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004143 return cxstring::createDup(E->getName().getAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004144 OverloadedTemplateStorage *Ovl
4145 = Storage.get<OverloadedTemplateStorage*>();
4146 if (Ovl->size() == 0)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004147 return cxstring::createEmpty();
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004148 return cxstring::createDup((*Ovl->begin())->getNameAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004149 }
4150
4151 case CXCursor_VariableRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004152 const VarDecl *Var = getCursorVariableRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004153 assert(Var && "Missing variable decl");
4154
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004155 return cxstring::createDup(Var->getNameAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004156 }
4157
4158 default:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004159 return cxstring::createRef("<not implemented>");
Guy Benyei11169dd2012-12-18 14:30:41 +00004160 }
4161 }
4162
4163 if (clang_isExpression(C.kind)) {
Argyrios Kyrtzidis3227d862014-03-03 19:40:52 +00004164 const Expr *E = getCursorExpr(C);
4165
4166 if (C.kind == CXCursor_ObjCStringLiteral ||
4167 C.kind == CXCursor_StringLiteral) {
4168 const StringLiteral *SLit;
4169 if (const ObjCStringLiteral *OSL = dyn_cast<ObjCStringLiteral>(E)) {
4170 SLit = OSL->getString();
4171 } else {
4172 SLit = cast<StringLiteral>(E);
4173 }
4174 SmallString<256> Buf;
4175 llvm::raw_svector_ostream OS(Buf);
4176 SLit->outputString(OS);
4177 return cxstring::createDup(OS.str());
4178 }
4179
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004180 const Decl *D = getDeclFromExpr(getCursorExpr(C));
Guy Benyei11169dd2012-12-18 14:30:41 +00004181 if (D)
4182 return getDeclSpelling(D);
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004183 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004184 }
4185
4186 if (clang_isStatement(C.kind)) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00004187 const Stmt *S = getCursorStmt(C);
4188 if (const LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S))
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004189 return cxstring::createRef(Label->getName());
Guy Benyei11169dd2012-12-18 14:30:41 +00004190
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004191 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004192 }
4193
4194 if (C.kind == CXCursor_MacroExpansion)
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004195 return cxstring::createRef(getCursorMacroExpansion(C).getName()
Guy Benyei11169dd2012-12-18 14:30:41 +00004196 ->getNameStart());
4197
4198 if (C.kind == CXCursor_MacroDefinition)
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004199 return cxstring::createRef(getCursorMacroDefinition(C)->getName()
Guy Benyei11169dd2012-12-18 14:30:41 +00004200 ->getNameStart());
4201
4202 if (C.kind == CXCursor_InclusionDirective)
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004203 return cxstring::createDup(getCursorInclusionDirective(C)->getFileName());
Guy Benyei11169dd2012-12-18 14:30:41 +00004204
4205 if (clang_isDeclaration(C.kind))
4206 return getDeclSpelling(getCursorDecl(C));
4207
4208 if (C.kind == CXCursor_AnnotateAttr) {
Dmitri Gribenkoe4baea62013-01-26 18:08:08 +00004209 const AnnotateAttr *AA = cast<AnnotateAttr>(cxcursor::getCursorAttr(C));
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004210 return cxstring::createDup(AA->getAnnotation());
Guy Benyei11169dd2012-12-18 14:30:41 +00004211 }
4212
4213 if (C.kind == CXCursor_AsmLabelAttr) {
Dmitri Gribenkoe4baea62013-01-26 18:08:08 +00004214 const AsmLabelAttr *AA = cast<AsmLabelAttr>(cxcursor::getCursorAttr(C));
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004215 return cxstring::createDup(AA->getLabel());
Guy Benyei11169dd2012-12-18 14:30:41 +00004216 }
4217
Argyrios Kyrtzidis16834f12013-09-25 00:14:38 +00004218 if (C.kind == CXCursor_PackedAttr) {
4219 return cxstring::createRef("packed");
4220 }
4221
Saleem Abdulrasool79c69712015-09-05 18:53:43 +00004222 if (C.kind == CXCursor_VisibilityAttr) {
4223 const VisibilityAttr *AA = cast<VisibilityAttr>(cxcursor::getCursorAttr(C));
4224 switch (AA->getVisibility()) {
4225 case VisibilityAttr::VisibilityType::Default:
4226 return cxstring::createRef("default");
4227 case VisibilityAttr::VisibilityType::Hidden:
4228 return cxstring::createRef("hidden");
4229 case VisibilityAttr::VisibilityType::Protected:
4230 return cxstring::createRef("protected");
4231 }
4232 llvm_unreachable("unknown visibility type");
4233 }
4234
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004235 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004236}
4237
4238CXSourceRange clang_Cursor_getSpellingNameRange(CXCursor C,
4239 unsigned pieceIndex,
4240 unsigned options) {
4241 if (clang_Cursor_isNull(C))
4242 return clang_getNullRange();
4243
4244 ASTContext &Ctx = getCursorContext(C);
4245
4246 if (clang_isStatement(C.kind)) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00004247 const Stmt *S = getCursorStmt(C);
4248 if (const LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004249 if (pieceIndex > 0)
4250 return clang_getNullRange();
4251 return cxloc::translateSourceRange(Ctx, Label->getIdentLoc());
4252 }
4253
4254 return clang_getNullRange();
4255 }
4256
4257 if (C.kind == CXCursor_ObjCMessageExpr) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00004258 if (const ObjCMessageExpr *
Guy Benyei11169dd2012-12-18 14:30:41 +00004259 ME = dyn_cast_or_null<ObjCMessageExpr>(getCursorExpr(C))) {
4260 if (pieceIndex >= ME->getNumSelectorLocs())
4261 return clang_getNullRange();
4262 return cxloc::translateSourceRange(Ctx, ME->getSelectorLoc(pieceIndex));
4263 }
4264 }
4265
4266 if (C.kind == CXCursor_ObjCInstanceMethodDecl ||
4267 C.kind == CXCursor_ObjCClassMethodDecl) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004268 if (const ObjCMethodDecl *
Guy Benyei11169dd2012-12-18 14:30:41 +00004269 MD = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(C))) {
4270 if (pieceIndex >= MD->getNumSelectorLocs())
4271 return clang_getNullRange();
4272 return cxloc::translateSourceRange(Ctx, MD->getSelectorLoc(pieceIndex));
4273 }
4274 }
4275
4276 if (C.kind == CXCursor_ObjCCategoryDecl ||
4277 C.kind == CXCursor_ObjCCategoryImplDecl) {
4278 if (pieceIndex > 0)
4279 return clang_getNullRange();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004280 if (const ObjCCategoryDecl *
Guy Benyei11169dd2012-12-18 14:30:41 +00004281 CD = dyn_cast_or_null<ObjCCategoryDecl>(getCursorDecl(C)))
4282 return cxloc::translateSourceRange(Ctx, CD->getCategoryNameLoc());
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004283 if (const ObjCCategoryImplDecl *
Guy Benyei11169dd2012-12-18 14:30:41 +00004284 CID = dyn_cast_or_null<ObjCCategoryImplDecl>(getCursorDecl(C)))
4285 return cxloc::translateSourceRange(Ctx, CID->getCategoryNameLoc());
4286 }
4287
4288 if (C.kind == CXCursor_ModuleImportDecl) {
4289 if (pieceIndex > 0)
4290 return clang_getNullRange();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004291 if (const ImportDecl *ImportD =
4292 dyn_cast_or_null<ImportDecl>(getCursorDecl(C))) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004293 ArrayRef<SourceLocation> Locs = ImportD->getIdentifierLocs();
4294 if (!Locs.empty())
4295 return cxloc::translateSourceRange(Ctx,
4296 SourceRange(Locs.front(), Locs.back()));
4297 }
4298 return clang_getNullRange();
4299 }
4300
Argyrios Kyrtzidisa2a1e532014-08-26 20:23:26 +00004301 if (C.kind == CXCursor_CXXMethod || C.kind == CXCursor_Destructor ||
4302 C.kind == CXCursor_ConversionFunction) {
4303 if (pieceIndex > 0)
4304 return clang_getNullRange();
4305 if (const FunctionDecl *FD =
4306 dyn_cast_or_null<FunctionDecl>(getCursorDecl(C))) {
4307 DeclarationNameInfo FunctionName = FD->getNameInfo();
4308 return cxloc::translateSourceRange(Ctx, FunctionName.getSourceRange());
4309 }
4310 return clang_getNullRange();
4311 }
4312
Guy Benyei11169dd2012-12-18 14:30:41 +00004313 // FIXME: A CXCursor_InclusionDirective should give the location of the
4314 // filename, but we don't keep track of this.
4315
4316 // FIXME: A CXCursor_AnnotateAttr should give the location of the annotation
4317 // but we don't keep track of this.
4318
4319 // FIXME: A CXCursor_AsmLabelAttr should give the location of the label
4320 // but we don't keep track of this.
4321
4322 // Default handling, give the location of the cursor.
4323
4324 if (pieceIndex > 0)
4325 return clang_getNullRange();
4326
4327 CXSourceLocation CXLoc = clang_getCursorLocation(C);
4328 SourceLocation Loc = cxloc::translateSourceLocation(CXLoc);
4329 return cxloc::translateSourceRange(Ctx, Loc);
4330}
4331
Eli Bendersky44a206f2014-07-31 18:04:56 +00004332CXString clang_Cursor_getMangling(CXCursor C) {
4333 if (clang_isInvalid(C.kind) || !clang_isDeclaration(C.kind))
4334 return cxstring::createEmpty();
4335
Eli Bendersky44a206f2014-07-31 18:04:56 +00004336 // Mangling only works for functions and variables.
Eli Bendersky79759592014-08-01 15:01:10 +00004337 const Decl *D = getCursorDecl(C);
Eli Bendersky44a206f2014-07-31 18:04:56 +00004338 if (!D || !(isa<FunctionDecl>(D) || isa<VarDecl>(D)))
4339 return cxstring::createEmpty();
4340
Argyrios Kyrtzidisca741ce2016-02-14 22:30:14 +00004341 ASTContext &Ctx = D->getASTContext();
4342 index::CodegenNameGenerator CGNameGen(Ctx);
4343 return cxstring::createDup(CGNameGen.getName(D));
Eli Bendersky44a206f2014-07-31 18:04:56 +00004344}
4345
Saleem Abdulrasool60034432015-11-12 03:57:22 +00004346CXStringSet *clang_Cursor_getCXXManglings(CXCursor C) {
4347 if (clang_isInvalid(C.kind) || !clang_isDeclaration(C.kind))
4348 return nullptr;
4349
4350 const Decl *D = getCursorDecl(C);
4351 if (!(isa<CXXRecordDecl>(D) || isa<CXXMethodDecl>(D)))
4352 return nullptr;
4353
Argyrios Kyrtzidisca741ce2016-02-14 22:30:14 +00004354 ASTContext &Ctx = D->getASTContext();
4355 index::CodegenNameGenerator CGNameGen(Ctx);
4356 std::vector<std::string> Manglings = CGNameGen.getAllManglings(D);
Saleem Abdulrasool60034432015-11-12 03:57:22 +00004357 return cxstring::createSet(Manglings);
4358}
4359
Guy Benyei11169dd2012-12-18 14:30:41 +00004360CXString clang_getCursorDisplayName(CXCursor C) {
4361 if (!clang_isDeclaration(C.kind))
4362 return clang_getCursorSpelling(C);
4363
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004364 const Decl *D = getCursorDecl(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00004365 if (!D)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004366 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004367
4368 PrintingPolicy Policy = getCursorContext(C).getPrintingPolicy();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004369 if (const FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00004370 D = FunTmpl->getTemplatedDecl();
4371
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004372 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004373 SmallString<64> Str;
4374 llvm::raw_svector_ostream OS(Str);
4375 OS << *Function;
4376 if (Function->getPrimaryTemplate())
4377 OS << "<>";
4378 OS << "(";
4379 for (unsigned I = 0, N = Function->getNumParams(); I != N; ++I) {
4380 if (I)
4381 OS << ", ";
4382 OS << Function->getParamDecl(I)->getType().getAsString(Policy);
4383 }
4384
4385 if (Function->isVariadic()) {
4386 if (Function->getNumParams())
4387 OS << ", ";
4388 OS << "...";
4389 }
4390 OS << ")";
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004391 return cxstring::createDup(OS.str());
Guy Benyei11169dd2012-12-18 14:30:41 +00004392 }
4393
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004394 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004395 SmallString<64> Str;
4396 llvm::raw_svector_ostream OS(Str);
4397 OS << *ClassTemplate;
4398 OS << "<";
4399 TemplateParameterList *Params = ClassTemplate->getTemplateParameters();
4400 for (unsigned I = 0, N = Params->size(); I != N; ++I) {
4401 if (I)
4402 OS << ", ";
4403
4404 NamedDecl *Param = Params->getParam(I);
4405 if (Param->getIdentifier()) {
4406 OS << Param->getIdentifier()->getName();
4407 continue;
4408 }
4409
4410 // There is no parameter name, which makes this tricky. Try to come up
4411 // with something useful that isn't too long.
4412 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
4413 OS << (TTP->wasDeclaredWithTypename()? "typename" : "class");
4414 else if (NonTypeTemplateParmDecl *NTTP
4415 = dyn_cast<NonTypeTemplateParmDecl>(Param))
4416 OS << NTTP->getType().getAsString(Policy);
4417 else
4418 OS << "template<...> class";
4419 }
4420
4421 OS << ">";
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004422 return cxstring::createDup(OS.str());
Guy Benyei11169dd2012-12-18 14:30:41 +00004423 }
4424
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004425 if (const ClassTemplateSpecializationDecl *ClassSpec
Guy Benyei11169dd2012-12-18 14:30:41 +00004426 = dyn_cast<ClassTemplateSpecializationDecl>(D)) {
4427 // If the type was explicitly written, use that.
4428 if (TypeSourceInfo *TSInfo = ClassSpec->getTypeAsWritten())
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004429 return cxstring::createDup(TSInfo->getType().getAsString(Policy));
Guy Benyei11169dd2012-12-18 14:30:41 +00004430
Benjamin Kramer9170e912013-02-22 15:46:01 +00004431 SmallString<128> Str;
Guy Benyei11169dd2012-12-18 14:30:41 +00004432 llvm::raw_svector_ostream OS(Str);
4433 OS << *ClassSpec;
Benjamin Kramer9170e912013-02-22 15:46:01 +00004434 TemplateSpecializationType::PrintTemplateArgumentList(OS,
Guy Benyei11169dd2012-12-18 14:30:41 +00004435 ClassSpec->getTemplateArgs().data(),
4436 ClassSpec->getTemplateArgs().size(),
4437 Policy);
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004438 return cxstring::createDup(OS.str());
Guy Benyei11169dd2012-12-18 14:30:41 +00004439 }
4440
4441 return clang_getCursorSpelling(C);
4442}
4443
4444CXString clang_getCursorKindSpelling(enum CXCursorKind Kind) {
4445 switch (Kind) {
4446 case CXCursor_FunctionDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004447 return cxstring::createRef("FunctionDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004448 case CXCursor_TypedefDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004449 return cxstring::createRef("TypedefDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004450 case CXCursor_EnumDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004451 return cxstring::createRef("EnumDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004452 case CXCursor_EnumConstantDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004453 return cxstring::createRef("EnumConstantDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004454 case CXCursor_StructDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004455 return cxstring::createRef("StructDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004456 case CXCursor_UnionDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004457 return cxstring::createRef("UnionDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004458 case CXCursor_ClassDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004459 return cxstring::createRef("ClassDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004460 case CXCursor_FieldDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004461 return cxstring::createRef("FieldDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004462 case CXCursor_VarDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004463 return cxstring::createRef("VarDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004464 case CXCursor_ParmDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004465 return cxstring::createRef("ParmDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004466 case CXCursor_ObjCInterfaceDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004467 return cxstring::createRef("ObjCInterfaceDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004468 case CXCursor_ObjCCategoryDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004469 return cxstring::createRef("ObjCCategoryDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004470 case CXCursor_ObjCProtocolDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004471 return cxstring::createRef("ObjCProtocolDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004472 case CXCursor_ObjCPropertyDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004473 return cxstring::createRef("ObjCPropertyDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004474 case CXCursor_ObjCIvarDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004475 return cxstring::createRef("ObjCIvarDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004476 case CXCursor_ObjCInstanceMethodDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004477 return cxstring::createRef("ObjCInstanceMethodDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004478 case CXCursor_ObjCClassMethodDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004479 return cxstring::createRef("ObjCClassMethodDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004480 case CXCursor_ObjCImplementationDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004481 return cxstring::createRef("ObjCImplementationDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004482 case CXCursor_ObjCCategoryImplDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004483 return cxstring::createRef("ObjCCategoryImplDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004484 case CXCursor_CXXMethod:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004485 return cxstring::createRef("CXXMethod");
Guy Benyei11169dd2012-12-18 14:30:41 +00004486 case CXCursor_UnexposedDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004487 return cxstring::createRef("UnexposedDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004488 case CXCursor_ObjCSuperClassRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004489 return cxstring::createRef("ObjCSuperClassRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004490 case CXCursor_ObjCProtocolRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004491 return cxstring::createRef("ObjCProtocolRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004492 case CXCursor_ObjCClassRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004493 return cxstring::createRef("ObjCClassRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004494 case CXCursor_TypeRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004495 return cxstring::createRef("TypeRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004496 case CXCursor_TemplateRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004497 return cxstring::createRef("TemplateRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004498 case CXCursor_NamespaceRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004499 return cxstring::createRef("NamespaceRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004500 case CXCursor_MemberRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004501 return cxstring::createRef("MemberRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004502 case CXCursor_LabelRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004503 return cxstring::createRef("LabelRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004504 case CXCursor_OverloadedDeclRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004505 return cxstring::createRef("OverloadedDeclRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004506 case CXCursor_VariableRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004507 return cxstring::createRef("VariableRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004508 case CXCursor_IntegerLiteral:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004509 return cxstring::createRef("IntegerLiteral");
Guy Benyei11169dd2012-12-18 14:30:41 +00004510 case CXCursor_FloatingLiteral:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004511 return cxstring::createRef("FloatingLiteral");
Guy Benyei11169dd2012-12-18 14:30:41 +00004512 case CXCursor_ImaginaryLiteral:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004513 return cxstring::createRef("ImaginaryLiteral");
Guy Benyei11169dd2012-12-18 14:30:41 +00004514 case CXCursor_StringLiteral:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004515 return cxstring::createRef("StringLiteral");
Guy Benyei11169dd2012-12-18 14:30:41 +00004516 case CXCursor_CharacterLiteral:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004517 return cxstring::createRef("CharacterLiteral");
Guy Benyei11169dd2012-12-18 14:30:41 +00004518 case CXCursor_ParenExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004519 return cxstring::createRef("ParenExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004520 case CXCursor_UnaryOperator:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004521 return cxstring::createRef("UnaryOperator");
Guy Benyei11169dd2012-12-18 14:30:41 +00004522 case CXCursor_ArraySubscriptExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004523 return cxstring::createRef("ArraySubscriptExpr");
Alexey Bataev1a3320e2015-08-25 14:24:04 +00004524 case CXCursor_OMPArraySectionExpr:
4525 return cxstring::createRef("OMPArraySectionExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004526 case CXCursor_BinaryOperator:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004527 return cxstring::createRef("BinaryOperator");
Guy Benyei11169dd2012-12-18 14:30:41 +00004528 case CXCursor_CompoundAssignOperator:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004529 return cxstring::createRef("CompoundAssignOperator");
Guy Benyei11169dd2012-12-18 14:30:41 +00004530 case CXCursor_ConditionalOperator:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004531 return cxstring::createRef("ConditionalOperator");
Guy Benyei11169dd2012-12-18 14:30:41 +00004532 case CXCursor_CStyleCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004533 return cxstring::createRef("CStyleCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004534 case CXCursor_CompoundLiteralExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004535 return cxstring::createRef("CompoundLiteralExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004536 case CXCursor_InitListExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004537 return cxstring::createRef("InitListExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004538 case CXCursor_AddrLabelExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004539 return cxstring::createRef("AddrLabelExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004540 case CXCursor_StmtExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004541 return cxstring::createRef("StmtExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004542 case CXCursor_GenericSelectionExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004543 return cxstring::createRef("GenericSelectionExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004544 case CXCursor_GNUNullExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004545 return cxstring::createRef("GNUNullExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004546 case CXCursor_CXXStaticCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004547 return cxstring::createRef("CXXStaticCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004548 case CXCursor_CXXDynamicCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004549 return cxstring::createRef("CXXDynamicCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004550 case CXCursor_CXXReinterpretCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004551 return cxstring::createRef("CXXReinterpretCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004552 case CXCursor_CXXConstCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004553 return cxstring::createRef("CXXConstCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004554 case CXCursor_CXXFunctionalCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004555 return cxstring::createRef("CXXFunctionalCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004556 case CXCursor_CXXTypeidExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004557 return cxstring::createRef("CXXTypeidExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004558 case CXCursor_CXXBoolLiteralExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004559 return cxstring::createRef("CXXBoolLiteralExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004560 case CXCursor_CXXNullPtrLiteralExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004561 return cxstring::createRef("CXXNullPtrLiteralExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004562 case CXCursor_CXXThisExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004563 return cxstring::createRef("CXXThisExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004564 case CXCursor_CXXThrowExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004565 return cxstring::createRef("CXXThrowExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004566 case CXCursor_CXXNewExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004567 return cxstring::createRef("CXXNewExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004568 case CXCursor_CXXDeleteExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004569 return cxstring::createRef("CXXDeleteExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004570 case CXCursor_UnaryExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004571 return cxstring::createRef("UnaryExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004572 case CXCursor_ObjCStringLiteral:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004573 return cxstring::createRef("ObjCStringLiteral");
Guy Benyei11169dd2012-12-18 14:30:41 +00004574 case CXCursor_ObjCBoolLiteralExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004575 return cxstring::createRef("ObjCBoolLiteralExpr");
Argyrios Kyrtzidisc2233be2013-04-23 17:57:17 +00004576 case CXCursor_ObjCSelfExpr:
4577 return cxstring::createRef("ObjCSelfExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004578 case CXCursor_ObjCEncodeExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004579 return cxstring::createRef("ObjCEncodeExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004580 case CXCursor_ObjCSelectorExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004581 return cxstring::createRef("ObjCSelectorExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004582 case CXCursor_ObjCProtocolExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004583 return cxstring::createRef("ObjCProtocolExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004584 case CXCursor_ObjCBridgedCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004585 return cxstring::createRef("ObjCBridgedCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004586 case CXCursor_BlockExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004587 return cxstring::createRef("BlockExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004588 case CXCursor_PackExpansionExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004589 return cxstring::createRef("PackExpansionExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004590 case CXCursor_SizeOfPackExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004591 return cxstring::createRef("SizeOfPackExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004592 case CXCursor_LambdaExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004593 return cxstring::createRef("LambdaExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004594 case CXCursor_UnexposedExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004595 return cxstring::createRef("UnexposedExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004596 case CXCursor_DeclRefExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004597 return cxstring::createRef("DeclRefExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004598 case CXCursor_MemberRefExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004599 return cxstring::createRef("MemberRefExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004600 case CXCursor_CallExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004601 return cxstring::createRef("CallExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004602 case CXCursor_ObjCMessageExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004603 return cxstring::createRef("ObjCMessageExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004604 case CXCursor_UnexposedStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004605 return cxstring::createRef("UnexposedStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004606 case CXCursor_DeclStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004607 return cxstring::createRef("DeclStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004608 case CXCursor_LabelStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004609 return cxstring::createRef("LabelStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004610 case CXCursor_CompoundStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004611 return cxstring::createRef("CompoundStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004612 case CXCursor_CaseStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004613 return cxstring::createRef("CaseStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004614 case CXCursor_DefaultStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004615 return cxstring::createRef("DefaultStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004616 case CXCursor_IfStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004617 return cxstring::createRef("IfStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004618 case CXCursor_SwitchStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004619 return cxstring::createRef("SwitchStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004620 case CXCursor_WhileStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004621 return cxstring::createRef("WhileStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004622 case CXCursor_DoStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004623 return cxstring::createRef("DoStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004624 case CXCursor_ForStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004625 return cxstring::createRef("ForStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004626 case CXCursor_GotoStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004627 return cxstring::createRef("GotoStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004628 case CXCursor_IndirectGotoStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004629 return cxstring::createRef("IndirectGotoStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004630 case CXCursor_ContinueStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004631 return cxstring::createRef("ContinueStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004632 case CXCursor_BreakStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004633 return cxstring::createRef("BreakStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004634 case CXCursor_ReturnStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004635 return cxstring::createRef("ReturnStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004636 case CXCursor_GCCAsmStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004637 return cxstring::createRef("GCCAsmStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004638 case CXCursor_MSAsmStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004639 return cxstring::createRef("MSAsmStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004640 case CXCursor_ObjCAtTryStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004641 return cxstring::createRef("ObjCAtTryStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004642 case CXCursor_ObjCAtCatchStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004643 return cxstring::createRef("ObjCAtCatchStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004644 case CXCursor_ObjCAtFinallyStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004645 return cxstring::createRef("ObjCAtFinallyStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004646 case CXCursor_ObjCAtThrowStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004647 return cxstring::createRef("ObjCAtThrowStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004648 case CXCursor_ObjCAtSynchronizedStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004649 return cxstring::createRef("ObjCAtSynchronizedStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004650 case CXCursor_ObjCAutoreleasePoolStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004651 return cxstring::createRef("ObjCAutoreleasePoolStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004652 case CXCursor_ObjCForCollectionStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004653 return cxstring::createRef("ObjCForCollectionStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004654 case CXCursor_CXXCatchStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004655 return cxstring::createRef("CXXCatchStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004656 case CXCursor_CXXTryStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004657 return cxstring::createRef("CXXTryStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004658 case CXCursor_CXXForRangeStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004659 return cxstring::createRef("CXXForRangeStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004660 case CXCursor_SEHTryStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004661 return cxstring::createRef("SEHTryStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004662 case CXCursor_SEHExceptStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004663 return cxstring::createRef("SEHExceptStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004664 case CXCursor_SEHFinallyStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004665 return cxstring::createRef("SEHFinallyStmt");
Nico Weber9b982072014-07-07 00:12:30 +00004666 case CXCursor_SEHLeaveStmt:
4667 return cxstring::createRef("SEHLeaveStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004668 case CXCursor_NullStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004669 return cxstring::createRef("NullStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004670 case CXCursor_InvalidFile:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004671 return cxstring::createRef("InvalidFile");
Guy Benyei11169dd2012-12-18 14:30:41 +00004672 case CXCursor_InvalidCode:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004673 return cxstring::createRef("InvalidCode");
Guy Benyei11169dd2012-12-18 14:30:41 +00004674 case CXCursor_NoDeclFound:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004675 return cxstring::createRef("NoDeclFound");
Guy Benyei11169dd2012-12-18 14:30:41 +00004676 case CXCursor_NotImplemented:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004677 return cxstring::createRef("NotImplemented");
Guy Benyei11169dd2012-12-18 14:30:41 +00004678 case CXCursor_TranslationUnit:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004679 return cxstring::createRef("TranslationUnit");
Guy Benyei11169dd2012-12-18 14:30:41 +00004680 case CXCursor_UnexposedAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004681 return cxstring::createRef("UnexposedAttr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004682 case CXCursor_IBActionAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004683 return cxstring::createRef("attribute(ibaction)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004684 case CXCursor_IBOutletAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004685 return cxstring::createRef("attribute(iboutlet)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004686 case CXCursor_IBOutletCollectionAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004687 return cxstring::createRef("attribute(iboutletcollection)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004688 case CXCursor_CXXFinalAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004689 return cxstring::createRef("attribute(final)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004690 case CXCursor_CXXOverrideAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004691 return cxstring::createRef("attribute(override)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004692 case CXCursor_AnnotateAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004693 return cxstring::createRef("attribute(annotate)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004694 case CXCursor_AsmLabelAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004695 return cxstring::createRef("asm label");
Argyrios Kyrtzidis16834f12013-09-25 00:14:38 +00004696 case CXCursor_PackedAttr:
4697 return cxstring::createRef("attribute(packed)");
Joey Gouly81228382014-05-01 15:41:58 +00004698 case CXCursor_PureAttr:
4699 return cxstring::createRef("attribute(pure)");
4700 case CXCursor_ConstAttr:
4701 return cxstring::createRef("attribute(const)");
4702 case CXCursor_NoDuplicateAttr:
4703 return cxstring::createRef("attribute(noduplicate)");
Eli Bendersky2581e662014-05-28 19:29:58 +00004704 case CXCursor_CUDAConstantAttr:
4705 return cxstring::createRef("attribute(constant)");
4706 case CXCursor_CUDADeviceAttr:
4707 return cxstring::createRef("attribute(device)");
4708 case CXCursor_CUDAGlobalAttr:
4709 return cxstring::createRef("attribute(global)");
4710 case CXCursor_CUDAHostAttr:
4711 return cxstring::createRef("attribute(host)");
Eli Bendersky9b071472014-08-08 14:59:00 +00004712 case CXCursor_CUDASharedAttr:
4713 return cxstring::createRef("attribute(shared)");
Saleem Abdulrasool79c69712015-09-05 18:53:43 +00004714 case CXCursor_VisibilityAttr:
4715 return cxstring::createRef("attribute(visibility)");
Saleem Abdulrasool8aa0b802015-12-10 18:45:18 +00004716 case CXCursor_DLLExport:
4717 return cxstring::createRef("attribute(dllexport)");
4718 case CXCursor_DLLImport:
4719 return cxstring::createRef("attribute(dllimport)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004720 case CXCursor_PreprocessingDirective:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004721 return cxstring::createRef("preprocessing directive");
Guy Benyei11169dd2012-12-18 14:30:41 +00004722 case CXCursor_MacroDefinition:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004723 return cxstring::createRef("macro definition");
Guy Benyei11169dd2012-12-18 14:30:41 +00004724 case CXCursor_MacroExpansion:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004725 return cxstring::createRef("macro expansion");
Guy Benyei11169dd2012-12-18 14:30:41 +00004726 case CXCursor_InclusionDirective:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004727 return cxstring::createRef("inclusion directive");
Guy Benyei11169dd2012-12-18 14:30:41 +00004728 case CXCursor_Namespace:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004729 return cxstring::createRef("Namespace");
Guy Benyei11169dd2012-12-18 14:30:41 +00004730 case CXCursor_LinkageSpec:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004731 return cxstring::createRef("LinkageSpec");
Guy Benyei11169dd2012-12-18 14:30:41 +00004732 case CXCursor_CXXBaseSpecifier:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004733 return cxstring::createRef("C++ base class specifier");
Guy Benyei11169dd2012-12-18 14:30:41 +00004734 case CXCursor_Constructor:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004735 return cxstring::createRef("CXXConstructor");
Guy Benyei11169dd2012-12-18 14:30:41 +00004736 case CXCursor_Destructor:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004737 return cxstring::createRef("CXXDestructor");
Guy Benyei11169dd2012-12-18 14:30:41 +00004738 case CXCursor_ConversionFunction:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004739 return cxstring::createRef("CXXConversion");
Guy Benyei11169dd2012-12-18 14:30:41 +00004740 case CXCursor_TemplateTypeParameter:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004741 return cxstring::createRef("TemplateTypeParameter");
Guy Benyei11169dd2012-12-18 14:30:41 +00004742 case CXCursor_NonTypeTemplateParameter:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004743 return cxstring::createRef("NonTypeTemplateParameter");
Guy Benyei11169dd2012-12-18 14:30:41 +00004744 case CXCursor_TemplateTemplateParameter:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004745 return cxstring::createRef("TemplateTemplateParameter");
Guy Benyei11169dd2012-12-18 14:30:41 +00004746 case CXCursor_FunctionTemplate:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004747 return cxstring::createRef("FunctionTemplate");
Guy Benyei11169dd2012-12-18 14:30:41 +00004748 case CXCursor_ClassTemplate:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004749 return cxstring::createRef("ClassTemplate");
Guy Benyei11169dd2012-12-18 14:30:41 +00004750 case CXCursor_ClassTemplatePartialSpecialization:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004751 return cxstring::createRef("ClassTemplatePartialSpecialization");
Guy Benyei11169dd2012-12-18 14:30:41 +00004752 case CXCursor_NamespaceAlias:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004753 return cxstring::createRef("NamespaceAlias");
Guy Benyei11169dd2012-12-18 14:30:41 +00004754 case CXCursor_UsingDirective:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004755 return cxstring::createRef("UsingDirective");
Guy Benyei11169dd2012-12-18 14:30:41 +00004756 case CXCursor_UsingDeclaration:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004757 return cxstring::createRef("UsingDeclaration");
Guy Benyei11169dd2012-12-18 14:30:41 +00004758 case CXCursor_TypeAliasDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004759 return cxstring::createRef("TypeAliasDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004760 case CXCursor_ObjCSynthesizeDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004761 return cxstring::createRef("ObjCSynthesizeDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004762 case CXCursor_ObjCDynamicDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004763 return cxstring::createRef("ObjCDynamicDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004764 case CXCursor_CXXAccessSpecifier:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004765 return cxstring::createRef("CXXAccessSpecifier");
Guy Benyei11169dd2012-12-18 14:30:41 +00004766 case CXCursor_ModuleImportDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004767 return cxstring::createRef("ModuleImport");
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00004768 case CXCursor_OMPParallelDirective:
Alexey Bataev1b59ab52014-02-27 08:29:12 +00004769 return cxstring::createRef("OMPParallelDirective");
4770 case CXCursor_OMPSimdDirective:
4771 return cxstring::createRef("OMPSimdDirective");
Alexey Bataevf29276e2014-06-18 04:14:57 +00004772 case CXCursor_OMPForDirective:
4773 return cxstring::createRef("OMPForDirective");
Alexander Musmanf82886e2014-09-18 05:12:34 +00004774 case CXCursor_OMPForSimdDirective:
4775 return cxstring::createRef("OMPForSimdDirective");
Alexey Bataevd3f8dd22014-06-25 11:44:49 +00004776 case CXCursor_OMPSectionsDirective:
4777 return cxstring::createRef("OMPSectionsDirective");
Alexey Bataev1e0498a2014-06-26 08:21:58 +00004778 case CXCursor_OMPSectionDirective:
4779 return cxstring::createRef("OMPSectionDirective");
Alexey Bataevd1e40fb2014-06-26 12:05:45 +00004780 case CXCursor_OMPSingleDirective:
4781 return cxstring::createRef("OMPSingleDirective");
Alexander Musman80c22892014-07-17 08:54:58 +00004782 case CXCursor_OMPMasterDirective:
4783 return cxstring::createRef("OMPMasterDirective");
Alexander Musmand9ed09f2014-07-21 09:42:05 +00004784 case CXCursor_OMPCriticalDirective:
4785 return cxstring::createRef("OMPCriticalDirective");
Alexey Bataev4acb8592014-07-07 13:01:15 +00004786 case CXCursor_OMPParallelForDirective:
4787 return cxstring::createRef("OMPParallelForDirective");
Alexander Musmane4e893b2014-09-23 09:33:00 +00004788 case CXCursor_OMPParallelForSimdDirective:
4789 return cxstring::createRef("OMPParallelForSimdDirective");
Alexey Bataev84d0b3e2014-07-08 08:12:03 +00004790 case CXCursor_OMPParallelSectionsDirective:
4791 return cxstring::createRef("OMPParallelSectionsDirective");
Alexey Bataev9c2e8ee2014-07-11 11:25:16 +00004792 case CXCursor_OMPTaskDirective:
4793 return cxstring::createRef("OMPTaskDirective");
Alexey Bataev68446b72014-07-18 07:47:19 +00004794 case CXCursor_OMPTaskyieldDirective:
4795 return cxstring::createRef("OMPTaskyieldDirective");
Alexey Bataev4d1dfea2014-07-18 09:11:51 +00004796 case CXCursor_OMPBarrierDirective:
4797 return cxstring::createRef("OMPBarrierDirective");
Alexey Bataev2df347a2014-07-18 10:17:07 +00004798 case CXCursor_OMPTaskwaitDirective:
4799 return cxstring::createRef("OMPTaskwaitDirective");
Alexey Bataevc30dd2d2015-06-18 12:14:09 +00004800 case CXCursor_OMPTaskgroupDirective:
4801 return cxstring::createRef("OMPTaskgroupDirective");
Alexey Bataev6125da92014-07-21 11:26:11 +00004802 case CXCursor_OMPFlushDirective:
4803 return cxstring::createRef("OMPFlushDirective");
Alexey Bataev9fb6e642014-07-22 06:45:04 +00004804 case CXCursor_OMPOrderedDirective:
4805 return cxstring::createRef("OMPOrderedDirective");
Alexey Bataev0162e452014-07-22 10:10:35 +00004806 case CXCursor_OMPAtomicDirective:
4807 return cxstring::createRef("OMPAtomicDirective");
Alexey Bataev0bd520b2014-09-19 08:19:49 +00004808 case CXCursor_OMPTargetDirective:
4809 return cxstring::createRef("OMPTargetDirective");
Michael Wong65f367f2015-07-21 13:44:28 +00004810 case CXCursor_OMPTargetDataDirective:
4811 return cxstring::createRef("OMPTargetDataDirective");
Samuel Antaodf67fc42016-01-19 19:15:56 +00004812 case CXCursor_OMPTargetEnterDataDirective:
4813 return cxstring::createRef("OMPTargetEnterDataDirective");
Samuel Antao72590762016-01-19 20:04:50 +00004814 case CXCursor_OMPTargetExitDataDirective:
4815 return cxstring::createRef("OMPTargetExitDataDirective");
Arpith Chacko Jacobe955b3d2016-01-26 18:48:41 +00004816 case CXCursor_OMPTargetParallelDirective:
4817 return cxstring::createRef("OMPTargetParallelDirective");
Arpith Chacko Jacob05bebb52016-02-03 15:46:42 +00004818 case CXCursor_OMPTargetParallelForDirective:
4819 return cxstring::createRef("OMPTargetParallelForDirective");
Alexey Bataev13314bf2014-10-09 04:18:56 +00004820 case CXCursor_OMPTeamsDirective:
4821 return cxstring::createRef("OMPTeamsDirective");
Alexey Bataev6d4ed052015-07-01 06:57:41 +00004822 case CXCursor_OMPCancellationPointDirective:
4823 return cxstring::createRef("OMPCancellationPointDirective");
Alexey Bataev80909872015-07-02 11:25:17 +00004824 case CXCursor_OMPCancelDirective:
4825 return cxstring::createRef("OMPCancelDirective");
Alexey Bataev49f6e782015-12-01 04:18:41 +00004826 case CXCursor_OMPTaskLoopDirective:
4827 return cxstring::createRef("OMPTaskLoopDirective");
Alexey Bataev0a6ed842015-12-03 09:40:15 +00004828 case CXCursor_OMPTaskLoopSimdDirective:
4829 return cxstring::createRef("OMPTaskLoopSimdDirective");
Carlo Bertolli6200a3d2015-12-14 14:51:25 +00004830 case CXCursor_OMPDistributeDirective:
4831 return cxstring::createRef("OMPDistributeDirective");
Francisco Lopes da Silva975a9f62015-01-21 16:24:11 +00004832 case CXCursor_OverloadCandidate:
4833 return cxstring::createRef("OverloadCandidate");
Sergey Kalinichev8f3b1872015-11-15 13:48:32 +00004834 case CXCursor_TypeAliasTemplateDecl:
4835 return cxstring::createRef("TypeAliasTemplateDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004836 }
4837
4838 llvm_unreachable("Unhandled CXCursorKind");
4839}
4840
4841struct GetCursorData {
4842 SourceLocation TokenBeginLoc;
4843 bool PointsAtMacroArgExpansion;
4844 bool VisitedObjCPropertyImplDecl;
4845 SourceLocation VisitedDeclaratorDeclStartLoc;
4846 CXCursor &BestCursor;
4847
4848 GetCursorData(SourceManager &SM,
4849 SourceLocation tokenBegin, CXCursor &outputCursor)
4850 : TokenBeginLoc(tokenBegin), BestCursor(outputCursor) {
4851 PointsAtMacroArgExpansion = SM.isMacroArgExpansion(tokenBegin);
4852 VisitedObjCPropertyImplDecl = false;
4853 }
4854};
4855
4856static enum CXChildVisitResult GetCursorVisitor(CXCursor cursor,
4857 CXCursor parent,
4858 CXClientData client_data) {
4859 GetCursorData *Data = static_cast<GetCursorData *>(client_data);
4860 CXCursor *BestCursor = &Data->BestCursor;
4861
4862 // If we point inside a macro argument we should provide info of what the
4863 // token is so use the actual cursor, don't replace it with a macro expansion
4864 // cursor.
4865 if (cursor.kind == CXCursor_MacroExpansion && Data->PointsAtMacroArgExpansion)
4866 return CXChildVisit_Recurse;
4867
4868 if (clang_isDeclaration(cursor.kind)) {
4869 // Avoid having the implicit methods override the property decls.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004870 if (const ObjCMethodDecl *MD
Guy Benyei11169dd2012-12-18 14:30:41 +00004871 = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(cursor))) {
4872 if (MD->isImplicit())
4873 return CXChildVisit_Break;
4874
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004875 } else if (const ObjCInterfaceDecl *ID
Guy Benyei11169dd2012-12-18 14:30:41 +00004876 = dyn_cast_or_null<ObjCInterfaceDecl>(getCursorDecl(cursor))) {
4877 // Check that when we have multiple @class references in the same line,
4878 // that later ones do not override the previous ones.
4879 // If we have:
4880 // @class Foo, Bar;
4881 // source ranges for both start at '@', so 'Bar' will end up overriding
4882 // 'Foo' even though the cursor location was at 'Foo'.
4883 if (BestCursor->kind == CXCursor_ObjCInterfaceDecl ||
4884 BestCursor->kind == CXCursor_ObjCClassRef)
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004885 if (const ObjCInterfaceDecl *PrevID
Guy Benyei11169dd2012-12-18 14:30:41 +00004886 = dyn_cast_or_null<ObjCInterfaceDecl>(getCursorDecl(*BestCursor))){
4887 if (PrevID != ID &&
4888 !PrevID->isThisDeclarationADefinition() &&
4889 !ID->isThisDeclarationADefinition())
4890 return CXChildVisit_Break;
4891 }
4892
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004893 } else if (const DeclaratorDecl *DD
Guy Benyei11169dd2012-12-18 14:30:41 +00004894 = dyn_cast_or_null<DeclaratorDecl>(getCursorDecl(cursor))) {
4895 SourceLocation StartLoc = DD->getSourceRange().getBegin();
4896 // Check that when we have multiple declarators in the same line,
4897 // that later ones do not override the previous ones.
4898 // If we have:
4899 // int Foo, Bar;
4900 // source ranges for both start at 'int', so 'Bar' will end up overriding
4901 // 'Foo' even though the cursor location was at 'Foo'.
4902 if (Data->VisitedDeclaratorDeclStartLoc == StartLoc)
4903 return CXChildVisit_Break;
4904 Data->VisitedDeclaratorDeclStartLoc = StartLoc;
4905
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004906 } else if (const ObjCPropertyImplDecl *PropImp
Guy Benyei11169dd2012-12-18 14:30:41 +00004907 = dyn_cast_or_null<ObjCPropertyImplDecl>(getCursorDecl(cursor))) {
4908 (void)PropImp;
4909 // Check that when we have multiple @synthesize in the same line,
4910 // that later ones do not override the previous ones.
4911 // If we have:
4912 // @synthesize Foo, Bar;
4913 // source ranges for both start at '@', so 'Bar' will end up overriding
4914 // 'Foo' even though the cursor location was at 'Foo'.
4915 if (Data->VisitedObjCPropertyImplDecl)
4916 return CXChildVisit_Break;
4917 Data->VisitedObjCPropertyImplDecl = true;
4918 }
4919 }
4920
4921 if (clang_isExpression(cursor.kind) &&
4922 clang_isDeclaration(BestCursor->kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004923 if (const Decl *D = getCursorDecl(*BestCursor)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004924 // Avoid having the cursor of an expression replace the declaration cursor
4925 // when the expression source range overlaps the declaration range.
4926 // This can happen for C++ constructor expressions whose range generally
4927 // include the variable declaration, e.g.:
4928 // MyCXXClass foo; // Make sure pointing at 'foo' returns a VarDecl cursor.
4929 if (D->getLocation().isValid() && Data->TokenBeginLoc.isValid() &&
4930 D->getLocation() == Data->TokenBeginLoc)
4931 return CXChildVisit_Break;
4932 }
4933 }
4934
4935 // If our current best cursor is the construction of a temporary object,
4936 // don't replace that cursor with a type reference, because we want
4937 // clang_getCursor() to point at the constructor.
4938 if (clang_isExpression(BestCursor->kind) &&
4939 isa<CXXTemporaryObjectExpr>(getCursorExpr(*BestCursor)) &&
4940 cursor.kind == CXCursor_TypeRef) {
4941 // Keep the cursor pointing at CXXTemporaryObjectExpr but also mark it
4942 // as having the actual point on the type reference.
4943 *BestCursor = getTypeRefedCallExprCursor(*BestCursor);
4944 return CXChildVisit_Recurse;
4945 }
Douglas Gregore9d95f12015-07-07 03:57:35 +00004946
4947 // If we already have an Objective-C superclass reference, don't
4948 // update it further.
4949 if (BestCursor->kind == CXCursor_ObjCSuperClassRef)
4950 return CXChildVisit_Break;
4951
Guy Benyei11169dd2012-12-18 14:30:41 +00004952 *BestCursor = cursor;
4953 return CXChildVisit_Recurse;
4954}
4955
4956CXCursor clang_getCursor(CXTranslationUnit TU, CXSourceLocation Loc) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00004957 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00004958 LOG_BAD_TU(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00004959 return clang_getNullCursor();
Dmitri Gribenko256454f2014-02-11 14:34:14 +00004960 }
Guy Benyei11169dd2012-12-18 14:30:41 +00004961
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00004962 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00004963 ASTUnit::ConcurrencyCheck Check(*CXXUnit);
4964
4965 SourceLocation SLoc = cxloc::translateSourceLocation(Loc);
4966 CXCursor Result = cxcursor::getCursor(TU, SLoc);
4967
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00004968 LOG_FUNC_SECTION {
Guy Benyei11169dd2012-12-18 14:30:41 +00004969 CXFile SearchFile;
4970 unsigned SearchLine, SearchColumn;
4971 CXFile ResultFile;
4972 unsigned ResultLine, ResultColumn;
4973 CXString SearchFileName, ResultFileName, KindSpelling, USR;
4974 const char *IsDef = clang_isCursorDefinition(Result)? " (Definition)" : "";
4975 CXSourceLocation ResultLoc = clang_getCursorLocation(Result);
Craig Topper69186e72014-06-08 08:38:04 +00004976
4977 clang_getFileLocation(Loc, &SearchFile, &SearchLine, &SearchColumn,
4978 nullptr);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00004979 clang_getFileLocation(ResultLoc, &ResultFile, &ResultLine,
Craig Topper69186e72014-06-08 08:38:04 +00004980 &ResultColumn, nullptr);
Guy Benyei11169dd2012-12-18 14:30:41 +00004981 SearchFileName = clang_getFileName(SearchFile);
4982 ResultFileName = clang_getFileName(ResultFile);
4983 KindSpelling = clang_getCursorKindSpelling(Result.kind);
4984 USR = clang_getCursorUSR(Result);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00004985 *Log << llvm::format("(%s:%d:%d) = %s",
4986 clang_getCString(SearchFileName), SearchLine, SearchColumn,
4987 clang_getCString(KindSpelling))
4988 << llvm::format("(%s:%d:%d):%s%s",
4989 clang_getCString(ResultFileName), ResultLine, ResultColumn,
4990 clang_getCString(USR), IsDef);
Guy Benyei11169dd2012-12-18 14:30:41 +00004991 clang_disposeString(SearchFileName);
4992 clang_disposeString(ResultFileName);
4993 clang_disposeString(KindSpelling);
4994 clang_disposeString(USR);
4995
4996 CXCursor Definition = clang_getCursorDefinition(Result);
4997 if (!clang_equalCursors(Definition, clang_getNullCursor())) {
4998 CXSourceLocation DefinitionLoc = clang_getCursorLocation(Definition);
4999 CXString DefinitionKindSpelling
5000 = clang_getCursorKindSpelling(Definition.kind);
5001 CXFile DefinitionFile;
5002 unsigned DefinitionLine, DefinitionColumn;
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00005003 clang_getFileLocation(DefinitionLoc, &DefinitionFile,
Craig Topper69186e72014-06-08 08:38:04 +00005004 &DefinitionLine, &DefinitionColumn, nullptr);
Guy Benyei11169dd2012-12-18 14:30:41 +00005005 CXString DefinitionFileName = clang_getFileName(DefinitionFile);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00005006 *Log << llvm::format(" -> %s(%s:%d:%d)",
5007 clang_getCString(DefinitionKindSpelling),
5008 clang_getCString(DefinitionFileName),
5009 DefinitionLine, DefinitionColumn);
Guy Benyei11169dd2012-12-18 14:30:41 +00005010 clang_disposeString(DefinitionFileName);
5011 clang_disposeString(DefinitionKindSpelling);
5012 }
5013 }
5014
5015 return Result;
5016}
5017
5018CXCursor clang_getNullCursor(void) {
5019 return MakeCXCursorInvalid(CXCursor_InvalidFile);
5020}
5021
5022unsigned clang_equalCursors(CXCursor X, CXCursor Y) {
Argyrios Kyrtzidisbf1be592013-01-08 18:23:28 +00005023 // Clear out the "FirstInDeclGroup" part in a declaration cursor, since we
5024 // can't set consistently. For example, when visiting a DeclStmt we will set
5025 // it but we don't set it on the result of clang_getCursorDefinition for
5026 // a reference of the same declaration.
5027 // FIXME: Setting "FirstInDeclGroup" in CXCursors is a hack that only works
5028 // when visiting a DeclStmt currently, the AST should be enhanced to be able
5029 // to provide that kind of info.
5030 if (clang_isDeclaration(X.kind))
Craig Topper69186e72014-06-08 08:38:04 +00005031 X.data[1] = nullptr;
Argyrios Kyrtzidisbf1be592013-01-08 18:23:28 +00005032 if (clang_isDeclaration(Y.kind))
Craig Topper69186e72014-06-08 08:38:04 +00005033 Y.data[1] = nullptr;
Argyrios Kyrtzidisbf1be592013-01-08 18:23:28 +00005034
Guy Benyei11169dd2012-12-18 14:30:41 +00005035 return X == Y;
5036}
5037
5038unsigned clang_hashCursor(CXCursor C) {
5039 unsigned Index = 0;
5040 if (clang_isExpression(C.kind) || clang_isStatement(C.kind))
5041 Index = 1;
5042
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005043 return llvm::DenseMapInfo<std::pair<unsigned, const void*> >::getHashValue(
Guy Benyei11169dd2012-12-18 14:30:41 +00005044 std::make_pair(C.kind, C.data[Index]));
5045}
5046
5047unsigned clang_isInvalid(enum CXCursorKind K) {
5048 return K >= CXCursor_FirstInvalid && K <= CXCursor_LastInvalid;
5049}
5050
5051unsigned clang_isDeclaration(enum CXCursorKind K) {
5052 return (K >= CXCursor_FirstDecl && K <= CXCursor_LastDecl) ||
5053 (K >= CXCursor_FirstExtraDecl && K <= CXCursor_LastExtraDecl);
5054}
5055
5056unsigned clang_isReference(enum CXCursorKind K) {
5057 return K >= CXCursor_FirstRef && K <= CXCursor_LastRef;
5058}
5059
5060unsigned clang_isExpression(enum CXCursorKind K) {
5061 return K >= CXCursor_FirstExpr && K <= CXCursor_LastExpr;
5062}
5063
5064unsigned clang_isStatement(enum CXCursorKind K) {
5065 return K >= CXCursor_FirstStmt && K <= CXCursor_LastStmt;
5066}
5067
5068unsigned clang_isAttribute(enum CXCursorKind K) {
5069 return K >= CXCursor_FirstAttr && K <= CXCursor_LastAttr;
5070}
5071
5072unsigned clang_isTranslationUnit(enum CXCursorKind K) {
5073 return K == CXCursor_TranslationUnit;
5074}
5075
5076unsigned clang_isPreprocessing(enum CXCursorKind K) {
5077 return K >= CXCursor_FirstPreprocessing && K <= CXCursor_LastPreprocessing;
5078}
5079
5080unsigned clang_isUnexposed(enum CXCursorKind K) {
5081 switch (K) {
5082 case CXCursor_UnexposedDecl:
5083 case CXCursor_UnexposedExpr:
5084 case CXCursor_UnexposedStmt:
5085 case CXCursor_UnexposedAttr:
5086 return true;
5087 default:
5088 return false;
5089 }
5090}
5091
5092CXCursorKind clang_getCursorKind(CXCursor C) {
5093 return C.kind;
5094}
5095
5096CXSourceLocation clang_getCursorLocation(CXCursor C) {
5097 if (clang_isReference(C.kind)) {
5098 switch (C.kind) {
5099 case CXCursor_ObjCSuperClassRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005100 std::pair<const ObjCInterfaceDecl *, SourceLocation> P
Guy Benyei11169dd2012-12-18 14:30:41 +00005101 = getCursorObjCSuperClassRef(C);
5102 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5103 }
5104
5105 case CXCursor_ObjCProtocolRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005106 std::pair<const ObjCProtocolDecl *, SourceLocation> P
Guy Benyei11169dd2012-12-18 14:30:41 +00005107 = getCursorObjCProtocolRef(C);
5108 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5109 }
5110
5111 case CXCursor_ObjCClassRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005112 std::pair<const ObjCInterfaceDecl *, SourceLocation> P
Guy Benyei11169dd2012-12-18 14:30:41 +00005113 = getCursorObjCClassRef(C);
5114 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5115 }
5116
5117 case CXCursor_TypeRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005118 std::pair<const TypeDecl *, SourceLocation> P = getCursorTypeRef(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005119 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5120 }
5121
5122 case CXCursor_TemplateRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005123 std::pair<const TemplateDecl *, SourceLocation> P =
5124 getCursorTemplateRef(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005125 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5126 }
5127
5128 case CXCursor_NamespaceRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005129 std::pair<const NamedDecl *, SourceLocation> P = getCursorNamespaceRef(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005130 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5131 }
5132
5133 case CXCursor_MemberRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005134 std::pair<const FieldDecl *, SourceLocation> P = getCursorMemberRef(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005135 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5136 }
5137
5138 case CXCursor_VariableRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005139 std::pair<const VarDecl *, SourceLocation> P = getCursorVariableRef(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005140 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5141 }
5142
5143 case CXCursor_CXXBaseSpecifier: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005144 const CXXBaseSpecifier *BaseSpec = getCursorCXXBaseSpecifier(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005145 if (!BaseSpec)
5146 return clang_getNullLocation();
5147
5148 if (TypeSourceInfo *TSInfo = BaseSpec->getTypeSourceInfo())
5149 return cxloc::translateSourceLocation(getCursorContext(C),
5150 TSInfo->getTypeLoc().getBeginLoc());
5151
5152 return cxloc::translateSourceLocation(getCursorContext(C),
5153 BaseSpec->getLocStart());
5154 }
5155
5156 case CXCursor_LabelRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005157 std::pair<const LabelStmt *, SourceLocation> P = getCursorLabelRef(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005158 return cxloc::translateSourceLocation(getCursorContext(C), P.second);
5159 }
5160
5161 case CXCursor_OverloadedDeclRef:
5162 return cxloc::translateSourceLocation(getCursorContext(C),
5163 getCursorOverloadedDeclRef(C).second);
5164
5165 default:
5166 // FIXME: Need a way to enumerate all non-reference cases.
5167 llvm_unreachable("Missed a reference kind");
5168 }
5169 }
5170
5171 if (clang_isExpression(C.kind))
5172 return cxloc::translateSourceLocation(getCursorContext(C),
5173 getLocationFromExpr(getCursorExpr(C)));
5174
5175 if (clang_isStatement(C.kind))
5176 return cxloc::translateSourceLocation(getCursorContext(C),
5177 getCursorStmt(C)->getLocStart());
5178
5179 if (C.kind == CXCursor_PreprocessingDirective) {
5180 SourceLocation L = cxcursor::getCursorPreprocessingDirective(C).getBegin();
5181 return cxloc::translateSourceLocation(getCursorContext(C), L);
5182 }
5183
5184 if (C.kind == CXCursor_MacroExpansion) {
5185 SourceLocation L
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00005186 = cxcursor::getCursorMacroExpansion(C).getSourceRange().getBegin();
Guy Benyei11169dd2012-12-18 14:30:41 +00005187 return cxloc::translateSourceLocation(getCursorContext(C), L);
5188 }
5189
5190 if (C.kind == CXCursor_MacroDefinition) {
5191 SourceLocation L = cxcursor::getCursorMacroDefinition(C)->getLocation();
5192 return cxloc::translateSourceLocation(getCursorContext(C), L);
5193 }
5194
5195 if (C.kind == CXCursor_InclusionDirective) {
5196 SourceLocation L
5197 = cxcursor::getCursorInclusionDirective(C)->getSourceRange().getBegin();
5198 return cxloc::translateSourceLocation(getCursorContext(C), L);
5199 }
5200
Argyrios Kyrtzidis16834f12013-09-25 00:14:38 +00005201 if (clang_isAttribute(C.kind)) {
5202 SourceLocation L
5203 = cxcursor::getCursorAttr(C)->getLocation();
5204 return cxloc::translateSourceLocation(getCursorContext(C), L);
5205 }
5206
Guy Benyei11169dd2012-12-18 14:30:41 +00005207 if (!clang_isDeclaration(C.kind))
5208 return clang_getNullLocation();
5209
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005210 const Decl *D = getCursorDecl(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005211 if (!D)
5212 return clang_getNullLocation();
5213
5214 SourceLocation Loc = D->getLocation();
5215 // FIXME: Multiple variables declared in a single declaration
5216 // currently lack the information needed to correctly determine their
5217 // ranges when accounting for the type-specifier. We use context
5218 // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
5219 // and if so, whether it is the first decl.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005220 if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005221 if (!cxcursor::isFirstInDeclGroup(C))
5222 Loc = VD->getLocation();
5223 }
5224
5225 // For ObjC methods, give the start location of the method name.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005226 if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00005227 Loc = MD->getSelectorStartLoc();
5228
5229 return cxloc::translateSourceLocation(getCursorContext(C), Loc);
5230}
5231
5232} // end extern "C"
5233
5234CXCursor cxcursor::getCursor(CXTranslationUnit TU, SourceLocation SLoc) {
5235 assert(TU);
5236
5237 // Guard against an invalid SourceLocation, or we may assert in one
5238 // of the following calls.
5239 if (SLoc.isInvalid())
5240 return clang_getNullCursor();
5241
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00005242 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00005243
5244 // Translate the given source location to make it point at the beginning of
5245 // the token under the cursor.
5246 SLoc = Lexer::GetBeginningOfToken(SLoc, CXXUnit->getSourceManager(),
5247 CXXUnit->getASTContext().getLangOpts());
5248
5249 CXCursor Result = MakeCXCursorInvalid(CXCursor_NoDeclFound);
5250 if (SLoc.isValid()) {
5251 GetCursorData ResultData(CXXUnit->getSourceManager(), SLoc, Result);
5252 CursorVisitor CursorVis(TU, GetCursorVisitor, &ResultData,
5253 /*VisitPreprocessorLast=*/true,
5254 /*VisitIncludedEntities=*/false,
5255 SourceLocation(SLoc));
5256 CursorVis.visitFileRegion();
5257 }
5258
5259 return Result;
5260}
5261
5262static SourceRange getRawCursorExtent(CXCursor C) {
5263 if (clang_isReference(C.kind)) {
5264 switch (C.kind) {
5265 case CXCursor_ObjCSuperClassRef:
5266 return getCursorObjCSuperClassRef(C).second;
5267
5268 case CXCursor_ObjCProtocolRef:
5269 return getCursorObjCProtocolRef(C).second;
5270
5271 case CXCursor_ObjCClassRef:
5272 return getCursorObjCClassRef(C).second;
5273
5274 case CXCursor_TypeRef:
5275 return getCursorTypeRef(C).second;
5276
5277 case CXCursor_TemplateRef:
5278 return getCursorTemplateRef(C).second;
5279
5280 case CXCursor_NamespaceRef:
5281 return getCursorNamespaceRef(C).second;
5282
5283 case CXCursor_MemberRef:
5284 return getCursorMemberRef(C).second;
5285
5286 case CXCursor_CXXBaseSpecifier:
5287 return getCursorCXXBaseSpecifier(C)->getSourceRange();
5288
5289 case CXCursor_LabelRef:
5290 return getCursorLabelRef(C).second;
5291
5292 case CXCursor_OverloadedDeclRef:
5293 return getCursorOverloadedDeclRef(C).second;
5294
5295 case CXCursor_VariableRef:
5296 return getCursorVariableRef(C).second;
5297
5298 default:
5299 // FIXME: Need a way to enumerate all non-reference cases.
5300 llvm_unreachable("Missed a reference kind");
5301 }
5302 }
5303
5304 if (clang_isExpression(C.kind))
5305 return getCursorExpr(C)->getSourceRange();
5306
5307 if (clang_isStatement(C.kind))
5308 return getCursorStmt(C)->getSourceRange();
5309
5310 if (clang_isAttribute(C.kind))
5311 return getCursorAttr(C)->getRange();
5312
5313 if (C.kind == CXCursor_PreprocessingDirective)
5314 return cxcursor::getCursorPreprocessingDirective(C);
5315
5316 if (C.kind == CXCursor_MacroExpansion) {
5317 ASTUnit *TU = getCursorASTUnit(C);
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00005318 SourceRange Range = cxcursor::getCursorMacroExpansion(C).getSourceRange();
Guy Benyei11169dd2012-12-18 14:30:41 +00005319 return TU->mapRangeFromPreamble(Range);
5320 }
5321
5322 if (C.kind == CXCursor_MacroDefinition) {
5323 ASTUnit *TU = getCursorASTUnit(C);
5324 SourceRange Range = cxcursor::getCursorMacroDefinition(C)->getSourceRange();
5325 return TU->mapRangeFromPreamble(Range);
5326 }
5327
5328 if (C.kind == CXCursor_InclusionDirective) {
5329 ASTUnit *TU = getCursorASTUnit(C);
5330 SourceRange Range = cxcursor::getCursorInclusionDirective(C)->getSourceRange();
5331 return TU->mapRangeFromPreamble(Range);
5332 }
5333
5334 if (C.kind == CXCursor_TranslationUnit) {
5335 ASTUnit *TU = getCursorASTUnit(C);
5336 FileID MainID = TU->getSourceManager().getMainFileID();
5337 SourceLocation Start = TU->getSourceManager().getLocForStartOfFile(MainID);
5338 SourceLocation End = TU->getSourceManager().getLocForEndOfFile(MainID);
5339 return SourceRange(Start, End);
5340 }
5341
5342 if (clang_isDeclaration(C.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005343 const Decl *D = cxcursor::getCursorDecl(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005344 if (!D)
5345 return SourceRange();
5346
5347 SourceRange R = D->getSourceRange();
5348 // FIXME: Multiple variables declared in a single declaration
5349 // currently lack the information needed to correctly determine their
5350 // ranges when accounting for the type-specifier. We use context
5351 // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
5352 // and if so, whether it is the first decl.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005353 if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005354 if (!cxcursor::isFirstInDeclGroup(C))
5355 R.setBegin(VD->getLocation());
5356 }
5357 return R;
5358 }
5359 return SourceRange();
5360}
5361
5362/// \brief Retrieves the "raw" cursor extent, which is then extended to include
5363/// the decl-specifier-seq for declarations.
5364static SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr) {
5365 if (clang_isDeclaration(C.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005366 const Decl *D = cxcursor::getCursorDecl(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005367 if (!D)
5368 return SourceRange();
5369
5370 SourceRange R = D->getSourceRange();
5371
5372 // Adjust the start of the location for declarations preceded by
5373 // declaration specifiers.
5374 SourceLocation StartLoc;
5375 if (const DeclaratorDecl *DD = dyn_cast<DeclaratorDecl>(D)) {
5376 if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
5377 StartLoc = TI->getTypeLoc().getLocStart();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005378 } else if (const TypedefDecl *Typedef = dyn_cast<TypedefDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005379 if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
5380 StartLoc = TI->getTypeLoc().getLocStart();
5381 }
5382
5383 if (StartLoc.isValid() && R.getBegin().isValid() &&
5384 SrcMgr.isBeforeInTranslationUnit(StartLoc, R.getBegin()))
5385 R.setBegin(StartLoc);
5386
5387 // FIXME: Multiple variables declared in a single declaration
5388 // currently lack the information needed to correctly determine their
5389 // ranges when accounting for the type-specifier. We use context
5390 // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
5391 // and if so, whether it is the first decl.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005392 if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005393 if (!cxcursor::isFirstInDeclGroup(C))
5394 R.setBegin(VD->getLocation());
5395 }
5396
5397 return R;
5398 }
5399
5400 return getRawCursorExtent(C);
5401}
5402
5403extern "C" {
5404
5405CXSourceRange clang_getCursorExtent(CXCursor C) {
5406 SourceRange R = getRawCursorExtent(C);
5407 if (R.isInvalid())
5408 return clang_getNullRange();
5409
5410 return cxloc::translateSourceRange(getCursorContext(C), R);
5411}
5412
5413CXCursor clang_getCursorReferenced(CXCursor C) {
5414 if (clang_isInvalid(C.kind))
5415 return clang_getNullCursor();
5416
5417 CXTranslationUnit tu = getCursorTU(C);
5418 if (clang_isDeclaration(C.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005419 const Decl *D = getCursorDecl(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005420 if (!D)
5421 return clang_getNullCursor();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005422 if (const UsingDecl *Using = dyn_cast<UsingDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00005423 return MakeCursorOverloadedDeclRef(Using, D->getLocation(), tu);
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005424 if (const ObjCPropertyImplDecl *PropImpl =
5425 dyn_cast<ObjCPropertyImplDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00005426 if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
5427 return MakeCXCursor(Property, tu);
5428
5429 return C;
5430 }
5431
5432 if (clang_isExpression(C.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005433 const Expr *E = getCursorExpr(C);
5434 const Decl *D = getDeclFromExpr(E);
Guy Benyei11169dd2012-12-18 14:30:41 +00005435 if (D) {
5436 CXCursor declCursor = MakeCXCursor(D, tu);
5437 declCursor = getSelectorIdentifierCursor(getSelectorIdentifierIndex(C),
5438 declCursor);
5439 return declCursor;
5440 }
5441
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005442 if (const OverloadExpr *Ovl = dyn_cast_or_null<OverloadExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00005443 return MakeCursorOverloadedDeclRef(Ovl, tu);
5444
5445 return clang_getNullCursor();
5446 }
5447
5448 if (clang_isStatement(C.kind)) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00005449 const Stmt *S = getCursorStmt(C);
5450 if (const GotoStmt *Goto = dyn_cast_or_null<GotoStmt>(S))
Guy Benyei11169dd2012-12-18 14:30:41 +00005451 if (LabelDecl *label = Goto->getLabel())
5452 if (LabelStmt *labelS = label->getStmt())
5453 return MakeCXCursor(labelS, getCursorDecl(C), tu);
5454
5455 return clang_getNullCursor();
5456 }
Richard Smith66a81862015-05-04 02:25:31 +00005457
Guy Benyei11169dd2012-12-18 14:30:41 +00005458 if (C.kind == CXCursor_MacroExpansion) {
Richard Smith66a81862015-05-04 02:25:31 +00005459 if (const MacroDefinitionRecord *Def =
5460 getCursorMacroExpansion(C).getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00005461 return MakeMacroDefinitionCursor(Def, tu);
5462 }
5463
5464 if (!clang_isReference(C.kind))
5465 return clang_getNullCursor();
5466
5467 switch (C.kind) {
5468 case CXCursor_ObjCSuperClassRef:
5469 return MakeCXCursor(getCursorObjCSuperClassRef(C).first, tu);
5470
5471 case CXCursor_ObjCProtocolRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005472 const ObjCProtocolDecl *Prot = getCursorObjCProtocolRef(C).first;
5473 if (const ObjCProtocolDecl *Def = Prot->getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00005474 return MakeCXCursor(Def, tu);
5475
5476 return MakeCXCursor(Prot, tu);
5477 }
5478
5479 case CXCursor_ObjCClassRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005480 const ObjCInterfaceDecl *Class = getCursorObjCClassRef(C).first;
5481 if (const ObjCInterfaceDecl *Def = Class->getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00005482 return MakeCXCursor(Def, tu);
5483
5484 return MakeCXCursor(Class, tu);
5485 }
5486
5487 case CXCursor_TypeRef:
5488 return MakeCXCursor(getCursorTypeRef(C).first, tu );
5489
5490 case CXCursor_TemplateRef:
5491 return MakeCXCursor(getCursorTemplateRef(C).first, tu );
5492
5493 case CXCursor_NamespaceRef:
5494 return MakeCXCursor(getCursorNamespaceRef(C).first, tu );
5495
5496 case CXCursor_MemberRef:
5497 return MakeCXCursor(getCursorMemberRef(C).first, tu );
5498
5499 case CXCursor_CXXBaseSpecifier: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005500 const CXXBaseSpecifier *B = cxcursor::getCursorCXXBaseSpecifier(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005501 return clang_getTypeDeclaration(cxtype::MakeCXType(B->getType(),
5502 tu ));
5503 }
5504
5505 case CXCursor_LabelRef:
5506 // FIXME: We end up faking the "parent" declaration here because we
5507 // don't want to make CXCursor larger.
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00005508 return MakeCXCursor(getCursorLabelRef(C).first,
5509 cxtu::getASTUnit(tu)->getASTContext()
5510 .getTranslationUnitDecl(),
Guy Benyei11169dd2012-12-18 14:30:41 +00005511 tu);
5512
5513 case CXCursor_OverloadedDeclRef:
5514 return C;
5515
5516 case CXCursor_VariableRef:
5517 return MakeCXCursor(getCursorVariableRef(C).first, tu);
5518
5519 default:
5520 // We would prefer to enumerate all non-reference cursor kinds here.
5521 llvm_unreachable("Unhandled reference cursor kind");
5522 }
5523}
5524
5525CXCursor clang_getCursorDefinition(CXCursor C) {
5526 if (clang_isInvalid(C.kind))
5527 return clang_getNullCursor();
5528
5529 CXTranslationUnit TU = getCursorTU(C);
5530
5531 bool WasReference = false;
5532 if (clang_isReference(C.kind) || clang_isExpression(C.kind)) {
5533 C = clang_getCursorReferenced(C);
5534 WasReference = true;
5535 }
5536
5537 if (C.kind == CXCursor_MacroExpansion)
5538 return clang_getCursorReferenced(C);
5539
5540 if (!clang_isDeclaration(C.kind))
5541 return clang_getNullCursor();
5542
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005543 const Decl *D = getCursorDecl(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005544 if (!D)
5545 return clang_getNullCursor();
5546
5547 switch (D->getKind()) {
5548 // Declaration kinds that don't really separate the notions of
5549 // declaration and definition.
5550 case Decl::Namespace:
5551 case Decl::Typedef:
5552 case Decl::TypeAlias:
5553 case Decl::TypeAliasTemplate:
5554 case Decl::TemplateTypeParm:
5555 case Decl::EnumConstant:
5556 case Decl::Field:
John McCall5e77d762013-04-16 07:28:30 +00005557 case Decl::MSProperty:
Guy Benyei11169dd2012-12-18 14:30:41 +00005558 case Decl::IndirectField:
5559 case Decl::ObjCIvar:
5560 case Decl::ObjCAtDefsField:
5561 case Decl::ImplicitParam:
5562 case Decl::ParmVar:
5563 case Decl::NonTypeTemplateParm:
5564 case Decl::TemplateTemplateParm:
5565 case Decl::ObjCCategoryImpl:
5566 case Decl::ObjCImplementation:
5567 case Decl::AccessSpec:
5568 case Decl::LinkageSpec:
5569 case Decl::ObjCPropertyImpl:
5570 case Decl::FileScopeAsm:
5571 case Decl::StaticAssert:
5572 case Decl::Block:
Tareq A. Siraj6dfa25a2013-04-16 19:37:38 +00005573 case Decl::Captured:
Alexey Bataev4244be22016-02-11 05:35:55 +00005574 case Decl::OMPCapturedExpr:
Guy Benyei11169dd2012-12-18 14:30:41 +00005575 case Decl::Label: // FIXME: Is this right??
5576 case Decl::ClassScopeFunctionSpecialization:
5577 case Decl::Import:
Alexey Bataeva769e072013-03-22 06:34:35 +00005578 case Decl::OMPThreadPrivate:
Alexey Bataev94a4f0c2016-03-03 05:21:39 +00005579 case Decl::OMPDeclareReduction:
Douglas Gregor85f3f952015-07-07 03:57:15 +00005580 case Decl::ObjCTypeParam:
David Majnemerd9b1a4f2015-11-04 03:40:30 +00005581 case Decl::BuiltinTemplate:
Nico Weber66220292016-03-02 17:28:48 +00005582 case Decl::PragmaComment:
Nico Webercbbaeb12016-03-02 19:28:54 +00005583 case Decl::PragmaDetectMismatch:
Guy Benyei11169dd2012-12-18 14:30:41 +00005584 return C;
5585
5586 // Declaration kinds that don't make any sense here, but are
5587 // nonetheless harmless.
David Blaikief005d3c2013-02-22 17:44:58 +00005588 case Decl::Empty:
Guy Benyei11169dd2012-12-18 14:30:41 +00005589 case Decl::TranslationUnit:
Richard Smithf19e1272015-03-07 00:04:49 +00005590 case Decl::ExternCContext:
Guy Benyei11169dd2012-12-18 14:30:41 +00005591 break;
5592
5593 // Declaration kinds for which the definition is not resolvable.
5594 case Decl::UnresolvedUsingTypename:
5595 case Decl::UnresolvedUsingValue:
5596 break;
5597
5598 case Decl::UsingDirective:
5599 return MakeCXCursor(cast<UsingDirectiveDecl>(D)->getNominatedNamespace(),
5600 TU);
5601
5602 case Decl::NamespaceAlias:
5603 return MakeCXCursor(cast<NamespaceAliasDecl>(D)->getNamespace(), TU);
5604
5605 case Decl::Enum:
5606 case Decl::Record:
5607 case Decl::CXXRecord:
5608 case Decl::ClassTemplateSpecialization:
5609 case Decl::ClassTemplatePartialSpecialization:
5610 if (TagDecl *Def = cast<TagDecl>(D)->getDefinition())
5611 return MakeCXCursor(Def, TU);
5612 return clang_getNullCursor();
5613
5614 case Decl::Function:
5615 case Decl::CXXMethod:
5616 case Decl::CXXConstructor:
5617 case Decl::CXXDestructor:
5618 case Decl::CXXConversion: {
Craig Topper69186e72014-06-08 08:38:04 +00005619 const FunctionDecl *Def = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00005620 if (cast<FunctionDecl>(D)->getBody(Def))
Dmitri Gribenko9c256e32013-01-14 00:46:27 +00005621 return MakeCXCursor(Def, TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00005622 return clang_getNullCursor();
5623 }
5624
Larisse Voufo39a1e502013-08-06 01:03:05 +00005625 case Decl::Var:
5626 case Decl::VarTemplateSpecialization:
5627 case Decl::VarTemplatePartialSpecialization: {
Guy Benyei11169dd2012-12-18 14:30:41 +00005628 // Ask the variable if it has a definition.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005629 if (const VarDecl *Def = cast<VarDecl>(D)->getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00005630 return MakeCXCursor(Def, TU);
5631 return clang_getNullCursor();
5632 }
5633
5634 case Decl::FunctionTemplate: {
Craig Topper69186e72014-06-08 08:38:04 +00005635 const FunctionDecl *Def = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00005636 if (cast<FunctionTemplateDecl>(D)->getTemplatedDecl()->getBody(Def))
5637 return MakeCXCursor(Def->getDescribedFunctionTemplate(), TU);
5638 return clang_getNullCursor();
5639 }
5640
5641 case Decl::ClassTemplate: {
5642 if (RecordDecl *Def = cast<ClassTemplateDecl>(D)->getTemplatedDecl()
5643 ->getDefinition())
5644 return MakeCXCursor(cast<CXXRecordDecl>(Def)->getDescribedClassTemplate(),
5645 TU);
5646 return clang_getNullCursor();
5647 }
5648
Larisse Voufo39a1e502013-08-06 01:03:05 +00005649 case Decl::VarTemplate: {
5650 if (VarDecl *Def =
5651 cast<VarTemplateDecl>(D)->getTemplatedDecl()->getDefinition())
5652 return MakeCXCursor(cast<VarDecl>(Def)->getDescribedVarTemplate(), TU);
5653 return clang_getNullCursor();
5654 }
5655
Guy Benyei11169dd2012-12-18 14:30:41 +00005656 case Decl::Using:
5657 return MakeCursorOverloadedDeclRef(cast<UsingDecl>(D),
5658 D->getLocation(), TU);
5659
5660 case Decl::UsingShadow:
5661 return clang_getCursorDefinition(
5662 MakeCXCursor(cast<UsingShadowDecl>(D)->getTargetDecl(),
5663 TU));
5664
5665 case Decl::ObjCMethod: {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005666 const ObjCMethodDecl *Method = cast<ObjCMethodDecl>(D);
Guy Benyei11169dd2012-12-18 14:30:41 +00005667 if (Method->isThisDeclarationADefinition())
5668 return C;
5669
5670 // Dig out the method definition in the associated
5671 // @implementation, if we have it.
5672 // FIXME: The ASTs should make finding the definition easier.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005673 if (const ObjCInterfaceDecl *Class
Guy Benyei11169dd2012-12-18 14:30:41 +00005674 = dyn_cast<ObjCInterfaceDecl>(Method->getDeclContext()))
5675 if (ObjCImplementationDecl *ClassImpl = Class->getImplementation())
5676 if (ObjCMethodDecl *Def = ClassImpl->getMethod(Method->getSelector(),
5677 Method->isInstanceMethod()))
5678 if (Def->isThisDeclarationADefinition())
5679 return MakeCXCursor(Def, TU);
5680
5681 return clang_getNullCursor();
5682 }
5683
5684 case Decl::ObjCCategory:
5685 if (ObjCCategoryImplDecl *Impl
5686 = cast<ObjCCategoryDecl>(D)->getImplementation())
5687 return MakeCXCursor(Impl, TU);
5688 return clang_getNullCursor();
5689
5690 case Decl::ObjCProtocol:
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005691 if (const ObjCProtocolDecl *Def = cast<ObjCProtocolDecl>(D)->getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00005692 return MakeCXCursor(Def, TU);
5693 return clang_getNullCursor();
5694
5695 case Decl::ObjCInterface: {
5696 // There are two notions of a "definition" for an Objective-C
5697 // class: the interface and its implementation. When we resolved a
5698 // reference to an Objective-C class, produce the @interface as
5699 // the definition; when we were provided with the interface,
5700 // produce the @implementation as the definition.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005701 const ObjCInterfaceDecl *IFace = cast<ObjCInterfaceDecl>(D);
Guy Benyei11169dd2012-12-18 14:30:41 +00005702 if (WasReference) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005703 if (const ObjCInterfaceDecl *Def = IFace->getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00005704 return MakeCXCursor(Def, TU);
5705 } else if (ObjCImplementationDecl *Impl = IFace->getImplementation())
5706 return MakeCXCursor(Impl, TU);
5707 return clang_getNullCursor();
5708 }
5709
5710 case Decl::ObjCProperty:
5711 // FIXME: We don't really know where to find the
5712 // ObjCPropertyImplDecls that implement this property.
5713 return clang_getNullCursor();
5714
5715 case Decl::ObjCCompatibleAlias:
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005716 if (const ObjCInterfaceDecl *Class
Guy Benyei11169dd2012-12-18 14:30:41 +00005717 = cast<ObjCCompatibleAliasDecl>(D)->getClassInterface())
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005718 if (const ObjCInterfaceDecl *Def = Class->getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00005719 return MakeCXCursor(Def, TU);
5720
5721 return clang_getNullCursor();
5722
5723 case Decl::Friend:
5724 if (NamedDecl *Friend = cast<FriendDecl>(D)->getFriendDecl())
5725 return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
5726 return clang_getNullCursor();
5727
5728 case Decl::FriendTemplate:
5729 if (NamedDecl *Friend = cast<FriendTemplateDecl>(D)->getFriendDecl())
5730 return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
5731 return clang_getNullCursor();
5732 }
5733
5734 return clang_getNullCursor();
5735}
5736
5737unsigned clang_isCursorDefinition(CXCursor C) {
5738 if (!clang_isDeclaration(C.kind))
5739 return 0;
5740
5741 return clang_getCursorDefinition(C) == C;
5742}
5743
5744CXCursor clang_getCanonicalCursor(CXCursor C) {
5745 if (!clang_isDeclaration(C.kind))
5746 return C;
5747
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005748 if (const Decl *D = getCursorDecl(C)) {
5749 if (const ObjCCategoryImplDecl *CatImplD = dyn_cast<ObjCCategoryImplDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00005750 if (ObjCCategoryDecl *CatD = CatImplD->getCategoryDecl())
5751 return MakeCXCursor(CatD, getCursorTU(C));
5752
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005753 if (const ObjCImplDecl *ImplD = dyn_cast<ObjCImplDecl>(D))
5754 if (const ObjCInterfaceDecl *IFD = ImplD->getClassInterface())
Guy Benyei11169dd2012-12-18 14:30:41 +00005755 return MakeCXCursor(IFD, getCursorTU(C));
5756
5757 return MakeCXCursor(D->getCanonicalDecl(), getCursorTU(C));
5758 }
5759
5760 return C;
5761}
5762
5763int clang_Cursor_getObjCSelectorIndex(CXCursor cursor) {
5764 return cxcursor::getSelectorIdentifierIndexAndLoc(cursor).first;
5765}
5766
5767unsigned clang_getNumOverloadedDecls(CXCursor C) {
5768 if (C.kind != CXCursor_OverloadedDeclRef)
5769 return 0;
5770
5771 OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005772 if (const OverloadExpr *E = Storage.dyn_cast<const OverloadExpr *>())
Guy Benyei11169dd2012-12-18 14:30:41 +00005773 return E->getNumDecls();
5774
5775 if (OverloadedTemplateStorage *S
5776 = Storage.dyn_cast<OverloadedTemplateStorage*>())
5777 return S->size();
5778
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005779 const Decl *D = Storage.get<const Decl *>();
5780 if (const UsingDecl *Using = dyn_cast<UsingDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00005781 return Using->shadow_size();
5782
5783 return 0;
5784}
5785
5786CXCursor clang_getOverloadedDecl(CXCursor cursor, unsigned index) {
5787 if (cursor.kind != CXCursor_OverloadedDeclRef)
5788 return clang_getNullCursor();
5789
5790 if (index >= clang_getNumOverloadedDecls(cursor))
5791 return clang_getNullCursor();
5792
5793 CXTranslationUnit TU = getCursorTU(cursor);
5794 OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(cursor).first;
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005795 if (const OverloadExpr *E = Storage.dyn_cast<const OverloadExpr *>())
Guy Benyei11169dd2012-12-18 14:30:41 +00005796 return MakeCXCursor(E->decls_begin()[index], TU);
5797
5798 if (OverloadedTemplateStorage *S
5799 = Storage.dyn_cast<OverloadedTemplateStorage*>())
5800 return MakeCXCursor(S->begin()[index], TU);
5801
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005802 const Decl *D = Storage.get<const Decl *>();
5803 if (const UsingDecl *Using = dyn_cast<UsingDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005804 // FIXME: This is, unfortunately, linear time.
5805 UsingDecl::shadow_iterator Pos = Using->shadow_begin();
5806 std::advance(Pos, index);
5807 return MakeCXCursor(cast<UsingShadowDecl>(*Pos)->getTargetDecl(), TU);
5808 }
5809
5810 return clang_getNullCursor();
5811}
5812
5813void clang_getDefinitionSpellingAndExtent(CXCursor C,
5814 const char **startBuf,
5815 const char **endBuf,
5816 unsigned *startLine,
5817 unsigned *startColumn,
5818 unsigned *endLine,
5819 unsigned *endColumn) {
5820 assert(getCursorDecl(C) && "CXCursor has null decl");
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005821 const FunctionDecl *FD = dyn_cast<FunctionDecl>(getCursorDecl(C));
Guy Benyei11169dd2012-12-18 14:30:41 +00005822 CompoundStmt *Body = dyn_cast<CompoundStmt>(FD->getBody());
5823
5824 SourceManager &SM = FD->getASTContext().getSourceManager();
5825 *startBuf = SM.getCharacterData(Body->getLBracLoc());
5826 *endBuf = SM.getCharacterData(Body->getRBracLoc());
5827 *startLine = SM.getSpellingLineNumber(Body->getLBracLoc());
5828 *startColumn = SM.getSpellingColumnNumber(Body->getLBracLoc());
5829 *endLine = SM.getSpellingLineNumber(Body->getRBracLoc());
5830 *endColumn = SM.getSpellingColumnNumber(Body->getRBracLoc());
5831}
5832
5833
5834CXSourceRange clang_getCursorReferenceNameRange(CXCursor C, unsigned NameFlags,
5835 unsigned PieceIndex) {
5836 RefNamePieces Pieces;
5837
5838 switch (C.kind) {
5839 case CXCursor_MemberRefExpr:
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00005840 if (const MemberExpr *E = dyn_cast<MemberExpr>(getCursorExpr(C)))
Guy Benyei11169dd2012-12-18 14:30:41 +00005841 Pieces = buildPieces(NameFlags, true, E->getMemberNameInfo(),
5842 E->getQualifierLoc().getSourceRange());
5843 break;
5844
5845 case CXCursor_DeclRefExpr:
James Y Knight04ec5bf2015-12-24 02:59:37 +00005846 if (const DeclRefExpr *E = dyn_cast<DeclRefExpr>(getCursorExpr(C))) {
5847 SourceRange TemplateArgLoc(E->getLAngleLoc(), E->getRAngleLoc());
5848 Pieces =
5849 buildPieces(NameFlags, false, E->getNameInfo(),
5850 E->getQualifierLoc().getSourceRange(), &TemplateArgLoc);
5851 }
Guy Benyei11169dd2012-12-18 14:30:41 +00005852 break;
5853
5854 case CXCursor_CallExpr:
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00005855 if (const CXXOperatorCallExpr *OCE =
Guy Benyei11169dd2012-12-18 14:30:41 +00005856 dyn_cast<CXXOperatorCallExpr>(getCursorExpr(C))) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00005857 const Expr *Callee = OCE->getCallee();
5858 if (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Callee))
Guy Benyei11169dd2012-12-18 14:30:41 +00005859 Callee = ICE->getSubExpr();
5860
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00005861 if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Callee))
Guy Benyei11169dd2012-12-18 14:30:41 +00005862 Pieces = buildPieces(NameFlags, false, DRE->getNameInfo(),
5863 DRE->getQualifierLoc().getSourceRange());
5864 }
5865 break;
5866
5867 default:
5868 break;
5869 }
5870
5871 if (Pieces.empty()) {
5872 if (PieceIndex == 0)
5873 return clang_getCursorExtent(C);
5874 } else if (PieceIndex < Pieces.size()) {
5875 SourceRange R = Pieces[PieceIndex];
5876 if (R.isValid())
5877 return cxloc::translateSourceRange(getCursorContext(C), R);
5878 }
5879
5880 return clang_getNullRange();
5881}
5882
5883void clang_enableStackTraces(void) {
5884 llvm::sys::PrintStackTraceOnErrorSignal();
5885}
5886
5887void clang_executeOnThread(void (*fn)(void*), void *user_data,
5888 unsigned stack_size) {
5889 llvm::llvm_execute_on_thread(fn, user_data, stack_size);
5890}
5891
5892} // end: extern "C"
5893
5894//===----------------------------------------------------------------------===//
5895// Token-based Operations.
5896//===----------------------------------------------------------------------===//
5897
5898/* CXToken layout:
5899 * int_data[0]: a CXTokenKind
5900 * int_data[1]: starting token location
5901 * int_data[2]: token length
5902 * int_data[3]: reserved
5903 * ptr_data: for identifiers and keywords, an IdentifierInfo*.
5904 * otherwise unused.
5905 */
5906extern "C" {
5907
5908CXTokenKind clang_getTokenKind(CXToken CXTok) {
5909 return static_cast<CXTokenKind>(CXTok.int_data[0]);
5910}
5911
5912CXString clang_getTokenSpelling(CXTranslationUnit TU, CXToken CXTok) {
5913 switch (clang_getTokenKind(CXTok)) {
5914 case CXToken_Identifier:
5915 case CXToken_Keyword:
5916 // We know we have an IdentifierInfo*, so use that.
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005917 return cxstring::createRef(static_cast<IdentifierInfo *>(CXTok.ptr_data)
Guy Benyei11169dd2012-12-18 14:30:41 +00005918 ->getNameStart());
5919
5920 case CXToken_Literal: {
5921 // We have stashed the starting pointer in the ptr_data field. Use it.
5922 const char *Text = static_cast<const char *>(CXTok.ptr_data);
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00005923 return cxstring::createDup(StringRef(Text, CXTok.int_data[2]));
Guy Benyei11169dd2012-12-18 14:30:41 +00005924 }
5925
5926 case CXToken_Punctuation:
5927 case CXToken_Comment:
5928 break;
5929 }
5930
Dmitri Gribenko852d6222014-02-11 15:02:48 +00005931 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00005932 LOG_BAD_TU(TU);
5933 return cxstring::createEmpty();
5934 }
5935
Guy Benyei11169dd2012-12-18 14:30:41 +00005936 // We have to find the starting buffer pointer the hard way, by
5937 // deconstructing the source location.
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00005938 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00005939 if (!CXXUnit)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00005940 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00005941
5942 SourceLocation Loc = SourceLocation::getFromRawEncoding(CXTok.int_data[1]);
5943 std::pair<FileID, unsigned> LocInfo
5944 = CXXUnit->getSourceManager().getDecomposedSpellingLoc(Loc);
5945 bool Invalid = false;
5946 StringRef Buffer
5947 = CXXUnit->getSourceManager().getBufferData(LocInfo.first, &Invalid);
5948 if (Invalid)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00005949 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00005950
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00005951 return cxstring::createDup(Buffer.substr(LocInfo.second, CXTok.int_data[2]));
Guy Benyei11169dd2012-12-18 14:30:41 +00005952}
5953
5954CXSourceLocation clang_getTokenLocation(CXTranslationUnit TU, CXToken CXTok) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00005955 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00005956 LOG_BAD_TU(TU);
5957 return clang_getNullLocation();
5958 }
5959
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00005960 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00005961 if (!CXXUnit)
5962 return clang_getNullLocation();
5963
5964 return cxloc::translateSourceLocation(CXXUnit->getASTContext(),
5965 SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
5966}
5967
5968CXSourceRange clang_getTokenExtent(CXTranslationUnit TU, CXToken CXTok) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00005969 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00005970 LOG_BAD_TU(TU);
5971 return clang_getNullRange();
5972 }
5973
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00005974 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00005975 if (!CXXUnit)
5976 return clang_getNullRange();
5977
5978 return cxloc::translateSourceRange(CXXUnit->getASTContext(),
5979 SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
5980}
5981
5982static void getTokens(ASTUnit *CXXUnit, SourceRange Range,
5983 SmallVectorImpl<CXToken> &CXTokens) {
5984 SourceManager &SourceMgr = CXXUnit->getSourceManager();
5985 std::pair<FileID, unsigned> BeginLocInfo
Argyrios Kyrtzidis5d47a9b2013-02-13 18:33:28 +00005986 = SourceMgr.getDecomposedSpellingLoc(Range.getBegin());
Guy Benyei11169dd2012-12-18 14:30:41 +00005987 std::pair<FileID, unsigned> EndLocInfo
Argyrios Kyrtzidis5d47a9b2013-02-13 18:33:28 +00005988 = SourceMgr.getDecomposedSpellingLoc(Range.getEnd());
Guy Benyei11169dd2012-12-18 14:30:41 +00005989
5990 // Cannot tokenize across files.
5991 if (BeginLocInfo.first != EndLocInfo.first)
5992 return;
5993
5994 // Create a lexer
5995 bool Invalid = false;
5996 StringRef Buffer
5997 = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
5998 if (Invalid)
5999 return;
6000
6001 Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
6002 CXXUnit->getASTContext().getLangOpts(),
6003 Buffer.begin(), Buffer.data() + BeginLocInfo.second, Buffer.end());
6004 Lex.SetCommentRetentionState(true);
6005
6006 // Lex tokens until we hit the end of the range.
6007 const char *EffectiveBufferEnd = Buffer.data() + EndLocInfo.second;
6008 Token Tok;
6009 bool previousWasAt = false;
6010 do {
6011 // Lex the next token
6012 Lex.LexFromRawLexer(Tok);
6013 if (Tok.is(tok::eof))
6014 break;
6015
6016 // Initialize the CXToken.
6017 CXToken CXTok;
6018
6019 // - Common fields
6020 CXTok.int_data[1] = Tok.getLocation().getRawEncoding();
6021 CXTok.int_data[2] = Tok.getLength();
6022 CXTok.int_data[3] = 0;
6023
6024 // - Kind-specific fields
6025 if (Tok.isLiteral()) {
6026 CXTok.int_data[0] = CXToken_Literal;
Dmitri Gribenkof9304482013-01-23 15:56:07 +00006027 CXTok.ptr_data = const_cast<char *>(Tok.getLiteralData());
Guy Benyei11169dd2012-12-18 14:30:41 +00006028 } else if (Tok.is(tok::raw_identifier)) {
6029 // Lookup the identifier to determine whether we have a keyword.
6030 IdentifierInfo *II
6031 = CXXUnit->getPreprocessor().LookUpIdentifierInfo(Tok);
6032
6033 if ((II->getObjCKeywordID() != tok::objc_not_keyword) && previousWasAt) {
6034 CXTok.int_data[0] = CXToken_Keyword;
6035 }
6036 else {
6037 CXTok.int_data[0] = Tok.is(tok::identifier)
6038 ? CXToken_Identifier
6039 : CXToken_Keyword;
6040 }
6041 CXTok.ptr_data = II;
6042 } else if (Tok.is(tok::comment)) {
6043 CXTok.int_data[0] = CXToken_Comment;
Craig Topper69186e72014-06-08 08:38:04 +00006044 CXTok.ptr_data = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00006045 } else {
6046 CXTok.int_data[0] = CXToken_Punctuation;
Craig Topper69186e72014-06-08 08:38:04 +00006047 CXTok.ptr_data = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00006048 }
6049 CXTokens.push_back(CXTok);
6050 previousWasAt = Tok.is(tok::at);
6051 } while (Lex.getBufferLocation() <= EffectiveBufferEnd);
6052}
6053
6054void clang_tokenize(CXTranslationUnit TU, CXSourceRange Range,
6055 CXToken **Tokens, unsigned *NumTokens) {
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00006056 LOG_FUNC_SECTION {
6057 *Log << TU << ' ' << Range;
6058 }
6059
Guy Benyei11169dd2012-12-18 14:30:41 +00006060 if (Tokens)
Craig Topper69186e72014-06-08 08:38:04 +00006061 *Tokens = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00006062 if (NumTokens)
6063 *NumTokens = 0;
6064
Dmitri Gribenko852d6222014-02-11 15:02:48 +00006065 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00006066 LOG_BAD_TU(TU);
Argyrios Kyrtzidis0e95fca2013-04-04 22:40:59 +00006067 return;
Dmitri Gribenko256454f2014-02-11 14:34:14 +00006068 }
Argyrios Kyrtzidis0e95fca2013-04-04 22:40:59 +00006069
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006070 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00006071 if (!CXXUnit || !Tokens || !NumTokens)
6072 return;
6073
6074 ASTUnit::ConcurrencyCheck Check(*CXXUnit);
6075
6076 SourceRange R = cxloc::translateCXSourceRange(Range);
6077 if (R.isInvalid())
6078 return;
6079
6080 SmallVector<CXToken, 32> CXTokens;
6081 getTokens(CXXUnit, R, CXTokens);
6082
6083 if (CXTokens.empty())
6084 return;
6085
6086 *Tokens = (CXToken *)malloc(sizeof(CXToken) * CXTokens.size());
6087 memmove(*Tokens, CXTokens.data(), sizeof(CXToken) * CXTokens.size());
6088 *NumTokens = CXTokens.size();
6089}
6090
6091void clang_disposeTokens(CXTranslationUnit TU,
6092 CXToken *Tokens, unsigned NumTokens) {
6093 free(Tokens);
6094}
6095
6096} // end: extern "C"
6097
6098//===----------------------------------------------------------------------===//
6099// Token annotation APIs.
6100//===----------------------------------------------------------------------===//
6101
Guy Benyei11169dd2012-12-18 14:30:41 +00006102static enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
6103 CXCursor parent,
6104 CXClientData client_data);
6105static bool AnnotateTokensPostChildrenVisitor(CXCursor cursor,
6106 CXClientData client_data);
6107
6108namespace {
6109class AnnotateTokensWorker {
Guy Benyei11169dd2012-12-18 14:30:41 +00006110 CXToken *Tokens;
6111 CXCursor *Cursors;
6112 unsigned NumTokens;
6113 unsigned TokIdx;
6114 unsigned PreprocessingTokIdx;
6115 CursorVisitor AnnotateVis;
6116 SourceManager &SrcMgr;
6117 bool HasContextSensitiveKeywords;
6118
6119 struct PostChildrenInfo {
6120 CXCursor Cursor;
6121 SourceRange CursorRange;
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006122 unsigned BeforeReachingCursorIdx;
Guy Benyei11169dd2012-12-18 14:30:41 +00006123 unsigned BeforeChildrenTokenIdx;
6124 };
Dmitri Gribenkof8579502013-01-12 19:30:44 +00006125 SmallVector<PostChildrenInfo, 8> PostChildrenInfos;
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006126
6127 CXToken &getTok(unsigned Idx) {
6128 assert(Idx < NumTokens);
6129 return Tokens[Idx];
6130 }
6131 const CXToken &getTok(unsigned Idx) const {
6132 assert(Idx < NumTokens);
6133 return Tokens[Idx];
6134 }
Guy Benyei11169dd2012-12-18 14:30:41 +00006135 bool MoreTokens() const { return TokIdx < NumTokens; }
6136 unsigned NextToken() const { return TokIdx; }
6137 void AdvanceToken() { ++TokIdx; }
6138 SourceLocation GetTokenLoc(unsigned tokI) {
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006139 return SourceLocation::getFromRawEncoding(getTok(tokI).int_data[1]);
Guy Benyei11169dd2012-12-18 14:30:41 +00006140 }
6141 bool isFunctionMacroToken(unsigned tokI) const {
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006142 return getTok(tokI).int_data[3] != 0;
Guy Benyei11169dd2012-12-18 14:30:41 +00006143 }
6144 SourceLocation getFunctionMacroTokenLoc(unsigned tokI) const {
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006145 return SourceLocation::getFromRawEncoding(getTok(tokI).int_data[3]);
Guy Benyei11169dd2012-12-18 14:30:41 +00006146 }
6147
6148 void annotateAndAdvanceTokens(CXCursor, RangeComparisonResult, SourceRange);
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006149 bool annotateAndAdvanceFunctionMacroTokens(CXCursor, RangeComparisonResult,
Guy Benyei11169dd2012-12-18 14:30:41 +00006150 SourceRange);
6151
6152public:
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006153 AnnotateTokensWorker(CXToken *tokens, CXCursor *cursors, unsigned numTokens,
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006154 CXTranslationUnit TU, SourceRange RegionOfInterest)
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006155 : Tokens(tokens), Cursors(cursors),
Guy Benyei11169dd2012-12-18 14:30:41 +00006156 NumTokens(numTokens), TokIdx(0), PreprocessingTokIdx(0),
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006157 AnnotateVis(TU,
Guy Benyei11169dd2012-12-18 14:30:41 +00006158 AnnotateTokensVisitor, this,
6159 /*VisitPreprocessorLast=*/true,
6160 /*VisitIncludedEntities=*/false,
6161 RegionOfInterest,
6162 /*VisitDeclsOnly=*/false,
6163 AnnotateTokensPostChildrenVisitor),
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006164 SrcMgr(cxtu::getASTUnit(TU)->getSourceManager()),
Guy Benyei11169dd2012-12-18 14:30:41 +00006165 HasContextSensitiveKeywords(false) { }
6166
6167 void VisitChildren(CXCursor C) { AnnotateVis.VisitChildren(C); }
6168 enum CXChildVisitResult Visit(CXCursor cursor, CXCursor parent);
6169 bool postVisitChildren(CXCursor cursor);
6170 void AnnotateTokens();
6171
6172 /// \brief Determine whether the annotator saw any cursors that have
6173 /// context-sensitive keywords.
6174 bool hasContextSensitiveKeywords() const {
6175 return HasContextSensitiveKeywords;
6176 }
6177
6178 ~AnnotateTokensWorker() {
6179 assert(PostChildrenInfos.empty());
6180 }
6181};
Alexander Kornienkoab9db512015-06-22 23:07:51 +00006182}
Guy Benyei11169dd2012-12-18 14:30:41 +00006183
6184void AnnotateTokensWorker::AnnotateTokens() {
6185 // Walk the AST within the region of interest, annotating tokens
6186 // along the way.
6187 AnnotateVis.visitFileRegion();
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006188}
Guy Benyei11169dd2012-12-18 14:30:41 +00006189
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006190static inline void updateCursorAnnotation(CXCursor &Cursor,
6191 const CXCursor &updateC) {
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006192 if (clang_isInvalid(updateC.kind) || !clang_isInvalid(Cursor.kind))
Guy Benyei11169dd2012-12-18 14:30:41 +00006193 return;
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006194 Cursor = updateC;
Guy Benyei11169dd2012-12-18 14:30:41 +00006195}
6196
6197/// \brief It annotates and advances tokens with a cursor until the comparison
6198//// between the cursor location and the source range is the same as
6199/// \arg compResult.
6200///
6201/// Pass RangeBefore to annotate tokens with a cursor until a range is reached.
6202/// Pass RangeOverlap to annotate tokens inside a range.
6203void AnnotateTokensWorker::annotateAndAdvanceTokens(CXCursor updateC,
6204 RangeComparisonResult compResult,
6205 SourceRange range) {
6206 while (MoreTokens()) {
6207 const unsigned I = NextToken();
6208 if (isFunctionMacroToken(I))
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006209 if (!annotateAndAdvanceFunctionMacroTokens(updateC, compResult, range))
6210 return;
Guy Benyei11169dd2012-12-18 14:30:41 +00006211
6212 SourceLocation TokLoc = GetTokenLoc(I);
6213 if (LocationCompare(SrcMgr, TokLoc, range) == compResult) {
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006214 updateCursorAnnotation(Cursors[I], updateC);
Guy Benyei11169dd2012-12-18 14:30:41 +00006215 AdvanceToken();
6216 continue;
6217 }
6218 break;
6219 }
6220}
6221
6222/// \brief Special annotation handling for macro argument tokens.
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006223/// \returns true if it advanced beyond all macro tokens, false otherwise.
6224bool AnnotateTokensWorker::annotateAndAdvanceFunctionMacroTokens(
Guy Benyei11169dd2012-12-18 14:30:41 +00006225 CXCursor updateC,
6226 RangeComparisonResult compResult,
6227 SourceRange range) {
6228 assert(MoreTokens());
6229 assert(isFunctionMacroToken(NextToken()) &&
6230 "Should be called only for macro arg tokens");
6231
6232 // This works differently than annotateAndAdvanceTokens; because expanded
6233 // macro arguments can have arbitrary translation-unit source order, we do not
6234 // advance the token index one by one until a token fails the range test.
6235 // We only advance once past all of the macro arg tokens if all of them
6236 // pass the range test. If one of them fails we keep the token index pointing
6237 // at the start of the macro arg tokens so that the failing token will be
6238 // annotated by a subsequent annotation try.
6239
6240 bool atLeastOneCompFail = false;
6241
6242 unsigned I = NextToken();
6243 for (; I < NumTokens && isFunctionMacroToken(I); ++I) {
6244 SourceLocation TokLoc = getFunctionMacroTokenLoc(I);
6245 if (TokLoc.isFileID())
6246 continue; // not macro arg token, it's parens or comma.
6247 if (LocationCompare(SrcMgr, TokLoc, range) == compResult) {
6248 if (clang_isInvalid(clang_getCursorKind(Cursors[I])))
6249 Cursors[I] = updateC;
6250 } else
6251 atLeastOneCompFail = true;
6252 }
6253
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006254 if (atLeastOneCompFail)
6255 return false;
6256
6257 TokIdx = I; // All of the tokens were handled, advance beyond all of them.
6258 return true;
Guy Benyei11169dd2012-12-18 14:30:41 +00006259}
6260
6261enum CXChildVisitResult
6262AnnotateTokensWorker::Visit(CXCursor cursor, CXCursor parent) {
Guy Benyei11169dd2012-12-18 14:30:41 +00006263 SourceRange cursorRange = getRawCursorExtent(cursor);
6264 if (cursorRange.isInvalid())
6265 return CXChildVisit_Recurse;
6266
6267 if (!HasContextSensitiveKeywords) {
6268 // Objective-C properties can have context-sensitive keywords.
6269 if (cursor.kind == CXCursor_ObjCPropertyDecl) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006270 if (const ObjCPropertyDecl *Property
Guy Benyei11169dd2012-12-18 14:30:41 +00006271 = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(cursor)))
6272 HasContextSensitiveKeywords = Property->getPropertyAttributesAsWritten() != 0;
6273 }
6274 // Objective-C methods can have context-sensitive keywords.
6275 else if (cursor.kind == CXCursor_ObjCInstanceMethodDecl ||
6276 cursor.kind == CXCursor_ObjCClassMethodDecl) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006277 if (const ObjCMethodDecl *Method
Guy Benyei11169dd2012-12-18 14:30:41 +00006278 = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(cursor))) {
6279 if (Method->getObjCDeclQualifier())
6280 HasContextSensitiveKeywords = true;
6281 else {
Aaron Ballman43b68be2014-03-07 17:50:17 +00006282 for (const auto *P : Method->params()) {
6283 if (P->getObjCDeclQualifier()) {
Guy Benyei11169dd2012-12-18 14:30:41 +00006284 HasContextSensitiveKeywords = true;
6285 break;
6286 }
6287 }
6288 }
6289 }
6290 }
6291 // C++ methods can have context-sensitive keywords.
6292 else if (cursor.kind == CXCursor_CXXMethod) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006293 if (const CXXMethodDecl *Method
Guy Benyei11169dd2012-12-18 14:30:41 +00006294 = dyn_cast_or_null<CXXMethodDecl>(getCursorDecl(cursor))) {
6295 if (Method->hasAttr<FinalAttr>() || Method->hasAttr<OverrideAttr>())
6296 HasContextSensitiveKeywords = true;
6297 }
6298 }
6299 // C++ classes can have context-sensitive keywords.
6300 else if (cursor.kind == CXCursor_StructDecl ||
6301 cursor.kind == CXCursor_ClassDecl ||
6302 cursor.kind == CXCursor_ClassTemplate ||
6303 cursor.kind == CXCursor_ClassTemplatePartialSpecialization) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006304 if (const Decl *D = getCursorDecl(cursor))
Guy Benyei11169dd2012-12-18 14:30:41 +00006305 if (D->hasAttr<FinalAttr>())
6306 HasContextSensitiveKeywords = true;
6307 }
6308 }
Argyrios Kyrtzidis990b3862013-06-04 18:24:30 +00006309
6310 // Don't override a property annotation with its getter/setter method.
6311 if (cursor.kind == CXCursor_ObjCInstanceMethodDecl &&
6312 parent.kind == CXCursor_ObjCPropertyDecl)
6313 return CXChildVisit_Continue;
Guy Benyei11169dd2012-12-18 14:30:41 +00006314
6315 if (clang_isPreprocessing(cursor.kind)) {
6316 // Items in the preprocessing record are kept separate from items in
6317 // declarations, so we keep a separate token index.
6318 unsigned SavedTokIdx = TokIdx;
6319 TokIdx = PreprocessingTokIdx;
6320
6321 // Skip tokens up until we catch up to the beginning of the preprocessing
6322 // entry.
6323 while (MoreTokens()) {
6324 const unsigned I = NextToken();
6325 SourceLocation TokLoc = GetTokenLoc(I);
6326 switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
6327 case RangeBefore:
6328 AdvanceToken();
6329 continue;
6330 case RangeAfter:
6331 case RangeOverlap:
6332 break;
6333 }
6334 break;
6335 }
6336
6337 // Look at all of the tokens within this range.
6338 while (MoreTokens()) {
6339 const unsigned I = NextToken();
6340 SourceLocation TokLoc = GetTokenLoc(I);
6341 switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
6342 case RangeBefore:
6343 llvm_unreachable("Infeasible");
6344 case RangeAfter:
6345 break;
6346 case RangeOverlap:
Argyrios Kyrtzidis5d47a9b2013-02-13 18:33:28 +00006347 // For macro expansions, just note where the beginning of the macro
6348 // expansion occurs.
6349 if (cursor.kind == CXCursor_MacroExpansion) {
6350 if (TokLoc == cursorRange.getBegin())
6351 Cursors[I] = cursor;
6352 AdvanceToken();
6353 break;
6354 }
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006355 // We may have already annotated macro names inside macro definitions.
6356 if (Cursors[I].kind != CXCursor_MacroExpansion)
6357 Cursors[I] = cursor;
Guy Benyei11169dd2012-12-18 14:30:41 +00006358 AdvanceToken();
Guy Benyei11169dd2012-12-18 14:30:41 +00006359 continue;
6360 }
6361 break;
6362 }
6363
6364 // Save the preprocessing token index; restore the non-preprocessing
6365 // token index.
6366 PreprocessingTokIdx = TokIdx;
6367 TokIdx = SavedTokIdx;
6368 return CXChildVisit_Recurse;
6369 }
6370
6371 if (cursorRange.isInvalid())
6372 return CXChildVisit_Continue;
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006373
6374 unsigned BeforeReachingCursorIdx = NextToken();
Guy Benyei11169dd2012-12-18 14:30:41 +00006375 const enum CXCursorKind cursorK = clang_getCursorKind(cursor);
Guy Benyei11169dd2012-12-18 14:30:41 +00006376 const enum CXCursorKind K = clang_getCursorKind(parent);
6377 const CXCursor updateC =
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006378 (clang_isInvalid(K) || K == CXCursor_TranslationUnit ||
6379 // Attributes are annotated out-of-order, skip tokens until we reach it.
6380 clang_isAttribute(cursor.kind))
Guy Benyei11169dd2012-12-18 14:30:41 +00006381 ? clang_getNullCursor() : parent;
6382
6383 annotateAndAdvanceTokens(updateC, RangeBefore, cursorRange);
6384
6385 // Avoid having the cursor of an expression "overwrite" the annotation of the
6386 // variable declaration that it belongs to.
6387 // This can happen for C++ constructor expressions whose range generally
6388 // include the variable declaration, e.g.:
6389 // MyCXXClass foo; // Make sure we don't annotate 'foo' as a CallExpr cursor.
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006390 if (clang_isExpression(cursorK) && MoreTokens()) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00006391 const Expr *E = getCursorExpr(cursor);
Dmitri Gribenkoa1691182013-01-26 18:12:08 +00006392 if (const Decl *D = getCursorParentDecl(cursor)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00006393 const unsigned I = NextToken();
6394 if (E->getLocStart().isValid() && D->getLocation().isValid() &&
6395 E->getLocStart() == D->getLocation() &&
6396 E->getLocStart() == GetTokenLoc(I)) {
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006397 updateCursorAnnotation(Cursors[I], updateC);
Guy Benyei11169dd2012-12-18 14:30:41 +00006398 AdvanceToken();
6399 }
6400 }
6401 }
6402
6403 // Before recursing into the children keep some state that we are going
6404 // to use in the AnnotateTokensWorker::postVisitChildren callback to do some
6405 // extra work after the child nodes are visited.
6406 // Note that we don't call VisitChildren here to avoid traversing statements
6407 // code-recursively which can blow the stack.
6408
6409 PostChildrenInfo Info;
6410 Info.Cursor = cursor;
6411 Info.CursorRange = cursorRange;
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006412 Info.BeforeReachingCursorIdx = BeforeReachingCursorIdx;
Guy Benyei11169dd2012-12-18 14:30:41 +00006413 Info.BeforeChildrenTokenIdx = NextToken();
6414 PostChildrenInfos.push_back(Info);
6415
6416 return CXChildVisit_Recurse;
6417}
6418
6419bool AnnotateTokensWorker::postVisitChildren(CXCursor cursor) {
6420 if (PostChildrenInfos.empty())
6421 return false;
6422 const PostChildrenInfo &Info = PostChildrenInfos.back();
6423 if (!clang_equalCursors(Info.Cursor, cursor))
6424 return false;
6425
6426 const unsigned BeforeChildren = Info.BeforeChildrenTokenIdx;
6427 const unsigned AfterChildren = NextToken();
6428 SourceRange cursorRange = Info.CursorRange;
6429
6430 // Scan the tokens that are at the end of the cursor, but are not captured
6431 // but the child cursors.
6432 annotateAndAdvanceTokens(cursor, RangeOverlap, cursorRange);
6433
6434 // Scan the tokens that are at the beginning of the cursor, but are not
6435 // capture by the child cursors.
6436 for (unsigned I = BeforeChildren; I != AfterChildren; ++I) {
6437 if (!clang_isInvalid(clang_getCursorKind(Cursors[I])))
6438 break;
6439
6440 Cursors[I] = cursor;
6441 }
6442
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006443 // Attributes are annotated out-of-order, rewind TokIdx to when we first
6444 // encountered the attribute cursor.
6445 if (clang_isAttribute(cursor.kind))
6446 TokIdx = Info.BeforeReachingCursorIdx;
6447
Guy Benyei11169dd2012-12-18 14:30:41 +00006448 PostChildrenInfos.pop_back();
6449 return false;
6450}
6451
6452static enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
6453 CXCursor parent,
6454 CXClientData client_data) {
6455 return static_cast<AnnotateTokensWorker*>(client_data)->Visit(cursor, parent);
6456}
6457
6458static bool AnnotateTokensPostChildrenVisitor(CXCursor cursor,
6459 CXClientData client_data) {
6460 return static_cast<AnnotateTokensWorker*>(client_data)->
6461 postVisitChildren(cursor);
6462}
6463
6464namespace {
6465
6466/// \brief Uses the macro expansions in the preprocessing record to find
6467/// and mark tokens that are macro arguments. This info is used by the
6468/// AnnotateTokensWorker.
6469class MarkMacroArgTokensVisitor {
6470 SourceManager &SM;
6471 CXToken *Tokens;
6472 unsigned NumTokens;
6473 unsigned CurIdx;
6474
6475public:
6476 MarkMacroArgTokensVisitor(SourceManager &SM,
6477 CXToken *tokens, unsigned numTokens)
6478 : SM(SM), Tokens(tokens), NumTokens(numTokens), CurIdx(0) { }
6479
6480 CXChildVisitResult visit(CXCursor cursor, CXCursor parent) {
6481 if (cursor.kind != CXCursor_MacroExpansion)
6482 return CXChildVisit_Continue;
6483
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00006484 SourceRange macroRange = getCursorMacroExpansion(cursor).getSourceRange();
Guy Benyei11169dd2012-12-18 14:30:41 +00006485 if (macroRange.getBegin() == macroRange.getEnd())
6486 return CXChildVisit_Continue; // it's not a function macro.
6487
6488 for (; CurIdx < NumTokens; ++CurIdx) {
6489 if (!SM.isBeforeInTranslationUnit(getTokenLoc(CurIdx),
6490 macroRange.getBegin()))
6491 break;
6492 }
6493
6494 if (CurIdx == NumTokens)
6495 return CXChildVisit_Break;
6496
6497 for (; CurIdx < NumTokens; ++CurIdx) {
6498 SourceLocation tokLoc = getTokenLoc(CurIdx);
6499 if (!SM.isBeforeInTranslationUnit(tokLoc, macroRange.getEnd()))
6500 break;
6501
6502 setFunctionMacroTokenLoc(CurIdx, SM.getMacroArgExpandedLocation(tokLoc));
6503 }
6504
6505 if (CurIdx == NumTokens)
6506 return CXChildVisit_Break;
6507
6508 return CXChildVisit_Continue;
6509 }
6510
6511private:
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006512 CXToken &getTok(unsigned Idx) {
6513 assert(Idx < NumTokens);
6514 return Tokens[Idx];
6515 }
6516 const CXToken &getTok(unsigned Idx) const {
6517 assert(Idx < NumTokens);
6518 return Tokens[Idx];
6519 }
6520
Guy Benyei11169dd2012-12-18 14:30:41 +00006521 SourceLocation getTokenLoc(unsigned tokI) {
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006522 return SourceLocation::getFromRawEncoding(getTok(tokI).int_data[1]);
Guy Benyei11169dd2012-12-18 14:30:41 +00006523 }
6524
6525 void setFunctionMacroTokenLoc(unsigned tokI, SourceLocation loc) {
6526 // The third field is reserved and currently not used. Use it here
6527 // to mark macro arg expanded tokens with their expanded locations.
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006528 getTok(tokI).int_data[3] = loc.getRawEncoding();
Guy Benyei11169dd2012-12-18 14:30:41 +00006529 }
6530};
6531
6532} // end anonymous namespace
6533
6534static CXChildVisitResult
6535MarkMacroArgTokensVisitorDelegate(CXCursor cursor, CXCursor parent,
6536 CXClientData client_data) {
6537 return static_cast<MarkMacroArgTokensVisitor*>(client_data)->visit(cursor,
6538 parent);
6539}
6540
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006541/// \brief Used by \c annotatePreprocessorTokens.
6542/// \returns true if lexing was finished, false otherwise.
6543static bool lexNext(Lexer &Lex, Token &Tok,
6544 unsigned &NextIdx, unsigned NumTokens) {
6545 if (NextIdx >= NumTokens)
6546 return true;
6547
6548 ++NextIdx;
6549 Lex.LexFromRawLexer(Tok);
Alexander Kornienko1a9f1842015-12-28 15:24:08 +00006550 return Tok.is(tok::eof);
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006551}
6552
Guy Benyei11169dd2012-12-18 14:30:41 +00006553static void annotatePreprocessorTokens(CXTranslationUnit TU,
6554 SourceRange RegionOfInterest,
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006555 CXCursor *Cursors,
6556 CXToken *Tokens,
6557 unsigned NumTokens) {
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006558 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00006559
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006560 Preprocessor &PP = CXXUnit->getPreprocessor();
Guy Benyei11169dd2012-12-18 14:30:41 +00006561 SourceManager &SourceMgr = CXXUnit->getSourceManager();
6562 std::pair<FileID, unsigned> BeginLocInfo
Argyrios Kyrtzidis5d47a9b2013-02-13 18:33:28 +00006563 = SourceMgr.getDecomposedSpellingLoc(RegionOfInterest.getBegin());
Guy Benyei11169dd2012-12-18 14:30:41 +00006564 std::pair<FileID, unsigned> EndLocInfo
Argyrios Kyrtzidis5d47a9b2013-02-13 18:33:28 +00006565 = SourceMgr.getDecomposedSpellingLoc(RegionOfInterest.getEnd());
Guy Benyei11169dd2012-12-18 14:30:41 +00006566
6567 if (BeginLocInfo.first != EndLocInfo.first)
6568 return;
6569
6570 StringRef Buffer;
6571 bool Invalid = false;
6572 Buffer = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
6573 if (Buffer.empty() || Invalid)
6574 return;
6575
6576 Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
6577 CXXUnit->getASTContext().getLangOpts(),
6578 Buffer.begin(), Buffer.data() + BeginLocInfo.second,
6579 Buffer.end());
6580 Lex.SetCommentRetentionState(true);
6581
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006582 unsigned NextIdx = 0;
Guy Benyei11169dd2012-12-18 14:30:41 +00006583 // Lex tokens in raw mode until we hit the end of the range, to avoid
6584 // entering #includes or expanding macros.
6585 while (true) {
6586 Token Tok;
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006587 if (lexNext(Lex, Tok, NextIdx, NumTokens))
6588 break;
6589 unsigned TokIdx = NextIdx-1;
6590 assert(Tok.getLocation() ==
6591 SourceLocation::getFromRawEncoding(Tokens[TokIdx].int_data[1]));
Guy Benyei11169dd2012-12-18 14:30:41 +00006592
6593 reprocess:
6594 if (Tok.is(tok::hash) && Tok.isAtStartOfLine()) {
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006595 // We have found a preprocessing directive. Annotate the tokens
6596 // appropriately.
Guy Benyei11169dd2012-12-18 14:30:41 +00006597 //
6598 // FIXME: Some simple tests here could identify macro definitions and
6599 // #undefs, to provide specific cursor kinds for those.
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006600
6601 SourceLocation BeginLoc = Tok.getLocation();
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006602 if (lexNext(Lex, Tok, NextIdx, NumTokens))
6603 break;
6604
Craig Topper69186e72014-06-08 08:38:04 +00006605 MacroInfo *MI = nullptr;
Alp Toker2d57cea2014-05-17 04:53:25 +00006606 if (Tok.is(tok::raw_identifier) && Tok.getRawIdentifier() == "define") {
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006607 if (lexNext(Lex, Tok, NextIdx, NumTokens))
6608 break;
6609
6610 if (Tok.is(tok::raw_identifier)) {
Alp Toker2d57cea2014-05-17 04:53:25 +00006611 IdentifierInfo &II =
6612 PP.getIdentifierTable().get(Tok.getRawIdentifier());
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006613 SourceLocation MappedTokLoc =
6614 CXXUnit->mapLocationToPreamble(Tok.getLocation());
6615 MI = getMacroInfo(II, MappedTokLoc, TU);
6616 }
6617 }
6618
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006619 bool finished = false;
Guy Benyei11169dd2012-12-18 14:30:41 +00006620 do {
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006621 if (lexNext(Lex, Tok, NextIdx, NumTokens)) {
6622 finished = true;
6623 break;
6624 }
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006625 // If we are in a macro definition, check if the token was ever a
6626 // macro name and annotate it if that's the case.
6627 if (MI) {
6628 SourceLocation SaveLoc = Tok.getLocation();
6629 Tok.setLocation(CXXUnit->mapLocationToPreamble(SaveLoc));
Richard Smith66a81862015-05-04 02:25:31 +00006630 MacroDefinitionRecord *MacroDef =
6631 checkForMacroInMacroDefinition(MI, Tok, TU);
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006632 Tok.setLocation(SaveLoc);
6633 if (MacroDef)
Richard Smith66a81862015-05-04 02:25:31 +00006634 Cursors[NextIdx - 1] =
6635 MakeMacroExpansionCursor(MacroDef, Tok.getLocation(), TU);
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006636 }
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006637 } while (!Tok.isAtStartOfLine());
6638
6639 unsigned LastIdx = finished ? NextIdx-1 : NextIdx-2;
6640 assert(TokIdx <= LastIdx);
6641 SourceLocation EndLoc =
6642 SourceLocation::getFromRawEncoding(Tokens[LastIdx].int_data[1]);
6643 CXCursor Cursor =
6644 MakePreprocessingDirectiveCursor(SourceRange(BeginLoc, EndLoc), TU);
6645
6646 for (; TokIdx <= LastIdx; ++TokIdx)
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006647 updateCursorAnnotation(Cursors[TokIdx], Cursor);
Guy Benyei11169dd2012-12-18 14:30:41 +00006648
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006649 if (finished)
6650 break;
6651 goto reprocess;
Guy Benyei11169dd2012-12-18 14:30:41 +00006652 }
Guy Benyei11169dd2012-12-18 14:30:41 +00006653 }
6654}
6655
6656// This gets run a separate thread to avoid stack blowout.
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00006657static void clang_annotateTokensImpl(CXTranslationUnit TU, ASTUnit *CXXUnit,
6658 CXToken *Tokens, unsigned NumTokens,
6659 CXCursor *Cursors) {
Dmitri Gribenko183436e2013-01-26 21:49:50 +00006660 CIndexer *CXXIdx = TU->CIdx;
Guy Benyei11169dd2012-12-18 14:30:41 +00006661 if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForEditing))
6662 setThreadBackgroundPriority();
6663
6664 // Determine the region of interest, which contains all of the tokens.
6665 SourceRange RegionOfInterest;
6666 RegionOfInterest.setBegin(
6667 cxloc::translateSourceLocation(clang_getTokenLocation(TU, Tokens[0])));
6668 RegionOfInterest.setEnd(
6669 cxloc::translateSourceLocation(clang_getTokenLocation(TU,
6670 Tokens[NumTokens-1])));
6671
Guy Benyei11169dd2012-12-18 14:30:41 +00006672 // Relex the tokens within the source range to look for preprocessing
6673 // directives.
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006674 annotatePreprocessorTokens(TU, RegionOfInterest, Cursors, Tokens, NumTokens);
Argyrios Kyrtzidis5d47a9b2013-02-13 18:33:28 +00006675
6676 // If begin location points inside a macro argument, set it to the expansion
6677 // location so we can have the full context when annotating semantically.
6678 {
6679 SourceManager &SM = CXXUnit->getSourceManager();
6680 SourceLocation Loc =
6681 SM.getMacroArgExpandedLocation(RegionOfInterest.getBegin());
6682 if (Loc.isMacroID())
6683 RegionOfInterest.setBegin(SM.getExpansionLoc(Loc));
6684 }
6685
Guy Benyei11169dd2012-12-18 14:30:41 +00006686 if (CXXUnit->getPreprocessor().getPreprocessingRecord()) {
6687 // Search and mark tokens that are macro argument expansions.
6688 MarkMacroArgTokensVisitor Visitor(CXXUnit->getSourceManager(),
6689 Tokens, NumTokens);
6690 CursorVisitor MacroArgMarker(TU,
6691 MarkMacroArgTokensVisitorDelegate, &Visitor,
6692 /*VisitPreprocessorLast=*/true,
6693 /*VisitIncludedEntities=*/false,
6694 RegionOfInterest);
6695 MacroArgMarker.visitPreprocessedEntitiesInRegion();
6696 }
6697
6698 // Annotate all of the source locations in the region of interest that map to
6699 // a specific cursor.
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006700 AnnotateTokensWorker W(Tokens, Cursors, NumTokens, TU, RegionOfInterest);
Guy Benyei11169dd2012-12-18 14:30:41 +00006701
6702 // FIXME: We use a ridiculous stack size here because the data-recursion
6703 // algorithm uses a large stack frame than the non-data recursive version,
6704 // and AnnotationTokensWorker currently transforms the data-recursion
6705 // algorithm back into a traditional recursion by explicitly calling
6706 // VisitChildren(). We will need to remove this explicit recursive call.
6707 W.AnnotateTokens();
6708
6709 // If we ran into any entities that involve context-sensitive keywords,
6710 // take another pass through the tokens to mark them as such.
6711 if (W.hasContextSensitiveKeywords()) {
6712 for (unsigned I = 0; I != NumTokens; ++I) {
6713 if (clang_getTokenKind(Tokens[I]) != CXToken_Identifier)
6714 continue;
6715
6716 if (Cursors[I].kind == CXCursor_ObjCPropertyDecl) {
6717 IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006718 if (const ObjCPropertyDecl *Property
Guy Benyei11169dd2012-12-18 14:30:41 +00006719 = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(Cursors[I]))) {
6720 if (Property->getPropertyAttributesAsWritten() != 0 &&
6721 llvm::StringSwitch<bool>(II->getName())
6722 .Case("readonly", true)
6723 .Case("assign", true)
6724 .Case("unsafe_unretained", true)
6725 .Case("readwrite", true)
6726 .Case("retain", true)
6727 .Case("copy", true)
6728 .Case("nonatomic", true)
6729 .Case("atomic", true)
6730 .Case("getter", true)
6731 .Case("setter", true)
6732 .Case("strong", true)
6733 .Case("weak", true)
6734 .Default(false))
6735 Tokens[I].int_data[0] = CXToken_Keyword;
6736 }
6737 continue;
6738 }
6739
6740 if (Cursors[I].kind == CXCursor_ObjCInstanceMethodDecl ||
6741 Cursors[I].kind == CXCursor_ObjCClassMethodDecl) {
6742 IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
6743 if (llvm::StringSwitch<bool>(II->getName())
6744 .Case("in", true)
6745 .Case("out", true)
6746 .Case("inout", true)
6747 .Case("oneway", true)
6748 .Case("bycopy", true)
6749 .Case("byref", true)
6750 .Default(false))
6751 Tokens[I].int_data[0] = CXToken_Keyword;
6752 continue;
6753 }
6754
6755 if (Cursors[I].kind == CXCursor_CXXFinalAttr ||
6756 Cursors[I].kind == CXCursor_CXXOverrideAttr) {
6757 Tokens[I].int_data[0] = CXToken_Keyword;
6758 continue;
6759 }
6760 }
6761 }
6762}
6763
6764extern "C" {
6765
6766void clang_annotateTokens(CXTranslationUnit TU,
6767 CXToken *Tokens, unsigned NumTokens,
6768 CXCursor *Cursors) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00006769 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00006770 LOG_BAD_TU(TU);
6771 return;
6772 }
6773 if (NumTokens == 0 || !Tokens || !Cursors) {
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00006774 LOG_FUNC_SECTION { *Log << "<null input>"; }
Guy Benyei11169dd2012-12-18 14:30:41 +00006775 return;
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00006776 }
6777
6778 LOG_FUNC_SECTION {
6779 *Log << TU << ' ';
6780 CXSourceLocation bloc = clang_getTokenLocation(TU, Tokens[0]);
6781 CXSourceLocation eloc = clang_getTokenLocation(TU, Tokens[NumTokens-1]);
6782 *Log << clang_getRange(bloc, eloc);
6783 }
Guy Benyei11169dd2012-12-18 14:30:41 +00006784
6785 // Any token we don't specifically annotate will have a NULL cursor.
6786 CXCursor C = clang_getNullCursor();
6787 for (unsigned I = 0; I != NumTokens; ++I)
6788 Cursors[I] = C;
6789
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006790 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00006791 if (!CXXUnit)
6792 return;
6793
6794 ASTUnit::ConcurrencyCheck Check(*CXXUnit);
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00006795
6796 auto AnnotateTokensImpl = [=]() {
6797 clang_annotateTokensImpl(TU, CXXUnit, Tokens, NumTokens, Cursors);
6798 };
Guy Benyei11169dd2012-12-18 14:30:41 +00006799 llvm::CrashRecoveryContext CRC;
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00006800 if (!RunSafely(CRC, AnnotateTokensImpl, GetSafetyThreadStackSize() * 2)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00006801 fprintf(stderr, "libclang: crash detected while annotating tokens\n");
6802 }
6803}
6804
6805} // end: extern "C"
6806
6807//===----------------------------------------------------------------------===//
6808// Operations for querying linkage of a cursor.
6809//===----------------------------------------------------------------------===//
6810
6811extern "C" {
6812CXLinkageKind clang_getCursorLinkage(CXCursor cursor) {
6813 if (!clang_isDeclaration(cursor.kind))
6814 return CXLinkage_Invalid;
6815
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006816 const Decl *D = cxcursor::getCursorDecl(cursor);
6817 if (const NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D))
Rafael Espindola3ae00052013-05-13 00:12:11 +00006818 switch (ND->getLinkageInternal()) {
Rafael Espindola50df3a02013-05-25 17:16:20 +00006819 case NoLinkage:
6820 case VisibleNoLinkage: return CXLinkage_NoLinkage;
Guy Benyei11169dd2012-12-18 14:30:41 +00006821 case InternalLinkage: return CXLinkage_Internal;
6822 case UniqueExternalLinkage: return CXLinkage_UniqueExternal;
6823 case ExternalLinkage: return CXLinkage_External;
6824 };
6825
6826 return CXLinkage_Invalid;
6827}
6828} // end: extern "C"
6829
6830//===----------------------------------------------------------------------===//
Ehsan Akhgari93697fa2015-11-23 19:56:46 +00006831// Operations for querying visibility of a cursor.
6832//===----------------------------------------------------------------------===//
6833
6834extern "C" {
6835CXVisibilityKind clang_getCursorVisibility(CXCursor cursor) {
6836 if (!clang_isDeclaration(cursor.kind))
6837 return CXVisibility_Invalid;
6838
6839 const Decl *D = cxcursor::getCursorDecl(cursor);
6840 if (const NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D))
6841 switch (ND->getVisibility()) {
6842 case HiddenVisibility: return CXVisibility_Hidden;
6843 case ProtectedVisibility: return CXVisibility_Protected;
6844 case DefaultVisibility: return CXVisibility_Default;
6845 };
6846
6847 return CXVisibility_Invalid;
6848}
6849} // end: extern "C"
6850
6851//===----------------------------------------------------------------------===//
Guy Benyei11169dd2012-12-18 14:30:41 +00006852// Operations for querying language of a cursor.
6853//===----------------------------------------------------------------------===//
6854
6855static CXLanguageKind getDeclLanguage(const Decl *D) {
6856 if (!D)
6857 return CXLanguage_C;
6858
6859 switch (D->getKind()) {
6860 default:
6861 break;
6862 case Decl::ImplicitParam:
6863 case Decl::ObjCAtDefsField:
6864 case Decl::ObjCCategory:
6865 case Decl::ObjCCategoryImpl:
6866 case Decl::ObjCCompatibleAlias:
6867 case Decl::ObjCImplementation:
6868 case Decl::ObjCInterface:
6869 case Decl::ObjCIvar:
6870 case Decl::ObjCMethod:
6871 case Decl::ObjCProperty:
6872 case Decl::ObjCPropertyImpl:
6873 case Decl::ObjCProtocol:
Douglas Gregor85f3f952015-07-07 03:57:15 +00006874 case Decl::ObjCTypeParam:
Guy Benyei11169dd2012-12-18 14:30:41 +00006875 return CXLanguage_ObjC;
6876 case Decl::CXXConstructor:
6877 case Decl::CXXConversion:
6878 case Decl::CXXDestructor:
6879 case Decl::CXXMethod:
6880 case Decl::CXXRecord:
6881 case Decl::ClassTemplate:
6882 case Decl::ClassTemplatePartialSpecialization:
6883 case Decl::ClassTemplateSpecialization:
6884 case Decl::Friend:
6885 case Decl::FriendTemplate:
6886 case Decl::FunctionTemplate:
6887 case Decl::LinkageSpec:
6888 case Decl::Namespace:
6889 case Decl::NamespaceAlias:
6890 case Decl::NonTypeTemplateParm:
6891 case Decl::StaticAssert:
6892 case Decl::TemplateTemplateParm:
6893 case Decl::TemplateTypeParm:
6894 case Decl::UnresolvedUsingTypename:
6895 case Decl::UnresolvedUsingValue:
6896 case Decl::Using:
6897 case Decl::UsingDirective:
6898 case Decl::UsingShadow:
6899 return CXLanguage_CPlusPlus;
6900 }
6901
6902 return CXLanguage_C;
6903}
6904
6905extern "C" {
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006906
6907static CXAvailabilityKind getCursorAvailabilityForDecl(const Decl *D) {
6908 if (isa<FunctionDecl>(D) && cast<FunctionDecl>(D)->isDeleted())
Manuel Klimek8e3a7ed2015-09-25 17:53:16 +00006909 return CXAvailability_NotAvailable;
Guy Benyei11169dd2012-12-18 14:30:41 +00006910
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006911 switch (D->getAvailability()) {
6912 case AR_Available:
6913 case AR_NotYetIntroduced:
6914 if (const EnumConstantDecl *EnumConst = dyn_cast<EnumConstantDecl>(D))
Benjamin Kramer656363d2013-10-15 18:53:18 +00006915 return getCursorAvailabilityForDecl(
6916 cast<Decl>(EnumConst->getDeclContext()));
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006917 return CXAvailability_Available;
6918
6919 case AR_Deprecated:
6920 return CXAvailability_Deprecated;
6921
6922 case AR_Unavailable:
6923 return CXAvailability_NotAvailable;
6924 }
Benjamin Kramer656363d2013-10-15 18:53:18 +00006925
6926 llvm_unreachable("Unknown availability kind!");
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006927}
6928
Guy Benyei11169dd2012-12-18 14:30:41 +00006929enum CXAvailabilityKind clang_getCursorAvailability(CXCursor cursor) {
6930 if (clang_isDeclaration(cursor.kind))
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006931 if (const Decl *D = cxcursor::getCursorDecl(cursor))
6932 return getCursorAvailabilityForDecl(D);
Guy Benyei11169dd2012-12-18 14:30:41 +00006933
6934 return CXAvailability_Available;
6935}
6936
6937static CXVersion convertVersion(VersionTuple In) {
6938 CXVersion Out = { -1, -1, -1 };
6939 if (In.empty())
6940 return Out;
6941
6942 Out.Major = In.getMajor();
6943
NAKAMURA Takumic2b5d1f2013-02-21 02:32:34 +00006944 Optional<unsigned> Minor = In.getMinor();
6945 if (Minor.hasValue())
Guy Benyei11169dd2012-12-18 14:30:41 +00006946 Out.Minor = *Minor;
6947 else
6948 return Out;
6949
NAKAMURA Takumic2b5d1f2013-02-21 02:32:34 +00006950 Optional<unsigned> Subminor = In.getSubminor();
6951 if (Subminor.hasValue())
Guy Benyei11169dd2012-12-18 14:30:41 +00006952 Out.Subminor = *Subminor;
6953
6954 return Out;
6955}
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006956
6957static int getCursorPlatformAvailabilityForDecl(const Decl *D,
6958 int *always_deprecated,
6959 CXString *deprecated_message,
6960 int *always_unavailable,
6961 CXString *unavailable_message,
6962 CXPlatformAvailability *availability,
6963 int availability_size) {
6964 bool HadAvailAttr = false;
6965 int N = 0;
Aaron Ballmanb97112e2014-03-08 22:19:01 +00006966 for (auto A : D->attrs()) {
6967 if (DeprecatedAttr *Deprecated = dyn_cast<DeprecatedAttr>(A)) {
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006968 HadAvailAttr = true;
6969 if (always_deprecated)
6970 *always_deprecated = 1;
Nico Weberaacf0312014-04-24 05:16:45 +00006971 if (deprecated_message) {
Argyrios Kyrtzidisedfe07f2014-04-24 06:05:40 +00006972 clang_disposeString(*deprecated_message);
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006973 *deprecated_message = cxstring::createDup(Deprecated->getMessage());
Nico Weberaacf0312014-04-24 05:16:45 +00006974 }
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006975 continue;
6976 }
6977
Aaron Ballmanb97112e2014-03-08 22:19:01 +00006978 if (UnavailableAttr *Unavailable = dyn_cast<UnavailableAttr>(A)) {
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006979 HadAvailAttr = true;
6980 if (always_unavailable)
6981 *always_unavailable = 1;
6982 if (unavailable_message) {
Argyrios Kyrtzidisedfe07f2014-04-24 06:05:40 +00006983 clang_disposeString(*unavailable_message);
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006984 *unavailable_message = cxstring::createDup(Unavailable->getMessage());
6985 }
6986 continue;
6987 }
6988
Aaron Ballmanb97112e2014-03-08 22:19:01 +00006989 if (AvailabilityAttr *Avail = dyn_cast<AvailabilityAttr>(A)) {
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006990 HadAvailAttr = true;
6991 if (N < availability_size) {
6992 availability[N].Platform
6993 = cxstring::createDup(Avail->getPlatform()->getName());
6994 availability[N].Introduced = convertVersion(Avail->getIntroduced());
6995 availability[N].Deprecated = convertVersion(Avail->getDeprecated());
6996 availability[N].Obsoleted = convertVersion(Avail->getObsoleted());
6997 availability[N].Unavailable = Avail->getUnavailable();
6998 availability[N].Message = cxstring::createDup(Avail->getMessage());
6999 }
7000 ++N;
7001 }
7002 }
7003
7004 if (!HadAvailAttr)
7005 if (const EnumConstantDecl *EnumConst = dyn_cast<EnumConstantDecl>(D))
7006 return getCursorPlatformAvailabilityForDecl(
7007 cast<Decl>(EnumConst->getDeclContext()),
7008 always_deprecated,
7009 deprecated_message,
7010 always_unavailable,
7011 unavailable_message,
7012 availability,
7013 availability_size);
Guy Benyei11169dd2012-12-18 14:30:41 +00007014
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007015 return N;
7016}
7017
Guy Benyei11169dd2012-12-18 14:30:41 +00007018int clang_getCursorPlatformAvailability(CXCursor cursor,
7019 int *always_deprecated,
7020 CXString *deprecated_message,
7021 int *always_unavailable,
7022 CXString *unavailable_message,
7023 CXPlatformAvailability *availability,
7024 int availability_size) {
7025 if (always_deprecated)
7026 *always_deprecated = 0;
7027 if (deprecated_message)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00007028 *deprecated_message = cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00007029 if (always_unavailable)
7030 *always_unavailable = 0;
7031 if (unavailable_message)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00007032 *unavailable_message = cxstring::createEmpty();
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007033
Guy Benyei11169dd2012-12-18 14:30:41 +00007034 if (!clang_isDeclaration(cursor.kind))
7035 return 0;
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007036
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007037 const Decl *D = cxcursor::getCursorDecl(cursor);
Guy Benyei11169dd2012-12-18 14:30:41 +00007038 if (!D)
7039 return 0;
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007040
7041 return getCursorPlatformAvailabilityForDecl(D, always_deprecated,
7042 deprecated_message,
7043 always_unavailable,
7044 unavailable_message,
7045 availability,
7046 availability_size);
Guy Benyei11169dd2012-12-18 14:30:41 +00007047}
7048
7049void clang_disposeCXPlatformAvailability(CXPlatformAvailability *availability) {
7050 clang_disposeString(availability->Platform);
7051 clang_disposeString(availability->Message);
7052}
7053
7054CXLanguageKind clang_getCursorLanguage(CXCursor cursor) {
7055 if (clang_isDeclaration(cursor.kind))
7056 return getDeclLanguage(cxcursor::getCursorDecl(cursor));
7057
7058 return CXLanguage_Invalid;
7059}
7060
7061 /// \brief If the given cursor is the "templated" declaration
7062 /// descibing a class or function template, return the class or
7063 /// function template.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007064static const Decl *maybeGetTemplateCursor(const Decl *D) {
Guy Benyei11169dd2012-12-18 14:30:41 +00007065 if (!D)
Craig Topper69186e72014-06-08 08:38:04 +00007066 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007067
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007068 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00007069 if (FunctionTemplateDecl *FunTmpl = FD->getDescribedFunctionTemplate())
7070 return FunTmpl;
7071
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007072 if (const CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00007073 if (ClassTemplateDecl *ClassTmpl = RD->getDescribedClassTemplate())
7074 return ClassTmpl;
7075
7076 return D;
7077}
7078
Argyrios Kyrtzidis4e0854f2014-10-15 17:05:31 +00007079
7080enum CX_StorageClass clang_Cursor_getStorageClass(CXCursor C) {
7081 StorageClass sc = SC_None;
7082 const Decl *D = getCursorDecl(C);
7083 if (D) {
7084 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) {
7085 sc = FD->getStorageClass();
7086 } else if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
7087 sc = VD->getStorageClass();
7088 } else {
7089 return CX_SC_Invalid;
7090 }
7091 } else {
7092 return CX_SC_Invalid;
7093 }
7094 switch (sc) {
7095 case SC_None:
7096 return CX_SC_None;
7097 case SC_Extern:
7098 return CX_SC_Extern;
7099 case SC_Static:
7100 return CX_SC_Static;
7101 case SC_PrivateExtern:
7102 return CX_SC_PrivateExtern;
Argyrios Kyrtzidis4e0854f2014-10-15 17:05:31 +00007103 case SC_Auto:
7104 return CX_SC_Auto;
7105 case SC_Register:
7106 return CX_SC_Register;
Argyrios Kyrtzidis4e0854f2014-10-15 17:05:31 +00007107 }
Kaelyn Takataab61e702014-10-15 18:03:26 +00007108 llvm_unreachable("Unhandled storage class!");
Argyrios Kyrtzidis4e0854f2014-10-15 17:05:31 +00007109}
7110
Guy Benyei11169dd2012-12-18 14:30:41 +00007111CXCursor clang_getCursorSemanticParent(CXCursor cursor) {
7112 if (clang_isDeclaration(cursor.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007113 if (const Decl *D = getCursorDecl(cursor)) {
7114 const DeclContext *DC = D->getDeclContext();
Guy Benyei11169dd2012-12-18 14:30:41 +00007115 if (!DC)
7116 return clang_getNullCursor();
7117
7118 return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
7119 getCursorTU(cursor));
7120 }
7121 }
7122
7123 if (clang_isStatement(cursor.kind) || clang_isExpression(cursor.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007124 if (const Decl *D = getCursorDecl(cursor))
Guy Benyei11169dd2012-12-18 14:30:41 +00007125 return MakeCXCursor(D, getCursorTU(cursor));
7126 }
7127
7128 return clang_getNullCursor();
7129}
7130
7131CXCursor clang_getCursorLexicalParent(CXCursor cursor) {
7132 if (clang_isDeclaration(cursor.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007133 if (const Decl *D = getCursorDecl(cursor)) {
7134 const DeclContext *DC = D->getLexicalDeclContext();
Guy Benyei11169dd2012-12-18 14:30:41 +00007135 if (!DC)
7136 return clang_getNullCursor();
7137
7138 return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
7139 getCursorTU(cursor));
7140 }
7141 }
7142
7143 // FIXME: Note that we can't easily compute the lexical context of a
7144 // statement or expression, so we return nothing.
7145 return clang_getNullCursor();
7146}
7147
7148CXFile clang_getIncludedFile(CXCursor cursor) {
7149 if (cursor.kind != CXCursor_InclusionDirective)
Craig Topper69186e72014-06-08 08:38:04 +00007150 return nullptr;
7151
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00007152 const InclusionDirective *ID = getCursorInclusionDirective(cursor);
Dmitri Gribenkof9304482013-01-23 15:56:07 +00007153 return const_cast<FileEntry *>(ID->getFile());
Guy Benyei11169dd2012-12-18 14:30:41 +00007154}
7155
Argyrios Kyrtzidis9adfd8a2013-04-18 22:15:49 +00007156unsigned clang_Cursor_getObjCPropertyAttributes(CXCursor C, unsigned reserved) {
7157 if (C.kind != CXCursor_ObjCPropertyDecl)
7158 return CXObjCPropertyAttr_noattr;
7159
7160 unsigned Result = CXObjCPropertyAttr_noattr;
7161 const ObjCPropertyDecl *PD = dyn_cast<ObjCPropertyDecl>(getCursorDecl(C));
7162 ObjCPropertyDecl::PropertyAttributeKind Attr =
7163 PD->getPropertyAttributesAsWritten();
7164
7165#define SET_CXOBJCPROP_ATTR(A) \
7166 if (Attr & ObjCPropertyDecl::OBJC_PR_##A) \
7167 Result |= CXObjCPropertyAttr_##A
7168 SET_CXOBJCPROP_ATTR(readonly);
7169 SET_CXOBJCPROP_ATTR(getter);
7170 SET_CXOBJCPROP_ATTR(assign);
7171 SET_CXOBJCPROP_ATTR(readwrite);
7172 SET_CXOBJCPROP_ATTR(retain);
7173 SET_CXOBJCPROP_ATTR(copy);
7174 SET_CXOBJCPROP_ATTR(nonatomic);
7175 SET_CXOBJCPROP_ATTR(setter);
7176 SET_CXOBJCPROP_ATTR(atomic);
7177 SET_CXOBJCPROP_ATTR(weak);
7178 SET_CXOBJCPROP_ATTR(strong);
7179 SET_CXOBJCPROP_ATTR(unsafe_unretained);
7180#undef SET_CXOBJCPROP_ATTR
7181
7182 return Result;
7183}
7184
Argyrios Kyrtzidis9d9bc012013-04-18 23:29:12 +00007185unsigned clang_Cursor_getObjCDeclQualifiers(CXCursor C) {
7186 if (!clang_isDeclaration(C.kind))
7187 return CXObjCDeclQualifier_None;
7188
7189 Decl::ObjCDeclQualifier QT = Decl::OBJC_TQ_None;
7190 const Decl *D = getCursorDecl(C);
7191 if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D))
7192 QT = MD->getObjCDeclQualifier();
7193 else if (const ParmVarDecl *PD = dyn_cast<ParmVarDecl>(D))
7194 QT = PD->getObjCDeclQualifier();
7195 if (QT == Decl::OBJC_TQ_None)
7196 return CXObjCDeclQualifier_None;
7197
7198 unsigned Result = CXObjCDeclQualifier_None;
7199 if (QT & Decl::OBJC_TQ_In) Result |= CXObjCDeclQualifier_In;
7200 if (QT & Decl::OBJC_TQ_Inout) Result |= CXObjCDeclQualifier_Inout;
7201 if (QT & Decl::OBJC_TQ_Out) Result |= CXObjCDeclQualifier_Out;
7202 if (QT & Decl::OBJC_TQ_Bycopy) Result |= CXObjCDeclQualifier_Bycopy;
7203 if (QT & Decl::OBJC_TQ_Byref) Result |= CXObjCDeclQualifier_Byref;
7204 if (QT & Decl::OBJC_TQ_Oneway) Result |= CXObjCDeclQualifier_Oneway;
7205
7206 return Result;
7207}
7208
Argyrios Kyrtzidis7b50fc52013-07-05 20:44:37 +00007209unsigned clang_Cursor_isObjCOptional(CXCursor C) {
7210 if (!clang_isDeclaration(C.kind))
7211 return 0;
7212
7213 const Decl *D = getCursorDecl(C);
7214 if (const ObjCPropertyDecl *PD = dyn_cast<ObjCPropertyDecl>(D))
7215 return PD->getPropertyImplementation() == ObjCPropertyDecl::Optional;
7216 if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D))
7217 return MD->getImplementationControl() == ObjCMethodDecl::Optional;
7218
7219 return 0;
7220}
7221
Argyrios Kyrtzidis23814e42013-04-18 23:53:05 +00007222unsigned clang_Cursor_isVariadic(CXCursor C) {
7223 if (!clang_isDeclaration(C.kind))
7224 return 0;
7225
7226 const Decl *D = getCursorDecl(C);
7227 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
7228 return FD->isVariadic();
7229 if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D))
7230 return MD->isVariadic();
7231
7232 return 0;
7233}
7234
Guy Benyei11169dd2012-12-18 14:30:41 +00007235CXSourceRange clang_Cursor_getCommentRange(CXCursor C) {
7236 if (!clang_isDeclaration(C.kind))
7237 return clang_getNullRange();
7238
7239 const Decl *D = getCursorDecl(C);
7240 ASTContext &Context = getCursorContext(C);
7241 const RawComment *RC = Context.getRawCommentForAnyRedecl(D);
7242 if (!RC)
7243 return clang_getNullRange();
7244
7245 return cxloc::translateSourceRange(Context, RC->getSourceRange());
7246}
7247
7248CXString clang_Cursor_getRawCommentText(CXCursor C) {
7249 if (!clang_isDeclaration(C.kind))
Dmitri Gribenkof98dfba2013-02-01 14:13:32 +00007250 return cxstring::createNull();
Guy Benyei11169dd2012-12-18 14:30:41 +00007251
7252 const Decl *D = getCursorDecl(C);
7253 ASTContext &Context = getCursorContext(C);
7254 const RawComment *RC = Context.getRawCommentForAnyRedecl(D);
7255 StringRef RawText = RC ? RC->getRawText(Context.getSourceManager()) :
7256 StringRef();
7257
7258 // Don't duplicate the string because RawText points directly into source
7259 // code.
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00007260 return cxstring::createRef(RawText);
Guy Benyei11169dd2012-12-18 14:30:41 +00007261}
7262
7263CXString clang_Cursor_getBriefCommentText(CXCursor C) {
7264 if (!clang_isDeclaration(C.kind))
Dmitri Gribenkof98dfba2013-02-01 14:13:32 +00007265 return cxstring::createNull();
Guy Benyei11169dd2012-12-18 14:30:41 +00007266
7267 const Decl *D = getCursorDecl(C);
7268 const ASTContext &Context = getCursorContext(C);
7269 const RawComment *RC = Context.getRawCommentForAnyRedecl(D);
7270
7271 if (RC) {
7272 StringRef BriefText = RC->getBriefText(Context);
7273
7274 // Don't duplicate the string because RawComment ensures that this memory
7275 // will not go away.
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00007276 return cxstring::createRef(BriefText);
Guy Benyei11169dd2012-12-18 14:30:41 +00007277 }
7278
Dmitri Gribenkof98dfba2013-02-01 14:13:32 +00007279 return cxstring::createNull();
Guy Benyei11169dd2012-12-18 14:30:41 +00007280}
7281
Guy Benyei11169dd2012-12-18 14:30:41 +00007282CXModule clang_Cursor_getModule(CXCursor C) {
7283 if (C.kind == CXCursor_ModuleImportDecl) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007284 if (const ImportDecl *ImportD =
7285 dyn_cast_or_null<ImportDecl>(getCursorDecl(C)))
Guy Benyei11169dd2012-12-18 14:30:41 +00007286 return ImportD->getImportedModule();
7287 }
7288
Craig Topper69186e72014-06-08 08:38:04 +00007289 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007290}
7291
Argyrios Kyrtzidisf6d49c32014-05-14 23:14:37 +00007292CXModule clang_getModuleForFile(CXTranslationUnit TU, CXFile File) {
7293 if (isNotUsableTU(TU)) {
7294 LOG_BAD_TU(TU);
7295 return nullptr;
7296 }
7297 if (!File)
7298 return nullptr;
7299 FileEntry *FE = static_cast<FileEntry *>(File);
7300
7301 ASTUnit &Unit = *cxtu::getASTUnit(TU);
7302 HeaderSearch &HS = Unit.getPreprocessor().getHeaderSearchInfo();
7303 ModuleMap::KnownHeader Header = HS.findModuleForHeader(FE);
7304
Richard Smithfeb54b62014-10-23 02:01:19 +00007305 return Header.getModule();
Argyrios Kyrtzidisf6d49c32014-05-14 23:14:37 +00007306}
7307
Argyrios Kyrtzidis12fdb9e2013-04-26 22:47:49 +00007308CXFile clang_Module_getASTFile(CXModule CXMod) {
7309 if (!CXMod)
Craig Topper69186e72014-06-08 08:38:04 +00007310 return nullptr;
Argyrios Kyrtzidis12fdb9e2013-04-26 22:47:49 +00007311 Module *Mod = static_cast<Module*>(CXMod);
7312 return const_cast<FileEntry *>(Mod->getASTFile());
7313}
7314
Guy Benyei11169dd2012-12-18 14:30:41 +00007315CXModule clang_Module_getParent(CXModule CXMod) {
7316 if (!CXMod)
Craig Topper69186e72014-06-08 08:38:04 +00007317 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007318 Module *Mod = static_cast<Module*>(CXMod);
7319 return Mod->Parent;
7320}
7321
7322CXString clang_Module_getName(CXModule CXMod) {
7323 if (!CXMod)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00007324 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00007325 Module *Mod = static_cast<Module*>(CXMod);
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00007326 return cxstring::createDup(Mod->Name);
Guy Benyei11169dd2012-12-18 14:30:41 +00007327}
7328
7329CXString clang_Module_getFullName(CXModule CXMod) {
7330 if (!CXMod)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00007331 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00007332 Module *Mod = static_cast<Module*>(CXMod);
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00007333 return cxstring::createDup(Mod->getFullModuleName());
Guy Benyei11169dd2012-12-18 14:30:41 +00007334}
7335
Argyrios Kyrtzidis884337f2014-05-15 04:44:25 +00007336int clang_Module_isSystem(CXModule CXMod) {
7337 if (!CXMod)
7338 return 0;
7339 Module *Mod = static_cast<Module*>(CXMod);
7340 return Mod->IsSystem;
7341}
7342
Argyrios Kyrtzidis3c5305c2013-03-13 21:13:43 +00007343unsigned clang_Module_getNumTopLevelHeaders(CXTranslationUnit TU,
7344 CXModule CXMod) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00007345 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00007346 LOG_BAD_TU(TU);
7347 return 0;
7348 }
7349 if (!CXMod)
Guy Benyei11169dd2012-12-18 14:30:41 +00007350 return 0;
7351 Module *Mod = static_cast<Module*>(CXMod);
Argyrios Kyrtzidis3c5305c2013-03-13 21:13:43 +00007352 FileManager &FileMgr = cxtu::getASTUnit(TU)->getFileManager();
7353 ArrayRef<const FileEntry *> TopHeaders = Mod->getTopHeaders(FileMgr);
7354 return TopHeaders.size();
Guy Benyei11169dd2012-12-18 14:30:41 +00007355}
7356
Argyrios Kyrtzidis3c5305c2013-03-13 21:13:43 +00007357CXFile clang_Module_getTopLevelHeader(CXTranslationUnit TU,
7358 CXModule CXMod, unsigned Index) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00007359 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00007360 LOG_BAD_TU(TU);
Craig Topper69186e72014-06-08 08:38:04 +00007361 return nullptr;
Dmitri Gribenko256454f2014-02-11 14:34:14 +00007362 }
7363 if (!CXMod)
Craig Topper69186e72014-06-08 08:38:04 +00007364 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007365 Module *Mod = static_cast<Module*>(CXMod);
Argyrios Kyrtzidis3c5305c2013-03-13 21:13:43 +00007366 FileManager &FileMgr = cxtu::getASTUnit(TU)->getFileManager();
Guy Benyei11169dd2012-12-18 14:30:41 +00007367
Argyrios Kyrtzidis3c5305c2013-03-13 21:13:43 +00007368 ArrayRef<const FileEntry *> TopHeaders = Mod->getTopHeaders(FileMgr);
7369 if (Index < TopHeaders.size())
7370 return const_cast<FileEntry *>(TopHeaders[Index]);
Guy Benyei11169dd2012-12-18 14:30:41 +00007371
Craig Topper69186e72014-06-08 08:38:04 +00007372 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007373}
7374
7375} // end: extern "C"
7376
7377//===----------------------------------------------------------------------===//
7378// C++ AST instrospection.
7379//===----------------------------------------------------------------------===//
7380
7381extern "C" {
Saleem Abdulrasool6ea75db2015-10-27 15:50:22 +00007382unsigned clang_CXXField_isMutable(CXCursor C) {
7383 if (!clang_isDeclaration(C.kind))
7384 return 0;
7385
7386 if (const auto D = cxcursor::getCursorDecl(C))
7387 if (const auto FD = dyn_cast_or_null<FieldDecl>(D))
7388 return FD->isMutable() ? 1 : 0;
7389 return 0;
7390}
7391
Dmitri Gribenko62770be2013-05-17 18:38:35 +00007392unsigned clang_CXXMethod_isPureVirtual(CXCursor C) {
7393 if (!clang_isDeclaration(C.kind))
7394 return 0;
7395
Dmitri Gribenko62770be2013-05-17 18:38:35 +00007396 const Decl *D = cxcursor::getCursorDecl(C);
Alp Tokera2794f92014-01-22 07:29:52 +00007397 const CXXMethodDecl *Method =
Craig Topper69186e72014-06-08 08:38:04 +00007398 D ? dyn_cast_or_null<CXXMethodDecl>(D->getAsFunction()) : nullptr;
Dmitri Gribenko62770be2013-05-17 18:38:35 +00007399 return (Method && Method->isVirtual() && Method->isPure()) ? 1 : 0;
7400}
7401
Dmitri Gribenkoe570ede2014-04-07 14:59:13 +00007402unsigned clang_CXXMethod_isConst(CXCursor C) {
7403 if (!clang_isDeclaration(C.kind))
7404 return 0;
7405
7406 const Decl *D = cxcursor::getCursorDecl(C);
7407 const CXXMethodDecl *Method =
Craig Topper69186e72014-06-08 08:38:04 +00007408 D ? dyn_cast_or_null<CXXMethodDecl>(D->getAsFunction()) : nullptr;
Dmitri Gribenkoe570ede2014-04-07 14:59:13 +00007409 return (Method && (Method->getTypeQualifiers() & Qualifiers::Const)) ? 1 : 0;
7410}
7411
Guy Benyei11169dd2012-12-18 14:30:41 +00007412unsigned clang_CXXMethod_isStatic(CXCursor C) {
7413 if (!clang_isDeclaration(C.kind))
7414 return 0;
7415
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007416 const Decl *D = cxcursor::getCursorDecl(C);
Alp Tokera2794f92014-01-22 07:29:52 +00007417 const CXXMethodDecl *Method =
Craig Topper69186e72014-06-08 08:38:04 +00007418 D ? dyn_cast_or_null<CXXMethodDecl>(D->getAsFunction()) : nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007419 return (Method && Method->isStatic()) ? 1 : 0;
7420}
7421
7422unsigned clang_CXXMethod_isVirtual(CXCursor C) {
7423 if (!clang_isDeclaration(C.kind))
7424 return 0;
7425
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007426 const Decl *D = cxcursor::getCursorDecl(C);
Alp Tokera2794f92014-01-22 07:29:52 +00007427 const CXXMethodDecl *Method =
Craig Topper69186e72014-06-08 08:38:04 +00007428 D ? dyn_cast_or_null<CXXMethodDecl>(D->getAsFunction()) : nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007429 return (Method && Method->isVirtual()) ? 1 : 0;
7430}
7431} // end: extern "C"
7432
7433//===----------------------------------------------------------------------===//
7434// Attribute introspection.
7435//===----------------------------------------------------------------------===//
7436
7437extern "C" {
7438CXType clang_getIBOutletCollectionType(CXCursor C) {
7439 if (C.kind != CXCursor_IBOutletCollectionAttr)
7440 return cxtype::MakeCXType(QualType(), cxcursor::getCursorTU(C));
7441
Dmitri Gribenkoe4baea62013-01-26 18:08:08 +00007442 const IBOutletCollectionAttr *A =
Guy Benyei11169dd2012-12-18 14:30:41 +00007443 cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(C));
7444
7445 return cxtype::MakeCXType(A->getInterface(), cxcursor::getCursorTU(C));
7446}
7447} // end: extern "C"
7448
7449//===----------------------------------------------------------------------===//
7450// Inspecting memory usage.
7451//===----------------------------------------------------------------------===//
7452
7453typedef std::vector<CXTUResourceUsageEntry> MemUsageEntries;
7454
7455static inline void createCXTUResourceUsageEntry(MemUsageEntries &entries,
7456 enum CXTUResourceUsageKind k,
7457 unsigned long amount) {
7458 CXTUResourceUsageEntry entry = { k, amount };
7459 entries.push_back(entry);
7460}
7461
7462extern "C" {
7463
7464const char *clang_getTUResourceUsageName(CXTUResourceUsageKind kind) {
7465 const char *str = "";
7466 switch (kind) {
7467 case CXTUResourceUsage_AST:
7468 str = "ASTContext: expressions, declarations, and types";
7469 break;
7470 case CXTUResourceUsage_Identifiers:
7471 str = "ASTContext: identifiers";
7472 break;
7473 case CXTUResourceUsage_Selectors:
7474 str = "ASTContext: selectors";
7475 break;
7476 case CXTUResourceUsage_GlobalCompletionResults:
7477 str = "Code completion: cached global results";
7478 break;
7479 case CXTUResourceUsage_SourceManagerContentCache:
7480 str = "SourceManager: content cache allocator";
7481 break;
7482 case CXTUResourceUsage_AST_SideTables:
7483 str = "ASTContext: side tables";
7484 break;
7485 case CXTUResourceUsage_SourceManager_Membuffer_Malloc:
7486 str = "SourceManager: malloc'ed memory buffers";
7487 break;
7488 case CXTUResourceUsage_SourceManager_Membuffer_MMap:
7489 str = "SourceManager: mmap'ed memory buffers";
7490 break;
7491 case CXTUResourceUsage_ExternalASTSource_Membuffer_Malloc:
7492 str = "ExternalASTSource: malloc'ed memory buffers";
7493 break;
7494 case CXTUResourceUsage_ExternalASTSource_Membuffer_MMap:
7495 str = "ExternalASTSource: mmap'ed memory buffers";
7496 break;
7497 case CXTUResourceUsage_Preprocessor:
7498 str = "Preprocessor: malloc'ed memory";
7499 break;
7500 case CXTUResourceUsage_PreprocessingRecord:
7501 str = "Preprocessor: PreprocessingRecord";
7502 break;
7503 case CXTUResourceUsage_SourceManager_DataStructures:
7504 str = "SourceManager: data structures and tables";
7505 break;
7506 case CXTUResourceUsage_Preprocessor_HeaderSearch:
7507 str = "Preprocessor: header search tables";
7508 break;
7509 }
7510 return str;
7511}
7512
7513CXTUResourceUsage clang_getCXTUResourceUsage(CXTranslationUnit TU) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00007514 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00007515 LOG_BAD_TU(TU);
Craig Topper69186e72014-06-08 08:38:04 +00007516 CXTUResourceUsage usage = { (void*) nullptr, 0, nullptr };
Guy Benyei11169dd2012-12-18 14:30:41 +00007517 return usage;
7518 }
7519
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00007520 ASTUnit *astUnit = cxtu::getASTUnit(TU);
Ahmed Charlesb8984322014-03-07 20:03:18 +00007521 std::unique_ptr<MemUsageEntries> entries(new MemUsageEntries());
Guy Benyei11169dd2012-12-18 14:30:41 +00007522 ASTContext &astContext = astUnit->getASTContext();
7523
7524 // How much memory is used by AST nodes and types?
7525 createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_AST,
7526 (unsigned long) astContext.getASTAllocatedMemory());
7527
7528 // How much memory is used by identifiers?
7529 createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_Identifiers,
7530 (unsigned long) astContext.Idents.getAllocator().getTotalMemory());
7531
7532 // How much memory is used for selectors?
7533 createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_Selectors,
7534 (unsigned long) astContext.Selectors.getTotalMemory());
7535
7536 // How much memory is used by ASTContext's side tables?
7537 createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_AST_SideTables,
7538 (unsigned long) astContext.getSideTableAllocatedMemory());
7539
7540 // How much memory is used for caching global code completion results?
7541 unsigned long completionBytes = 0;
7542 if (GlobalCodeCompletionAllocator *completionAllocator =
Alp Tokerf994cef2014-07-05 03:08:06 +00007543 astUnit->getCachedCompletionAllocator().get()) {
Guy Benyei11169dd2012-12-18 14:30:41 +00007544 completionBytes = completionAllocator->getTotalMemory();
7545 }
7546 createCXTUResourceUsageEntry(*entries,
7547 CXTUResourceUsage_GlobalCompletionResults,
7548 completionBytes);
7549
7550 // How much memory is being used by SourceManager's content cache?
7551 createCXTUResourceUsageEntry(*entries,
7552 CXTUResourceUsage_SourceManagerContentCache,
7553 (unsigned long) astContext.getSourceManager().getContentCacheSize());
7554
7555 // How much memory is being used by the MemoryBuffer's in SourceManager?
7556 const SourceManager::MemoryBufferSizes &srcBufs =
7557 astUnit->getSourceManager().getMemoryBufferSizes();
7558
7559 createCXTUResourceUsageEntry(*entries,
7560 CXTUResourceUsage_SourceManager_Membuffer_Malloc,
7561 (unsigned long) srcBufs.malloc_bytes);
7562 createCXTUResourceUsageEntry(*entries,
7563 CXTUResourceUsage_SourceManager_Membuffer_MMap,
7564 (unsigned long) srcBufs.mmap_bytes);
7565 createCXTUResourceUsageEntry(*entries,
7566 CXTUResourceUsage_SourceManager_DataStructures,
7567 (unsigned long) astContext.getSourceManager()
7568 .getDataStructureSizes());
7569
7570 // How much memory is being used by the ExternalASTSource?
7571 if (ExternalASTSource *esrc = astContext.getExternalSource()) {
7572 const ExternalASTSource::MemoryBufferSizes &sizes =
7573 esrc->getMemoryBufferSizes();
7574
7575 createCXTUResourceUsageEntry(*entries,
7576 CXTUResourceUsage_ExternalASTSource_Membuffer_Malloc,
7577 (unsigned long) sizes.malloc_bytes);
7578 createCXTUResourceUsageEntry(*entries,
7579 CXTUResourceUsage_ExternalASTSource_Membuffer_MMap,
7580 (unsigned long) sizes.mmap_bytes);
7581 }
7582
7583 // How much memory is being used by the Preprocessor?
7584 Preprocessor &pp = astUnit->getPreprocessor();
7585 createCXTUResourceUsageEntry(*entries,
7586 CXTUResourceUsage_Preprocessor,
7587 pp.getTotalMemory());
7588
7589 if (PreprocessingRecord *pRec = pp.getPreprocessingRecord()) {
7590 createCXTUResourceUsageEntry(*entries,
7591 CXTUResourceUsage_PreprocessingRecord,
7592 pRec->getTotalMemory());
7593 }
7594
7595 createCXTUResourceUsageEntry(*entries,
7596 CXTUResourceUsage_Preprocessor_HeaderSearch,
7597 pp.getHeaderSearchInfo().getTotalMemory());
Craig Topper69186e72014-06-08 08:38:04 +00007598
Guy Benyei11169dd2012-12-18 14:30:41 +00007599 CXTUResourceUsage usage = { (void*) entries.get(),
7600 (unsigned) entries->size(),
Alexander Kornienko6ee521c2015-01-23 15:36:10 +00007601 !entries->empty() ? &(*entries)[0] : nullptr };
Ahmed Charles9a16beb2014-03-07 19:33:25 +00007602 entries.release();
Guy Benyei11169dd2012-12-18 14:30:41 +00007603 return usage;
7604}
7605
7606void clang_disposeCXTUResourceUsage(CXTUResourceUsage usage) {
7607 if (usage.data)
7608 delete (MemUsageEntries*) usage.data;
7609}
7610
Argyrios Kyrtzidis0e282ef2013-12-06 18:55:45 +00007611CXSourceRangeList *clang_getSkippedRanges(CXTranslationUnit TU, CXFile file) {
7612 CXSourceRangeList *skipped = new CXSourceRangeList;
Argyrios Kyrtzidis9ef57752013-12-05 08:19:32 +00007613 skipped->count = 0;
Craig Topper69186e72014-06-08 08:38:04 +00007614 skipped->ranges = nullptr;
Argyrios Kyrtzidis9ef57752013-12-05 08:19:32 +00007615
Dmitri Gribenko852d6222014-02-11 15:02:48 +00007616 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00007617 LOG_BAD_TU(TU);
7618 return skipped;
7619 }
7620
Argyrios Kyrtzidis9ef57752013-12-05 08:19:32 +00007621 if (!file)
7622 return skipped;
7623
7624 ASTUnit *astUnit = cxtu::getASTUnit(TU);
7625 PreprocessingRecord *ppRec = astUnit->getPreprocessor().getPreprocessingRecord();
7626 if (!ppRec)
7627 return skipped;
7628
7629 ASTContext &Ctx = astUnit->getASTContext();
7630 SourceManager &sm = Ctx.getSourceManager();
7631 FileEntry *fileEntry = static_cast<FileEntry *>(file);
7632 FileID wantedFileID = sm.translateFile(fileEntry);
7633
7634 const std::vector<SourceRange> &SkippedRanges = ppRec->getSkippedRanges();
7635 std::vector<SourceRange> wantedRanges;
7636 for (std::vector<SourceRange>::const_iterator i = SkippedRanges.begin(), ei = SkippedRanges.end();
7637 i != ei; ++i) {
7638 if (sm.getFileID(i->getBegin()) == wantedFileID || sm.getFileID(i->getEnd()) == wantedFileID)
7639 wantedRanges.push_back(*i);
7640 }
7641
7642 skipped->count = wantedRanges.size();
7643 skipped->ranges = new CXSourceRange[skipped->count];
7644 for (unsigned i = 0, ei = skipped->count; i != ei; ++i)
7645 skipped->ranges[i] = cxloc::translateSourceRange(Ctx, wantedRanges[i]);
7646
7647 return skipped;
7648}
7649
Argyrios Kyrtzidis0e282ef2013-12-06 18:55:45 +00007650void clang_disposeSourceRangeList(CXSourceRangeList *ranges) {
7651 if (ranges) {
7652 delete[] ranges->ranges;
7653 delete ranges;
Argyrios Kyrtzidis9ef57752013-12-05 08:19:32 +00007654 }
7655}
7656
Guy Benyei11169dd2012-12-18 14:30:41 +00007657} // end extern "C"
7658
7659void clang::PrintLibclangResourceUsage(CXTranslationUnit TU) {
7660 CXTUResourceUsage Usage = clang_getCXTUResourceUsage(TU);
7661 for (unsigned I = 0; I != Usage.numEntries; ++I)
7662 fprintf(stderr, " %s: %lu\n",
7663 clang_getTUResourceUsageName(Usage.entries[I].kind),
7664 Usage.entries[I].amount);
7665
7666 clang_disposeCXTUResourceUsage(Usage);
7667}
7668
7669//===----------------------------------------------------------------------===//
7670// Misc. utility functions.
7671//===----------------------------------------------------------------------===//
7672
7673/// Default to using an 8 MB stack size on "safety" threads.
7674static unsigned SafetyStackThreadSize = 8 << 20;
7675
7676namespace clang {
7677
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00007678bool RunSafely(llvm::CrashRecoveryContext &CRC, llvm::function_ref<void()> Fn,
Guy Benyei11169dd2012-12-18 14:30:41 +00007679 unsigned Size) {
7680 if (!Size)
7681 Size = GetSafetyThreadStackSize();
7682 if (Size)
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00007683 return CRC.RunSafelyOnThread(Fn, Size);
7684 return CRC.RunSafely(Fn);
Guy Benyei11169dd2012-12-18 14:30:41 +00007685}
7686
7687unsigned GetSafetyThreadStackSize() {
7688 return SafetyStackThreadSize;
7689}
7690
7691void SetSafetyThreadStackSize(unsigned Value) {
7692 SafetyStackThreadSize = Value;
7693}
7694
Alexander Kornienkoab9db512015-06-22 23:07:51 +00007695}
Guy Benyei11169dd2012-12-18 14:30:41 +00007696
7697void clang::setThreadBackgroundPriority() {
7698 if (getenv("LIBCLANG_BGPRIO_DISABLE"))
7699 return;
7700
Alp Toker1a86ad22014-07-06 06:24:00 +00007701#ifdef USE_DARWIN_THREADS
Guy Benyei11169dd2012-12-18 14:30:41 +00007702 setpriority(PRIO_DARWIN_THREAD, 0, PRIO_DARWIN_BG);
7703#endif
7704}
7705
7706void cxindex::printDiagsToStderr(ASTUnit *Unit) {
7707 if (!Unit)
7708 return;
7709
7710 for (ASTUnit::stored_diag_iterator D = Unit->stored_diag_begin(),
7711 DEnd = Unit->stored_diag_end();
7712 D != DEnd; ++D) {
Ben Langmuir749323f2014-04-22 17:40:12 +00007713 CXStoredDiagnostic Diag(*D, Unit->getLangOpts());
Guy Benyei11169dd2012-12-18 14:30:41 +00007714 CXString Msg = clang_formatDiagnostic(&Diag,
7715 clang_defaultDiagnosticDisplayOptions());
7716 fprintf(stderr, "%s\n", clang_getCString(Msg));
7717 clang_disposeString(Msg);
7718 }
7719#ifdef LLVM_ON_WIN32
7720 // On Windows, force a flush, since there may be multiple copies of
7721 // stderr and stdout in the file system, all with different buffers
7722 // but writing to the same device.
7723 fflush(stderr);
7724#endif
7725}
7726
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007727MacroInfo *cxindex::getMacroInfo(const IdentifierInfo &II,
7728 SourceLocation MacroDefLoc,
7729 CXTranslationUnit TU){
7730 if (MacroDefLoc.isInvalid() || !TU)
Craig Topper69186e72014-06-08 08:38:04 +00007731 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007732 if (!II.hadMacroDefinition())
Craig Topper69186e72014-06-08 08:38:04 +00007733 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007734
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00007735 ASTUnit *Unit = cxtu::getASTUnit(TU);
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00007736 Preprocessor &PP = Unit->getPreprocessor();
Richard Smith20e883e2015-04-29 23:20:19 +00007737 MacroDirective *MD = PP.getLocalMacroDirectiveHistory(&II);
Argyrios Kyrtzidisb6210df2013-03-26 17:17:01 +00007738 if (MD) {
7739 for (MacroDirective::DefInfo
7740 Def = MD->getDefinition(); Def; Def = Def.getPreviousDefinition()) {
7741 if (MacroDefLoc == Def.getMacroInfo()->getDefinitionLoc())
7742 return Def.getMacroInfo();
7743 }
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007744 }
7745
Craig Topper69186e72014-06-08 08:38:04 +00007746 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007747}
7748
Richard Smith66a81862015-05-04 02:25:31 +00007749const MacroInfo *cxindex::getMacroInfo(const MacroDefinitionRecord *MacroDef,
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00007750 CXTranslationUnit TU) {
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007751 if (!MacroDef || !TU)
Craig Topper69186e72014-06-08 08:38:04 +00007752 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007753 const IdentifierInfo *II = MacroDef->getName();
7754 if (!II)
Craig Topper69186e72014-06-08 08:38:04 +00007755 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007756
7757 return getMacroInfo(*II, MacroDef->getLocation(), TU);
7758}
7759
Richard Smith66a81862015-05-04 02:25:31 +00007760MacroDefinitionRecord *
7761cxindex::checkForMacroInMacroDefinition(const MacroInfo *MI, const Token &Tok,
7762 CXTranslationUnit TU) {
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007763 if (!MI || !TU)
Craig Topper69186e72014-06-08 08:38:04 +00007764 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007765 if (Tok.isNot(tok::raw_identifier))
Craig Topper69186e72014-06-08 08:38:04 +00007766 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007767
7768 if (MI->getNumTokens() == 0)
Craig Topper69186e72014-06-08 08:38:04 +00007769 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007770 SourceRange DefRange(MI->getReplacementToken(0).getLocation(),
7771 MI->getDefinitionEndLoc());
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00007772 ASTUnit *Unit = cxtu::getASTUnit(TU);
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007773
7774 // Check that the token is inside the definition and not its argument list.
7775 SourceManager &SM = Unit->getSourceManager();
7776 if (SM.isBeforeInTranslationUnit(Tok.getLocation(), DefRange.getBegin()))
Craig Topper69186e72014-06-08 08:38:04 +00007777 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007778 if (SM.isBeforeInTranslationUnit(DefRange.getEnd(), Tok.getLocation()))
Craig Topper69186e72014-06-08 08:38:04 +00007779 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007780
7781 Preprocessor &PP = Unit->getPreprocessor();
7782 PreprocessingRecord *PPRec = PP.getPreprocessingRecord();
7783 if (!PPRec)
Craig Topper69186e72014-06-08 08:38:04 +00007784 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007785
Alp Toker2d57cea2014-05-17 04:53:25 +00007786 IdentifierInfo &II = PP.getIdentifierTable().get(Tok.getRawIdentifier());
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007787 if (!II.hadMacroDefinition())
Craig Topper69186e72014-06-08 08:38:04 +00007788 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007789
7790 // Check that the identifier is not one of the macro arguments.
7791 if (std::find(MI->arg_begin(), MI->arg_end(), &II) != MI->arg_end())
Craig Topper69186e72014-06-08 08:38:04 +00007792 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007793
Richard Smith20e883e2015-04-29 23:20:19 +00007794 MacroDirective *InnerMD = PP.getLocalMacroDirectiveHistory(&II);
Argyrios Kyrtzidis09c9e812013-02-20 00:54:57 +00007795 if (!InnerMD)
Craig Topper69186e72014-06-08 08:38:04 +00007796 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007797
Argyrios Kyrtzidisb6210df2013-03-26 17:17:01 +00007798 return PPRec->findMacroDefinition(InnerMD->getMacroInfo());
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007799}
7800
Richard Smith66a81862015-05-04 02:25:31 +00007801MacroDefinitionRecord *
7802cxindex::checkForMacroInMacroDefinition(const MacroInfo *MI, SourceLocation Loc,
7803 CXTranslationUnit TU) {
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007804 if (Loc.isInvalid() || !MI || !TU)
Craig Topper69186e72014-06-08 08:38:04 +00007805 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007806
7807 if (MI->getNumTokens() == 0)
Craig Topper69186e72014-06-08 08:38:04 +00007808 return nullptr;
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00007809 ASTUnit *Unit = cxtu::getASTUnit(TU);
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007810 Preprocessor &PP = Unit->getPreprocessor();
7811 if (!PP.getPreprocessingRecord())
Craig Topper69186e72014-06-08 08:38:04 +00007812 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007813 Loc = Unit->getSourceManager().getSpellingLoc(Loc);
7814 Token Tok;
7815 if (PP.getRawToken(Loc, Tok))
Craig Topper69186e72014-06-08 08:38:04 +00007816 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007817
7818 return checkForMacroInMacroDefinition(MI, Tok, TU);
7819}
7820
Guy Benyei11169dd2012-12-18 14:30:41 +00007821extern "C" {
7822
7823CXString clang_getClangVersion() {
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00007824 return cxstring::createDup(getClangFullVersion());
Guy Benyei11169dd2012-12-18 14:30:41 +00007825}
7826
7827} // end: extern "C"
7828
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007829Logger &cxindex::Logger::operator<<(CXTranslationUnit TU) {
7830 if (TU) {
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00007831 if (ASTUnit *Unit = cxtu::getASTUnit(TU)) {
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007832 LogOS << '<' << Unit->getMainFileName() << '>';
Argyrios Kyrtzidis37f2ab42013-03-05 20:21:14 +00007833 if (Unit->isMainFileAST())
7834 LogOS << " (" << Unit->getASTFileName() << ')';
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007835 return *this;
7836 }
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00007837 } else {
7838 LogOS << "<NULL TU>";
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007839 }
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007840 return *this;
7841}
7842
Argyrios Kyrtzidisba4b5f82013-03-08 02:32:26 +00007843Logger &cxindex::Logger::operator<<(const FileEntry *FE) {
7844 *this << FE->getName();
7845 return *this;
7846}
7847
7848Logger &cxindex::Logger::operator<<(CXCursor cursor) {
7849 CXString cursorName = clang_getCursorDisplayName(cursor);
7850 *this << cursorName << "@" << clang_getCursorLocation(cursor);
7851 clang_disposeString(cursorName);
7852 return *this;
7853}
7854
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007855Logger &cxindex::Logger::operator<<(CXSourceLocation Loc) {
7856 CXFile File;
7857 unsigned Line, Column;
Craig Topper69186e72014-06-08 08:38:04 +00007858 clang_getFileLocation(Loc, &File, &Line, &Column, nullptr);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007859 CXString FileName = clang_getFileName(File);
7860 *this << llvm::format("(%s:%d:%d)", clang_getCString(FileName), Line, Column);
7861 clang_disposeString(FileName);
7862 return *this;
7863}
7864
7865Logger &cxindex::Logger::operator<<(CXSourceRange range) {
7866 CXSourceLocation BLoc = clang_getRangeStart(range);
7867 CXSourceLocation ELoc = clang_getRangeEnd(range);
7868
7869 CXFile BFile;
7870 unsigned BLine, BColumn;
Craig Topper69186e72014-06-08 08:38:04 +00007871 clang_getFileLocation(BLoc, &BFile, &BLine, &BColumn, nullptr);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007872
7873 CXFile EFile;
7874 unsigned ELine, EColumn;
Craig Topper69186e72014-06-08 08:38:04 +00007875 clang_getFileLocation(ELoc, &EFile, &ELine, &EColumn, nullptr);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007876
7877 CXString BFileName = clang_getFileName(BFile);
7878 if (BFile == EFile) {
7879 *this << llvm::format("[%s %d:%d-%d:%d]", clang_getCString(BFileName),
7880 BLine, BColumn, ELine, EColumn);
7881 } else {
7882 CXString EFileName = clang_getFileName(EFile);
7883 *this << llvm::format("[%s:%d:%d - ", clang_getCString(BFileName),
7884 BLine, BColumn)
7885 << llvm::format("%s:%d:%d]", clang_getCString(EFileName),
7886 ELine, EColumn);
7887 clang_disposeString(EFileName);
7888 }
7889 clang_disposeString(BFileName);
7890 return *this;
7891}
7892
7893Logger &cxindex::Logger::operator<<(CXString Str) {
7894 *this << clang_getCString(Str);
7895 return *this;
7896}
7897
7898Logger &cxindex::Logger::operator<<(const llvm::format_object_base &Fmt) {
7899 LogOS << Fmt;
7900 return *this;
7901}
7902
Chandler Carruth37ad2582014-06-27 15:14:39 +00007903static llvm::ManagedStatic<llvm::sys::Mutex> LoggingMutex;
7904
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007905cxindex::Logger::~Logger() {
Chandler Carruth37ad2582014-06-27 15:14:39 +00007906 llvm::sys::ScopedLock L(*LoggingMutex);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007907
7908 static llvm::TimeRecord sBeginTR = llvm::TimeRecord::getCurrentTime();
7909
Dmitri Gribenkof8579502013-01-12 19:30:44 +00007910 raw_ostream &OS = llvm::errs();
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007911 OS << "[libclang:" << Name << ':';
7912
Alp Toker1a86ad22014-07-06 06:24:00 +00007913#ifdef USE_DARWIN_THREADS
7914 // TODO: Portability.
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007915 mach_port_t tid = pthread_mach_thread_np(pthread_self());
7916 OS << tid << ':';
7917#endif
7918
7919 llvm::TimeRecord TR = llvm::TimeRecord::getCurrentTime();
7920 OS << llvm::format("%7.4f] ", TR.getWallTime() - sBeginTR.getWallTime());
Yaron Keren09fb7c62015-03-10 07:33:23 +00007921 OS << Msg << '\n';
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007922
7923 if (Trace) {
Zachary Turner1fe2a8d2015-03-05 19:15:09 +00007924 llvm::sys::PrintStackTrace(OS);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007925 OS << "--------------------------------------------------\n";
7926 }
7927}
Benjamin Kramerc1ffdab2016-03-03 08:58:18 +00007928
7929#ifdef CLANG_TOOL_EXTRA_BUILD
7930// This anchor is used to force the linker to link the clang-tidy plugin.
7931extern volatile int ClangTidyPluginAnchorSource;
7932static int LLVM_ATTRIBUTE_UNUSED ClangTidyPluginAnchorDestination =
7933 ClangTidyPluginAnchorSource;
7934#endif