blob: 7a64de2ad3e9fe5c4fc525267feab031325a17e8 [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
Guy Benyei11169dd2012-12-18 14:30:41 +000015#include "CIndexDiagnostic.h"
Mehdi Amini9670f842016-07-18 19:02:11 +000016#include "CIndexer.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"
Emilio Cobos Alvarez485ad422017-04-28 15:56:39 +000029#include "clang/Basic/TargetInfo.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000030#include "clang/Basic/Version.h"
31#include "clang/Frontend/ASTUnit.h"
32#include "clang/Frontend/CompilerInstance.h"
33#include "clang/Frontend/FrontendDiagnostic.h"
Argyrios Kyrtzidisca741ce2016-02-14 22:30:14 +000034#include "clang/Index/CodegenNameGenerator.h"
Dmitri Gribenko9e605112013-11-13 22:16:51 +000035#include "clang/Index/CommentToXML.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000036#include "clang/Lex/HeaderSearch.h"
37#include "clang/Lex/Lexer.h"
38#include "clang/Lex/PreprocessingRecord.h"
39#include "clang/Lex/Preprocessor.h"
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +000040#include "clang/Serialization/SerializationDiagnostic.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000041#include "llvm/ADT/Optional.h"
42#include "llvm/ADT/STLExtras.h"
43#include "llvm/ADT/StringSwitch.h"
Alp Toker1d257e12014-06-04 03:28:55 +000044#include "llvm/Config/llvm-config.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000045#include "llvm/Support/Compiler.h"
46#include "llvm/Support/CrashRecoveryContext.h"
Chandler Carruth4b417452013-01-19 08:09:44 +000047#include "llvm/Support/Format.h"
Chandler Carruth37ad2582014-06-27 15:14:39 +000048#include "llvm/Support/ManagedStatic.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000049#include "llvm/Support/MemoryBuffer.h"
50#include "llvm/Support/Mutex.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000051#include "llvm/Support/Program.h"
52#include "llvm/Support/SaveAndRestore.h"
53#include "llvm/Support/Signals.h"
Adrian Prantlbc068582015-07-08 01:00:30 +000054#include "llvm/Support/TargetSelect.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000055#include "llvm/Support/Threading.h"
56#include "llvm/Support/Timer.h"
57#include "llvm/Support/raw_ostream.h"
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +000058
Alp Toker1a86ad22014-07-06 06:24:00 +000059#if LLVM_ENABLE_THREADS != 0 && defined(__APPLE__)
60#define USE_DARWIN_THREADS
61#endif
62
63#ifdef USE_DARWIN_THREADS
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +000064#include <pthread.h>
65#endif
Guy Benyei11169dd2012-12-18 14:30:41 +000066
67using namespace clang;
68using namespace clang::cxcursor;
Guy Benyei11169dd2012-12-18 14:30:41 +000069using namespace clang::cxtu;
70using namespace clang::cxindex;
71
David Blaikieea4395e2017-01-06 19:49:01 +000072CXTranslationUnit cxtu::MakeCXTranslationUnit(CIndexer *CIdx,
73 std::unique_ptr<ASTUnit> AU) {
Dmitri Gribenkod36209e2013-01-26 21:32:42 +000074 if (!AU)
Craig Topper69186e72014-06-08 08:38:04 +000075 return nullptr;
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +000076 assert(CIdx);
Guy Benyei11169dd2012-12-18 14:30:41 +000077 CXTranslationUnit D = new CXTranslationUnitImpl();
78 D->CIdx = CIdx;
David Blaikieea4395e2017-01-06 19:49:01 +000079 D->TheASTUnit = AU.release();
Dmitri Gribenko74895212013-02-03 13:52:47 +000080 D->StringPool = new cxstring::CXStringPool();
Craig Topper69186e72014-06-08 08:38:04 +000081 D->Diagnostics = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +000082 D->OverridenCursorsPool = createOverridenCXCursorsPool();
Craig Topper69186e72014-06-08 08:38:04 +000083 D->CommentToXML = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +000084 return D;
85}
86
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +000087bool cxtu::isASTReadError(ASTUnit *AU) {
88 for (ASTUnit::stored_diag_iterator D = AU->stored_diag_begin(),
89 DEnd = AU->stored_diag_end();
90 D != DEnd; ++D) {
91 if (D->getLevel() >= DiagnosticsEngine::Error &&
92 DiagnosticIDs::getCategoryNumberForDiag(D->getID()) ==
93 diag::DiagCat_AST_Deserialization_Issue)
94 return true;
95 }
96 return false;
97}
98
Guy Benyei11169dd2012-12-18 14:30:41 +000099cxtu::CXTUOwner::~CXTUOwner() {
100 if (TU)
101 clang_disposeTranslationUnit(TU);
102}
103
104/// \brief Compare two source ranges to determine their relative position in
105/// the translation unit.
106static RangeComparisonResult RangeCompare(SourceManager &SM,
107 SourceRange R1,
108 SourceRange R2) {
109 assert(R1.isValid() && "First range is invalid?");
110 assert(R2.isValid() && "Second range is invalid?");
111 if (R1.getEnd() != R2.getBegin() &&
112 SM.isBeforeInTranslationUnit(R1.getEnd(), R2.getBegin()))
113 return RangeBefore;
114 if (R2.getEnd() != R1.getBegin() &&
115 SM.isBeforeInTranslationUnit(R2.getEnd(), R1.getBegin()))
116 return RangeAfter;
117 return RangeOverlap;
118}
119
120/// \brief Determine if a source location falls within, before, or after a
121/// a given source range.
122static RangeComparisonResult LocationCompare(SourceManager &SM,
123 SourceLocation L, SourceRange R) {
124 assert(R.isValid() && "First range is invalid?");
125 assert(L.isValid() && "Second range is invalid?");
126 if (L == R.getBegin() || L == R.getEnd())
127 return RangeOverlap;
128 if (SM.isBeforeInTranslationUnit(L, R.getBegin()))
129 return RangeBefore;
130 if (SM.isBeforeInTranslationUnit(R.getEnd(), L))
131 return RangeAfter;
132 return RangeOverlap;
133}
134
135/// \brief Translate a Clang source range into a CIndex source range.
136///
137/// Clang internally represents ranges where the end location points to the
138/// start of the token at the end. However, for external clients it is more
139/// useful to have a CXSourceRange be a proper half-open interval. This routine
140/// does the appropriate translation.
141CXSourceRange cxloc::translateSourceRange(const SourceManager &SM,
142 const LangOptions &LangOpts,
143 const CharSourceRange &R) {
144 // We want the last character in this location, so we will adjust the
145 // location accordingly.
146 SourceLocation EndLoc = R.getEnd();
147 if (EndLoc.isValid() && EndLoc.isMacroID() && !SM.isMacroArgExpansion(EndLoc))
148 EndLoc = SM.getExpansionRange(EndLoc).second;
Yaron Keren8b563662015-10-03 10:46:20 +0000149 if (R.isTokenRange() && EndLoc.isValid()) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000150 unsigned Length = Lexer::MeasureTokenLength(SM.getSpellingLoc(EndLoc),
151 SM, LangOpts);
152 EndLoc = EndLoc.getLocWithOffset(Length);
153 }
154
Bill Wendlingeade3622013-01-23 08:25:41 +0000155 CXSourceRange Result = {
Dmitri Gribenkof9304482013-01-23 15:56:07 +0000156 { &SM, &LangOpts },
Bill Wendlingeade3622013-01-23 08:25:41 +0000157 R.getBegin().getRawEncoding(),
158 EndLoc.getRawEncoding()
159 };
Guy Benyei11169dd2012-12-18 14:30:41 +0000160 return Result;
161}
162
163//===----------------------------------------------------------------------===//
164// Cursor visitor.
165//===----------------------------------------------------------------------===//
166
167static SourceRange getRawCursorExtent(CXCursor C);
168static SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr);
169
170
171RangeComparisonResult CursorVisitor::CompareRegionOfInterest(SourceRange R) {
172 return RangeCompare(AU->getSourceManager(), R, RegionOfInterest);
173}
174
175/// \brief Visit the given cursor and, if requested by the visitor,
176/// its children.
177///
178/// \param Cursor the cursor to visit.
179///
180/// \param CheckedRegionOfInterest if true, then the caller already checked
181/// that this cursor is within the region of interest.
182///
183/// \returns true if the visitation should be aborted, false if it
184/// should continue.
185bool CursorVisitor::Visit(CXCursor Cursor, bool CheckedRegionOfInterest) {
186 if (clang_isInvalid(Cursor.kind))
187 return false;
188
189 if (clang_isDeclaration(Cursor.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +0000190 const Decl *D = getCursorDecl(Cursor);
Guy Benyei11169dd2012-12-18 14:30:41 +0000191 if (!D) {
192 assert(0 && "Invalid declaration cursor");
193 return true; // abort.
194 }
195
196 // Ignore implicit declarations, unless it's an objc method because
197 // currently we should report implicit methods for properties when indexing.
198 if (D->isImplicit() && !isa<ObjCMethodDecl>(D))
199 return false;
200 }
201
202 // If we have a range of interest, and this cursor doesn't intersect with it,
203 // we're done.
204 if (RegionOfInterest.isValid() && !CheckedRegionOfInterest) {
205 SourceRange Range = getRawCursorExtent(Cursor);
206 if (Range.isInvalid() || CompareRegionOfInterest(Range))
207 return false;
208 }
209
210 switch (Visitor(Cursor, Parent, ClientData)) {
211 case CXChildVisit_Break:
212 return true;
213
214 case CXChildVisit_Continue:
215 return false;
216
217 case CXChildVisit_Recurse: {
218 bool ret = VisitChildren(Cursor);
219 if (PostChildrenVisitor)
220 if (PostChildrenVisitor(Cursor, ClientData))
221 return true;
222 return ret;
223 }
224 }
225
226 llvm_unreachable("Invalid CXChildVisitResult!");
227}
228
229static bool visitPreprocessedEntitiesInRange(SourceRange R,
230 PreprocessingRecord &PPRec,
231 CursorVisitor &Visitor) {
232 SourceManager &SM = Visitor.getASTUnit()->getSourceManager();
233 FileID FID;
234
235 if (!Visitor.shouldVisitIncludedEntities()) {
236 // If the begin/end of the range lie in the same FileID, do the optimization
237 // where we skip preprocessed entities that do not come from the same FileID.
238 FID = SM.getFileID(SM.getFileLoc(R.getBegin()));
239 if (FID != SM.getFileID(SM.getFileLoc(R.getEnd())))
240 FID = FileID();
241 }
242
Benjamin Kramerb4ef6682015-02-06 17:25:10 +0000243 const auto &Entities = PPRec.getPreprocessedEntitiesInRange(R);
244 return Visitor.visitPreprocessedEntities(Entities.begin(), Entities.end(),
Guy Benyei11169dd2012-12-18 14:30:41 +0000245 PPRec, FID);
246}
247
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000248bool CursorVisitor::visitFileRegion() {
Guy Benyei11169dd2012-12-18 14:30:41 +0000249 if (RegionOfInterest.isInvalid())
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000250 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000251
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +0000252 ASTUnit *Unit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +0000253 SourceManager &SM = Unit->getSourceManager();
254
255 std::pair<FileID, unsigned>
256 Begin = SM.getDecomposedLoc(SM.getFileLoc(RegionOfInterest.getBegin())),
257 End = SM.getDecomposedLoc(SM.getFileLoc(RegionOfInterest.getEnd()));
258
259 if (End.first != Begin.first) {
260 // If the end does not reside in the same file, try to recover by
261 // picking the end of the file of begin location.
262 End.first = Begin.first;
263 End.second = SM.getFileIDSize(Begin.first);
264 }
265
266 assert(Begin.first == End.first);
267 if (Begin.second > End.second)
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000268 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000269
270 FileID File = Begin.first;
271 unsigned Offset = Begin.second;
272 unsigned Length = End.second - Begin.second;
273
274 if (!VisitDeclsOnly && !VisitPreprocessorLast)
275 if (visitPreprocessedEntitiesInRegion())
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000276 return true; // visitation break.
Guy Benyei11169dd2012-12-18 14:30:41 +0000277
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000278 if (visitDeclsFromFileRegion(File, Offset, Length))
279 return true; // visitation break.
Guy Benyei11169dd2012-12-18 14:30:41 +0000280
281 if (!VisitDeclsOnly && VisitPreprocessorLast)
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000282 return visitPreprocessedEntitiesInRegion();
283
284 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000285}
286
287static bool isInLexicalContext(Decl *D, DeclContext *DC) {
288 if (!DC)
289 return false;
290
291 for (DeclContext *DeclDC = D->getLexicalDeclContext();
292 DeclDC; DeclDC = DeclDC->getLexicalParent()) {
293 if (DeclDC == DC)
294 return true;
295 }
296 return false;
297}
298
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000299bool CursorVisitor::visitDeclsFromFileRegion(FileID File,
Guy Benyei11169dd2012-12-18 14:30:41 +0000300 unsigned Offset, unsigned Length) {
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +0000301 ASTUnit *Unit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +0000302 SourceManager &SM = Unit->getSourceManager();
303 SourceRange Range = RegionOfInterest;
304
305 SmallVector<Decl *, 16> Decls;
306 Unit->findFileRegionDecls(File, Offset, Length, Decls);
307
308 // If we didn't find any file level decls for the file, try looking at the
309 // file that it was included from.
310 while (Decls.empty() || Decls.front()->isTopLevelDeclInObjCContainer()) {
311 bool Invalid = false;
312 const SrcMgr::SLocEntry &SLEntry = SM.getSLocEntry(File, &Invalid);
313 if (Invalid)
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000314 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000315
316 SourceLocation Outer;
317 if (SLEntry.isFile())
318 Outer = SLEntry.getFile().getIncludeLoc();
319 else
320 Outer = SLEntry.getExpansion().getExpansionLocStart();
321 if (Outer.isInvalid())
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000322 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000323
Benjamin Kramer867ea1d2014-03-02 13:01:17 +0000324 std::tie(File, Offset) = SM.getDecomposedExpansionLoc(Outer);
Guy Benyei11169dd2012-12-18 14:30:41 +0000325 Length = 0;
326 Unit->findFileRegionDecls(File, Offset, Length, Decls);
327 }
328
329 assert(!Decls.empty());
330
331 bool VisitedAtLeastOnce = false;
Craig Topper69186e72014-06-08 08:38:04 +0000332 DeclContext *CurDC = nullptr;
Craig Topper2341c0d2013-07-04 03:08:24 +0000333 SmallVectorImpl<Decl *>::iterator DIt = Decls.begin();
334 for (SmallVectorImpl<Decl *>::iterator DE = Decls.end(); DIt != DE; ++DIt) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000335 Decl *D = *DIt;
336 if (D->getSourceRange().isInvalid())
337 continue;
338
339 if (isInLexicalContext(D, CurDC))
340 continue;
341
342 CurDC = dyn_cast<DeclContext>(D);
343
344 if (TagDecl *TD = dyn_cast<TagDecl>(D))
345 if (!TD->isFreeStanding())
346 continue;
347
348 RangeComparisonResult CompRes = RangeCompare(SM, D->getSourceRange(),Range);
349 if (CompRes == RangeBefore)
350 continue;
351 if (CompRes == RangeAfter)
352 break;
353
354 assert(CompRes == RangeOverlap);
355 VisitedAtLeastOnce = true;
356
357 if (isa<ObjCContainerDecl>(D)) {
358 FileDI_current = &DIt;
359 FileDE_current = DE;
360 } else {
Craig Topper69186e72014-06-08 08:38:04 +0000361 FileDI_current = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +0000362 }
363
364 if (Visit(MakeCXCursor(D, TU, Range), /*CheckedRegionOfInterest=*/true))
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000365 return true; // visitation break.
Guy Benyei11169dd2012-12-18 14:30:41 +0000366 }
367
368 if (VisitedAtLeastOnce)
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000369 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000370
371 // No Decls overlapped with the range. Move up the lexical context until there
372 // is a context that contains the range or we reach the translation unit
373 // level.
374 DeclContext *DC = DIt == Decls.begin() ? (*DIt)->getLexicalDeclContext()
375 : (*(DIt-1))->getLexicalDeclContext();
376
377 while (DC && !DC->isTranslationUnit()) {
378 Decl *D = cast<Decl>(DC);
379 SourceRange CurDeclRange = D->getSourceRange();
380 if (CurDeclRange.isInvalid())
381 break;
382
383 if (RangeCompare(SM, CurDeclRange, Range) == RangeOverlap) {
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000384 if (Visit(MakeCXCursor(D, TU, Range), /*CheckedRegionOfInterest=*/true))
385 return true; // visitation break.
Guy Benyei11169dd2012-12-18 14:30:41 +0000386 }
387
388 DC = D->getLexicalDeclContext();
389 }
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000390
391 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000392}
393
394bool CursorVisitor::visitPreprocessedEntitiesInRegion() {
395 if (!AU->getPreprocessor().getPreprocessingRecord())
396 return false;
397
398 PreprocessingRecord &PPRec
399 = *AU->getPreprocessor().getPreprocessingRecord();
400 SourceManager &SM = AU->getSourceManager();
401
402 if (RegionOfInterest.isValid()) {
403 SourceRange MappedRange = AU->mapRangeToPreamble(RegionOfInterest);
404 SourceLocation B = MappedRange.getBegin();
405 SourceLocation E = MappedRange.getEnd();
406
407 if (AU->isInPreambleFileID(B)) {
408 if (SM.isLoadedSourceLocation(E))
409 return visitPreprocessedEntitiesInRange(SourceRange(B, E),
410 PPRec, *this);
411
412 // Beginning of range lies in the preamble but it also extends beyond
413 // it into the main file. Split the range into 2 parts, one covering
414 // the preamble and another covering the main file. This allows subsequent
415 // calls to visitPreprocessedEntitiesInRange to accept a source range that
416 // lies in the same FileID, allowing it to skip preprocessed entities that
417 // do not come from the same FileID.
418 bool breaked =
419 visitPreprocessedEntitiesInRange(
420 SourceRange(B, AU->getEndOfPreambleFileID()),
421 PPRec, *this);
422 if (breaked) return true;
423 return visitPreprocessedEntitiesInRange(
424 SourceRange(AU->getStartOfMainFileID(), E),
425 PPRec, *this);
426 }
427
428 return visitPreprocessedEntitiesInRange(SourceRange(B, E), PPRec, *this);
429 }
430
431 bool OnlyLocalDecls
432 = !AU->isMainFileAST() && AU->getOnlyLocalDecls();
433
434 if (OnlyLocalDecls)
435 return visitPreprocessedEntities(PPRec.local_begin(), PPRec.local_end(),
436 PPRec);
437
438 return visitPreprocessedEntities(PPRec.begin(), PPRec.end(), PPRec);
439}
440
441template<typename InputIterator>
442bool CursorVisitor::visitPreprocessedEntities(InputIterator First,
443 InputIterator Last,
444 PreprocessingRecord &PPRec,
445 FileID FID) {
446 for (; First != Last; ++First) {
447 if (!FID.isInvalid() && !PPRec.isEntityInFileID(First, FID))
448 continue;
449
450 PreprocessedEntity *PPE = *First;
Argyrios Kyrtzidis1030f262013-05-07 20:37:17 +0000451 if (!PPE)
452 continue;
453
Guy Benyei11169dd2012-12-18 14:30:41 +0000454 if (MacroExpansion *ME = dyn_cast<MacroExpansion>(PPE)) {
455 if (Visit(MakeMacroExpansionCursor(ME, TU)))
456 return true;
Richard Smith66a81862015-05-04 02:25:31 +0000457
Guy Benyei11169dd2012-12-18 14:30:41 +0000458 continue;
459 }
Richard Smith66a81862015-05-04 02:25:31 +0000460
461 if (MacroDefinitionRecord *MD = dyn_cast<MacroDefinitionRecord>(PPE)) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000462 if (Visit(MakeMacroDefinitionCursor(MD, TU)))
463 return true;
Richard Smith66a81862015-05-04 02:25:31 +0000464
Guy Benyei11169dd2012-12-18 14:30:41 +0000465 continue;
466 }
467
468 if (InclusionDirective *ID = dyn_cast<InclusionDirective>(PPE)) {
469 if (Visit(MakeInclusionDirectiveCursor(ID, TU)))
470 return true;
471
472 continue;
473 }
474 }
475
476 return false;
477}
478
479/// \brief Visit the children of the given cursor.
480///
481/// \returns true if the visitation should be aborted, false if it
482/// should continue.
483bool CursorVisitor::VisitChildren(CXCursor Cursor) {
484 if (clang_isReference(Cursor.kind) &&
485 Cursor.kind != CXCursor_CXXBaseSpecifier) {
486 // By definition, references have no children.
487 return false;
488 }
489
490 // Set the Parent field to Cursor, then back to its old value once we're
491 // done.
492 SetParentRAII SetParent(Parent, StmtParent, Cursor);
493
494 if (clang_isDeclaration(Cursor.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +0000495 Decl *D = const_cast<Decl *>(getCursorDecl(Cursor));
Guy Benyei11169dd2012-12-18 14:30:41 +0000496 if (!D)
497 return false;
498
499 return VisitAttributes(D) || Visit(D);
500 }
501
502 if (clang_isStatement(Cursor.kind)) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +0000503 if (const Stmt *S = getCursorStmt(Cursor))
Guy Benyei11169dd2012-12-18 14:30:41 +0000504 return Visit(S);
505
506 return false;
507 }
508
509 if (clang_isExpression(Cursor.kind)) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +0000510 if (const Expr *E = getCursorExpr(Cursor))
Guy Benyei11169dd2012-12-18 14:30:41 +0000511 return Visit(E);
512
513 return false;
514 }
515
516 if (clang_isTranslationUnit(Cursor.kind)) {
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +0000517 CXTranslationUnit TU = getCursorTU(Cursor);
518 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +0000519
520 int VisitOrder[2] = { VisitPreprocessorLast, !VisitPreprocessorLast };
521 for (unsigned I = 0; I != 2; ++I) {
522 if (VisitOrder[I]) {
523 if (!CXXUnit->isMainFileAST() && CXXUnit->getOnlyLocalDecls() &&
524 RegionOfInterest.isInvalid()) {
525 for (ASTUnit::top_level_iterator TL = CXXUnit->top_level_begin(),
526 TLEnd = CXXUnit->top_level_end();
527 TL != TLEnd; ++TL) {
Argyrios Kyrtzidise7c91042016-07-01 19:10:54 +0000528 const Optional<bool> V = handleDeclForVisitation(*TL);
529 if (!V.hasValue())
530 continue;
531 return V.getValue();
Guy Benyei11169dd2012-12-18 14:30:41 +0000532 }
533 } else if (VisitDeclContext(
534 CXXUnit->getASTContext().getTranslationUnitDecl()))
535 return true;
536 continue;
537 }
538
539 // Walk the preprocessing record.
540 if (CXXUnit->getPreprocessor().getPreprocessingRecord())
541 visitPreprocessedEntitiesInRegion();
542 }
543
544 return false;
545 }
546
547 if (Cursor.kind == CXCursor_CXXBaseSpecifier) {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +0000548 if (const CXXBaseSpecifier *Base = getCursorCXXBaseSpecifier(Cursor)) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000549 if (TypeSourceInfo *BaseTSInfo = Base->getTypeSourceInfo()) {
550 return Visit(BaseTSInfo->getTypeLoc());
551 }
552 }
553 }
554
555 if (Cursor.kind == CXCursor_IBOutletCollectionAttr) {
Dmitri Gribenkoe4baea62013-01-26 18:08:08 +0000556 const IBOutletCollectionAttr *A =
Guy Benyei11169dd2012-12-18 14:30:41 +0000557 cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(Cursor));
Richard Smithb1f9a282013-10-31 01:56:18 +0000558 if (const ObjCObjectType *ObjT = A->getInterface()->getAs<ObjCObjectType>())
Richard Smithb87c4652013-10-31 21:23:20 +0000559 return Visit(cxcursor::MakeCursorObjCClassRef(
560 ObjT->getInterface(),
561 A->getInterfaceLoc()->getTypeLoc().getLocStart(), TU));
Guy Benyei11169dd2012-12-18 14:30:41 +0000562 }
563
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +0000564 // If pointing inside a macro definition, check if the token is an identifier
565 // that was ever defined as a macro. In such a case, create a "pseudo" macro
566 // expansion cursor for that token.
567 SourceLocation BeginLoc = RegionOfInterest.getBegin();
568 if (Cursor.kind == CXCursor_MacroDefinition &&
569 BeginLoc == RegionOfInterest.getEnd()) {
570 SourceLocation Loc = AU->mapLocationToPreamble(BeginLoc);
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +0000571 const MacroInfo *MI =
572 getMacroInfo(cxcursor::getCursorMacroDefinition(Cursor), TU);
Richard Smith66a81862015-05-04 02:25:31 +0000573 if (MacroDefinitionRecord *MacroDef =
574 checkForMacroInMacroDefinition(MI, Loc, TU))
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +0000575 return Visit(cxcursor::MakeMacroExpansionCursor(MacroDef, BeginLoc, TU));
576 }
577
Guy Benyei11169dd2012-12-18 14:30:41 +0000578 // Nothing to visit at the moment.
579 return false;
580}
581
582bool CursorVisitor::VisitBlockDecl(BlockDecl *B) {
583 if (TypeSourceInfo *TSInfo = B->getSignatureAsWritten())
584 if (Visit(TSInfo->getTypeLoc()))
585 return true;
586
587 if (Stmt *Body = B->getBody())
588 return Visit(MakeCXCursor(Body, StmtParent, TU, RegionOfInterest));
589
590 return false;
591}
592
Ted Kremenek03325582013-02-21 01:29:01 +0000593Optional<bool> CursorVisitor::shouldVisitCursor(CXCursor Cursor) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000594 if (RegionOfInterest.isValid()) {
595 SourceRange Range = getFullCursorExtent(Cursor, AU->getSourceManager());
596 if (Range.isInvalid())
David Blaikie7a30dc52013-02-21 01:47:18 +0000597 return None;
Guy Benyei11169dd2012-12-18 14:30:41 +0000598
599 switch (CompareRegionOfInterest(Range)) {
600 case RangeBefore:
601 // This declaration comes before the region of interest; skip it.
David Blaikie7a30dc52013-02-21 01:47:18 +0000602 return None;
Guy Benyei11169dd2012-12-18 14:30:41 +0000603
604 case RangeAfter:
605 // This declaration comes after the region of interest; we're done.
606 return false;
607
608 case RangeOverlap:
609 // This declaration overlaps the region of interest; visit it.
610 break;
611 }
612 }
613 return true;
614}
615
616bool CursorVisitor::VisitDeclContext(DeclContext *DC) {
617 DeclContext::decl_iterator I = DC->decls_begin(), E = DC->decls_end();
618
619 // FIXME: Eventually remove. This part of a hack to support proper
620 // iteration over all Decls contained lexically within an ObjC container.
621 SaveAndRestore<DeclContext::decl_iterator*> DI_saved(DI_current, &I);
622 SaveAndRestore<DeclContext::decl_iterator> DE_saved(DE_current, E);
623
624 for ( ; I != E; ++I) {
625 Decl *D = *I;
626 if (D->getLexicalDeclContext() != DC)
627 continue;
Argyrios Kyrtzidise7c91042016-07-01 19:10:54 +0000628 const Optional<bool> V = handleDeclForVisitation(D);
Guy Benyei11169dd2012-12-18 14:30:41 +0000629 if (!V.hasValue())
630 continue;
Argyrios Kyrtzidise7c91042016-07-01 19:10:54 +0000631 return V.getValue();
Guy Benyei11169dd2012-12-18 14:30:41 +0000632 }
633 return false;
634}
635
Argyrios Kyrtzidise7c91042016-07-01 19:10:54 +0000636Optional<bool> CursorVisitor::handleDeclForVisitation(const Decl *D) {
637 CXCursor Cursor = MakeCXCursor(D, TU, RegionOfInterest);
638
639 // Ignore synthesized ivars here, otherwise if we have something like:
640 // @synthesize prop = _prop;
641 // and '_prop' is not declared, we will encounter a '_prop' ivar before
642 // encountering the 'prop' synthesize declaration and we will think that
643 // we passed the region-of-interest.
644 if (auto *ivarD = dyn_cast<ObjCIvarDecl>(D)) {
645 if (ivarD->getSynthesize())
646 return None;
647 }
648
649 // FIXME: ObjCClassRef/ObjCProtocolRef for forward class/protocol
650 // declarations is a mismatch with the compiler semantics.
651 if (Cursor.kind == CXCursor_ObjCInterfaceDecl) {
652 auto *ID = cast<ObjCInterfaceDecl>(D);
653 if (!ID->isThisDeclarationADefinition())
654 Cursor = MakeCursorObjCClassRef(ID, ID->getLocation(), TU);
655
656 } else if (Cursor.kind == CXCursor_ObjCProtocolDecl) {
657 auto *PD = cast<ObjCProtocolDecl>(D);
658 if (!PD->isThisDeclarationADefinition())
659 Cursor = MakeCursorObjCProtocolRef(PD, PD->getLocation(), TU);
660 }
661
662 const Optional<bool> V = shouldVisitCursor(Cursor);
663 if (!V.hasValue())
664 return None;
665 if (!V.getValue())
666 return false;
667 if (Visit(Cursor, true))
668 return true;
669 return None;
670}
671
Guy Benyei11169dd2012-12-18 14:30:41 +0000672bool CursorVisitor::VisitTranslationUnitDecl(TranslationUnitDecl *D) {
673 llvm_unreachable("Translation units are visited directly by Visit()");
674}
675
Sergey Kalinichev8f3b1872015-11-15 13:48:32 +0000676bool CursorVisitor::VisitTypeAliasTemplateDecl(TypeAliasTemplateDecl *D) {
677 if (VisitTemplateParameters(D->getTemplateParameters()))
678 return true;
679
680 return Visit(MakeCXCursor(D->getTemplatedDecl(), TU, RegionOfInterest));
681}
682
Guy Benyei11169dd2012-12-18 14:30:41 +0000683bool CursorVisitor::VisitTypeAliasDecl(TypeAliasDecl *D) {
684 if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
685 return Visit(TSInfo->getTypeLoc());
686
687 return false;
688}
689
690bool CursorVisitor::VisitTypedefDecl(TypedefDecl *D) {
691 if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
692 return Visit(TSInfo->getTypeLoc());
693
694 return false;
695}
696
697bool CursorVisitor::VisitTagDecl(TagDecl *D) {
698 return VisitDeclContext(D);
699}
700
701bool CursorVisitor::VisitClassTemplateSpecializationDecl(
702 ClassTemplateSpecializationDecl *D) {
703 bool ShouldVisitBody = false;
704 switch (D->getSpecializationKind()) {
705 case TSK_Undeclared:
706 case TSK_ImplicitInstantiation:
707 // Nothing to visit
708 return false;
709
710 case TSK_ExplicitInstantiationDeclaration:
711 case TSK_ExplicitInstantiationDefinition:
712 break;
713
714 case TSK_ExplicitSpecialization:
715 ShouldVisitBody = true;
716 break;
717 }
718
719 // Visit the template arguments used in the specialization.
720 if (TypeSourceInfo *SpecType = D->getTypeAsWritten()) {
721 TypeLoc TL = SpecType->getTypeLoc();
David Blaikie6adc78e2013-02-18 22:06:02 +0000722 if (TemplateSpecializationTypeLoc TSTLoc =
723 TL.getAs<TemplateSpecializationTypeLoc>()) {
724 for (unsigned I = 0, N = TSTLoc.getNumArgs(); I != N; ++I)
725 if (VisitTemplateArgumentLoc(TSTLoc.getArgLoc(I)))
Guy Benyei11169dd2012-12-18 14:30:41 +0000726 return true;
727 }
728 }
Alexander Kornienko1a9f1842015-12-28 15:24:08 +0000729
730 return ShouldVisitBody && VisitCXXRecordDecl(D);
Guy Benyei11169dd2012-12-18 14:30:41 +0000731}
732
733bool CursorVisitor::VisitClassTemplatePartialSpecializationDecl(
734 ClassTemplatePartialSpecializationDecl *D) {
735 // FIXME: Visit the "outer" template parameter lists on the TagDecl
736 // before visiting these template parameters.
737 if (VisitTemplateParameters(D->getTemplateParameters()))
738 return true;
739
740 // Visit the partial specialization arguments.
Enea Zaffanella6dbe1872013-08-10 07:24:53 +0000741 const ASTTemplateArgumentListInfo *Info = D->getTemplateArgsAsWritten();
742 const TemplateArgumentLoc *TemplateArgs = Info->getTemplateArgs();
743 for (unsigned I = 0, N = Info->NumTemplateArgs; I != N; ++I)
Guy Benyei11169dd2012-12-18 14:30:41 +0000744 if (VisitTemplateArgumentLoc(TemplateArgs[I]))
745 return true;
746
747 return VisitCXXRecordDecl(D);
748}
749
750bool CursorVisitor::VisitTemplateTypeParmDecl(TemplateTypeParmDecl *D) {
751 // Visit the default argument.
752 if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
753 if (TypeSourceInfo *DefArg = D->getDefaultArgumentInfo())
754 if (Visit(DefArg->getTypeLoc()))
755 return true;
756
757 return false;
758}
759
760bool CursorVisitor::VisitEnumConstantDecl(EnumConstantDecl *D) {
761 if (Expr *Init = D->getInitExpr())
762 return Visit(MakeCXCursor(Init, StmtParent, TU, RegionOfInterest));
763 return false;
764}
765
766bool CursorVisitor::VisitDeclaratorDecl(DeclaratorDecl *DD) {
Argyrios Kyrtzidis2ec76742013-04-05 21:04:10 +0000767 unsigned NumParamList = DD->getNumTemplateParameterLists();
768 for (unsigned i = 0; i < NumParamList; i++) {
769 TemplateParameterList* Params = DD->getTemplateParameterList(i);
770 if (VisitTemplateParameters(Params))
771 return true;
772 }
773
Guy Benyei11169dd2012-12-18 14:30:41 +0000774 if (TypeSourceInfo *TSInfo = DD->getTypeSourceInfo())
775 if (Visit(TSInfo->getTypeLoc()))
776 return true;
777
778 // Visit the nested-name-specifier, if present.
779 if (NestedNameSpecifierLoc QualifierLoc = DD->getQualifierLoc())
780 if (VisitNestedNameSpecifierLoc(QualifierLoc))
781 return true;
782
783 return false;
784}
785
Benjamin Kramer4cadf292014-03-07 21:51:58 +0000786/// \brief Compare two base or member initializers based on their source order.
787static int CompareCXXCtorInitializers(CXXCtorInitializer *const *X,
788 CXXCtorInitializer *const *Y) {
789 return (*X)->getSourceOrder() - (*Y)->getSourceOrder();
790}
791
Guy Benyei11169dd2012-12-18 14:30:41 +0000792bool CursorVisitor::VisitFunctionDecl(FunctionDecl *ND) {
Argyrios Kyrtzidis2ec76742013-04-05 21:04:10 +0000793 unsigned NumParamList = ND->getNumTemplateParameterLists();
794 for (unsigned i = 0; i < NumParamList; i++) {
795 TemplateParameterList* Params = ND->getTemplateParameterList(i);
796 if (VisitTemplateParameters(Params))
797 return true;
798 }
799
Guy Benyei11169dd2012-12-18 14:30:41 +0000800 if (TypeSourceInfo *TSInfo = ND->getTypeSourceInfo()) {
801 // Visit the function declaration's syntactic components in the order
802 // written. This requires a bit of work.
803 TypeLoc TL = TSInfo->getTypeLoc().IgnoreParens();
David Blaikie6adc78e2013-02-18 22:06:02 +0000804 FunctionTypeLoc FTL = TL.getAs<FunctionTypeLoc>();
Guy Benyei11169dd2012-12-18 14:30:41 +0000805
806 // If we have a function declared directly (without the use of a typedef),
807 // visit just the return type. Otherwise, just visit the function's type
808 // now.
Alp Toker42a16a62014-01-25 23:51:36 +0000809 if ((FTL && !isa<CXXConversionDecl>(ND) && Visit(FTL.getReturnLoc())) ||
Guy Benyei11169dd2012-12-18 14:30:41 +0000810 (!FTL && Visit(TL)))
811 return true;
812
813 // Visit the nested-name-specifier, if present.
814 if (NestedNameSpecifierLoc QualifierLoc = ND->getQualifierLoc())
815 if (VisitNestedNameSpecifierLoc(QualifierLoc))
816 return true;
817
818 // Visit the declaration name.
Argyrios Kyrtzidis4a4d2b42014-02-09 08:13:47 +0000819 if (!isa<CXXDestructorDecl>(ND))
820 if (VisitDeclarationNameInfo(ND->getNameInfo()))
821 return true;
Guy Benyei11169dd2012-12-18 14:30:41 +0000822
823 // FIXME: Visit explicitly-specified template arguments!
824
825 // Visit the function parameters, if we have a function type.
David Blaikie6adc78e2013-02-18 22:06:02 +0000826 if (FTL && VisitFunctionTypeLoc(FTL, true))
Guy Benyei11169dd2012-12-18 14:30:41 +0000827 return true;
828
Bill Wendling44426052012-12-20 19:22:21 +0000829 // FIXME: Attributes?
Guy Benyei11169dd2012-12-18 14:30:41 +0000830 }
831
832 if (ND->doesThisDeclarationHaveABody() && !ND->isLateTemplateParsed()) {
833 if (CXXConstructorDecl *Constructor = dyn_cast<CXXConstructorDecl>(ND)) {
834 // Find the initializers that were written in the source.
835 SmallVector<CXXCtorInitializer *, 4> WrittenInits;
Aaron Ballman0ad78302014-03-13 17:34:31 +0000836 for (auto *I : Constructor->inits()) {
837 if (!I->isWritten())
Guy Benyei11169dd2012-12-18 14:30:41 +0000838 continue;
839
Aaron Ballman0ad78302014-03-13 17:34:31 +0000840 WrittenInits.push_back(I);
Guy Benyei11169dd2012-12-18 14:30:41 +0000841 }
842
843 // Sort the initializers in source order
Benjamin Kramer4cadf292014-03-07 21:51:58 +0000844 llvm::array_pod_sort(WrittenInits.begin(), WrittenInits.end(),
845 &CompareCXXCtorInitializers);
846
Guy Benyei11169dd2012-12-18 14:30:41 +0000847 // Visit the initializers in source order
848 for (unsigned I = 0, N = WrittenInits.size(); I != N; ++I) {
849 CXXCtorInitializer *Init = WrittenInits[I];
850 if (Init->isAnyMemberInitializer()) {
851 if (Visit(MakeCursorMemberRef(Init->getAnyMember(),
852 Init->getMemberLocation(), TU)))
853 return true;
854 } else if (TypeSourceInfo *TInfo = Init->getTypeSourceInfo()) {
855 if (Visit(TInfo->getTypeLoc()))
856 return true;
857 }
858
859 // Visit the initializer value.
860 if (Expr *Initializer = Init->getInit())
861 if (Visit(MakeCXCursor(Initializer, ND, TU, RegionOfInterest)))
862 return true;
863 }
864 }
865
866 if (Visit(MakeCXCursor(ND->getBody(), StmtParent, TU, RegionOfInterest)))
867 return true;
868 }
869
870 return false;
871}
872
873bool CursorVisitor::VisitFieldDecl(FieldDecl *D) {
874 if (VisitDeclaratorDecl(D))
875 return true;
876
877 if (Expr *BitWidth = D->getBitWidth())
878 return Visit(MakeCXCursor(BitWidth, StmtParent, TU, RegionOfInterest));
879
880 return false;
881}
882
883bool CursorVisitor::VisitVarDecl(VarDecl *D) {
884 if (VisitDeclaratorDecl(D))
885 return true;
886
887 if (Expr *Init = D->getInit())
888 return Visit(MakeCXCursor(Init, StmtParent, TU, RegionOfInterest));
889
890 return false;
891}
892
893bool CursorVisitor::VisitNonTypeTemplateParmDecl(NonTypeTemplateParmDecl *D) {
894 if (VisitDeclaratorDecl(D))
895 return true;
896
897 if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
898 if (Expr *DefArg = D->getDefaultArgument())
899 return Visit(MakeCXCursor(DefArg, StmtParent, TU, RegionOfInterest));
900
901 return false;
902}
903
904bool CursorVisitor::VisitFunctionTemplateDecl(FunctionTemplateDecl *D) {
905 // FIXME: Visit the "outer" template parameter lists on the FunctionDecl
906 // before visiting these template parameters.
907 if (VisitTemplateParameters(D->getTemplateParameters()))
908 return true;
909
910 return VisitFunctionDecl(D->getTemplatedDecl());
911}
912
913bool CursorVisitor::VisitClassTemplateDecl(ClassTemplateDecl *D) {
914 // FIXME: Visit the "outer" template parameter lists on the TagDecl
915 // before visiting these template parameters.
916 if (VisitTemplateParameters(D->getTemplateParameters()))
917 return true;
918
919 return VisitCXXRecordDecl(D->getTemplatedDecl());
920}
921
922bool CursorVisitor::VisitTemplateTemplateParmDecl(TemplateTemplateParmDecl *D) {
923 if (VisitTemplateParameters(D->getTemplateParameters()))
924 return true;
925
926 if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited() &&
927 VisitTemplateArgumentLoc(D->getDefaultArgument()))
928 return true;
929
930 return false;
931}
932
Douglas Gregor9bda6cf2015-07-07 03:58:14 +0000933bool CursorVisitor::VisitObjCTypeParamDecl(ObjCTypeParamDecl *D) {
934 // Visit the bound, if it's explicit.
935 if (D->hasExplicitBound()) {
936 if (auto TInfo = D->getTypeSourceInfo()) {
937 if (Visit(TInfo->getTypeLoc()))
938 return true;
939 }
940 }
941
942 return false;
943}
944
Guy Benyei11169dd2012-12-18 14:30:41 +0000945bool CursorVisitor::VisitObjCMethodDecl(ObjCMethodDecl *ND) {
Alp Toker314cc812014-01-25 16:55:45 +0000946 if (TypeSourceInfo *TSInfo = ND->getReturnTypeSourceInfo())
Guy Benyei11169dd2012-12-18 14:30:41 +0000947 if (Visit(TSInfo->getTypeLoc()))
948 return true;
949
David Majnemer59f77922016-06-24 04:05:48 +0000950 for (const auto *P : ND->parameters()) {
Aaron Ballman43b68be2014-03-07 17:50:17 +0000951 if (Visit(MakeCXCursor(P, TU, RegionOfInterest)))
Guy Benyei11169dd2012-12-18 14:30:41 +0000952 return true;
953 }
954
Alexander Kornienko1a9f1842015-12-28 15:24:08 +0000955 return ND->isThisDeclarationADefinition() &&
956 Visit(MakeCXCursor(ND->getBody(), StmtParent, TU, RegionOfInterest));
Guy Benyei11169dd2012-12-18 14:30:41 +0000957}
958
959template <typename DeclIt>
960static void addRangedDeclsInContainer(DeclIt *DI_current, DeclIt DE_current,
961 SourceManager &SM, SourceLocation EndLoc,
962 SmallVectorImpl<Decl *> &Decls) {
963 DeclIt next = *DI_current;
964 while (++next != DE_current) {
965 Decl *D_next = *next;
966 if (!D_next)
967 break;
968 SourceLocation L = D_next->getLocStart();
969 if (!L.isValid())
970 break;
971 if (SM.isBeforeInTranslationUnit(L, EndLoc)) {
972 *DI_current = next;
973 Decls.push_back(D_next);
974 continue;
975 }
976 break;
977 }
978}
979
Guy Benyei11169dd2012-12-18 14:30:41 +0000980bool CursorVisitor::VisitObjCContainerDecl(ObjCContainerDecl *D) {
981 // FIXME: Eventually convert back to just 'VisitDeclContext()'. Essentially
982 // an @implementation can lexically contain Decls that are not properly
983 // nested in the AST. When we identify such cases, we need to retrofit
984 // this nesting here.
985 if (!DI_current && !FileDI_current)
986 return VisitDeclContext(D);
987
988 // Scan the Decls that immediately come after the container
989 // in the current DeclContext. If any fall within the
990 // container's lexical region, stash them into a vector
991 // for later processing.
992 SmallVector<Decl *, 24> DeclsInContainer;
993 SourceLocation EndLoc = D->getSourceRange().getEnd();
994 SourceManager &SM = AU->getSourceManager();
995 if (EndLoc.isValid()) {
996 if (DI_current) {
997 addRangedDeclsInContainer(DI_current, DE_current, SM, EndLoc,
998 DeclsInContainer);
999 } else {
1000 addRangedDeclsInContainer(FileDI_current, FileDE_current, SM, EndLoc,
1001 DeclsInContainer);
1002 }
1003 }
1004
1005 // The common case.
1006 if (DeclsInContainer.empty())
1007 return VisitDeclContext(D);
1008
1009 // Get all the Decls in the DeclContext, and sort them with the
1010 // additional ones we've collected. Then visit them.
Aaron Ballman629afae2014-03-07 19:56:05 +00001011 for (auto *SubDecl : D->decls()) {
1012 if (!SubDecl || SubDecl->getLexicalDeclContext() != D ||
1013 SubDecl->getLocStart().isInvalid())
Guy Benyei11169dd2012-12-18 14:30:41 +00001014 continue;
Aaron Ballman629afae2014-03-07 19:56:05 +00001015 DeclsInContainer.push_back(SubDecl);
Guy Benyei11169dd2012-12-18 14:30:41 +00001016 }
1017
1018 // Now sort the Decls so that they appear in lexical order.
1019 std::sort(DeclsInContainer.begin(), DeclsInContainer.end(),
Benjamin Kramerbbdd7642014-03-01 14:48:57 +00001020 [&SM](Decl *A, Decl *B) {
1021 SourceLocation L_A = A->getLocStart();
1022 SourceLocation L_B = B->getLocStart();
1023 assert(L_A.isValid() && L_B.isValid());
1024 return SM.isBeforeInTranslationUnit(L_A, L_B);
1025 });
Guy Benyei11169dd2012-12-18 14:30:41 +00001026
1027 // Now visit the decls.
1028 for (SmallVectorImpl<Decl*>::iterator I = DeclsInContainer.begin(),
1029 E = DeclsInContainer.end(); I != E; ++I) {
1030 CXCursor Cursor = MakeCXCursor(*I, TU, RegionOfInterest);
Ted Kremenek03325582013-02-21 01:29:01 +00001031 const Optional<bool> &V = shouldVisitCursor(Cursor);
Guy Benyei11169dd2012-12-18 14:30:41 +00001032 if (!V.hasValue())
1033 continue;
1034 if (!V.getValue())
1035 return false;
1036 if (Visit(Cursor, true))
1037 return true;
1038 }
1039 return false;
1040}
1041
1042bool CursorVisitor::VisitObjCCategoryDecl(ObjCCategoryDecl *ND) {
1043 if (Visit(MakeCursorObjCClassRef(ND->getClassInterface(), ND->getLocation(),
1044 TU)))
1045 return true;
1046
Douglas Gregore9d95f12015-07-07 03:57:35 +00001047 if (VisitObjCTypeParamList(ND->getTypeParamList()))
1048 return true;
1049
Guy Benyei11169dd2012-12-18 14:30:41 +00001050 ObjCCategoryDecl::protocol_loc_iterator PL = ND->protocol_loc_begin();
1051 for (ObjCCategoryDecl::protocol_iterator I = ND->protocol_begin(),
1052 E = ND->protocol_end(); I != E; ++I, ++PL)
1053 if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1054 return true;
1055
1056 return VisitObjCContainerDecl(ND);
1057}
1058
1059bool CursorVisitor::VisitObjCProtocolDecl(ObjCProtocolDecl *PID) {
1060 if (!PID->isThisDeclarationADefinition())
1061 return Visit(MakeCursorObjCProtocolRef(PID, PID->getLocation(), TU));
1062
1063 ObjCProtocolDecl::protocol_loc_iterator PL = PID->protocol_loc_begin();
1064 for (ObjCProtocolDecl::protocol_iterator I = PID->protocol_begin(),
1065 E = PID->protocol_end(); I != E; ++I, ++PL)
1066 if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1067 return true;
1068
1069 return VisitObjCContainerDecl(PID);
1070}
1071
1072bool CursorVisitor::VisitObjCPropertyDecl(ObjCPropertyDecl *PD) {
1073 if (PD->getTypeSourceInfo() && Visit(PD->getTypeSourceInfo()->getTypeLoc()))
1074 return true;
1075
1076 // FIXME: This implements a workaround with @property declarations also being
1077 // installed in the DeclContext for the @interface. Eventually this code
1078 // should be removed.
1079 ObjCCategoryDecl *CDecl = dyn_cast<ObjCCategoryDecl>(PD->getDeclContext());
1080 if (!CDecl || !CDecl->IsClassExtension())
1081 return false;
1082
1083 ObjCInterfaceDecl *ID = CDecl->getClassInterface();
1084 if (!ID)
1085 return false;
1086
1087 IdentifierInfo *PropertyId = PD->getIdentifier();
1088 ObjCPropertyDecl *prevDecl =
Manman Ren5b786402016-01-28 18:49:28 +00001089 ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(ID), PropertyId,
1090 PD->getQueryKind());
Guy Benyei11169dd2012-12-18 14:30:41 +00001091
1092 if (!prevDecl)
1093 return false;
1094
1095 // Visit synthesized methods since they will be skipped when visiting
1096 // the @interface.
1097 if (ObjCMethodDecl *MD = prevDecl->getGetterMethodDecl())
1098 if (MD->isPropertyAccessor() && MD->getLexicalDeclContext() == CDecl)
1099 if (Visit(MakeCXCursor(MD, TU, RegionOfInterest)))
1100 return true;
1101
1102 if (ObjCMethodDecl *MD = prevDecl->getSetterMethodDecl())
1103 if (MD->isPropertyAccessor() && MD->getLexicalDeclContext() == CDecl)
1104 if (Visit(MakeCXCursor(MD, TU, RegionOfInterest)))
1105 return true;
1106
1107 return false;
1108}
1109
Douglas Gregore9d95f12015-07-07 03:57:35 +00001110bool CursorVisitor::VisitObjCTypeParamList(ObjCTypeParamList *typeParamList) {
1111 if (!typeParamList)
1112 return false;
1113
1114 for (auto *typeParam : *typeParamList) {
1115 // Visit the type parameter.
1116 if (Visit(MakeCXCursor(typeParam, TU, RegionOfInterest)))
1117 return true;
Douglas Gregore9d95f12015-07-07 03:57:35 +00001118 }
1119
1120 return false;
1121}
1122
Guy Benyei11169dd2012-12-18 14:30:41 +00001123bool CursorVisitor::VisitObjCInterfaceDecl(ObjCInterfaceDecl *D) {
1124 if (!D->isThisDeclarationADefinition()) {
1125 // Forward declaration is treated like a reference.
1126 return Visit(MakeCursorObjCClassRef(D, D->getLocation(), TU));
1127 }
1128
Douglas Gregore9d95f12015-07-07 03:57:35 +00001129 // Objective-C type parameters.
1130 if (VisitObjCTypeParamList(D->getTypeParamListAsWritten()))
1131 return true;
1132
Guy Benyei11169dd2012-12-18 14:30:41 +00001133 // Issue callbacks for super class.
1134 if (D->getSuperClass() &&
1135 Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1136 D->getSuperClassLoc(),
1137 TU)))
1138 return true;
1139
Douglas Gregore9d95f12015-07-07 03:57:35 +00001140 if (TypeSourceInfo *SuperClassTInfo = D->getSuperClassTInfo())
1141 if (Visit(SuperClassTInfo->getTypeLoc()))
1142 return true;
1143
Guy Benyei11169dd2012-12-18 14:30:41 +00001144 ObjCInterfaceDecl::protocol_loc_iterator PL = D->protocol_loc_begin();
1145 for (ObjCInterfaceDecl::protocol_iterator I = D->protocol_begin(),
1146 E = D->protocol_end(); I != E; ++I, ++PL)
1147 if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1148 return true;
1149
1150 return VisitObjCContainerDecl(D);
1151}
1152
1153bool CursorVisitor::VisitObjCImplDecl(ObjCImplDecl *D) {
1154 return VisitObjCContainerDecl(D);
1155}
1156
1157bool CursorVisitor::VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D) {
1158 // 'ID' could be null when dealing with invalid code.
1159 if (ObjCInterfaceDecl *ID = D->getClassInterface())
1160 if (Visit(MakeCursorObjCClassRef(ID, D->getLocation(), TU)))
1161 return true;
1162
1163 return VisitObjCImplDecl(D);
1164}
1165
1166bool CursorVisitor::VisitObjCImplementationDecl(ObjCImplementationDecl *D) {
1167#if 0
1168 // Issue callbacks for super class.
1169 // FIXME: No source location information!
1170 if (D->getSuperClass() &&
1171 Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1172 D->getSuperClassLoc(),
1173 TU)))
1174 return true;
1175#endif
1176
1177 return VisitObjCImplDecl(D);
1178}
1179
1180bool CursorVisitor::VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *PD) {
1181 if (ObjCIvarDecl *Ivar = PD->getPropertyIvarDecl())
1182 if (PD->isIvarNameSpecified())
1183 return Visit(MakeCursorMemberRef(Ivar, PD->getPropertyIvarDeclLoc(), TU));
1184
1185 return false;
1186}
1187
1188bool CursorVisitor::VisitNamespaceDecl(NamespaceDecl *D) {
1189 return VisitDeclContext(D);
1190}
1191
1192bool CursorVisitor::VisitNamespaceAliasDecl(NamespaceAliasDecl *D) {
1193 // Visit nested-name-specifier.
1194 if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1195 if (VisitNestedNameSpecifierLoc(QualifierLoc))
1196 return true;
1197
1198 return Visit(MakeCursorNamespaceRef(D->getAliasedNamespace(),
1199 D->getTargetNameLoc(), TU));
1200}
1201
1202bool CursorVisitor::VisitUsingDecl(UsingDecl *D) {
1203 // Visit nested-name-specifier.
1204 if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1205 if (VisitNestedNameSpecifierLoc(QualifierLoc))
1206 return true;
1207 }
1208
1209 if (Visit(MakeCursorOverloadedDeclRef(D, D->getLocation(), TU)))
1210 return true;
1211
1212 return VisitDeclarationNameInfo(D->getNameInfo());
1213}
1214
1215bool CursorVisitor::VisitUsingDirectiveDecl(UsingDirectiveDecl *D) {
1216 // Visit nested-name-specifier.
1217 if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1218 if (VisitNestedNameSpecifierLoc(QualifierLoc))
1219 return true;
1220
1221 return Visit(MakeCursorNamespaceRef(D->getNominatedNamespaceAsWritten(),
1222 D->getIdentLocation(), TU));
1223}
1224
1225bool CursorVisitor::VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D) {
1226 // Visit nested-name-specifier.
1227 if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1228 if (VisitNestedNameSpecifierLoc(QualifierLoc))
1229 return true;
1230 }
1231
1232 return VisitDeclarationNameInfo(D->getNameInfo());
1233}
1234
1235bool CursorVisitor::VisitUnresolvedUsingTypenameDecl(
1236 UnresolvedUsingTypenameDecl *D) {
1237 // Visit nested-name-specifier.
1238 if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1239 if (VisitNestedNameSpecifierLoc(QualifierLoc))
1240 return true;
1241
1242 return false;
1243}
1244
Olivier Goffart81978012016-06-09 16:15:55 +00001245bool CursorVisitor::VisitStaticAssertDecl(StaticAssertDecl *D) {
1246 if (Visit(MakeCXCursor(D->getAssertExpr(), StmtParent, TU, RegionOfInterest)))
1247 return true;
Richard Trieuf3b77662016-09-13 01:37:01 +00001248 if (StringLiteral *Message = D->getMessage())
1249 if (Visit(MakeCXCursor(Message, StmtParent, TU, RegionOfInterest)))
1250 return true;
Olivier Goffart81978012016-06-09 16:15:55 +00001251 return false;
1252}
1253
Olivier Goffartd211c642016-11-04 06:29:27 +00001254bool CursorVisitor::VisitFriendDecl(FriendDecl *D) {
1255 if (NamedDecl *FriendD = D->getFriendDecl()) {
1256 if (Visit(MakeCXCursor(FriendD, TU, RegionOfInterest)))
1257 return true;
1258 } else if (TypeSourceInfo *TI = D->getFriendType()) {
1259 if (Visit(TI->getTypeLoc()))
1260 return true;
1261 }
1262 return false;
1263}
1264
Guy Benyei11169dd2012-12-18 14:30:41 +00001265bool CursorVisitor::VisitDeclarationNameInfo(DeclarationNameInfo Name) {
1266 switch (Name.getName().getNameKind()) {
1267 case clang::DeclarationName::Identifier:
1268 case clang::DeclarationName::CXXLiteralOperatorName:
Richard Smith35845152017-02-07 01:37:30 +00001269 case clang::DeclarationName::CXXDeductionGuideName:
Guy Benyei11169dd2012-12-18 14:30:41 +00001270 case clang::DeclarationName::CXXOperatorName:
1271 case clang::DeclarationName::CXXUsingDirective:
1272 return false;
Richard Smith35845152017-02-07 01:37:30 +00001273
Guy Benyei11169dd2012-12-18 14:30:41 +00001274 case clang::DeclarationName::CXXConstructorName:
1275 case clang::DeclarationName::CXXDestructorName:
1276 case clang::DeclarationName::CXXConversionFunctionName:
1277 if (TypeSourceInfo *TSInfo = Name.getNamedTypeInfo())
1278 return Visit(TSInfo->getTypeLoc());
1279 return false;
1280
1281 case clang::DeclarationName::ObjCZeroArgSelector:
1282 case clang::DeclarationName::ObjCOneArgSelector:
1283 case clang::DeclarationName::ObjCMultiArgSelector:
1284 // FIXME: Per-identifier location info?
1285 return false;
1286 }
1287
1288 llvm_unreachable("Invalid DeclarationName::Kind!");
1289}
1290
1291bool CursorVisitor::VisitNestedNameSpecifier(NestedNameSpecifier *NNS,
1292 SourceRange Range) {
1293 // FIXME: This whole routine is a hack to work around the lack of proper
1294 // source information in nested-name-specifiers (PR5791). Since we do have
1295 // a beginning source location, we can visit the first component of the
1296 // nested-name-specifier, if it's a single-token component.
1297 if (!NNS)
1298 return false;
1299
1300 // Get the first component in the nested-name-specifier.
1301 while (NestedNameSpecifier *Prefix = NNS->getPrefix())
1302 NNS = Prefix;
1303
1304 switch (NNS->getKind()) {
1305 case NestedNameSpecifier::Namespace:
1306 return Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(), Range.getBegin(),
1307 TU));
1308
1309 case NestedNameSpecifier::NamespaceAlias:
1310 return Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
1311 Range.getBegin(), TU));
1312
1313 case NestedNameSpecifier::TypeSpec: {
1314 // If the type has a form where we know that the beginning of the source
1315 // range matches up with a reference cursor. Visit the appropriate reference
1316 // cursor.
1317 const Type *T = NNS->getAsType();
1318 if (const TypedefType *Typedef = dyn_cast<TypedefType>(T))
1319 return Visit(MakeCursorTypeRef(Typedef->getDecl(), Range.getBegin(), TU));
1320 if (const TagType *Tag = dyn_cast<TagType>(T))
1321 return Visit(MakeCursorTypeRef(Tag->getDecl(), Range.getBegin(), TU));
1322 if (const TemplateSpecializationType *TST
1323 = dyn_cast<TemplateSpecializationType>(T))
1324 return VisitTemplateName(TST->getTemplateName(), Range.getBegin());
1325 break;
1326 }
1327
1328 case NestedNameSpecifier::TypeSpecWithTemplate:
1329 case NestedNameSpecifier::Global:
1330 case NestedNameSpecifier::Identifier:
Nikola Smiljanic67860242014-09-26 00:28:20 +00001331 case NestedNameSpecifier::Super:
Guy Benyei11169dd2012-12-18 14:30:41 +00001332 break;
1333 }
1334
1335 return false;
1336}
1337
1338bool
1339CursorVisitor::VisitNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
1340 SmallVector<NestedNameSpecifierLoc, 4> Qualifiers;
1341 for (; Qualifier; Qualifier = Qualifier.getPrefix())
1342 Qualifiers.push_back(Qualifier);
1343
1344 while (!Qualifiers.empty()) {
1345 NestedNameSpecifierLoc Q = Qualifiers.pop_back_val();
1346 NestedNameSpecifier *NNS = Q.getNestedNameSpecifier();
1347 switch (NNS->getKind()) {
1348 case NestedNameSpecifier::Namespace:
1349 if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(),
1350 Q.getLocalBeginLoc(),
1351 TU)))
1352 return true;
1353
1354 break;
1355
1356 case NestedNameSpecifier::NamespaceAlias:
1357 if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
1358 Q.getLocalBeginLoc(),
1359 TU)))
1360 return true;
1361
1362 break;
1363
1364 case NestedNameSpecifier::TypeSpec:
1365 case NestedNameSpecifier::TypeSpecWithTemplate:
1366 if (Visit(Q.getTypeLoc()))
1367 return true;
1368
1369 break;
1370
1371 case NestedNameSpecifier::Global:
1372 case NestedNameSpecifier::Identifier:
Nikola Smiljanic67860242014-09-26 00:28:20 +00001373 case NestedNameSpecifier::Super:
Guy Benyei11169dd2012-12-18 14:30:41 +00001374 break;
1375 }
1376 }
1377
1378 return false;
1379}
1380
1381bool CursorVisitor::VisitTemplateParameters(
1382 const TemplateParameterList *Params) {
1383 if (!Params)
1384 return false;
1385
1386 for (TemplateParameterList::const_iterator P = Params->begin(),
1387 PEnd = Params->end();
1388 P != PEnd; ++P) {
1389 if (Visit(MakeCXCursor(*P, TU, RegionOfInterest)))
1390 return true;
1391 }
1392
1393 return false;
1394}
1395
1396bool CursorVisitor::VisitTemplateName(TemplateName Name, SourceLocation Loc) {
1397 switch (Name.getKind()) {
1398 case TemplateName::Template:
1399 return Visit(MakeCursorTemplateRef(Name.getAsTemplateDecl(), Loc, TU));
1400
1401 case TemplateName::OverloadedTemplate:
1402 // Visit the overloaded template set.
1403 if (Visit(MakeCursorOverloadedDeclRef(Name, Loc, TU)))
1404 return true;
1405
1406 return false;
1407
1408 case TemplateName::DependentTemplate:
1409 // FIXME: Visit nested-name-specifier.
1410 return false;
1411
1412 case TemplateName::QualifiedTemplate:
1413 // FIXME: Visit nested-name-specifier.
1414 return Visit(MakeCursorTemplateRef(
1415 Name.getAsQualifiedTemplateName()->getDecl(),
1416 Loc, TU));
1417
1418 case TemplateName::SubstTemplateTemplateParm:
1419 return Visit(MakeCursorTemplateRef(
1420 Name.getAsSubstTemplateTemplateParm()->getParameter(),
1421 Loc, TU));
1422
1423 case TemplateName::SubstTemplateTemplateParmPack:
1424 return Visit(MakeCursorTemplateRef(
1425 Name.getAsSubstTemplateTemplateParmPack()->getParameterPack(),
1426 Loc, TU));
1427 }
1428
1429 llvm_unreachable("Invalid TemplateName::Kind!");
1430}
1431
1432bool CursorVisitor::VisitTemplateArgumentLoc(const TemplateArgumentLoc &TAL) {
1433 switch (TAL.getArgument().getKind()) {
1434 case TemplateArgument::Null:
1435 case TemplateArgument::Integral:
1436 case TemplateArgument::Pack:
1437 return false;
1438
1439 case TemplateArgument::Type:
1440 if (TypeSourceInfo *TSInfo = TAL.getTypeSourceInfo())
1441 return Visit(TSInfo->getTypeLoc());
1442 return false;
1443
1444 case TemplateArgument::Declaration:
1445 if (Expr *E = TAL.getSourceDeclExpression())
1446 return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1447 return false;
1448
1449 case TemplateArgument::NullPtr:
1450 if (Expr *E = TAL.getSourceNullPtrExpression())
1451 return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1452 return false;
1453
1454 case TemplateArgument::Expression:
1455 if (Expr *E = TAL.getSourceExpression())
1456 return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1457 return false;
1458
1459 case TemplateArgument::Template:
1460 case TemplateArgument::TemplateExpansion:
1461 if (VisitNestedNameSpecifierLoc(TAL.getTemplateQualifierLoc()))
1462 return true;
1463
1464 return VisitTemplateName(TAL.getArgument().getAsTemplateOrTemplatePattern(),
1465 TAL.getTemplateNameLoc());
1466 }
1467
1468 llvm_unreachable("Invalid TemplateArgument::Kind!");
1469}
1470
1471bool CursorVisitor::VisitLinkageSpecDecl(LinkageSpecDecl *D) {
1472 return VisitDeclContext(D);
1473}
1474
1475bool CursorVisitor::VisitQualifiedTypeLoc(QualifiedTypeLoc TL) {
1476 return Visit(TL.getUnqualifiedLoc());
1477}
1478
1479bool CursorVisitor::VisitBuiltinTypeLoc(BuiltinTypeLoc TL) {
1480 ASTContext &Context = AU->getASTContext();
1481
1482 // Some builtin types (such as Objective-C's "id", "sel", and
1483 // "Class") have associated declarations. Create cursors for those.
1484 QualType VisitType;
1485 switch (TL.getTypePtr()->getKind()) {
1486
1487 case BuiltinType::Void:
1488 case BuiltinType::NullPtr:
1489 case BuiltinType::Dependent:
Alexey Bader954ba212016-04-08 13:40:33 +00001490#define IMAGE_TYPE(ImgType, Id, SingletonId, Access, Suffix) \
1491 case BuiltinType::Id:
Alexey Baderb62f1442016-04-13 08:33:41 +00001492#include "clang/Basic/OpenCLImageTypes.def"
NAKAMURA Takumi288c42e2013-02-07 12:47:42 +00001493 case BuiltinType::OCLSampler:
Guy Benyei1b4fb3e2013-01-20 12:31:11 +00001494 case BuiltinType::OCLEvent:
Alexey Bader9c8453f2015-09-15 11:18:52 +00001495 case BuiltinType::OCLClkEvent:
1496 case BuiltinType::OCLQueue:
Alexey Bader9c8453f2015-09-15 11:18:52 +00001497 case BuiltinType::OCLReserveID:
Guy Benyei11169dd2012-12-18 14:30:41 +00001498#define BUILTIN_TYPE(Id, SingletonId)
1499#define SIGNED_TYPE(Id, SingletonId) case BuiltinType::Id:
1500#define UNSIGNED_TYPE(Id, SingletonId) case BuiltinType::Id:
1501#define FLOATING_TYPE(Id, SingletonId) case BuiltinType::Id:
1502#define PLACEHOLDER_TYPE(Id, SingletonId) case BuiltinType::Id:
1503#include "clang/AST/BuiltinTypes.def"
1504 break;
1505
1506 case BuiltinType::ObjCId:
1507 VisitType = Context.getObjCIdType();
1508 break;
1509
1510 case BuiltinType::ObjCClass:
1511 VisitType = Context.getObjCClassType();
1512 break;
1513
1514 case BuiltinType::ObjCSel:
1515 VisitType = Context.getObjCSelType();
1516 break;
1517 }
1518
1519 if (!VisitType.isNull()) {
1520 if (const TypedefType *Typedef = VisitType->getAs<TypedefType>())
1521 return Visit(MakeCursorTypeRef(Typedef->getDecl(), TL.getBuiltinLoc(),
1522 TU));
1523 }
1524
1525 return false;
1526}
1527
1528bool CursorVisitor::VisitTypedefTypeLoc(TypedefTypeLoc TL) {
1529 return Visit(MakeCursorTypeRef(TL.getTypedefNameDecl(), TL.getNameLoc(), TU));
1530}
1531
1532bool CursorVisitor::VisitUnresolvedUsingTypeLoc(UnresolvedUsingTypeLoc TL) {
1533 return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1534}
1535
1536bool CursorVisitor::VisitTagTypeLoc(TagTypeLoc TL) {
1537 if (TL.isDefinition())
1538 return Visit(MakeCXCursor(TL.getDecl(), TU, RegionOfInterest));
1539
1540 return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1541}
1542
1543bool CursorVisitor::VisitTemplateTypeParmTypeLoc(TemplateTypeParmTypeLoc TL) {
1544 return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1545}
1546
1547bool CursorVisitor::VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL) {
Hans Wennborg59dbe862015-09-29 20:56:43 +00001548 return Visit(MakeCursorObjCClassRef(TL.getIFaceDecl(), TL.getNameLoc(), TU));
Guy Benyei11169dd2012-12-18 14:30:41 +00001549}
1550
Manman Rene6be26c2016-09-13 17:25:08 +00001551bool CursorVisitor::VisitObjCTypeParamTypeLoc(ObjCTypeParamTypeLoc TL) {
1552 if (Visit(MakeCursorTypeRef(TL.getDecl(), TL.getLocStart(), TU)))
1553 return true;
1554 for (unsigned I = 0, N = TL.getNumProtocols(); I != N; ++I) {
1555 if (Visit(MakeCursorObjCProtocolRef(TL.getProtocol(I), TL.getProtocolLoc(I),
1556 TU)))
1557 return true;
1558 }
1559
1560 return false;
1561}
1562
Guy Benyei11169dd2012-12-18 14:30:41 +00001563bool CursorVisitor::VisitObjCObjectTypeLoc(ObjCObjectTypeLoc TL) {
1564 if (TL.hasBaseTypeAsWritten() && Visit(TL.getBaseLoc()))
1565 return true;
1566
Douglas Gregore9d95f12015-07-07 03:57:35 +00001567 for (unsigned I = 0, N = TL.getNumTypeArgs(); I != N; ++I) {
1568 if (Visit(TL.getTypeArgTInfo(I)->getTypeLoc()))
1569 return true;
1570 }
1571
Guy Benyei11169dd2012-12-18 14:30:41 +00001572 for (unsigned I = 0, N = TL.getNumProtocols(); I != N; ++I) {
1573 if (Visit(MakeCursorObjCProtocolRef(TL.getProtocol(I), TL.getProtocolLoc(I),
1574 TU)))
1575 return true;
1576 }
1577
1578 return false;
1579}
1580
1581bool CursorVisitor::VisitObjCObjectPointerTypeLoc(ObjCObjectPointerTypeLoc TL) {
1582 return Visit(TL.getPointeeLoc());
1583}
1584
1585bool CursorVisitor::VisitParenTypeLoc(ParenTypeLoc TL) {
1586 return Visit(TL.getInnerLoc());
1587}
1588
1589bool CursorVisitor::VisitPointerTypeLoc(PointerTypeLoc TL) {
1590 return Visit(TL.getPointeeLoc());
1591}
1592
1593bool CursorVisitor::VisitBlockPointerTypeLoc(BlockPointerTypeLoc TL) {
1594 return Visit(TL.getPointeeLoc());
1595}
1596
1597bool CursorVisitor::VisitMemberPointerTypeLoc(MemberPointerTypeLoc TL) {
1598 return Visit(TL.getPointeeLoc());
1599}
1600
1601bool CursorVisitor::VisitLValueReferenceTypeLoc(LValueReferenceTypeLoc TL) {
1602 return Visit(TL.getPointeeLoc());
1603}
1604
1605bool CursorVisitor::VisitRValueReferenceTypeLoc(RValueReferenceTypeLoc TL) {
1606 return Visit(TL.getPointeeLoc());
1607}
1608
1609bool CursorVisitor::VisitAttributedTypeLoc(AttributedTypeLoc TL) {
1610 return Visit(TL.getModifiedLoc());
1611}
1612
1613bool CursorVisitor::VisitFunctionTypeLoc(FunctionTypeLoc TL,
1614 bool SkipResultType) {
Alp Toker42a16a62014-01-25 23:51:36 +00001615 if (!SkipResultType && Visit(TL.getReturnLoc()))
Guy Benyei11169dd2012-12-18 14:30:41 +00001616 return true;
1617
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00001618 for (unsigned I = 0, N = TL.getNumParams(); I != N; ++I)
1619 if (Decl *D = TL.getParam(I))
Guy Benyei11169dd2012-12-18 14:30:41 +00001620 if (Visit(MakeCXCursor(D, TU, RegionOfInterest)))
1621 return true;
1622
1623 return false;
1624}
1625
1626bool CursorVisitor::VisitArrayTypeLoc(ArrayTypeLoc TL) {
1627 if (Visit(TL.getElementLoc()))
1628 return true;
1629
1630 if (Expr *Size = TL.getSizeExpr())
1631 return Visit(MakeCXCursor(Size, StmtParent, TU, RegionOfInterest));
1632
1633 return false;
1634}
1635
Reid Kleckner8a365022013-06-24 17:51:48 +00001636bool CursorVisitor::VisitDecayedTypeLoc(DecayedTypeLoc TL) {
1637 return Visit(TL.getOriginalLoc());
1638}
1639
Reid Kleckner0503a872013-12-05 01:23:43 +00001640bool CursorVisitor::VisitAdjustedTypeLoc(AdjustedTypeLoc TL) {
1641 return Visit(TL.getOriginalLoc());
1642}
1643
Richard Smith600b5262017-01-26 20:40:47 +00001644bool CursorVisitor::VisitDeducedTemplateSpecializationTypeLoc(
1645 DeducedTemplateSpecializationTypeLoc TL) {
1646 if (VisitTemplateName(TL.getTypePtr()->getTemplateName(),
1647 TL.getTemplateNameLoc()))
1648 return true;
1649
1650 return false;
1651}
1652
Guy Benyei11169dd2012-12-18 14:30:41 +00001653bool CursorVisitor::VisitTemplateSpecializationTypeLoc(
1654 TemplateSpecializationTypeLoc TL) {
1655 // Visit the template name.
1656 if (VisitTemplateName(TL.getTypePtr()->getTemplateName(),
1657 TL.getTemplateNameLoc()))
1658 return true;
1659
1660 // Visit the template arguments.
1661 for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
1662 if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
1663 return true;
1664
1665 return false;
1666}
1667
1668bool CursorVisitor::VisitTypeOfExprTypeLoc(TypeOfExprTypeLoc TL) {
1669 return Visit(MakeCXCursor(TL.getUnderlyingExpr(), StmtParent, TU));
1670}
1671
1672bool CursorVisitor::VisitTypeOfTypeLoc(TypeOfTypeLoc TL) {
1673 if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
1674 return Visit(TSInfo->getTypeLoc());
1675
1676 return false;
1677}
1678
1679bool CursorVisitor::VisitUnaryTransformTypeLoc(UnaryTransformTypeLoc TL) {
1680 if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
1681 return Visit(TSInfo->getTypeLoc());
1682
1683 return false;
1684}
1685
1686bool CursorVisitor::VisitDependentNameTypeLoc(DependentNameTypeLoc TL) {
Hans Wennborg59dbe862015-09-29 20:56:43 +00001687 return VisitNestedNameSpecifierLoc(TL.getQualifierLoc());
Guy Benyei11169dd2012-12-18 14:30:41 +00001688}
1689
1690bool CursorVisitor::VisitDependentTemplateSpecializationTypeLoc(
1691 DependentTemplateSpecializationTypeLoc TL) {
1692 // Visit the nested-name-specifier, if there is one.
1693 if (TL.getQualifierLoc() &&
1694 VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
1695 return true;
1696
1697 // Visit the template arguments.
1698 for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
1699 if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
1700 return true;
1701
1702 return false;
1703}
1704
1705bool CursorVisitor::VisitElaboratedTypeLoc(ElaboratedTypeLoc TL) {
1706 if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
1707 return true;
1708
1709 return Visit(TL.getNamedTypeLoc());
1710}
1711
1712bool CursorVisitor::VisitPackExpansionTypeLoc(PackExpansionTypeLoc TL) {
1713 return Visit(TL.getPatternLoc());
1714}
1715
1716bool CursorVisitor::VisitDecltypeTypeLoc(DecltypeTypeLoc TL) {
1717 if (Expr *E = TL.getUnderlyingExpr())
1718 return Visit(MakeCXCursor(E, StmtParent, TU));
1719
1720 return false;
1721}
1722
1723bool CursorVisitor::VisitInjectedClassNameTypeLoc(InjectedClassNameTypeLoc TL) {
1724 return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1725}
1726
1727bool CursorVisitor::VisitAtomicTypeLoc(AtomicTypeLoc TL) {
1728 return Visit(TL.getValueLoc());
1729}
1730
Xiuli Pan9c14e282016-01-09 12:53:17 +00001731bool CursorVisitor::VisitPipeTypeLoc(PipeTypeLoc TL) {
1732 return Visit(TL.getValueLoc());
1733}
1734
Guy Benyei11169dd2012-12-18 14:30:41 +00001735#define DEFAULT_TYPELOC_IMPL(CLASS, PARENT) \
1736bool CursorVisitor::Visit##CLASS##TypeLoc(CLASS##TypeLoc TL) { \
1737 return Visit##PARENT##Loc(TL); \
1738}
1739
1740DEFAULT_TYPELOC_IMPL(Complex, Type)
1741DEFAULT_TYPELOC_IMPL(ConstantArray, ArrayType)
1742DEFAULT_TYPELOC_IMPL(IncompleteArray, ArrayType)
1743DEFAULT_TYPELOC_IMPL(VariableArray, ArrayType)
1744DEFAULT_TYPELOC_IMPL(DependentSizedArray, ArrayType)
1745DEFAULT_TYPELOC_IMPL(DependentSizedExtVector, Type)
1746DEFAULT_TYPELOC_IMPL(Vector, Type)
1747DEFAULT_TYPELOC_IMPL(ExtVector, VectorType)
1748DEFAULT_TYPELOC_IMPL(FunctionProto, FunctionType)
1749DEFAULT_TYPELOC_IMPL(FunctionNoProto, FunctionType)
1750DEFAULT_TYPELOC_IMPL(Record, TagType)
1751DEFAULT_TYPELOC_IMPL(Enum, TagType)
1752DEFAULT_TYPELOC_IMPL(SubstTemplateTypeParm, Type)
1753DEFAULT_TYPELOC_IMPL(SubstTemplateTypeParmPack, Type)
1754DEFAULT_TYPELOC_IMPL(Auto, Type)
1755
1756bool CursorVisitor::VisitCXXRecordDecl(CXXRecordDecl *D) {
1757 // Visit the nested-name-specifier, if present.
1758 if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1759 if (VisitNestedNameSpecifierLoc(QualifierLoc))
1760 return true;
1761
1762 if (D->isCompleteDefinition()) {
Aaron Ballman574705e2014-03-13 15:41:46 +00001763 for (const auto &I : D->bases()) {
1764 if (Visit(cxcursor::MakeCursorCXXBaseSpecifier(&I, TU)))
Guy Benyei11169dd2012-12-18 14:30:41 +00001765 return true;
1766 }
1767 }
1768
1769 return VisitTagDecl(D);
1770}
1771
1772bool CursorVisitor::VisitAttributes(Decl *D) {
Aaron Ballmanb97112e2014-03-08 22:19:01 +00001773 for (const auto *I : D->attrs())
1774 if (Visit(MakeCXCursor(I, D, TU)))
Guy Benyei11169dd2012-12-18 14:30:41 +00001775 return true;
1776
1777 return false;
1778}
1779
1780//===----------------------------------------------------------------------===//
1781// Data-recursive visitor methods.
1782//===----------------------------------------------------------------------===//
1783
1784namespace {
1785#define DEF_JOB(NAME, DATA, KIND)\
1786class NAME : public VisitorJob {\
1787public:\
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001788 NAME(const DATA *d, CXCursor parent) : \
1789 VisitorJob(parent, VisitorJob::KIND, d) {} \
Guy Benyei11169dd2012-12-18 14:30:41 +00001790 static bool classof(const VisitorJob *VJ) { return VJ->getKind() == KIND; }\
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001791 const DATA *get() const { return static_cast<const DATA*>(data[0]); }\
Guy Benyei11169dd2012-12-18 14:30:41 +00001792};
1793
1794DEF_JOB(StmtVisit, Stmt, StmtVisitKind)
1795DEF_JOB(MemberExprParts, MemberExpr, MemberExprPartsKind)
1796DEF_JOB(DeclRefExprParts, DeclRefExpr, DeclRefExprPartsKind)
1797DEF_JOB(OverloadExprParts, OverloadExpr, OverloadExprPartsKind)
Guy Benyei11169dd2012-12-18 14:30:41 +00001798DEF_JOB(SizeOfPackExprParts, SizeOfPackExpr, SizeOfPackExprPartsKind)
1799DEF_JOB(LambdaExprParts, LambdaExpr, LambdaExprPartsKind)
1800DEF_JOB(PostChildrenVisit, void, PostChildrenVisitKind)
1801#undef DEF_JOB
1802
James Y Knight04ec5bf2015-12-24 02:59:37 +00001803class ExplicitTemplateArgsVisit : public VisitorJob {
1804public:
1805 ExplicitTemplateArgsVisit(const TemplateArgumentLoc *Begin,
1806 const TemplateArgumentLoc *End, CXCursor parent)
1807 : VisitorJob(parent, VisitorJob::ExplicitTemplateArgsVisitKind, Begin,
1808 End) {}
1809 static bool classof(const VisitorJob *VJ) {
1810 return VJ->getKind() == ExplicitTemplateArgsVisitKind;
1811 }
1812 const TemplateArgumentLoc *begin() const {
1813 return static_cast<const TemplateArgumentLoc *>(data[0]);
1814 }
1815 const TemplateArgumentLoc *end() {
1816 return static_cast<const TemplateArgumentLoc *>(data[1]);
1817 }
1818};
Guy Benyei11169dd2012-12-18 14:30:41 +00001819class DeclVisit : public VisitorJob {
1820public:
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001821 DeclVisit(const Decl *D, CXCursor parent, bool isFirst) :
Guy Benyei11169dd2012-12-18 14:30:41 +00001822 VisitorJob(parent, VisitorJob::DeclVisitKind,
Craig Topper69186e72014-06-08 08:38:04 +00001823 D, isFirst ? (void*) 1 : (void*) nullptr) {}
Guy Benyei11169dd2012-12-18 14:30:41 +00001824 static bool classof(const VisitorJob *VJ) {
1825 return VJ->getKind() == DeclVisitKind;
1826 }
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001827 const Decl *get() const { return static_cast<const Decl *>(data[0]); }
Dmitri Gribenkoe5423a72015-03-23 19:23:50 +00001828 bool isFirst() const { return data[1] != nullptr; }
Guy Benyei11169dd2012-12-18 14:30:41 +00001829};
1830class TypeLocVisit : public VisitorJob {
1831public:
1832 TypeLocVisit(TypeLoc tl, CXCursor parent) :
1833 VisitorJob(parent, VisitorJob::TypeLocVisitKind,
1834 tl.getType().getAsOpaquePtr(), tl.getOpaqueData()) {}
1835
1836 static bool classof(const VisitorJob *VJ) {
1837 return VJ->getKind() == TypeLocVisitKind;
1838 }
1839
1840 TypeLoc get() const {
1841 QualType T = QualType::getFromOpaquePtr(data[0]);
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001842 return TypeLoc(T, const_cast<void *>(data[1]));
Guy Benyei11169dd2012-12-18 14:30:41 +00001843 }
1844};
1845
1846class LabelRefVisit : public VisitorJob {
1847public:
1848 LabelRefVisit(LabelDecl *LD, SourceLocation labelLoc, CXCursor parent)
1849 : VisitorJob(parent, VisitorJob::LabelRefVisitKind, LD,
1850 labelLoc.getPtrEncoding()) {}
1851
1852 static bool classof(const VisitorJob *VJ) {
1853 return VJ->getKind() == VisitorJob::LabelRefVisitKind;
1854 }
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001855 const LabelDecl *get() const {
1856 return static_cast<const LabelDecl *>(data[0]);
1857 }
Guy Benyei11169dd2012-12-18 14:30:41 +00001858 SourceLocation getLoc() const {
1859 return SourceLocation::getFromPtrEncoding(data[1]); }
1860};
1861
1862class NestedNameSpecifierLocVisit : public VisitorJob {
1863public:
1864 NestedNameSpecifierLocVisit(NestedNameSpecifierLoc Qualifier, CXCursor parent)
1865 : VisitorJob(parent, VisitorJob::NestedNameSpecifierLocVisitKind,
1866 Qualifier.getNestedNameSpecifier(),
1867 Qualifier.getOpaqueData()) { }
1868
1869 static bool classof(const VisitorJob *VJ) {
1870 return VJ->getKind() == VisitorJob::NestedNameSpecifierLocVisitKind;
1871 }
1872
1873 NestedNameSpecifierLoc get() const {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001874 return NestedNameSpecifierLoc(
1875 const_cast<NestedNameSpecifier *>(
1876 static_cast<const NestedNameSpecifier *>(data[0])),
1877 const_cast<void *>(data[1]));
Guy Benyei11169dd2012-12-18 14:30:41 +00001878 }
1879};
1880
1881class DeclarationNameInfoVisit : public VisitorJob {
1882public:
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001883 DeclarationNameInfoVisit(const Stmt *S, CXCursor parent)
Dmitri Gribenkodd7dacf2013-02-03 13:19:54 +00001884 : VisitorJob(parent, VisitorJob::DeclarationNameInfoVisitKind, S) {}
Guy Benyei11169dd2012-12-18 14:30:41 +00001885 static bool classof(const VisitorJob *VJ) {
1886 return VJ->getKind() == VisitorJob::DeclarationNameInfoVisitKind;
1887 }
1888 DeclarationNameInfo get() const {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001889 const Stmt *S = static_cast<const Stmt *>(data[0]);
Guy Benyei11169dd2012-12-18 14:30:41 +00001890 switch (S->getStmtClass()) {
1891 default:
1892 llvm_unreachable("Unhandled Stmt");
1893 case clang::Stmt::MSDependentExistsStmtClass:
1894 return cast<MSDependentExistsStmt>(S)->getNameInfo();
1895 case Stmt::CXXDependentScopeMemberExprClass:
1896 return cast<CXXDependentScopeMemberExpr>(S)->getMemberNameInfo();
1897 case Stmt::DependentScopeDeclRefExprClass:
1898 return cast<DependentScopeDeclRefExpr>(S)->getNameInfo();
Alexander Musmand9ed09f2014-07-21 09:42:05 +00001899 case Stmt::OMPCriticalDirectiveClass:
1900 return cast<OMPCriticalDirective>(S)->getDirectiveName();
Guy Benyei11169dd2012-12-18 14:30:41 +00001901 }
1902 }
1903};
1904class MemberRefVisit : public VisitorJob {
1905public:
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001906 MemberRefVisit(const FieldDecl *D, SourceLocation L, CXCursor parent)
Guy Benyei11169dd2012-12-18 14:30:41 +00001907 : VisitorJob(parent, VisitorJob::MemberRefVisitKind, D,
1908 L.getPtrEncoding()) {}
1909 static bool classof(const VisitorJob *VJ) {
1910 return VJ->getKind() == VisitorJob::MemberRefVisitKind;
1911 }
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001912 const FieldDecl *get() const {
1913 return static_cast<const FieldDecl *>(data[0]);
Guy Benyei11169dd2012-12-18 14:30:41 +00001914 }
1915 SourceLocation getLoc() const {
1916 return SourceLocation::getFromRawEncoding((unsigned)(uintptr_t) data[1]);
1917 }
1918};
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001919class EnqueueVisitor : public ConstStmtVisitor<EnqueueVisitor, void> {
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001920 friend class OMPClauseEnqueue;
Guy Benyei11169dd2012-12-18 14:30:41 +00001921 VisitorWorkList &WL;
1922 CXCursor Parent;
1923public:
1924 EnqueueVisitor(VisitorWorkList &wl, CXCursor parent)
1925 : WL(wl), Parent(parent) {}
1926
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001927 void VisitAddrLabelExpr(const AddrLabelExpr *E);
1928 void VisitBlockExpr(const BlockExpr *B);
1929 void VisitCompoundLiteralExpr(const CompoundLiteralExpr *E);
1930 void VisitCompoundStmt(const CompoundStmt *S);
1931 void VisitCXXDefaultArgExpr(const CXXDefaultArgExpr *E) { /* Do nothing. */ }
1932 void VisitMSDependentExistsStmt(const MSDependentExistsStmt *S);
1933 void VisitCXXDependentScopeMemberExpr(const CXXDependentScopeMemberExpr *E);
1934 void VisitCXXNewExpr(const CXXNewExpr *E);
1935 void VisitCXXScalarValueInitExpr(const CXXScalarValueInitExpr *E);
1936 void VisitCXXOperatorCallExpr(const CXXOperatorCallExpr *E);
1937 void VisitCXXPseudoDestructorExpr(const CXXPseudoDestructorExpr *E);
1938 void VisitCXXTemporaryObjectExpr(const CXXTemporaryObjectExpr *E);
1939 void VisitCXXTypeidExpr(const CXXTypeidExpr *E);
1940 void VisitCXXUnresolvedConstructExpr(const CXXUnresolvedConstructExpr *E);
1941 void VisitCXXUuidofExpr(const CXXUuidofExpr *E);
1942 void VisitCXXCatchStmt(const CXXCatchStmt *S);
Argyrios Kyrtzidis99891242014-11-13 09:03:21 +00001943 void VisitCXXForRangeStmt(const CXXForRangeStmt *S);
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001944 void VisitDeclRefExpr(const DeclRefExpr *D);
1945 void VisitDeclStmt(const DeclStmt *S);
1946 void VisitDependentScopeDeclRefExpr(const DependentScopeDeclRefExpr *E);
1947 void VisitDesignatedInitExpr(const DesignatedInitExpr *E);
1948 void VisitExplicitCastExpr(const ExplicitCastExpr *E);
1949 void VisitForStmt(const ForStmt *FS);
1950 void VisitGotoStmt(const GotoStmt *GS);
1951 void VisitIfStmt(const IfStmt *If);
1952 void VisitInitListExpr(const InitListExpr *IE);
1953 void VisitMemberExpr(const MemberExpr *M);
1954 void VisitOffsetOfExpr(const OffsetOfExpr *E);
1955 void VisitObjCEncodeExpr(const ObjCEncodeExpr *E);
1956 void VisitObjCMessageExpr(const ObjCMessageExpr *M);
1957 void VisitOverloadExpr(const OverloadExpr *E);
1958 void VisitUnaryExprOrTypeTraitExpr(const UnaryExprOrTypeTraitExpr *E);
1959 void VisitStmt(const Stmt *S);
1960 void VisitSwitchStmt(const SwitchStmt *S);
1961 void VisitWhileStmt(const WhileStmt *W);
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001962 void VisitTypeTraitExpr(const TypeTraitExpr *E);
1963 void VisitArrayTypeTraitExpr(const ArrayTypeTraitExpr *E);
1964 void VisitExpressionTraitExpr(const ExpressionTraitExpr *E);
1965 void VisitUnresolvedMemberExpr(const UnresolvedMemberExpr *U);
1966 void VisitVAArgExpr(const VAArgExpr *E);
1967 void VisitSizeOfPackExpr(const SizeOfPackExpr *E);
1968 void VisitPseudoObjectExpr(const PseudoObjectExpr *E);
1969 void VisitOpaqueValueExpr(const OpaqueValueExpr *E);
1970 void VisitLambdaExpr(const LambdaExpr *E);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001971 void VisitOMPExecutableDirective(const OMPExecutableDirective *D);
Alexander Musman3aaab662014-08-19 11:27:13 +00001972 void VisitOMPLoopDirective(const OMPLoopDirective *D);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001973 void VisitOMPParallelDirective(const OMPParallelDirective *D);
Alexey Bataev1b59ab52014-02-27 08:29:12 +00001974 void VisitOMPSimdDirective(const OMPSimdDirective *D);
Alexey Bataevf29276e2014-06-18 04:14:57 +00001975 void VisitOMPForDirective(const OMPForDirective *D);
Alexander Musmanf82886e2014-09-18 05:12:34 +00001976 void VisitOMPForSimdDirective(const OMPForSimdDirective *D);
Alexey Bataevd3f8dd22014-06-25 11:44:49 +00001977 void VisitOMPSectionsDirective(const OMPSectionsDirective *D);
Alexey Bataev1e0498a2014-06-26 08:21:58 +00001978 void VisitOMPSectionDirective(const OMPSectionDirective *D);
Alexey Bataevd1e40fb2014-06-26 12:05:45 +00001979 void VisitOMPSingleDirective(const OMPSingleDirective *D);
Alexander Musman80c22892014-07-17 08:54:58 +00001980 void VisitOMPMasterDirective(const OMPMasterDirective *D);
Alexander Musmand9ed09f2014-07-21 09:42:05 +00001981 void VisitOMPCriticalDirective(const OMPCriticalDirective *D);
Alexey Bataev4acb8592014-07-07 13:01:15 +00001982 void VisitOMPParallelForDirective(const OMPParallelForDirective *D);
Alexander Musmane4e893b2014-09-23 09:33:00 +00001983 void VisitOMPParallelForSimdDirective(const OMPParallelForSimdDirective *D);
Alexey Bataev84d0b3e2014-07-08 08:12:03 +00001984 void VisitOMPParallelSectionsDirective(const OMPParallelSectionsDirective *D);
Alexey Bataev9c2e8ee2014-07-11 11:25:16 +00001985 void VisitOMPTaskDirective(const OMPTaskDirective *D);
Alexey Bataev68446b72014-07-18 07:47:19 +00001986 void VisitOMPTaskyieldDirective(const OMPTaskyieldDirective *D);
Alexey Bataev4d1dfea2014-07-18 09:11:51 +00001987 void VisitOMPBarrierDirective(const OMPBarrierDirective *D);
Alexey Bataev2df347a2014-07-18 10:17:07 +00001988 void VisitOMPTaskwaitDirective(const OMPTaskwaitDirective *D);
Alexey Bataevc30dd2d2015-06-18 12:14:09 +00001989 void VisitOMPTaskgroupDirective(const OMPTaskgroupDirective *D);
Alexey Bataev6d4ed052015-07-01 06:57:41 +00001990 void
1991 VisitOMPCancellationPointDirective(const OMPCancellationPointDirective *D);
Alexey Bataev80909872015-07-02 11:25:17 +00001992 void VisitOMPCancelDirective(const OMPCancelDirective *D);
Alexey Bataev6125da92014-07-21 11:26:11 +00001993 void VisitOMPFlushDirective(const OMPFlushDirective *D);
Alexey Bataev9fb6e642014-07-22 06:45:04 +00001994 void VisitOMPOrderedDirective(const OMPOrderedDirective *D);
Alexey Bataev0162e452014-07-22 10:10:35 +00001995 void VisitOMPAtomicDirective(const OMPAtomicDirective *D);
Alexey Bataev0bd520b2014-09-19 08:19:49 +00001996 void VisitOMPTargetDirective(const OMPTargetDirective *D);
Michael Wong65f367f2015-07-21 13:44:28 +00001997 void VisitOMPTargetDataDirective(const OMPTargetDataDirective *D);
Samuel Antaodf67fc42016-01-19 19:15:56 +00001998 void VisitOMPTargetEnterDataDirective(const OMPTargetEnterDataDirective *D);
Samuel Antao72590762016-01-19 20:04:50 +00001999 void VisitOMPTargetExitDataDirective(const OMPTargetExitDataDirective *D);
Arpith Chacko Jacobe955b3d2016-01-26 18:48:41 +00002000 void VisitOMPTargetParallelDirective(const OMPTargetParallelDirective *D);
Arpith Chacko Jacob05bebb52016-02-03 15:46:42 +00002001 void
2002 VisitOMPTargetParallelForDirective(const OMPTargetParallelForDirective *D);
Alexey Bataev13314bf2014-10-09 04:18:56 +00002003 void VisitOMPTeamsDirective(const OMPTeamsDirective *D);
Alexey Bataev49f6e782015-12-01 04:18:41 +00002004 void VisitOMPTaskLoopDirective(const OMPTaskLoopDirective *D);
Alexey Bataev0a6ed842015-12-03 09:40:15 +00002005 void VisitOMPTaskLoopSimdDirective(const OMPTaskLoopSimdDirective *D);
Carlo Bertolli6200a3d2015-12-14 14:51:25 +00002006 void VisitOMPDistributeDirective(const OMPDistributeDirective *D);
Carlo Bertolli9925f152016-06-27 14:55:37 +00002007 void VisitOMPDistributeParallelForDirective(
2008 const OMPDistributeParallelForDirective *D);
Kelvin Li4a39add2016-07-05 05:00:15 +00002009 void VisitOMPDistributeParallelForSimdDirective(
2010 const OMPDistributeParallelForSimdDirective *D);
Kelvin Li787f3fc2016-07-06 04:45:38 +00002011 void VisitOMPDistributeSimdDirective(const OMPDistributeSimdDirective *D);
Kelvin Lia579b912016-07-14 02:54:56 +00002012 void VisitOMPTargetParallelForSimdDirective(
2013 const OMPTargetParallelForSimdDirective *D);
Kelvin Li986330c2016-07-20 22:57:10 +00002014 void VisitOMPTargetSimdDirective(const OMPTargetSimdDirective *D);
Kelvin Li02532872016-08-05 14:37:37 +00002015 void VisitOMPTeamsDistributeDirective(const OMPTeamsDistributeDirective *D);
Kelvin Li4e325f72016-10-25 12:50:55 +00002016 void VisitOMPTeamsDistributeSimdDirective(
2017 const OMPTeamsDistributeSimdDirective *D);
Kelvin Li579e41c2016-11-30 23:51:03 +00002018 void VisitOMPTeamsDistributeParallelForSimdDirective(
2019 const OMPTeamsDistributeParallelForSimdDirective *D);
Kelvin Li7ade93f2016-12-09 03:24:30 +00002020 void VisitOMPTeamsDistributeParallelForDirective(
2021 const OMPTeamsDistributeParallelForDirective *D);
Kelvin Libf594a52016-12-17 05:48:59 +00002022 void VisitOMPTargetTeamsDirective(const OMPTargetTeamsDirective *D);
Kelvin Li83c451e2016-12-25 04:52:54 +00002023 void VisitOMPTargetTeamsDistributeDirective(
2024 const OMPTargetTeamsDistributeDirective *D);
Kelvin Li80e8f562016-12-29 22:16:30 +00002025 void VisitOMPTargetTeamsDistributeParallelForDirective(
2026 const OMPTargetTeamsDistributeParallelForDirective *D);
Kelvin Li1851df52017-01-03 05:23:48 +00002027 void VisitOMPTargetTeamsDistributeParallelForSimdDirective(
2028 const OMPTargetTeamsDistributeParallelForSimdDirective *D);
Kelvin Lida681182017-01-10 18:08:18 +00002029 void VisitOMPTargetTeamsDistributeSimdDirective(
2030 const OMPTargetTeamsDistributeSimdDirective *D);
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002031
Guy Benyei11169dd2012-12-18 14:30:41 +00002032private:
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002033 void AddDeclarationNameInfo(const Stmt *S);
Guy Benyei11169dd2012-12-18 14:30:41 +00002034 void AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier);
James Y Knight04ec5bf2015-12-24 02:59:37 +00002035 void AddExplicitTemplateArgs(const TemplateArgumentLoc *A,
2036 unsigned NumTemplateArgs);
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002037 void AddMemberRef(const FieldDecl *D, SourceLocation L);
2038 void AddStmt(const Stmt *S);
2039 void AddDecl(const Decl *D, bool isFirst = true);
Guy Benyei11169dd2012-12-18 14:30:41 +00002040 void AddTypeLoc(TypeSourceInfo *TI);
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002041 void EnqueueChildren(const Stmt *S);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002042 void EnqueueChildren(const OMPClause *S);
Guy Benyei11169dd2012-12-18 14:30:41 +00002043};
Alexander Kornienkoab9db512015-06-22 23:07:51 +00002044} // end anonyous namespace
Guy Benyei11169dd2012-12-18 14:30:41 +00002045
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002046void EnqueueVisitor::AddDeclarationNameInfo(const Stmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002047 // 'S' should always be non-null, since it comes from the
2048 // statement we are visiting.
2049 WL.push_back(DeclarationNameInfoVisit(S, Parent));
2050}
2051
2052void
2053EnqueueVisitor::AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
2054 if (Qualifier)
2055 WL.push_back(NestedNameSpecifierLocVisit(Qualifier, Parent));
2056}
2057
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002058void EnqueueVisitor::AddStmt(const Stmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002059 if (S)
2060 WL.push_back(StmtVisit(S, Parent));
2061}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002062void EnqueueVisitor::AddDecl(const Decl *D, bool isFirst) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002063 if (D)
2064 WL.push_back(DeclVisit(D, Parent, isFirst));
2065}
James Y Knight04ec5bf2015-12-24 02:59:37 +00002066void EnqueueVisitor::AddExplicitTemplateArgs(const TemplateArgumentLoc *A,
2067 unsigned NumTemplateArgs) {
2068 WL.push_back(ExplicitTemplateArgsVisit(A, A + NumTemplateArgs, Parent));
Guy Benyei11169dd2012-12-18 14:30:41 +00002069}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002070void EnqueueVisitor::AddMemberRef(const FieldDecl *D, SourceLocation L) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002071 if (D)
2072 WL.push_back(MemberRefVisit(D, L, Parent));
2073}
2074void EnqueueVisitor::AddTypeLoc(TypeSourceInfo *TI) {
2075 if (TI)
2076 WL.push_back(TypeLocVisit(TI->getTypeLoc(), Parent));
2077 }
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002078void EnqueueVisitor::EnqueueChildren(const Stmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002079 unsigned size = WL.size();
Benjamin Kramer642f1732015-07-02 21:03:14 +00002080 for (const Stmt *SubStmt : S->children()) {
2081 AddStmt(SubStmt);
Guy Benyei11169dd2012-12-18 14:30:41 +00002082 }
2083 if (size == WL.size())
2084 return;
2085 // Now reverse the entries we just added. This will match the DFS
2086 // ordering performed by the worklist.
2087 VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
2088 std::reverse(I, E);
2089}
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002090namespace {
2091class OMPClauseEnqueue : public ConstOMPClauseVisitor<OMPClauseEnqueue> {
2092 EnqueueVisitor *Visitor;
Alexey Bataev756c1962013-09-24 03:17:45 +00002093 /// \brief Process clauses with list of variables.
2094 template <typename T>
2095 void VisitOMPClauseList(T *Node);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002096public:
2097 OMPClauseEnqueue(EnqueueVisitor *Visitor) : Visitor(Visitor) { }
2098#define OPENMP_CLAUSE(Name, Class) \
2099 void Visit##Class(const Class *C);
2100#include "clang/Basic/OpenMPKinds.def"
Alexey Bataev3392d762016-02-16 11:18:12 +00002101 void VisitOMPClauseWithPreInit(const OMPClauseWithPreInit *C);
Alexey Bataev005248a2016-02-25 05:25:57 +00002102 void VisitOMPClauseWithPostUpdate(const OMPClauseWithPostUpdate *C);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002103};
2104
Alexey Bataev3392d762016-02-16 11:18:12 +00002105void OMPClauseEnqueue::VisitOMPClauseWithPreInit(
2106 const OMPClauseWithPreInit *C) {
2107 Visitor->AddStmt(C->getPreInitStmt());
2108}
2109
Alexey Bataev005248a2016-02-25 05:25:57 +00002110void OMPClauseEnqueue::VisitOMPClauseWithPostUpdate(
2111 const OMPClauseWithPostUpdate *C) {
Alexey Bataev37e594c2016-03-04 07:21:16 +00002112 VisitOMPClauseWithPreInit(C);
Alexey Bataev005248a2016-02-25 05:25:57 +00002113 Visitor->AddStmt(C->getPostUpdateExpr());
2114}
2115
Alexey Bataevaadd52e2014-02-13 05:29:23 +00002116void OMPClauseEnqueue::VisitOMPIfClause(const OMPIfClause *C) {
Arpith Chacko Jacobfe4890a2017-01-18 20:40:48 +00002117 VisitOMPClauseWithPreInit(C);
Alexey Bataevaadd52e2014-02-13 05:29:23 +00002118 Visitor->AddStmt(C->getCondition());
2119}
2120
Alexey Bataev3778b602014-07-17 07:32:53 +00002121void OMPClauseEnqueue::VisitOMPFinalClause(const OMPFinalClause *C) {
2122 Visitor->AddStmt(C->getCondition());
2123}
2124
Alexey Bataev568a8332014-03-06 06:15:19 +00002125void OMPClauseEnqueue::VisitOMPNumThreadsClause(const OMPNumThreadsClause *C) {
Arpith Chacko Jacob33c849a2017-01-25 00:57:16 +00002126 VisitOMPClauseWithPreInit(C);
Alexey Bataev568a8332014-03-06 06:15:19 +00002127 Visitor->AddStmt(C->getNumThreads());
2128}
2129
Alexey Bataev62c87d22014-03-21 04:51:18 +00002130void OMPClauseEnqueue::VisitOMPSafelenClause(const OMPSafelenClause *C) {
2131 Visitor->AddStmt(C->getSafelen());
2132}
2133
Alexey Bataev66b15b52015-08-21 11:14:16 +00002134void OMPClauseEnqueue::VisitOMPSimdlenClause(const OMPSimdlenClause *C) {
2135 Visitor->AddStmt(C->getSimdlen());
2136}
2137
Alexander Musman8bd31e62014-05-27 15:12:19 +00002138void OMPClauseEnqueue::VisitOMPCollapseClause(const OMPCollapseClause *C) {
2139 Visitor->AddStmt(C->getNumForLoops());
2140}
2141
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002142void OMPClauseEnqueue::VisitOMPDefaultClause(const OMPDefaultClause *C) { }
Alexey Bataev756c1962013-09-24 03:17:45 +00002143
Alexey Bataevbcbadb62014-05-06 06:04:14 +00002144void OMPClauseEnqueue::VisitOMPProcBindClause(const OMPProcBindClause *C) { }
2145
Alexey Bataev56dafe82014-06-20 07:16:17 +00002146void OMPClauseEnqueue::VisitOMPScheduleClause(const OMPScheduleClause *C) {
Alexey Bataev3392d762016-02-16 11:18:12 +00002147 VisitOMPClauseWithPreInit(C);
Alexey Bataev56dafe82014-06-20 07:16:17 +00002148 Visitor->AddStmt(C->getChunkSize());
2149}
2150
Alexey Bataev10e775f2015-07-30 11:36:16 +00002151void OMPClauseEnqueue::VisitOMPOrderedClause(const OMPOrderedClause *C) {
2152 Visitor->AddStmt(C->getNumForLoops());
2153}
Alexey Bataev142e1fc2014-06-20 09:44:06 +00002154
Alexey Bataev236070f2014-06-20 11:19:47 +00002155void OMPClauseEnqueue::VisitOMPNowaitClause(const OMPNowaitClause *) {}
2156
Alexey Bataev7aea99a2014-07-17 12:19:31 +00002157void OMPClauseEnqueue::VisitOMPUntiedClause(const OMPUntiedClause *) {}
2158
Alexey Bataev74ba3a52014-07-17 12:47:03 +00002159void OMPClauseEnqueue::VisitOMPMergeableClause(const OMPMergeableClause *) {}
2160
Alexey Bataevf98b00c2014-07-23 02:27:21 +00002161void OMPClauseEnqueue::VisitOMPReadClause(const OMPReadClause *) {}
2162
Alexey Bataevdea47612014-07-23 07:46:59 +00002163void OMPClauseEnqueue::VisitOMPWriteClause(const OMPWriteClause *) {}
2164
Alexey Bataev67a4f222014-07-23 10:25:33 +00002165void OMPClauseEnqueue::VisitOMPUpdateClause(const OMPUpdateClause *) {}
2166
Alexey Bataev459dec02014-07-24 06:46:57 +00002167void OMPClauseEnqueue::VisitOMPCaptureClause(const OMPCaptureClause *) {}
2168
Alexey Bataev82bad8b2014-07-24 08:55:34 +00002169void OMPClauseEnqueue::VisitOMPSeqCstClause(const OMPSeqCstClause *) {}
2170
Alexey Bataev346265e2015-09-25 10:37:12 +00002171void OMPClauseEnqueue::VisitOMPThreadsClause(const OMPThreadsClause *) {}
2172
Alexey Bataevd14d1e62015-09-28 06:39:35 +00002173void OMPClauseEnqueue::VisitOMPSIMDClause(const OMPSIMDClause *) {}
2174
Alexey Bataevb825de12015-12-07 10:51:44 +00002175void OMPClauseEnqueue::VisitOMPNogroupClause(const OMPNogroupClause *) {}
2176
Michael Wonge710d542015-08-07 16:16:36 +00002177void OMPClauseEnqueue::VisitOMPDeviceClause(const OMPDeviceClause *C) {
2178 Visitor->AddStmt(C->getDevice());
2179}
2180
Kelvin Li099bb8c2015-11-24 20:50:12 +00002181void OMPClauseEnqueue::VisitOMPNumTeamsClause(const OMPNumTeamsClause *C) {
Arpith Chacko Jacobbc126342017-01-25 11:28:18 +00002182 VisitOMPClauseWithPreInit(C);
Kelvin Li099bb8c2015-11-24 20:50:12 +00002183 Visitor->AddStmt(C->getNumTeams());
2184}
2185
Kelvin Lia15fb1a2015-11-27 18:47:36 +00002186void OMPClauseEnqueue::VisitOMPThreadLimitClause(const OMPThreadLimitClause *C) {
Arpith Chacko Jacob7ecc0b72017-01-25 11:44:35 +00002187 VisitOMPClauseWithPreInit(C);
Kelvin Lia15fb1a2015-11-27 18:47:36 +00002188 Visitor->AddStmt(C->getThreadLimit());
2189}
2190
Alexey Bataeva0569352015-12-01 10:17:31 +00002191void OMPClauseEnqueue::VisitOMPPriorityClause(const OMPPriorityClause *C) {
2192 Visitor->AddStmt(C->getPriority());
2193}
2194
Alexey Bataev1fd4aed2015-12-07 12:52:51 +00002195void OMPClauseEnqueue::VisitOMPGrainsizeClause(const OMPGrainsizeClause *C) {
2196 Visitor->AddStmt(C->getGrainsize());
2197}
2198
Alexey Bataev382967a2015-12-08 12:06:20 +00002199void OMPClauseEnqueue::VisitOMPNumTasksClause(const OMPNumTasksClause *C) {
2200 Visitor->AddStmt(C->getNumTasks());
2201}
2202
Alexey Bataev28c75412015-12-15 08:19:24 +00002203void OMPClauseEnqueue::VisitOMPHintClause(const OMPHintClause *C) {
2204 Visitor->AddStmt(C->getHint());
2205}
2206
Alexey Bataev756c1962013-09-24 03:17:45 +00002207template<typename T>
2208void OMPClauseEnqueue::VisitOMPClauseList(T *Node) {
Alexey Bataev03b340a2014-10-21 03:16:40 +00002209 for (const auto *I : Node->varlists()) {
Aaron Ballman2205d2a2014-03-14 15:55:35 +00002210 Visitor->AddStmt(I);
Alexey Bataev03b340a2014-10-21 03:16:40 +00002211 }
Alexey Bataev756c1962013-09-24 03:17:45 +00002212}
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002213
2214void OMPClauseEnqueue::VisitOMPPrivateClause(const OMPPrivateClause *C) {
Alexey Bataev756c1962013-09-24 03:17:45 +00002215 VisitOMPClauseList(C);
Alexey Bataev03b340a2014-10-21 03:16:40 +00002216 for (const auto *E : C->private_copies()) {
2217 Visitor->AddStmt(E);
2218 }
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002219}
Alexey Bataevd5af8e42013-10-01 05:32:34 +00002220void OMPClauseEnqueue::VisitOMPFirstprivateClause(
2221 const OMPFirstprivateClause *C) {
2222 VisitOMPClauseList(C);
Alexey Bataev417089f2016-02-17 13:19:37 +00002223 VisitOMPClauseWithPreInit(C);
2224 for (const auto *E : C->private_copies()) {
2225 Visitor->AddStmt(E);
2226 }
2227 for (const auto *E : C->inits()) {
2228 Visitor->AddStmt(E);
2229 }
Alexey Bataevd5af8e42013-10-01 05:32:34 +00002230}
Alexander Musman1bb328c2014-06-04 13:06:39 +00002231void OMPClauseEnqueue::VisitOMPLastprivateClause(
2232 const OMPLastprivateClause *C) {
2233 VisitOMPClauseList(C);
Alexey Bataev005248a2016-02-25 05:25:57 +00002234 VisitOMPClauseWithPostUpdate(C);
Alexey Bataev38e89532015-04-16 04:54:05 +00002235 for (auto *E : C->private_copies()) {
2236 Visitor->AddStmt(E);
2237 }
2238 for (auto *E : C->source_exprs()) {
2239 Visitor->AddStmt(E);
2240 }
2241 for (auto *E : C->destination_exprs()) {
2242 Visitor->AddStmt(E);
2243 }
2244 for (auto *E : C->assignment_ops()) {
2245 Visitor->AddStmt(E);
2246 }
Alexander Musman1bb328c2014-06-04 13:06:39 +00002247}
Alexey Bataev758e55e2013-09-06 18:03:48 +00002248void OMPClauseEnqueue::VisitOMPSharedClause(const OMPSharedClause *C) {
Alexey Bataev756c1962013-09-24 03:17:45 +00002249 VisitOMPClauseList(C);
Alexey Bataev758e55e2013-09-06 18:03:48 +00002250}
Alexey Bataevc5e02582014-06-16 07:08:35 +00002251void OMPClauseEnqueue::VisitOMPReductionClause(const OMPReductionClause *C) {
2252 VisitOMPClauseList(C);
Alexey Bataev61205072016-03-02 04:57:40 +00002253 VisitOMPClauseWithPostUpdate(C);
Alexey Bataevf24e7b12015-10-08 09:10:53 +00002254 for (auto *E : C->privates()) {
2255 Visitor->AddStmt(E);
2256 }
Alexey Bataev794ba0d2015-04-10 10:43:45 +00002257 for (auto *E : C->lhs_exprs()) {
2258 Visitor->AddStmt(E);
2259 }
2260 for (auto *E : C->rhs_exprs()) {
2261 Visitor->AddStmt(E);
2262 }
2263 for (auto *E : C->reduction_ops()) {
2264 Visitor->AddStmt(E);
2265 }
Alexey Bataevc5e02582014-06-16 07:08:35 +00002266}
Alexey Bataev169d96a2017-07-18 20:17:46 +00002267void OMPClauseEnqueue::VisitOMPTaskReductionClause(
2268 const OMPTaskReductionClause *C) {
2269 VisitOMPClauseList(C);
2270 VisitOMPClauseWithPostUpdate(C);
2271 for (auto *E : C->privates()) {
2272 Visitor->AddStmt(E);
2273 }
2274 for (auto *E : C->lhs_exprs()) {
2275 Visitor->AddStmt(E);
2276 }
2277 for (auto *E : C->rhs_exprs()) {
2278 Visitor->AddStmt(E);
2279 }
2280 for (auto *E : C->reduction_ops()) {
2281 Visitor->AddStmt(E);
2282 }
2283}
Alexey Bataevfa312f32017-07-21 18:48:21 +00002284void OMPClauseEnqueue::VisitOMPInReductionClause(
2285 const OMPInReductionClause *C) {
2286 VisitOMPClauseList(C);
2287 VisitOMPClauseWithPostUpdate(C);
2288 for (auto *E : C->privates()) {
2289 Visitor->AddStmt(E);
2290 }
2291 for (auto *E : C->lhs_exprs()) {
2292 Visitor->AddStmt(E);
2293 }
2294 for (auto *E : C->rhs_exprs()) {
2295 Visitor->AddStmt(E);
2296 }
2297 for (auto *E : C->reduction_ops()) {
2298 Visitor->AddStmt(E);
2299 }
2300}
Alexander Musman8dba6642014-04-22 13:09:42 +00002301void OMPClauseEnqueue::VisitOMPLinearClause(const OMPLinearClause *C) {
2302 VisitOMPClauseList(C);
Alexey Bataev78849fb2016-03-09 09:49:00 +00002303 VisitOMPClauseWithPostUpdate(C);
Alexey Bataevbd9fec12015-08-18 06:47:21 +00002304 for (const auto *E : C->privates()) {
2305 Visitor->AddStmt(E);
2306 }
Alexander Musman3276a272015-03-21 10:12:56 +00002307 for (const auto *E : C->inits()) {
2308 Visitor->AddStmt(E);
2309 }
2310 for (const auto *E : C->updates()) {
2311 Visitor->AddStmt(E);
2312 }
2313 for (const auto *E : C->finals()) {
2314 Visitor->AddStmt(E);
2315 }
Alexander Musman8dba6642014-04-22 13:09:42 +00002316 Visitor->AddStmt(C->getStep());
Alexander Musman3276a272015-03-21 10:12:56 +00002317 Visitor->AddStmt(C->getCalcStep());
Alexander Musman8dba6642014-04-22 13:09:42 +00002318}
Alexander Musmanf0d76e72014-05-29 14:36:25 +00002319void OMPClauseEnqueue::VisitOMPAlignedClause(const OMPAlignedClause *C) {
2320 VisitOMPClauseList(C);
2321 Visitor->AddStmt(C->getAlignment());
2322}
Alexey Bataevd48bcd82014-03-31 03:36:38 +00002323void OMPClauseEnqueue::VisitOMPCopyinClause(const OMPCopyinClause *C) {
2324 VisitOMPClauseList(C);
Alexey Bataevf56f98c2015-04-16 05:39:01 +00002325 for (auto *E : C->source_exprs()) {
2326 Visitor->AddStmt(E);
2327 }
2328 for (auto *E : C->destination_exprs()) {
2329 Visitor->AddStmt(E);
2330 }
2331 for (auto *E : C->assignment_ops()) {
2332 Visitor->AddStmt(E);
2333 }
Alexey Bataevd48bcd82014-03-31 03:36:38 +00002334}
Alexey Bataevbae9a792014-06-27 10:37:06 +00002335void
2336OMPClauseEnqueue::VisitOMPCopyprivateClause(const OMPCopyprivateClause *C) {
2337 VisitOMPClauseList(C);
Alexey Bataeva63048e2015-03-23 06:18:07 +00002338 for (auto *E : C->source_exprs()) {
2339 Visitor->AddStmt(E);
2340 }
2341 for (auto *E : C->destination_exprs()) {
2342 Visitor->AddStmt(E);
2343 }
2344 for (auto *E : C->assignment_ops()) {
2345 Visitor->AddStmt(E);
2346 }
Alexey Bataevbae9a792014-06-27 10:37:06 +00002347}
Alexey Bataev6125da92014-07-21 11:26:11 +00002348void OMPClauseEnqueue::VisitOMPFlushClause(const OMPFlushClause *C) {
2349 VisitOMPClauseList(C);
2350}
Alexey Bataev1c2cfbc2015-06-23 14:25:19 +00002351void OMPClauseEnqueue::VisitOMPDependClause(const OMPDependClause *C) {
2352 VisitOMPClauseList(C);
2353}
Kelvin Li0bff7af2015-11-23 05:32:03 +00002354void OMPClauseEnqueue::VisitOMPMapClause(const OMPMapClause *C) {
2355 VisitOMPClauseList(C);
2356}
Carlo Bertollib4adf552016-01-15 18:50:31 +00002357void OMPClauseEnqueue::VisitOMPDistScheduleClause(
2358 const OMPDistScheduleClause *C) {
Alexey Bataev3392d762016-02-16 11:18:12 +00002359 VisitOMPClauseWithPreInit(C);
Carlo Bertollib4adf552016-01-15 18:50:31 +00002360 Visitor->AddStmt(C->getChunkSize());
Carlo Bertollib4adf552016-01-15 18:50:31 +00002361}
Alexey Bataev3392d762016-02-16 11:18:12 +00002362void OMPClauseEnqueue::VisitOMPDefaultmapClause(
2363 const OMPDefaultmapClause * /*C*/) {}
Samuel Antao661c0902016-05-26 17:39:58 +00002364void OMPClauseEnqueue::VisitOMPToClause(const OMPToClause *C) {
2365 VisitOMPClauseList(C);
2366}
Samuel Antaoec172c62016-05-26 17:49:04 +00002367void OMPClauseEnqueue::VisitOMPFromClause(const OMPFromClause *C) {
2368 VisitOMPClauseList(C);
2369}
Carlo Bertolli2404b172016-07-13 15:37:16 +00002370void OMPClauseEnqueue::VisitOMPUseDevicePtrClause(const OMPUseDevicePtrClause *C) {
2371 VisitOMPClauseList(C);
2372}
Carlo Bertolli70594e92016-07-13 17:16:49 +00002373void OMPClauseEnqueue::VisitOMPIsDevicePtrClause(const OMPIsDevicePtrClause *C) {
2374 VisitOMPClauseList(C);
2375}
Alexander Kornienkoab9db512015-06-22 23:07:51 +00002376}
Alexey Bataev756c1962013-09-24 03:17:45 +00002377
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002378void EnqueueVisitor::EnqueueChildren(const OMPClause *S) {
2379 unsigned size = WL.size();
2380 OMPClauseEnqueue Visitor(this);
2381 Visitor.Visit(S);
2382 if (size == WL.size())
2383 return;
2384 // Now reverse the entries we just added. This will match the DFS
2385 // ordering performed by the worklist.
2386 VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
2387 std::reverse(I, E);
2388}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002389void EnqueueVisitor::VisitAddrLabelExpr(const AddrLabelExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002390 WL.push_back(LabelRefVisit(E->getLabel(), E->getLabelLoc(), Parent));
2391}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002392void EnqueueVisitor::VisitBlockExpr(const BlockExpr *B) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002393 AddDecl(B->getBlockDecl());
2394}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002395void EnqueueVisitor::VisitCompoundLiteralExpr(const CompoundLiteralExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002396 EnqueueChildren(E);
2397 AddTypeLoc(E->getTypeSourceInfo());
2398}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002399void EnqueueVisitor::VisitCompoundStmt(const CompoundStmt *S) {
Pete Cooper57d3f142015-07-30 17:22:52 +00002400 for (auto &I : llvm::reverse(S->body()))
2401 AddStmt(I);
Guy Benyei11169dd2012-12-18 14:30:41 +00002402}
2403void EnqueueVisitor::
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002404VisitMSDependentExistsStmt(const MSDependentExistsStmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002405 AddStmt(S->getSubStmt());
2406 AddDeclarationNameInfo(S);
2407 if (NestedNameSpecifierLoc QualifierLoc = S->getQualifierLoc())
2408 AddNestedNameSpecifierLoc(QualifierLoc);
2409}
2410
2411void EnqueueVisitor::
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002412VisitCXXDependentScopeMemberExpr(const CXXDependentScopeMemberExpr *E) {
James Y Knight04ec5bf2015-12-24 02:59:37 +00002413 if (E->hasExplicitTemplateArgs())
2414 AddExplicitTemplateArgs(E->getTemplateArgs(), E->getNumTemplateArgs());
Guy Benyei11169dd2012-12-18 14:30:41 +00002415 AddDeclarationNameInfo(E);
2416 if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
2417 AddNestedNameSpecifierLoc(QualifierLoc);
2418 if (!E->isImplicitAccess())
2419 AddStmt(E->getBase());
2420}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002421void EnqueueVisitor::VisitCXXNewExpr(const CXXNewExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002422 // Enqueue the initializer , if any.
2423 AddStmt(E->getInitializer());
2424 // Enqueue the array size, if any.
2425 AddStmt(E->getArraySize());
2426 // Enqueue the allocated type.
2427 AddTypeLoc(E->getAllocatedTypeSourceInfo());
2428 // Enqueue the placement arguments.
2429 for (unsigned I = E->getNumPlacementArgs(); I > 0; --I)
2430 AddStmt(E->getPlacementArg(I-1));
2431}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002432void EnqueueVisitor::VisitCXXOperatorCallExpr(const CXXOperatorCallExpr *CE) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002433 for (unsigned I = CE->getNumArgs(); I > 1 /* Yes, this is 1 */; --I)
2434 AddStmt(CE->getArg(I-1));
2435 AddStmt(CE->getCallee());
2436 AddStmt(CE->getArg(0));
2437}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002438void EnqueueVisitor::VisitCXXPseudoDestructorExpr(
2439 const CXXPseudoDestructorExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002440 // Visit the name of the type being destroyed.
2441 AddTypeLoc(E->getDestroyedTypeInfo());
2442 // Visit the scope type that looks disturbingly like the nested-name-specifier
2443 // but isn't.
2444 AddTypeLoc(E->getScopeTypeInfo());
2445 // Visit the nested-name-specifier.
2446 if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
2447 AddNestedNameSpecifierLoc(QualifierLoc);
2448 // Visit base expression.
2449 AddStmt(E->getBase());
2450}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002451void EnqueueVisitor::VisitCXXScalarValueInitExpr(
2452 const CXXScalarValueInitExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002453 AddTypeLoc(E->getTypeSourceInfo());
2454}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002455void EnqueueVisitor::VisitCXXTemporaryObjectExpr(
2456 const CXXTemporaryObjectExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002457 EnqueueChildren(E);
2458 AddTypeLoc(E->getTypeSourceInfo());
2459}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002460void EnqueueVisitor::VisitCXXTypeidExpr(const CXXTypeidExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002461 EnqueueChildren(E);
2462 if (E->isTypeOperand())
2463 AddTypeLoc(E->getTypeOperandSourceInfo());
2464}
2465
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002466void EnqueueVisitor::VisitCXXUnresolvedConstructExpr(
2467 const CXXUnresolvedConstructExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002468 EnqueueChildren(E);
2469 AddTypeLoc(E->getTypeSourceInfo());
2470}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002471void EnqueueVisitor::VisitCXXUuidofExpr(const CXXUuidofExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002472 EnqueueChildren(E);
2473 if (E->isTypeOperand())
2474 AddTypeLoc(E->getTypeOperandSourceInfo());
2475}
2476
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002477void EnqueueVisitor::VisitCXXCatchStmt(const CXXCatchStmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002478 EnqueueChildren(S);
2479 AddDecl(S->getExceptionDecl());
2480}
2481
Argyrios Kyrtzidis99891242014-11-13 09:03:21 +00002482void EnqueueVisitor::VisitCXXForRangeStmt(const CXXForRangeStmt *S) {
Argyrios Kyrtzidiscde70692014-11-13 09:50:19 +00002483 AddStmt(S->getBody());
Argyrios Kyrtzidis99891242014-11-13 09:03:21 +00002484 AddStmt(S->getRangeInit());
Argyrios Kyrtzidiscde70692014-11-13 09:50:19 +00002485 AddDecl(S->getLoopVariable());
Argyrios Kyrtzidis99891242014-11-13 09:03:21 +00002486}
2487
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002488void EnqueueVisitor::VisitDeclRefExpr(const DeclRefExpr *DR) {
James Y Knight04ec5bf2015-12-24 02:59:37 +00002489 if (DR->hasExplicitTemplateArgs())
2490 AddExplicitTemplateArgs(DR->getTemplateArgs(), DR->getNumTemplateArgs());
Guy Benyei11169dd2012-12-18 14:30:41 +00002491 WL.push_back(DeclRefExprParts(DR, Parent));
2492}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002493void EnqueueVisitor::VisitDependentScopeDeclRefExpr(
2494 const DependentScopeDeclRefExpr *E) {
James Y Knight04ec5bf2015-12-24 02:59:37 +00002495 if (E->hasExplicitTemplateArgs())
2496 AddExplicitTemplateArgs(E->getTemplateArgs(), E->getNumTemplateArgs());
Guy Benyei11169dd2012-12-18 14:30:41 +00002497 AddDeclarationNameInfo(E);
2498 AddNestedNameSpecifierLoc(E->getQualifierLoc());
2499}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002500void EnqueueVisitor::VisitDeclStmt(const DeclStmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002501 unsigned size = WL.size();
2502 bool isFirst = true;
Aaron Ballman535bbcc2014-03-14 17:01:24 +00002503 for (const auto *D : S->decls()) {
2504 AddDecl(D, isFirst);
Guy Benyei11169dd2012-12-18 14:30:41 +00002505 isFirst = false;
2506 }
2507 if (size == WL.size())
2508 return;
2509 // Now reverse the entries we just added. This will match the DFS
2510 // ordering performed by the worklist.
2511 VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
2512 std::reverse(I, E);
2513}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002514void EnqueueVisitor::VisitDesignatedInitExpr(const DesignatedInitExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002515 AddStmt(E->getInit());
David Majnemerf7e36092016-06-23 00:15:04 +00002516 for (const DesignatedInitExpr::Designator &D :
2517 llvm::reverse(E->designators())) {
2518 if (D.isFieldDesignator()) {
2519 if (FieldDecl *Field = D.getField())
2520 AddMemberRef(Field, D.getFieldLoc());
Guy Benyei11169dd2012-12-18 14:30:41 +00002521 continue;
2522 }
David Majnemerf7e36092016-06-23 00:15:04 +00002523 if (D.isArrayDesignator()) {
2524 AddStmt(E->getArrayIndex(D));
Guy Benyei11169dd2012-12-18 14:30:41 +00002525 continue;
2526 }
David Majnemerf7e36092016-06-23 00:15:04 +00002527 assert(D.isArrayRangeDesignator() && "Unknown designator kind");
2528 AddStmt(E->getArrayRangeEnd(D));
2529 AddStmt(E->getArrayRangeStart(D));
Guy Benyei11169dd2012-12-18 14:30:41 +00002530 }
2531}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002532void EnqueueVisitor::VisitExplicitCastExpr(const ExplicitCastExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002533 EnqueueChildren(E);
2534 AddTypeLoc(E->getTypeInfoAsWritten());
2535}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002536void EnqueueVisitor::VisitForStmt(const ForStmt *FS) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002537 AddStmt(FS->getBody());
2538 AddStmt(FS->getInc());
2539 AddStmt(FS->getCond());
2540 AddDecl(FS->getConditionVariable());
2541 AddStmt(FS->getInit());
2542}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002543void EnqueueVisitor::VisitGotoStmt(const GotoStmt *GS) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002544 WL.push_back(LabelRefVisit(GS->getLabel(), GS->getLabelLoc(), Parent));
2545}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002546void EnqueueVisitor::VisitIfStmt(const IfStmt *If) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002547 AddStmt(If->getElse());
2548 AddStmt(If->getThen());
2549 AddStmt(If->getCond());
2550 AddDecl(If->getConditionVariable());
2551}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002552void EnqueueVisitor::VisitInitListExpr(const InitListExpr *IE) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002553 // We care about the syntactic form of the initializer list, only.
2554 if (InitListExpr *Syntactic = IE->getSyntacticForm())
2555 IE = Syntactic;
2556 EnqueueChildren(IE);
2557}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002558void EnqueueVisitor::VisitMemberExpr(const MemberExpr *M) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002559 WL.push_back(MemberExprParts(M, Parent));
2560
2561 // If the base of the member access expression is an implicit 'this', don't
2562 // visit it.
2563 // FIXME: If we ever want to show these implicit accesses, this will be
2564 // unfortunate. However, clang_getCursor() relies on this behavior.
Argyrios Kyrtzidis58d0e7a2015-03-13 04:40:07 +00002565 if (M->isImplicitAccess())
2566 return;
2567
2568 // Ignore base anonymous struct/union fields, otherwise they will shadow the
2569 // real field that that we are interested in.
2570 if (auto *SubME = dyn_cast<MemberExpr>(M->getBase())) {
2571 if (auto *FD = dyn_cast_or_null<FieldDecl>(SubME->getMemberDecl())) {
2572 if (FD->isAnonymousStructOrUnion()) {
2573 AddStmt(SubME->getBase());
2574 return;
2575 }
2576 }
2577 }
2578
2579 AddStmt(M->getBase());
Guy Benyei11169dd2012-12-18 14:30:41 +00002580}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002581void EnqueueVisitor::VisitObjCEncodeExpr(const ObjCEncodeExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002582 AddTypeLoc(E->getEncodedTypeSourceInfo());
2583}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002584void EnqueueVisitor::VisitObjCMessageExpr(const ObjCMessageExpr *M) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002585 EnqueueChildren(M);
2586 AddTypeLoc(M->getClassReceiverTypeInfo());
2587}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002588void EnqueueVisitor::VisitOffsetOfExpr(const OffsetOfExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002589 // Visit the components of the offsetof expression.
2590 for (unsigned N = E->getNumComponents(), I = N; I > 0; --I) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002591 const OffsetOfNode &Node = E->getComponent(I-1);
2592 switch (Node.getKind()) {
2593 case OffsetOfNode::Array:
2594 AddStmt(E->getIndexExpr(Node.getArrayExprIndex()));
2595 break;
2596 case OffsetOfNode::Field:
2597 AddMemberRef(Node.getField(), Node.getSourceRange().getEnd());
2598 break;
2599 case OffsetOfNode::Identifier:
2600 case OffsetOfNode::Base:
2601 continue;
2602 }
2603 }
2604 // Visit the type into which we're computing the offset.
2605 AddTypeLoc(E->getTypeSourceInfo());
2606}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002607void EnqueueVisitor::VisitOverloadExpr(const OverloadExpr *E) {
James Y Knight04ec5bf2015-12-24 02:59:37 +00002608 if (E->hasExplicitTemplateArgs())
2609 AddExplicitTemplateArgs(E->getTemplateArgs(), E->getNumTemplateArgs());
Guy Benyei11169dd2012-12-18 14:30:41 +00002610 WL.push_back(OverloadExprParts(E, Parent));
2611}
2612void EnqueueVisitor::VisitUnaryExprOrTypeTraitExpr(
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002613 const UnaryExprOrTypeTraitExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002614 EnqueueChildren(E);
2615 if (E->isArgumentType())
2616 AddTypeLoc(E->getArgumentTypeInfo());
2617}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002618void EnqueueVisitor::VisitStmt(const Stmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002619 EnqueueChildren(S);
2620}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002621void EnqueueVisitor::VisitSwitchStmt(const SwitchStmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002622 AddStmt(S->getBody());
2623 AddStmt(S->getCond());
2624 AddDecl(S->getConditionVariable());
2625}
2626
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002627void EnqueueVisitor::VisitWhileStmt(const WhileStmt *W) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002628 AddStmt(W->getBody());
2629 AddStmt(W->getCond());
2630 AddDecl(W->getConditionVariable());
2631}
2632
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002633void EnqueueVisitor::VisitTypeTraitExpr(const TypeTraitExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002634 for (unsigned I = E->getNumArgs(); I > 0; --I)
2635 AddTypeLoc(E->getArg(I-1));
2636}
2637
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002638void EnqueueVisitor::VisitArrayTypeTraitExpr(const ArrayTypeTraitExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002639 AddTypeLoc(E->getQueriedTypeSourceInfo());
2640}
2641
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002642void EnqueueVisitor::VisitExpressionTraitExpr(const ExpressionTraitExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002643 EnqueueChildren(E);
2644}
2645
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002646void EnqueueVisitor::VisitUnresolvedMemberExpr(const UnresolvedMemberExpr *U) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002647 VisitOverloadExpr(U);
2648 if (!U->isImplicitAccess())
2649 AddStmt(U->getBase());
2650}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002651void EnqueueVisitor::VisitVAArgExpr(const VAArgExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002652 AddStmt(E->getSubExpr());
2653 AddTypeLoc(E->getWrittenTypeInfo());
2654}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002655void EnqueueVisitor::VisitSizeOfPackExpr(const SizeOfPackExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002656 WL.push_back(SizeOfPackExprParts(E, Parent));
2657}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002658void EnqueueVisitor::VisitOpaqueValueExpr(const OpaqueValueExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002659 // If the opaque value has a source expression, just transparently
2660 // visit that. This is useful for (e.g.) pseudo-object expressions.
2661 if (Expr *SourceExpr = E->getSourceExpr())
2662 return Visit(SourceExpr);
2663}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002664void EnqueueVisitor::VisitLambdaExpr(const LambdaExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002665 AddStmt(E->getBody());
2666 WL.push_back(LambdaExprParts(E, Parent));
2667}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002668void EnqueueVisitor::VisitPseudoObjectExpr(const PseudoObjectExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002669 // Treat the expression like its syntactic form.
2670 Visit(E->getSyntacticForm());
2671}
2672
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002673void EnqueueVisitor::VisitOMPExecutableDirective(
2674 const OMPExecutableDirective *D) {
2675 EnqueueChildren(D);
2676 for (ArrayRef<OMPClause *>::iterator I = D->clauses().begin(),
2677 E = D->clauses().end();
2678 I != E; ++I)
2679 EnqueueChildren(*I);
2680}
2681
Alexander Musman3aaab662014-08-19 11:27:13 +00002682void EnqueueVisitor::VisitOMPLoopDirective(const OMPLoopDirective *D) {
2683 VisitOMPExecutableDirective(D);
2684}
2685
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002686void EnqueueVisitor::VisitOMPParallelDirective(const OMPParallelDirective *D) {
2687 VisitOMPExecutableDirective(D);
2688}
2689
Alexey Bataev1b59ab52014-02-27 08:29:12 +00002690void EnqueueVisitor::VisitOMPSimdDirective(const OMPSimdDirective *D) {
Alexander Musman3aaab662014-08-19 11:27:13 +00002691 VisitOMPLoopDirective(D);
Alexey Bataev1b59ab52014-02-27 08:29:12 +00002692}
2693
Alexey Bataevf29276e2014-06-18 04:14:57 +00002694void EnqueueVisitor::VisitOMPForDirective(const OMPForDirective *D) {
Alexander Musman3aaab662014-08-19 11:27:13 +00002695 VisitOMPLoopDirective(D);
Alexey Bataevf29276e2014-06-18 04:14:57 +00002696}
2697
Alexander Musmanf82886e2014-09-18 05:12:34 +00002698void EnqueueVisitor::VisitOMPForSimdDirective(const OMPForSimdDirective *D) {
2699 VisitOMPLoopDirective(D);
2700}
2701
Alexey Bataevd3f8dd22014-06-25 11:44:49 +00002702void EnqueueVisitor::VisitOMPSectionsDirective(const OMPSectionsDirective *D) {
2703 VisitOMPExecutableDirective(D);
2704}
2705
Alexey Bataev1e0498a2014-06-26 08:21:58 +00002706void EnqueueVisitor::VisitOMPSectionDirective(const OMPSectionDirective *D) {
2707 VisitOMPExecutableDirective(D);
2708}
2709
Alexey Bataevd1e40fb2014-06-26 12:05:45 +00002710void EnqueueVisitor::VisitOMPSingleDirective(const OMPSingleDirective *D) {
2711 VisitOMPExecutableDirective(D);
2712}
2713
Alexander Musman80c22892014-07-17 08:54:58 +00002714void EnqueueVisitor::VisitOMPMasterDirective(const OMPMasterDirective *D) {
2715 VisitOMPExecutableDirective(D);
2716}
2717
Alexander Musmand9ed09f2014-07-21 09:42:05 +00002718void EnqueueVisitor::VisitOMPCriticalDirective(const OMPCriticalDirective *D) {
2719 VisitOMPExecutableDirective(D);
2720 AddDeclarationNameInfo(D);
2721}
2722
Alexey Bataev4acb8592014-07-07 13:01:15 +00002723void
2724EnqueueVisitor::VisitOMPParallelForDirective(const OMPParallelForDirective *D) {
Alexander Musman3aaab662014-08-19 11:27:13 +00002725 VisitOMPLoopDirective(D);
Alexey Bataev4acb8592014-07-07 13:01:15 +00002726}
2727
Alexander Musmane4e893b2014-09-23 09:33:00 +00002728void EnqueueVisitor::VisitOMPParallelForSimdDirective(
2729 const OMPParallelForSimdDirective *D) {
2730 VisitOMPLoopDirective(D);
2731}
2732
Alexey Bataev84d0b3e2014-07-08 08:12:03 +00002733void EnqueueVisitor::VisitOMPParallelSectionsDirective(
2734 const OMPParallelSectionsDirective *D) {
2735 VisitOMPExecutableDirective(D);
2736}
2737
Alexey Bataev9c2e8ee2014-07-11 11:25:16 +00002738void EnqueueVisitor::VisitOMPTaskDirective(const OMPTaskDirective *D) {
2739 VisitOMPExecutableDirective(D);
2740}
2741
Alexey Bataev68446b72014-07-18 07:47:19 +00002742void
2743EnqueueVisitor::VisitOMPTaskyieldDirective(const OMPTaskyieldDirective *D) {
2744 VisitOMPExecutableDirective(D);
2745}
2746
Alexey Bataev4d1dfea2014-07-18 09:11:51 +00002747void EnqueueVisitor::VisitOMPBarrierDirective(const OMPBarrierDirective *D) {
2748 VisitOMPExecutableDirective(D);
2749}
2750
Alexey Bataev2df347a2014-07-18 10:17:07 +00002751void EnqueueVisitor::VisitOMPTaskwaitDirective(const OMPTaskwaitDirective *D) {
2752 VisitOMPExecutableDirective(D);
2753}
2754
Alexey Bataevc30dd2d2015-06-18 12:14:09 +00002755void EnqueueVisitor::VisitOMPTaskgroupDirective(
2756 const OMPTaskgroupDirective *D) {
2757 VisitOMPExecutableDirective(D);
Alexey Bataev3b1b8952017-07-25 15:53:26 +00002758 if (const Expr *E = D->getReductionRef())
2759 VisitStmt(E);
Alexey Bataevc30dd2d2015-06-18 12:14:09 +00002760}
2761
Alexey Bataev6125da92014-07-21 11:26:11 +00002762void EnqueueVisitor::VisitOMPFlushDirective(const OMPFlushDirective *D) {
2763 VisitOMPExecutableDirective(D);
2764}
2765
Alexey Bataev9fb6e642014-07-22 06:45:04 +00002766void EnqueueVisitor::VisitOMPOrderedDirective(const OMPOrderedDirective *D) {
2767 VisitOMPExecutableDirective(D);
2768}
2769
Alexey Bataev0162e452014-07-22 10:10:35 +00002770void EnqueueVisitor::VisitOMPAtomicDirective(const OMPAtomicDirective *D) {
2771 VisitOMPExecutableDirective(D);
2772}
2773
Alexey Bataev0bd520b2014-09-19 08:19:49 +00002774void EnqueueVisitor::VisitOMPTargetDirective(const OMPTargetDirective *D) {
2775 VisitOMPExecutableDirective(D);
2776}
2777
Michael Wong65f367f2015-07-21 13:44:28 +00002778void EnqueueVisitor::VisitOMPTargetDataDirective(const
2779 OMPTargetDataDirective *D) {
2780 VisitOMPExecutableDirective(D);
2781}
2782
Samuel Antaodf67fc42016-01-19 19:15:56 +00002783void EnqueueVisitor::VisitOMPTargetEnterDataDirective(
2784 const OMPTargetEnterDataDirective *D) {
2785 VisitOMPExecutableDirective(D);
2786}
2787
Samuel Antao72590762016-01-19 20:04:50 +00002788void EnqueueVisitor::VisitOMPTargetExitDataDirective(
2789 const OMPTargetExitDataDirective *D) {
2790 VisitOMPExecutableDirective(D);
2791}
2792
Arpith Chacko Jacobe955b3d2016-01-26 18:48:41 +00002793void EnqueueVisitor::VisitOMPTargetParallelDirective(
2794 const OMPTargetParallelDirective *D) {
2795 VisitOMPExecutableDirective(D);
2796}
2797
Arpith Chacko Jacob05bebb52016-02-03 15:46:42 +00002798void EnqueueVisitor::VisitOMPTargetParallelForDirective(
2799 const OMPTargetParallelForDirective *D) {
2800 VisitOMPLoopDirective(D);
2801}
2802
Alexey Bataev13314bf2014-10-09 04:18:56 +00002803void EnqueueVisitor::VisitOMPTeamsDirective(const OMPTeamsDirective *D) {
2804 VisitOMPExecutableDirective(D);
2805}
2806
Alexey Bataev6d4ed052015-07-01 06:57:41 +00002807void EnqueueVisitor::VisitOMPCancellationPointDirective(
2808 const OMPCancellationPointDirective *D) {
2809 VisitOMPExecutableDirective(D);
2810}
2811
Alexey Bataev80909872015-07-02 11:25:17 +00002812void EnqueueVisitor::VisitOMPCancelDirective(const OMPCancelDirective *D) {
2813 VisitOMPExecutableDirective(D);
2814}
2815
Alexey Bataev49f6e782015-12-01 04:18:41 +00002816void EnqueueVisitor::VisitOMPTaskLoopDirective(const OMPTaskLoopDirective *D) {
2817 VisitOMPLoopDirective(D);
2818}
2819
Alexey Bataev0a6ed842015-12-03 09:40:15 +00002820void EnqueueVisitor::VisitOMPTaskLoopSimdDirective(
2821 const OMPTaskLoopSimdDirective *D) {
2822 VisitOMPLoopDirective(D);
2823}
2824
Carlo Bertolli6200a3d2015-12-14 14:51:25 +00002825void EnqueueVisitor::VisitOMPDistributeDirective(
2826 const OMPDistributeDirective *D) {
2827 VisitOMPLoopDirective(D);
2828}
2829
Carlo Bertolli9925f152016-06-27 14:55:37 +00002830void EnqueueVisitor::VisitOMPDistributeParallelForDirective(
2831 const OMPDistributeParallelForDirective *D) {
2832 VisitOMPLoopDirective(D);
2833}
2834
Kelvin Li4a39add2016-07-05 05:00:15 +00002835void EnqueueVisitor::VisitOMPDistributeParallelForSimdDirective(
2836 const OMPDistributeParallelForSimdDirective *D) {
2837 VisitOMPLoopDirective(D);
2838}
2839
Kelvin Li787f3fc2016-07-06 04:45:38 +00002840void EnqueueVisitor::VisitOMPDistributeSimdDirective(
2841 const OMPDistributeSimdDirective *D) {
2842 VisitOMPLoopDirective(D);
2843}
2844
Kelvin Lia579b912016-07-14 02:54:56 +00002845void EnqueueVisitor::VisitOMPTargetParallelForSimdDirective(
2846 const OMPTargetParallelForSimdDirective *D) {
2847 VisitOMPLoopDirective(D);
2848}
2849
Kelvin Li986330c2016-07-20 22:57:10 +00002850void EnqueueVisitor::VisitOMPTargetSimdDirective(
2851 const OMPTargetSimdDirective *D) {
2852 VisitOMPLoopDirective(D);
2853}
2854
Kelvin Li02532872016-08-05 14:37:37 +00002855void EnqueueVisitor::VisitOMPTeamsDistributeDirective(
2856 const OMPTeamsDistributeDirective *D) {
2857 VisitOMPLoopDirective(D);
2858}
2859
Kelvin Li4e325f72016-10-25 12:50:55 +00002860void EnqueueVisitor::VisitOMPTeamsDistributeSimdDirective(
2861 const OMPTeamsDistributeSimdDirective *D) {
2862 VisitOMPLoopDirective(D);
2863}
2864
Kelvin Li579e41c2016-11-30 23:51:03 +00002865void EnqueueVisitor::VisitOMPTeamsDistributeParallelForSimdDirective(
2866 const OMPTeamsDistributeParallelForSimdDirective *D) {
2867 VisitOMPLoopDirective(D);
2868}
2869
Kelvin Li7ade93f2016-12-09 03:24:30 +00002870void EnqueueVisitor::VisitOMPTeamsDistributeParallelForDirective(
2871 const OMPTeamsDistributeParallelForDirective *D) {
2872 VisitOMPLoopDirective(D);
2873}
2874
Kelvin Libf594a52016-12-17 05:48:59 +00002875void EnqueueVisitor::VisitOMPTargetTeamsDirective(
2876 const OMPTargetTeamsDirective *D) {
2877 VisitOMPExecutableDirective(D);
2878}
2879
Kelvin Li83c451e2016-12-25 04:52:54 +00002880void EnqueueVisitor::VisitOMPTargetTeamsDistributeDirective(
2881 const OMPTargetTeamsDistributeDirective *D) {
2882 VisitOMPLoopDirective(D);
2883}
2884
Kelvin Li80e8f562016-12-29 22:16:30 +00002885void EnqueueVisitor::VisitOMPTargetTeamsDistributeParallelForDirective(
2886 const OMPTargetTeamsDistributeParallelForDirective *D) {
2887 VisitOMPLoopDirective(D);
2888}
2889
Kelvin Li1851df52017-01-03 05:23:48 +00002890void EnqueueVisitor::VisitOMPTargetTeamsDistributeParallelForSimdDirective(
2891 const OMPTargetTeamsDistributeParallelForSimdDirective *D) {
2892 VisitOMPLoopDirective(D);
2893}
2894
Kelvin Lida681182017-01-10 18:08:18 +00002895void EnqueueVisitor::VisitOMPTargetTeamsDistributeSimdDirective(
2896 const OMPTargetTeamsDistributeSimdDirective *D) {
2897 VisitOMPLoopDirective(D);
2898}
2899
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002900void CursorVisitor::EnqueueWorkList(VisitorWorkList &WL, const Stmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002901 EnqueueVisitor(WL, MakeCXCursor(S, StmtParent, TU,RegionOfInterest)).Visit(S);
2902}
2903
2904bool CursorVisitor::IsInRegionOfInterest(CXCursor C) {
2905 if (RegionOfInterest.isValid()) {
2906 SourceRange Range = getRawCursorExtent(C);
2907 if (Range.isInvalid() || CompareRegionOfInterest(Range))
2908 return false;
2909 }
2910 return true;
2911}
2912
2913bool CursorVisitor::RunVisitorWorkList(VisitorWorkList &WL) {
2914 while (!WL.empty()) {
2915 // Dequeue the worklist item.
Robert Wilhelm25284cc2013-08-23 16:11:15 +00002916 VisitorJob LI = WL.pop_back_val();
Guy Benyei11169dd2012-12-18 14:30:41 +00002917
2918 // Set the Parent field, then back to its old value once we're done.
2919 SetParentRAII SetParent(Parent, StmtParent, LI.getParent());
2920
2921 switch (LI.getKind()) {
2922 case VisitorJob::DeclVisitKind: {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002923 const Decl *D = cast<DeclVisit>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002924 if (!D)
2925 continue;
2926
2927 // For now, perform default visitation for Decls.
2928 if (Visit(MakeCXCursor(D, TU, RegionOfInterest,
2929 cast<DeclVisit>(&LI)->isFirst())))
2930 return true;
2931
2932 continue;
2933 }
2934 case VisitorJob::ExplicitTemplateArgsVisitKind: {
James Y Knight04ec5bf2015-12-24 02:59:37 +00002935 for (const TemplateArgumentLoc &Arg :
2936 *cast<ExplicitTemplateArgsVisit>(&LI)) {
2937 if (VisitTemplateArgumentLoc(Arg))
Guy Benyei11169dd2012-12-18 14:30:41 +00002938 return true;
2939 }
2940 continue;
2941 }
2942 case VisitorJob::TypeLocVisitKind: {
2943 // Perform default visitation for TypeLocs.
2944 if (Visit(cast<TypeLocVisit>(&LI)->get()))
2945 return true;
2946 continue;
2947 }
2948 case VisitorJob::LabelRefVisitKind: {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002949 const LabelDecl *LS = cast<LabelRefVisit>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002950 if (LabelStmt *stmt = LS->getStmt()) {
2951 if (Visit(MakeCursorLabelRef(stmt, cast<LabelRefVisit>(&LI)->getLoc(),
2952 TU))) {
2953 return true;
2954 }
2955 }
2956 continue;
2957 }
2958
2959 case VisitorJob::NestedNameSpecifierLocVisitKind: {
2960 NestedNameSpecifierLocVisit *V = cast<NestedNameSpecifierLocVisit>(&LI);
2961 if (VisitNestedNameSpecifierLoc(V->get()))
2962 return true;
2963 continue;
2964 }
2965
2966 case VisitorJob::DeclarationNameInfoVisitKind: {
2967 if (VisitDeclarationNameInfo(cast<DeclarationNameInfoVisit>(&LI)
2968 ->get()))
2969 return true;
2970 continue;
2971 }
2972 case VisitorJob::MemberRefVisitKind: {
2973 MemberRefVisit *V = cast<MemberRefVisit>(&LI);
2974 if (Visit(MakeCursorMemberRef(V->get(), V->getLoc(), TU)))
2975 return true;
2976 continue;
2977 }
2978 case VisitorJob::StmtVisitKind: {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002979 const Stmt *S = cast<StmtVisit>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002980 if (!S)
2981 continue;
2982
2983 // Update the current cursor.
2984 CXCursor Cursor = MakeCXCursor(S, StmtParent, TU, RegionOfInterest);
2985 if (!IsInRegionOfInterest(Cursor))
2986 continue;
2987 switch (Visitor(Cursor, Parent, ClientData)) {
2988 case CXChildVisit_Break: return true;
2989 case CXChildVisit_Continue: break;
2990 case CXChildVisit_Recurse:
2991 if (PostChildrenVisitor)
Craig Topper69186e72014-06-08 08:38:04 +00002992 WL.push_back(PostChildrenVisit(nullptr, Cursor));
Guy Benyei11169dd2012-12-18 14:30:41 +00002993 EnqueueWorkList(WL, S);
2994 break;
2995 }
2996 continue;
2997 }
2998 case VisitorJob::MemberExprPartsKind: {
2999 // Handle the other pieces in the MemberExpr besides the base.
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003000 const MemberExpr *M = cast<MemberExprParts>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00003001
3002 // Visit the nested-name-specifier
3003 if (NestedNameSpecifierLoc QualifierLoc = M->getQualifierLoc())
3004 if (VisitNestedNameSpecifierLoc(QualifierLoc))
3005 return true;
3006
3007 // Visit the declaration name.
3008 if (VisitDeclarationNameInfo(M->getMemberNameInfo()))
3009 return true;
3010
3011 // Visit the explicitly-specified template arguments, if any.
3012 if (M->hasExplicitTemplateArgs()) {
3013 for (const TemplateArgumentLoc *Arg = M->getTemplateArgs(),
3014 *ArgEnd = Arg + M->getNumTemplateArgs();
3015 Arg != ArgEnd; ++Arg) {
3016 if (VisitTemplateArgumentLoc(*Arg))
3017 return true;
3018 }
3019 }
3020 continue;
3021 }
3022 case VisitorJob::DeclRefExprPartsKind: {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003023 const DeclRefExpr *DR = cast<DeclRefExprParts>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00003024 // Visit nested-name-specifier, if present.
3025 if (NestedNameSpecifierLoc QualifierLoc = DR->getQualifierLoc())
3026 if (VisitNestedNameSpecifierLoc(QualifierLoc))
3027 return true;
3028 // Visit declaration name.
3029 if (VisitDeclarationNameInfo(DR->getNameInfo()))
3030 return true;
3031 continue;
3032 }
3033 case VisitorJob::OverloadExprPartsKind: {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003034 const OverloadExpr *O = cast<OverloadExprParts>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00003035 // Visit the nested-name-specifier.
3036 if (NestedNameSpecifierLoc QualifierLoc = O->getQualifierLoc())
3037 if (VisitNestedNameSpecifierLoc(QualifierLoc))
3038 return true;
3039 // Visit the declaration name.
3040 if (VisitDeclarationNameInfo(O->getNameInfo()))
3041 return true;
3042 // Visit the overloaded declaration reference.
3043 if (Visit(MakeCursorOverloadedDeclRef(O, TU)))
3044 return true;
3045 continue;
3046 }
3047 case VisitorJob::SizeOfPackExprPartsKind: {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003048 const SizeOfPackExpr *E = cast<SizeOfPackExprParts>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00003049 NamedDecl *Pack = E->getPack();
3050 if (isa<TemplateTypeParmDecl>(Pack)) {
3051 if (Visit(MakeCursorTypeRef(cast<TemplateTypeParmDecl>(Pack),
3052 E->getPackLoc(), TU)))
3053 return true;
3054
3055 continue;
3056 }
3057
3058 if (isa<TemplateTemplateParmDecl>(Pack)) {
3059 if (Visit(MakeCursorTemplateRef(cast<TemplateTemplateParmDecl>(Pack),
3060 E->getPackLoc(), TU)))
3061 return true;
3062
3063 continue;
3064 }
3065
3066 // Non-type template parameter packs and function parameter packs are
3067 // treated like DeclRefExpr cursors.
3068 continue;
3069 }
3070
3071 case VisitorJob::LambdaExprPartsKind: {
3072 // Visit captures.
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003073 const LambdaExpr *E = cast<LambdaExprParts>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00003074 for (LambdaExpr::capture_iterator C = E->explicit_capture_begin(),
3075 CEnd = E->explicit_capture_end();
3076 C != CEnd; ++C) {
Richard Smithba71c082013-05-16 06:20:58 +00003077 // FIXME: Lambda init-captures.
3078 if (!C->capturesVariable())
Guy Benyei11169dd2012-12-18 14:30:41 +00003079 continue;
Richard Smithba71c082013-05-16 06:20:58 +00003080
Guy Benyei11169dd2012-12-18 14:30:41 +00003081 if (Visit(MakeCursorVariableRef(C->getCapturedVar(),
3082 C->getLocation(),
3083 TU)))
3084 return true;
3085 }
3086
3087 // Visit parameters and return type, if present.
3088 if (E->hasExplicitParameters() || E->hasExplicitResultType()) {
3089 TypeLoc TL = E->getCallOperator()->getTypeSourceInfo()->getTypeLoc();
3090 if (E->hasExplicitParameters() && E->hasExplicitResultType()) {
3091 // Visit the whole type.
3092 if (Visit(TL))
3093 return true;
David Blaikie6adc78e2013-02-18 22:06:02 +00003094 } else if (FunctionProtoTypeLoc Proto =
3095 TL.getAs<FunctionProtoTypeLoc>()) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003096 if (E->hasExplicitParameters()) {
3097 // Visit parameters.
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00003098 for (unsigned I = 0, N = Proto.getNumParams(); I != N; ++I)
3099 if (Visit(MakeCXCursor(Proto.getParam(I), TU)))
Guy Benyei11169dd2012-12-18 14:30:41 +00003100 return true;
3101 } else {
3102 // Visit result type.
Alp Toker42a16a62014-01-25 23:51:36 +00003103 if (Visit(Proto.getReturnLoc()))
Guy Benyei11169dd2012-12-18 14:30:41 +00003104 return true;
3105 }
3106 }
3107 }
3108 break;
3109 }
3110
3111 case VisitorJob::PostChildrenVisitKind:
3112 if (PostChildrenVisitor(Parent, ClientData))
3113 return true;
3114 break;
3115 }
3116 }
3117 return false;
3118}
3119
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003120bool CursorVisitor::Visit(const Stmt *S) {
Craig Topper69186e72014-06-08 08:38:04 +00003121 VisitorWorkList *WL = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00003122 if (!WorkListFreeList.empty()) {
3123 WL = WorkListFreeList.back();
3124 WL->clear();
3125 WorkListFreeList.pop_back();
3126 }
3127 else {
3128 WL = new VisitorWorkList();
3129 WorkListCache.push_back(WL);
3130 }
3131 EnqueueWorkList(*WL, S);
3132 bool result = RunVisitorWorkList(*WL);
3133 WorkListFreeList.push_back(WL);
3134 return result;
3135}
3136
3137namespace {
Dmitri Gribenkof8579502013-01-12 19:30:44 +00003138typedef SmallVector<SourceRange, 4> RefNamePieces;
James Y Knight04ec5bf2015-12-24 02:59:37 +00003139RefNamePieces buildPieces(unsigned NameFlags, bool IsMemberRefExpr,
3140 const DeclarationNameInfo &NI, SourceRange QLoc,
3141 const SourceRange *TemplateArgsLoc = nullptr) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003142 const bool WantQualifier = NameFlags & CXNameRange_WantQualifier;
3143 const bool WantTemplateArgs = NameFlags & CXNameRange_WantTemplateArgs;
3144 const bool WantSinglePiece = NameFlags & CXNameRange_WantSinglePiece;
3145
3146 const DeclarationName::NameKind Kind = NI.getName().getNameKind();
3147
3148 RefNamePieces Pieces;
3149
3150 if (WantQualifier && QLoc.isValid())
3151 Pieces.push_back(QLoc);
3152
3153 if (Kind != DeclarationName::CXXOperatorName || IsMemberRefExpr)
3154 Pieces.push_back(NI.getLoc());
James Y Knight04ec5bf2015-12-24 02:59:37 +00003155
3156 if (WantTemplateArgs && TemplateArgsLoc && TemplateArgsLoc->isValid())
3157 Pieces.push_back(*TemplateArgsLoc);
3158
Guy Benyei11169dd2012-12-18 14:30:41 +00003159 if (Kind == DeclarationName::CXXOperatorName) {
3160 Pieces.push_back(SourceLocation::getFromRawEncoding(
3161 NI.getInfo().CXXOperatorName.BeginOpNameLoc));
3162 Pieces.push_back(SourceLocation::getFromRawEncoding(
3163 NI.getInfo().CXXOperatorName.EndOpNameLoc));
3164 }
3165
3166 if (WantSinglePiece) {
3167 SourceRange R(Pieces.front().getBegin(), Pieces.back().getEnd());
3168 Pieces.clear();
3169 Pieces.push_back(R);
3170 }
3171
3172 return Pieces;
3173}
Alexander Kornienkoab9db512015-06-22 23:07:51 +00003174}
Guy Benyei11169dd2012-12-18 14:30:41 +00003175
3176//===----------------------------------------------------------------------===//
3177// Misc. API hooks.
3178//===----------------------------------------------------------------------===//
3179
Chad Rosier05c71aa2013-03-27 18:28:23 +00003180static void fatal_error_handler(void *user_data, const std::string& reason,
3181 bool gen_crash_diag) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003182 // Write the result out to stderr avoiding errs() because raw_ostreams can
3183 // call report_fatal_error.
3184 fprintf(stderr, "LIBCLANG FATAL ERROR: %s\n", reason.c_str());
3185 ::abort();
3186}
3187
Chandler Carruth66660742014-06-27 16:37:27 +00003188namespace {
3189struct RegisterFatalErrorHandler {
3190 RegisterFatalErrorHandler() {
3191 llvm::install_fatal_error_handler(fatal_error_handler, nullptr);
3192 }
3193};
3194}
3195
3196static llvm::ManagedStatic<RegisterFatalErrorHandler> RegisterFatalErrorHandlerOnce;
3197
Guy Benyei11169dd2012-12-18 14:30:41 +00003198CXIndex clang_createIndex(int excludeDeclarationsFromPCH,
3199 int displayDiagnostics) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003200 // We use crash recovery to make some of our APIs more reliable, implicitly
3201 // enable it.
Argyrios Kyrtzidis3701f542013-11-27 08:58:09 +00003202 if (!getenv("LIBCLANG_DISABLE_CRASH_RECOVERY"))
3203 llvm::CrashRecoveryContext::Enable();
Guy Benyei11169dd2012-12-18 14:30:41 +00003204
Chandler Carruth66660742014-06-27 16:37:27 +00003205 // Look through the managed static to trigger construction of the managed
3206 // static which registers our fatal error handler. This ensures it is only
3207 // registered once.
3208 (void)*RegisterFatalErrorHandlerOnce;
Guy Benyei11169dd2012-12-18 14:30:41 +00003209
Adrian Prantlbc068582015-07-08 01:00:30 +00003210 // Initialize targets for clang module support.
3211 llvm::InitializeAllTargets();
3212 llvm::InitializeAllTargetMCs();
3213 llvm::InitializeAllAsmPrinters();
3214 llvm::InitializeAllAsmParsers();
3215
Adrian Prantlfb2398d2015-07-17 01:19:54 +00003216 CIndexer *CIdxr = new CIndexer();
3217
Guy Benyei11169dd2012-12-18 14:30:41 +00003218 if (excludeDeclarationsFromPCH)
3219 CIdxr->setOnlyLocalDecls();
3220 if (displayDiagnostics)
3221 CIdxr->setDisplayDiagnostics();
3222
3223 if (getenv("LIBCLANG_BGPRIO_INDEX"))
3224 CIdxr->setCXGlobalOptFlags(CIdxr->getCXGlobalOptFlags() |
3225 CXGlobalOpt_ThreadBackgroundPriorityForIndexing);
3226 if (getenv("LIBCLANG_BGPRIO_EDIT"))
3227 CIdxr->setCXGlobalOptFlags(CIdxr->getCXGlobalOptFlags() |
3228 CXGlobalOpt_ThreadBackgroundPriorityForEditing);
3229
3230 return CIdxr;
3231}
3232
3233void clang_disposeIndex(CXIndex CIdx) {
3234 if (CIdx)
3235 delete static_cast<CIndexer *>(CIdx);
3236}
3237
3238void clang_CXIndex_setGlobalOptions(CXIndex CIdx, unsigned options) {
3239 if (CIdx)
3240 static_cast<CIndexer *>(CIdx)->setCXGlobalOptFlags(options);
3241}
3242
3243unsigned clang_CXIndex_getGlobalOptions(CXIndex CIdx) {
3244 if (CIdx)
3245 return static_cast<CIndexer *>(CIdx)->getCXGlobalOptFlags();
3246 return 0;
3247}
3248
3249void clang_toggleCrashRecovery(unsigned isEnabled) {
3250 if (isEnabled)
3251 llvm::CrashRecoveryContext::Enable();
3252 else
3253 llvm::CrashRecoveryContext::Disable();
3254}
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003255
Guy Benyei11169dd2012-12-18 14:30:41 +00003256CXTranslationUnit clang_createTranslationUnit(CXIndex CIdx,
3257 const char *ast_filename) {
Dmitri Gribenko8850cda2014-02-19 10:24:00 +00003258 CXTranslationUnit TU;
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003259 enum CXErrorCode Result =
3260 clang_createTranslationUnit2(CIdx, ast_filename, &TU);
Reid Klecknerfd48fc62014-02-12 23:56:20 +00003261 (void)Result;
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003262 assert((TU && Result == CXError_Success) ||
3263 (!TU && Result != CXError_Success));
3264 return TU;
3265}
3266
3267enum CXErrorCode clang_createTranslationUnit2(CXIndex CIdx,
3268 const char *ast_filename,
3269 CXTranslationUnit *out_TU) {
Dmitri Gribenko8850cda2014-02-19 10:24:00 +00003270 if (out_TU)
Craig Topper69186e72014-06-08 08:38:04 +00003271 *out_TU = nullptr;
Dmitri Gribenko8850cda2014-02-19 10:24:00 +00003272
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003273 if (!CIdx || !ast_filename || !out_TU)
3274 return CXError_InvalidArguments;
Guy Benyei11169dd2012-12-18 14:30:41 +00003275
Argyrios Kyrtzidis27021012013-05-24 22:24:07 +00003276 LOG_FUNC_SECTION {
3277 *Log << ast_filename;
3278 }
3279
Guy Benyei11169dd2012-12-18 14:30:41 +00003280 CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
3281 FileSystemOptions FileSystemOpts;
3282
Justin Bognerd512c1e2014-10-15 00:33:06 +00003283 IntrusiveRefCntPtr<DiagnosticsEngine> Diags =
3284 CompilerInstance::createDiagnostics(new DiagnosticOptions());
David Blaikie6f7382d2014-08-10 19:08:04 +00003285 std::unique_ptr<ASTUnit> AU = ASTUnit::LoadFromASTFile(
Richard Smithdbafb6c2017-06-29 23:23:46 +00003286 ast_filename, CXXIdx->getPCHContainerOperations()->getRawReader(),
3287 ASTUnit::LoadEverything, Diags,
Adrian Prantl6b21ab22015-08-27 19:46:20 +00003288 FileSystemOpts, /*UseDebugInfo=*/false,
3289 CXXIdx->getOnlyLocalDecls(), None,
David Blaikie6f7382d2014-08-10 19:08:04 +00003290 /*CaptureDiagnostics=*/true,
3291 /*AllowPCHWithCompilerErrors=*/true,
3292 /*UserFilesAreVolatile=*/true);
David Blaikieea4395e2017-01-06 19:49:01 +00003293 *out_TU = MakeCXTranslationUnit(CXXIdx, std::move(AU));
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003294 return *out_TU ? CXError_Success : CXError_Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00003295}
3296
3297unsigned clang_defaultEditingTranslationUnitOptions() {
3298 return CXTranslationUnit_PrecompiledPreamble |
3299 CXTranslationUnit_CacheCompletionResults;
3300}
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003301
Guy Benyei11169dd2012-12-18 14:30:41 +00003302CXTranslationUnit
3303clang_createTranslationUnitFromSourceFile(CXIndex CIdx,
3304 const char *source_filename,
3305 int num_command_line_args,
3306 const char * const *command_line_args,
3307 unsigned num_unsaved_files,
3308 struct CXUnsavedFile *unsaved_files) {
3309 unsigned Options = CXTranslationUnit_DetailedPreprocessingRecord;
3310 return clang_parseTranslationUnit(CIdx, source_filename,
3311 command_line_args, num_command_line_args,
3312 unsaved_files, num_unsaved_files,
3313 Options);
3314}
3315
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003316static CXErrorCode
3317clang_parseTranslationUnit_Impl(CXIndex CIdx, const char *source_filename,
3318 const char *const *command_line_args,
3319 int num_command_line_args,
3320 ArrayRef<CXUnsavedFile> unsaved_files,
3321 unsigned options, CXTranslationUnit *out_TU) {
Dmitri Gribenko1bf8d912014-02-18 15:20:02 +00003322 // Set up the initial return values.
3323 if (out_TU)
Craig Topper69186e72014-06-08 08:38:04 +00003324 *out_TU = nullptr;
Dmitri Gribenko1bf8d912014-02-18 15:20:02 +00003325
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003326 // Check arguments.
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003327 if (!CIdx || !out_TU)
3328 return CXError_InvalidArguments;
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003329
Guy Benyei11169dd2012-12-18 14:30:41 +00003330 CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
3331
3332 if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForIndexing))
3333 setThreadBackgroundPriority();
3334
3335 bool PrecompilePreamble = options & CXTranslationUnit_PrecompiledPreamble;
Benjamin Kramer5c248d82015-12-15 09:30:31 +00003336 bool CreatePreambleOnFirstParse =
3337 options & CXTranslationUnit_CreatePreambleOnFirstParse;
Guy Benyei11169dd2012-12-18 14:30:41 +00003338 // FIXME: Add a flag for modules.
3339 TranslationUnitKind TUKind
Argyrios Kyrtzidis735e92c2017-06-09 01:20:48 +00003340 = (options & (CXTranslationUnit_Incomplete |
3341 CXTranslationUnit_SingleFileParse))? TU_Prefix : TU_Complete;
Alp Toker8c8a8752013-12-03 06:53:35 +00003342 bool CacheCodeCompletionResults
Guy Benyei11169dd2012-12-18 14:30:41 +00003343 = options & CXTranslationUnit_CacheCompletionResults;
3344 bool IncludeBriefCommentsInCodeCompletion
3345 = options & CXTranslationUnit_IncludeBriefCommentsInCodeCompletion;
3346 bool SkipFunctionBodies = options & CXTranslationUnit_SkipFunctionBodies;
Argyrios Kyrtzidis735e92c2017-06-09 01:20:48 +00003347 bool SingleFileParse = options & CXTranslationUnit_SingleFileParse;
Guy Benyei11169dd2012-12-18 14:30:41 +00003348 bool ForSerialization = options & CXTranslationUnit_ForSerialization;
3349
3350 // Configure the diagnostics.
3351 IntrusiveRefCntPtr<DiagnosticsEngine>
Sean Silvaf1b49e22013-01-20 01:58:28 +00003352 Diags(CompilerInstance::createDiagnostics(new DiagnosticOptions));
Guy Benyei11169dd2012-12-18 14:30:41 +00003353
Manuel Klimek016c0242016-03-01 10:56:19 +00003354 if (options & CXTranslationUnit_KeepGoing)
Richard Smithe37391c2017-05-03 00:28:49 +00003355 Diags->setSuppressAfterFatalError(false);
Manuel Klimek016c0242016-03-01 10:56:19 +00003356
Guy Benyei11169dd2012-12-18 14:30:41 +00003357 // Recover resources if we crash before exiting this function.
3358 llvm::CrashRecoveryContextCleanupRegistrar<DiagnosticsEngine,
3359 llvm::CrashRecoveryContextReleaseRefCleanup<DiagnosticsEngine> >
Alp Tokerf994cef2014-07-05 03:08:06 +00003360 DiagCleanup(Diags.get());
Guy Benyei11169dd2012-12-18 14:30:41 +00003361
Ahmed Charlesb8984322014-03-07 20:03:18 +00003362 std::unique_ptr<std::vector<ASTUnit::RemappedFile>> RemappedFiles(
3363 new std::vector<ASTUnit::RemappedFile>());
Guy Benyei11169dd2012-12-18 14:30:41 +00003364
3365 // Recover resources if we crash before exiting this function.
3366 llvm::CrashRecoveryContextCleanupRegistrar<
3367 std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
3368
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003369 for (auto &UF : unsaved_files) {
Rafael Espindolad87f8d72014-08-27 20:03:29 +00003370 std::unique_ptr<llvm::MemoryBuffer> MB =
Alp Toker9d85b182014-07-07 01:23:14 +00003371 llvm::MemoryBuffer::getMemBufferCopy(getContents(UF), UF.Filename);
Rafael Espindolad87f8d72014-08-27 20:03:29 +00003372 RemappedFiles->push_back(std::make_pair(UF.Filename, MB.release()));
Guy Benyei11169dd2012-12-18 14:30:41 +00003373 }
3374
Ahmed Charlesb8984322014-03-07 20:03:18 +00003375 std::unique_ptr<std::vector<const char *>> Args(
3376 new std::vector<const char *>());
Guy Benyei11169dd2012-12-18 14:30:41 +00003377
3378 // Recover resources if we crash before exiting this method.
3379 llvm::CrashRecoveryContextCleanupRegistrar<std::vector<const char*> >
3380 ArgsCleanup(Args.get());
3381
3382 // Since the Clang C library is primarily used by batch tools dealing with
3383 // (often very broken) source code, where spell-checking can have a
3384 // significant negative impact on performance (particularly when
3385 // precompiled headers are involved), we disable it by default.
3386 // Only do this if we haven't found a spell-checking-related argument.
3387 bool FoundSpellCheckingArgument = false;
3388 for (int I = 0; I != num_command_line_args; ++I) {
3389 if (strcmp(command_line_args[I], "-fno-spell-checking") == 0 ||
3390 strcmp(command_line_args[I], "-fspell-checking") == 0) {
3391 FoundSpellCheckingArgument = true;
3392 break;
3393 }
3394 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003395 Args->insert(Args->end(), command_line_args,
3396 command_line_args + num_command_line_args);
3397
Benjamin Kramerc02670e2015-11-18 16:14:27 +00003398 if (!FoundSpellCheckingArgument)
3399 Args->insert(Args->begin() + 1, "-fno-spell-checking");
3400
Guy Benyei11169dd2012-12-18 14:30:41 +00003401 // The 'source_filename' argument is optional. If the caller does not
3402 // specify it then it is assumed that the source file is specified
3403 // in the actual argument list.
3404 // Put the source file after command_line_args otherwise if '-x' flag is
3405 // present it will be unused.
3406 if (source_filename)
3407 Args->push_back(source_filename);
3408
3409 // Do we need the detailed preprocessing record?
3410 if (options & CXTranslationUnit_DetailedPreprocessingRecord) {
3411 Args->push_back("-Xclang");
3412 Args->push_back("-detailed-preprocessing-record");
3413 }
Alex Lorenzcb006402017-04-27 13:47:03 +00003414
3415 // Suppress any editor placeholder diagnostics.
3416 Args->push_back("-fallow-editor-placeholders");
3417
Guy Benyei11169dd2012-12-18 14:30:41 +00003418 unsigned NumErrors = Diags->getClient()->getNumErrors();
Ahmed Charlesb8984322014-03-07 20:03:18 +00003419 std::unique_ptr<ASTUnit> ErrUnit;
Benjamin Kramer5c248d82015-12-15 09:30:31 +00003420 // Unless the user specified that they want the preamble on the first parse
3421 // set it up to be created on the first reparse. This makes the first parse
3422 // faster, trading for a slower (first) reparse.
3423 unsigned PrecompilePreambleAfterNParses =
3424 !PrecompilePreamble ? 0 : 2 - CreatePreambleOnFirstParse;
Ahmed Charlesb8984322014-03-07 20:03:18 +00003425 std::unique_ptr<ASTUnit> Unit(ASTUnit::LoadFromCommandLine(
Adrian Prantlbb165fb2015-06-20 18:53:08 +00003426 Args->data(), Args->data() + Args->size(),
3427 CXXIdx->getPCHContainerOperations(), Diags,
Ahmed Charlesb8984322014-03-07 20:03:18 +00003428 CXXIdx->getClangResourcesPath(), CXXIdx->getOnlyLocalDecls(),
3429 /*CaptureDiagnostics=*/true, *RemappedFiles.get(),
Benjamin Kramer5c248d82015-12-15 09:30:31 +00003430 /*RemappedFilesKeepOriginalName=*/true, PrecompilePreambleAfterNParses,
3431 TUKind, CacheCodeCompletionResults, IncludeBriefCommentsInCodeCompletion,
Argyrios Kyrtzidis735e92c2017-06-09 01:20:48 +00003432 /*AllowPCHWithCompilerErrors=*/true, SkipFunctionBodies, SingleFileParse,
Argyrios Kyrtzidisa3e2ff12015-11-20 03:36:21 +00003433 /*UserFilesAreVolatile=*/true, ForSerialization,
3434 CXXIdx->getPCHContainerOperations()->getRawReader().getFormat(),
3435 &ErrUnit));
Guy Benyei11169dd2012-12-18 14:30:41 +00003436
Artem Belevich0ff05cd2015-07-13 23:27:56 +00003437 // Early failures in LoadFromCommandLine may return with ErrUnit unset.
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003438 if (!Unit && !ErrUnit)
3439 return CXError_ASTReadError;
Artem Belevich0ff05cd2015-07-13 23:27:56 +00003440
Guy Benyei11169dd2012-12-18 14:30:41 +00003441 if (NumErrors != Diags->getClient()->getNumErrors()) {
3442 // Make sure to check that 'Unit' is non-NULL.
3443 if (CXXIdx->getDisplayDiagnostics())
3444 printDiagsToStderr(Unit ? Unit.get() : ErrUnit.get());
3445 }
3446
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003447 if (isASTReadError(Unit ? Unit.get() : ErrUnit.get()))
3448 return CXError_ASTReadError;
3449
David Blaikieea4395e2017-01-06 19:49:01 +00003450 *out_TU = MakeCXTranslationUnit(CXXIdx, std::move(Unit));
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003451 return *out_TU ? CXError_Success : CXError_Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00003452}
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003453
3454CXTranslationUnit
3455clang_parseTranslationUnit(CXIndex CIdx,
3456 const char *source_filename,
3457 const char *const *command_line_args,
3458 int num_command_line_args,
3459 struct CXUnsavedFile *unsaved_files,
3460 unsigned num_unsaved_files,
3461 unsigned options) {
3462 CXTranslationUnit TU;
3463 enum CXErrorCode Result = clang_parseTranslationUnit2(
3464 CIdx, source_filename, command_line_args, num_command_line_args,
3465 unsaved_files, num_unsaved_files, options, &TU);
Reid Kleckner6eaf05a2014-02-13 01:19:59 +00003466 (void)Result;
Dmitri Gribenko1bf8d912014-02-18 15:20:02 +00003467 assert((TU && Result == CXError_Success) ||
3468 (!TU && Result != CXError_Success));
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003469 return TU;
3470}
3471
3472enum CXErrorCode clang_parseTranslationUnit2(
Benjamin Kramerc02670e2015-11-18 16:14:27 +00003473 CXIndex CIdx, const char *source_filename,
3474 const char *const *command_line_args, int num_command_line_args,
3475 struct CXUnsavedFile *unsaved_files, unsigned num_unsaved_files,
3476 unsigned options, CXTranslationUnit *out_TU) {
3477 SmallVector<const char *, 4> Args;
3478 Args.push_back("clang");
3479 Args.append(command_line_args, command_line_args + num_command_line_args);
3480 return clang_parseTranslationUnit2FullArgv(
3481 CIdx, source_filename, Args.data(), Args.size(), unsaved_files,
3482 num_unsaved_files, options, out_TU);
3483}
3484
3485enum CXErrorCode clang_parseTranslationUnit2FullArgv(
3486 CXIndex CIdx, const char *source_filename,
3487 const char *const *command_line_args, int num_command_line_args,
3488 struct CXUnsavedFile *unsaved_files, unsigned num_unsaved_files,
3489 unsigned options, CXTranslationUnit *out_TU) {
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00003490 LOG_FUNC_SECTION {
3491 *Log << source_filename << ": ";
3492 for (int i = 0; i != num_command_line_args; ++i)
3493 *Log << command_line_args[i] << " ";
3494 }
3495
Alp Toker9d85b182014-07-07 01:23:14 +00003496 if (num_unsaved_files && !unsaved_files)
3497 return CXError_InvalidArguments;
3498
Alp Toker5c532982014-07-07 22:42:03 +00003499 CXErrorCode result = CXError_Failure;
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003500 auto ParseTranslationUnitImpl = [=, &result] {
3501 result = clang_parseTranslationUnit_Impl(
3502 CIdx, source_filename, command_line_args, num_command_line_args,
3503 llvm::makeArrayRef(unsaved_files, num_unsaved_files), options, out_TU);
3504 };
Guy Benyei11169dd2012-12-18 14:30:41 +00003505 llvm::CrashRecoveryContext CRC;
3506
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003507 if (!RunSafely(CRC, ParseTranslationUnitImpl)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003508 fprintf(stderr, "libclang: crash detected during parsing: {\n");
3509 fprintf(stderr, " 'source_filename' : '%s'\n", source_filename);
3510 fprintf(stderr, " 'command_line_args' : [");
3511 for (int i = 0; i != num_command_line_args; ++i) {
3512 if (i)
3513 fprintf(stderr, ", ");
3514 fprintf(stderr, "'%s'", command_line_args[i]);
3515 }
3516 fprintf(stderr, "],\n");
3517 fprintf(stderr, " 'unsaved_files' : [");
3518 for (unsigned i = 0; i != num_unsaved_files; ++i) {
3519 if (i)
3520 fprintf(stderr, ", ");
3521 fprintf(stderr, "('%s', '...', %ld)", unsaved_files[i].Filename,
3522 unsaved_files[i].Length);
3523 }
3524 fprintf(stderr, "],\n");
3525 fprintf(stderr, " 'options' : %d,\n", options);
3526 fprintf(stderr, "}\n");
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003527
3528 return CXError_Crashed;
Guy Benyei11169dd2012-12-18 14:30:41 +00003529 } else if (getenv("LIBCLANG_RESOURCE_USAGE")) {
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003530 if (CXTranslationUnit *TU = out_TU)
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003531 PrintLibclangResourceUsage(*TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00003532 }
Alp Toker5c532982014-07-07 22:42:03 +00003533
3534 return result;
Guy Benyei11169dd2012-12-18 14:30:41 +00003535}
3536
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003537CXString clang_Type_getObjCEncoding(CXType CT) {
3538 CXTranslationUnit tu = static_cast<CXTranslationUnit>(CT.data[1]);
3539 ASTContext &Ctx = getASTUnit(tu)->getASTContext();
3540 std::string encoding;
3541 Ctx.getObjCEncodingForType(QualType::getFromOpaquePtr(CT.data[0]),
3542 encoding);
3543
3544 return cxstring::createDup(encoding);
3545}
3546
3547static const IdentifierInfo *getMacroIdentifier(CXCursor C) {
3548 if (C.kind == CXCursor_MacroDefinition) {
3549 if (const MacroDefinitionRecord *MDR = getCursorMacroDefinition(C))
3550 return MDR->getName();
3551 } else if (C.kind == CXCursor_MacroExpansion) {
3552 MacroExpansionCursor ME = getCursorMacroExpansion(C);
3553 return ME.getName();
3554 }
3555 return nullptr;
3556}
3557
3558unsigned clang_Cursor_isMacroFunctionLike(CXCursor C) {
3559 const IdentifierInfo *II = getMacroIdentifier(C);
3560 if (!II) {
3561 return false;
3562 }
3563 ASTUnit *ASTU = getCursorASTUnit(C);
3564 Preprocessor &PP = ASTU->getPreprocessor();
3565 if (const MacroInfo *MI = PP.getMacroInfo(II))
3566 return MI->isFunctionLike();
3567 return false;
3568}
3569
3570unsigned clang_Cursor_isMacroBuiltin(CXCursor C) {
3571 const IdentifierInfo *II = getMacroIdentifier(C);
3572 if (!II) {
3573 return false;
3574 }
3575 ASTUnit *ASTU = getCursorASTUnit(C);
3576 Preprocessor &PP = ASTU->getPreprocessor();
3577 if (const MacroInfo *MI = PP.getMacroInfo(II))
3578 return MI->isBuiltinMacro();
3579 return false;
3580}
3581
3582unsigned clang_Cursor_isFunctionInlined(CXCursor C) {
3583 const Decl *D = getCursorDecl(C);
3584 const FunctionDecl *FD = dyn_cast_or_null<FunctionDecl>(D);
3585 if (!FD) {
3586 return false;
3587 }
3588 return FD->isInlined();
3589}
3590
3591static StringLiteral* getCFSTR_value(CallExpr *callExpr) {
3592 if (callExpr->getNumArgs() != 1) {
3593 return nullptr;
3594 }
3595
3596 StringLiteral *S = nullptr;
3597 auto *arg = callExpr->getArg(0);
3598 if (arg->getStmtClass() == Stmt::ImplicitCastExprClass) {
3599 ImplicitCastExpr *I = static_cast<ImplicitCastExpr *>(arg);
3600 auto *subExpr = I->getSubExprAsWritten();
3601
3602 if(subExpr->getStmtClass() != Stmt::StringLiteralClass){
3603 return nullptr;
3604 }
3605
3606 S = static_cast<StringLiteral *>(I->getSubExprAsWritten());
3607 } else if (arg->getStmtClass() == Stmt::StringLiteralClass) {
3608 S = static_cast<StringLiteral *>(callExpr->getArg(0));
3609 } else {
3610 return nullptr;
3611 }
3612 return S;
3613}
3614
David Blaikie59272572016-04-13 18:23:33 +00003615struct ExprEvalResult {
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003616 CXEvalResultKind EvalType;
3617 union {
Argyrios Kyrtzidis5dda1122016-12-01 23:41:27 +00003618 unsigned long long unsignedVal;
3619 long long intVal;
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003620 double floatVal;
3621 char *stringVal;
3622 } EvalData;
Argyrios Kyrtzidis5dda1122016-12-01 23:41:27 +00003623 bool IsUnsignedInt;
David Blaikie59272572016-04-13 18:23:33 +00003624 ~ExprEvalResult() {
3625 if (EvalType != CXEval_UnExposed && EvalType != CXEval_Float &&
3626 EvalType != CXEval_Int) {
3627 delete EvalData.stringVal;
3628 }
3629 }
3630};
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003631
3632void clang_EvalResult_dispose(CXEvalResult E) {
David Blaikie59272572016-04-13 18:23:33 +00003633 delete static_cast<ExprEvalResult *>(E);
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003634}
3635
3636CXEvalResultKind clang_EvalResult_getKind(CXEvalResult E) {
3637 if (!E) {
3638 return CXEval_UnExposed;
3639 }
3640 return ((ExprEvalResult *)E)->EvalType;
3641}
3642
3643int clang_EvalResult_getAsInt(CXEvalResult E) {
Argyrios Kyrtzidis5dda1122016-12-01 23:41:27 +00003644 return clang_EvalResult_getAsLongLong(E);
3645}
3646
3647long long clang_EvalResult_getAsLongLong(CXEvalResult E) {
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003648 if (!E) {
3649 return 0;
3650 }
Argyrios Kyrtzidis5dda1122016-12-01 23:41:27 +00003651 ExprEvalResult *Result = (ExprEvalResult*)E;
3652 if (Result->IsUnsignedInt)
3653 return Result->EvalData.unsignedVal;
3654 return Result->EvalData.intVal;
3655}
3656
3657unsigned clang_EvalResult_isUnsignedInt(CXEvalResult E) {
3658 return ((ExprEvalResult *)E)->IsUnsignedInt;
3659}
3660
3661unsigned long long clang_EvalResult_getAsUnsigned(CXEvalResult E) {
3662 if (!E) {
3663 return 0;
3664 }
3665
3666 ExprEvalResult *Result = (ExprEvalResult*)E;
3667 if (Result->IsUnsignedInt)
3668 return Result->EvalData.unsignedVal;
3669 return Result->EvalData.intVal;
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003670}
3671
3672double clang_EvalResult_getAsDouble(CXEvalResult E) {
3673 if (!E) {
3674 return 0;
3675 }
3676 return ((ExprEvalResult *)E)->EvalData.floatVal;
3677}
3678
3679const char* clang_EvalResult_getAsStr(CXEvalResult E) {
3680 if (!E) {
3681 return nullptr;
3682 }
3683 return ((ExprEvalResult *)E)->EvalData.stringVal;
3684}
3685
3686static const ExprEvalResult* evaluateExpr(Expr *expr, CXCursor C) {
3687 Expr::EvalResult ER;
3688 ASTContext &ctx = getCursorContext(C);
David Blaikiebbc00882016-04-13 18:36:19 +00003689 if (!expr)
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003690 return nullptr;
David Blaikiebbc00882016-04-13 18:36:19 +00003691
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003692 expr = expr->IgnoreParens();
David Blaikiebbc00882016-04-13 18:36:19 +00003693 if (!expr->EvaluateAsRValue(ER, ctx))
3694 return nullptr;
3695
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003696 QualType rettype;
3697 CallExpr *callExpr;
David Blaikie59272572016-04-13 18:23:33 +00003698 auto result = llvm::make_unique<ExprEvalResult>();
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003699 result->EvalType = CXEval_UnExposed;
Argyrios Kyrtzidis5dda1122016-12-01 23:41:27 +00003700 result->IsUnsignedInt = false;
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003701
David Blaikiebbc00882016-04-13 18:36:19 +00003702 if (ER.Val.isInt()) {
3703 result->EvalType = CXEval_Int;
Argyrios Kyrtzidis5dda1122016-12-01 23:41:27 +00003704
3705 auto& val = ER.Val.getInt();
3706 if (val.isUnsigned()) {
3707 result->IsUnsignedInt = true;
3708 result->EvalData.unsignedVal = val.getZExtValue();
3709 } else {
3710 result->EvalData.intVal = val.getExtValue();
3711 }
3712
David Blaikiebbc00882016-04-13 18:36:19 +00003713 return result.release();
3714 }
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003715
David Blaikiebbc00882016-04-13 18:36:19 +00003716 if (ER.Val.isFloat()) {
3717 llvm::SmallVector<char, 100> Buffer;
3718 ER.Val.getFloat().toString(Buffer);
3719 std::string floatStr(Buffer.data(), Buffer.size());
3720 result->EvalType = CXEval_Float;
3721 bool ignored;
3722 llvm::APFloat apFloat = ER.Val.getFloat();
Stephan Bergmann17c7f702016-12-14 11:57:17 +00003723 apFloat.convert(llvm::APFloat::IEEEdouble(),
David Blaikiebbc00882016-04-13 18:36:19 +00003724 llvm::APFloat::rmNearestTiesToEven, &ignored);
3725 result->EvalData.floatVal = apFloat.convertToDouble();
3726 return result.release();
3727 }
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003728
David Blaikiebbc00882016-04-13 18:36:19 +00003729 if (expr->getStmtClass() == Stmt::ImplicitCastExprClass) {
3730 const ImplicitCastExpr *I = dyn_cast<ImplicitCastExpr>(expr);
3731 auto *subExpr = I->getSubExprAsWritten();
3732 if (subExpr->getStmtClass() == Stmt::StringLiteralClass ||
3733 subExpr->getStmtClass() == Stmt::ObjCStringLiteralClass) {
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003734 const StringLiteral *StrE = nullptr;
3735 const ObjCStringLiteral *ObjCExpr;
David Blaikiebbc00882016-04-13 18:36:19 +00003736 ObjCExpr = dyn_cast<ObjCStringLiteral>(subExpr);
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003737
3738 if (ObjCExpr) {
3739 StrE = ObjCExpr->getString();
3740 result->EvalType = CXEval_ObjCStrLiteral;
3741 } else {
David Blaikiebbc00882016-04-13 18:36:19 +00003742 StrE = cast<StringLiteral>(I->getSubExprAsWritten());
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003743 result->EvalType = CXEval_StrLiteral;
3744 }
3745
3746 std::string strRef(StrE->getString().str());
David Blaikie59272572016-04-13 18:23:33 +00003747 result->EvalData.stringVal = new char[strRef.size() + 1];
David Blaikiebbc00882016-04-13 18:36:19 +00003748 strncpy((char *)result->EvalData.stringVal, strRef.c_str(),
3749 strRef.size());
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003750 result->EvalData.stringVal[strRef.size()] = '\0';
David Blaikie59272572016-04-13 18:23:33 +00003751 return result.release();
David Blaikiebbc00882016-04-13 18:36:19 +00003752 }
3753 } else if (expr->getStmtClass() == Stmt::ObjCStringLiteralClass ||
3754 expr->getStmtClass() == Stmt::StringLiteralClass) {
3755 const StringLiteral *StrE = nullptr;
3756 const ObjCStringLiteral *ObjCExpr;
3757 ObjCExpr = dyn_cast<ObjCStringLiteral>(expr);
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003758
David Blaikiebbc00882016-04-13 18:36:19 +00003759 if (ObjCExpr) {
3760 StrE = ObjCExpr->getString();
3761 result->EvalType = CXEval_ObjCStrLiteral;
3762 } else {
3763 StrE = cast<StringLiteral>(expr);
3764 result->EvalType = CXEval_StrLiteral;
3765 }
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003766
David Blaikiebbc00882016-04-13 18:36:19 +00003767 std::string strRef(StrE->getString().str());
3768 result->EvalData.stringVal = new char[strRef.size() + 1];
3769 strncpy((char *)result->EvalData.stringVal, strRef.c_str(), strRef.size());
3770 result->EvalData.stringVal[strRef.size()] = '\0';
3771 return result.release();
3772 }
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003773
David Blaikiebbc00882016-04-13 18:36:19 +00003774 if (expr->getStmtClass() == Stmt::CStyleCastExprClass) {
3775 CStyleCastExpr *CC = static_cast<CStyleCastExpr *>(expr);
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003776
David Blaikiebbc00882016-04-13 18:36:19 +00003777 rettype = CC->getType();
3778 if (rettype.getAsString() == "CFStringRef" &&
3779 CC->getSubExpr()->getStmtClass() == Stmt::CallExprClass) {
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003780
David Blaikiebbc00882016-04-13 18:36:19 +00003781 callExpr = static_cast<CallExpr *>(CC->getSubExpr());
3782 StringLiteral *S = getCFSTR_value(callExpr);
3783 if (S) {
3784 std::string strLiteral(S->getString().str());
3785 result->EvalType = CXEval_CFStr;
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003786
David Blaikiebbc00882016-04-13 18:36:19 +00003787 result->EvalData.stringVal = new char[strLiteral.size() + 1];
3788 strncpy((char *)result->EvalData.stringVal, strLiteral.c_str(),
3789 strLiteral.size());
3790 result->EvalData.stringVal[strLiteral.size()] = '\0';
David Blaikie59272572016-04-13 18:23:33 +00003791 return result.release();
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003792 }
3793 }
3794
David Blaikiebbc00882016-04-13 18:36:19 +00003795 } else if (expr->getStmtClass() == Stmt::CallExprClass) {
3796 callExpr = static_cast<CallExpr *>(expr);
3797 rettype = callExpr->getCallReturnType(ctx);
3798
3799 if (rettype->isVectorType() || callExpr->getNumArgs() > 1)
3800 return nullptr;
3801
3802 if (rettype->isIntegralType(ctx) || rettype->isRealFloatingType()) {
3803 if (callExpr->getNumArgs() == 1 &&
3804 !callExpr->getArg(0)->getType()->isIntegralType(ctx))
3805 return nullptr;
3806 } else if (rettype.getAsString() == "CFStringRef") {
3807
3808 StringLiteral *S = getCFSTR_value(callExpr);
3809 if (S) {
3810 std::string strLiteral(S->getString().str());
3811 result->EvalType = CXEval_CFStr;
3812 result->EvalData.stringVal = new char[strLiteral.size() + 1];
3813 strncpy((char *)result->EvalData.stringVal, strLiteral.c_str(),
3814 strLiteral.size());
3815 result->EvalData.stringVal[strLiteral.size()] = '\0';
3816 return result.release();
3817 }
3818 }
3819 } else if (expr->getStmtClass() == Stmt::DeclRefExprClass) {
3820 DeclRefExpr *D = static_cast<DeclRefExpr *>(expr);
3821 ValueDecl *V = D->getDecl();
3822 if (V->getKind() == Decl::Function) {
3823 std::string strName = V->getNameAsString();
3824 result->EvalType = CXEval_Other;
3825 result->EvalData.stringVal = new char[strName.size() + 1];
3826 strncpy(result->EvalData.stringVal, strName.c_str(), strName.size());
3827 result->EvalData.stringVal[strName.size()] = '\0';
3828 return result.release();
3829 }
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003830 }
3831
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003832 return nullptr;
3833}
3834
3835CXEvalResult clang_Cursor_Evaluate(CXCursor C) {
3836 const Decl *D = getCursorDecl(C);
3837 if (D) {
3838 const Expr *expr = nullptr;
3839 if (auto *Var = dyn_cast<VarDecl>(D)) {
3840 expr = Var->getInit();
3841 } else if (auto *Field = dyn_cast<FieldDecl>(D)) {
3842 expr = Field->getInClassInitializer();
3843 }
3844 if (expr)
Aaron Ballman01dc1572016-01-20 15:25:30 +00003845 return const_cast<CXEvalResult>(reinterpret_cast<const void *>(
3846 evaluateExpr(const_cast<Expr *>(expr), C)));
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003847 return nullptr;
3848 }
3849
3850 const CompoundStmt *compoundStmt = dyn_cast_or_null<CompoundStmt>(getCursorStmt(C));
3851 if (compoundStmt) {
3852 Expr *expr = nullptr;
3853 for (auto *bodyIterator : compoundStmt->body()) {
3854 if ((expr = dyn_cast<Expr>(bodyIterator))) {
3855 break;
3856 }
3857 }
3858 if (expr)
Aaron Ballman01dc1572016-01-20 15:25:30 +00003859 return const_cast<CXEvalResult>(
3860 reinterpret_cast<const void *>(evaluateExpr(expr, C)));
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003861 }
3862 return nullptr;
3863}
3864
3865unsigned clang_Cursor_hasAttrs(CXCursor C) {
3866 const Decl *D = getCursorDecl(C);
3867 if (!D) {
3868 return 0;
3869 }
3870
3871 if (D->hasAttrs()) {
3872 return 1;
3873 }
3874
3875 return 0;
3876}
Guy Benyei11169dd2012-12-18 14:30:41 +00003877unsigned clang_defaultSaveOptions(CXTranslationUnit TU) {
3878 return CXSaveTranslationUnit_None;
3879}
3880
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003881static CXSaveError clang_saveTranslationUnit_Impl(CXTranslationUnit TU,
3882 const char *FileName,
3883 unsigned options) {
3884 CIndexer *CXXIdx = TU->CIdx;
Guy Benyei11169dd2012-12-18 14:30:41 +00003885 if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForIndexing))
3886 setThreadBackgroundPriority();
3887
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003888 bool hadError = cxtu::getASTUnit(TU)->Save(FileName);
3889 return hadError ? CXSaveError_Unknown : CXSaveError_None;
Guy Benyei11169dd2012-12-18 14:30:41 +00003890}
3891
3892int clang_saveTranslationUnit(CXTranslationUnit TU, const char *FileName,
3893 unsigned options) {
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00003894 LOG_FUNC_SECTION {
3895 *Log << TU << ' ' << FileName;
3896 }
3897
Dmitri Gribenko852d6222014-02-11 15:02:48 +00003898 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003899 LOG_BAD_TU(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00003900 return CXSaveError_InvalidTU;
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003901 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003902
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003903 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00003904 ASTUnit::ConcurrencyCheck Check(*CXXUnit);
3905 if (!CXXUnit->hasSema())
3906 return CXSaveError_InvalidTU;
3907
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003908 CXSaveError result;
3909 auto SaveTranslationUnitImpl = [=, &result]() {
3910 result = clang_saveTranslationUnit_Impl(TU, FileName, options);
3911 };
Guy Benyei11169dd2012-12-18 14:30:41 +00003912
3913 if (!CXXUnit->getDiagnostics().hasUnrecoverableErrorOccurred() ||
3914 getenv("LIBCLANG_NOTHREADS")) {
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003915 SaveTranslationUnitImpl();
Guy Benyei11169dd2012-12-18 14:30:41 +00003916
3917 if (getenv("LIBCLANG_RESOURCE_USAGE"))
3918 PrintLibclangResourceUsage(TU);
3919
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003920 return result;
Guy Benyei11169dd2012-12-18 14:30:41 +00003921 }
3922
3923 // We have an AST that has invalid nodes due to compiler errors.
3924 // Use a crash recovery thread for protection.
3925
3926 llvm::CrashRecoveryContext CRC;
3927
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003928 if (!RunSafely(CRC, SaveTranslationUnitImpl)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003929 fprintf(stderr, "libclang: crash detected during AST saving: {\n");
3930 fprintf(stderr, " 'filename' : '%s'\n", FileName);
3931 fprintf(stderr, " 'options' : %d,\n", options);
3932 fprintf(stderr, "}\n");
3933
3934 return CXSaveError_Unknown;
3935
3936 } else if (getenv("LIBCLANG_RESOURCE_USAGE")) {
3937 PrintLibclangResourceUsage(TU);
3938 }
3939
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003940 return result;
Guy Benyei11169dd2012-12-18 14:30:41 +00003941}
3942
3943void clang_disposeTranslationUnit(CXTranslationUnit CTUnit) {
3944 if (CTUnit) {
3945 // If the translation unit has been marked as unsafe to free, just discard
3946 // it.
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003947 ASTUnit *Unit = cxtu::getASTUnit(CTUnit);
3948 if (Unit && Unit->isUnsafeToFree())
Guy Benyei11169dd2012-12-18 14:30:41 +00003949 return;
3950
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003951 delete cxtu::getASTUnit(CTUnit);
Dmitri Gribenkob95b3f12013-01-26 22:44:19 +00003952 delete CTUnit->StringPool;
Guy Benyei11169dd2012-12-18 14:30:41 +00003953 delete static_cast<CXDiagnosticSetImpl *>(CTUnit->Diagnostics);
3954 disposeOverridenCXCursorsPool(CTUnit->OverridenCursorsPool);
Dmitri Gribenko9e605112013-11-13 22:16:51 +00003955 delete CTUnit->CommentToXML;
Guy Benyei11169dd2012-12-18 14:30:41 +00003956 delete CTUnit;
3957 }
3958}
3959
Erik Verbruggen346066b2017-05-30 14:25:54 +00003960unsigned clang_suspendTranslationUnit(CXTranslationUnit CTUnit) {
3961 if (CTUnit) {
3962 ASTUnit *Unit = cxtu::getASTUnit(CTUnit);
3963
3964 if (Unit && Unit->isUnsafeToFree())
3965 return false;
3966
3967 Unit->ResetForParse();
3968 return true;
3969 }
3970
3971 return false;
3972}
3973
Guy Benyei11169dd2012-12-18 14:30:41 +00003974unsigned clang_defaultReparseOptions(CXTranslationUnit TU) {
3975 return CXReparse_None;
3976}
3977
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003978static CXErrorCode
3979clang_reparseTranslationUnit_Impl(CXTranslationUnit TU,
3980 ArrayRef<CXUnsavedFile> unsaved_files,
3981 unsigned options) {
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003982 // Check arguments.
Dmitri Gribenko852d6222014-02-11 15:02:48 +00003983 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003984 LOG_BAD_TU(TU);
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003985 return CXError_InvalidArguments;
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003986 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003987
3988 // Reset the associated diagnostics.
3989 delete static_cast<CXDiagnosticSetImpl*>(TU->Diagnostics);
Craig Topper69186e72014-06-08 08:38:04 +00003990 TU->Diagnostics = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00003991
Dmitri Gribenko183436e2013-01-26 21:49:50 +00003992 CIndexer *CXXIdx = TU->CIdx;
Guy Benyei11169dd2012-12-18 14:30:41 +00003993 if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForEditing))
3994 setThreadBackgroundPriority();
3995
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003996 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00003997 ASTUnit::ConcurrencyCheck Check(*CXXUnit);
Ahmed Charlesb8984322014-03-07 20:03:18 +00003998
3999 std::unique_ptr<std::vector<ASTUnit::RemappedFile>> RemappedFiles(
4000 new std::vector<ASTUnit::RemappedFile>());
4001
Guy Benyei11169dd2012-12-18 14:30:41 +00004002 // Recover resources if we crash before exiting this function.
4003 llvm::CrashRecoveryContextCleanupRegistrar<
4004 std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
Alp Toker9d85b182014-07-07 01:23:14 +00004005
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00004006 for (auto &UF : unsaved_files) {
Rafael Espindolad87f8d72014-08-27 20:03:29 +00004007 std::unique_ptr<llvm::MemoryBuffer> MB =
Alp Toker9d85b182014-07-07 01:23:14 +00004008 llvm::MemoryBuffer::getMemBufferCopy(getContents(UF), UF.Filename);
Rafael Espindolad87f8d72014-08-27 20:03:29 +00004009 RemappedFiles->push_back(std::make_pair(UF.Filename, MB.release()));
Guy Benyei11169dd2012-12-18 14:30:41 +00004010 }
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00004011
Adrian Prantlbb165fb2015-06-20 18:53:08 +00004012 if (!CXXUnit->Reparse(CXXIdx->getPCHContainerOperations(),
4013 *RemappedFiles.get()))
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00004014 return CXError_Success;
4015 if (isASTReadError(CXXUnit))
4016 return CXError_ASTReadError;
4017 return CXError_Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00004018}
4019
4020int clang_reparseTranslationUnit(CXTranslationUnit TU,
4021 unsigned num_unsaved_files,
4022 struct CXUnsavedFile *unsaved_files,
4023 unsigned options) {
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00004024 LOG_FUNC_SECTION {
4025 *Log << TU;
4026 }
4027
Alp Toker9d85b182014-07-07 01:23:14 +00004028 if (num_unsaved_files && !unsaved_files)
4029 return CXError_InvalidArguments;
4030
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00004031 CXErrorCode result;
4032 auto ReparseTranslationUnitImpl = [=, &result]() {
4033 result = clang_reparseTranslationUnit_Impl(
4034 TU, llvm::makeArrayRef(unsaved_files, num_unsaved_files), options);
4035 };
Guy Benyei11169dd2012-12-18 14:30:41 +00004036
4037 if (getenv("LIBCLANG_NOTHREADS")) {
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00004038 ReparseTranslationUnitImpl();
Alp Toker5c532982014-07-07 22:42:03 +00004039 return result;
Guy Benyei11169dd2012-12-18 14:30:41 +00004040 }
4041
4042 llvm::CrashRecoveryContext CRC;
4043
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00004044 if (!RunSafely(CRC, ReparseTranslationUnitImpl)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004045 fprintf(stderr, "libclang: crash detected during reparsing\n");
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00004046 cxtu::getASTUnit(TU)->setUnsafeToFree(true);
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00004047 return CXError_Crashed;
Guy Benyei11169dd2012-12-18 14:30:41 +00004048 } else if (getenv("LIBCLANG_RESOURCE_USAGE"))
4049 PrintLibclangResourceUsage(TU);
4050
Alp Toker5c532982014-07-07 22:42:03 +00004051 return result;
Guy Benyei11169dd2012-12-18 14:30:41 +00004052}
4053
4054
4055CXString clang_getTranslationUnitSpelling(CXTranslationUnit CTUnit) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00004056 if (isNotUsableTU(CTUnit)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00004057 LOG_BAD_TU(CTUnit);
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004058 return cxstring::createEmpty();
Dmitri Gribenko256454f2014-02-11 14:34:14 +00004059 }
Guy Benyei11169dd2012-12-18 14:30:41 +00004060
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00004061 ASTUnit *CXXUnit = cxtu::getASTUnit(CTUnit);
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004062 return cxstring::createDup(CXXUnit->getOriginalSourceFileName());
Guy Benyei11169dd2012-12-18 14:30:41 +00004063}
4064
4065CXCursor clang_getTranslationUnitCursor(CXTranslationUnit TU) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00004066 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00004067 LOG_BAD_TU(TU);
Argyrios Kyrtzidis0e95fca2013-04-04 22:40:59 +00004068 return clang_getNullCursor();
Dmitri Gribenko256454f2014-02-11 14:34:14 +00004069 }
Argyrios Kyrtzidis0e95fca2013-04-04 22:40:59 +00004070
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00004071 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00004072 return MakeCXCursor(CXXUnit->getASTContext().getTranslationUnitDecl(), TU);
4073}
4074
Emilio Cobos Alvarez485ad422017-04-28 15:56:39 +00004075CXTargetInfo clang_getTranslationUnitTargetInfo(CXTranslationUnit CTUnit) {
4076 if (isNotUsableTU(CTUnit)) {
4077 LOG_BAD_TU(CTUnit);
4078 return nullptr;
4079 }
4080
4081 CXTargetInfoImpl* impl = new CXTargetInfoImpl();
4082 impl->TranslationUnit = CTUnit;
4083 return impl;
4084}
4085
4086CXString clang_TargetInfo_getTriple(CXTargetInfo TargetInfo) {
4087 if (!TargetInfo)
4088 return cxstring::createEmpty();
4089
4090 CXTranslationUnit CTUnit = TargetInfo->TranslationUnit;
4091 assert(!isNotUsableTU(CTUnit) &&
4092 "Unexpected unusable translation unit in TargetInfo");
4093
4094 ASTUnit *CXXUnit = cxtu::getASTUnit(CTUnit);
4095 std::string Triple =
4096 CXXUnit->getASTContext().getTargetInfo().getTriple().normalize();
4097 return cxstring::createDup(Triple);
4098}
4099
4100int clang_TargetInfo_getPointerWidth(CXTargetInfo TargetInfo) {
4101 if (!TargetInfo)
4102 return -1;
4103
4104 CXTranslationUnit CTUnit = TargetInfo->TranslationUnit;
4105 assert(!isNotUsableTU(CTUnit) &&
4106 "Unexpected unusable translation unit in TargetInfo");
4107
4108 ASTUnit *CXXUnit = cxtu::getASTUnit(CTUnit);
4109 return CXXUnit->getASTContext().getTargetInfo().getMaxPointerWidth();
4110}
4111
4112void clang_TargetInfo_dispose(CXTargetInfo TargetInfo) {
4113 if (!TargetInfo)
4114 return;
4115
4116 delete TargetInfo;
4117}
4118
Guy Benyei11169dd2012-12-18 14:30:41 +00004119//===----------------------------------------------------------------------===//
4120// CXFile Operations.
4121//===----------------------------------------------------------------------===//
4122
Guy Benyei11169dd2012-12-18 14:30:41 +00004123CXString clang_getFileName(CXFile SFile) {
4124 if (!SFile)
Dmitri Gribenkof98dfba2013-02-01 14:13:32 +00004125 return cxstring::createNull();
Guy Benyei11169dd2012-12-18 14:30:41 +00004126
4127 FileEntry *FEnt = static_cast<FileEntry *>(SFile);
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004128 return cxstring::createRef(FEnt->getName());
Guy Benyei11169dd2012-12-18 14:30:41 +00004129}
4130
4131time_t clang_getFileTime(CXFile SFile) {
4132 if (!SFile)
4133 return 0;
4134
4135 FileEntry *FEnt = static_cast<FileEntry *>(SFile);
4136 return FEnt->getModificationTime();
4137}
4138
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00004139CXFile clang_getFile(CXTranslationUnit TU, const char *file_name) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00004140 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00004141 LOG_BAD_TU(TU);
Craig Topper69186e72014-06-08 08:38:04 +00004142 return nullptr;
Dmitri Gribenko256454f2014-02-11 14:34:14 +00004143 }
Guy Benyei11169dd2012-12-18 14:30:41 +00004144
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00004145 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00004146
4147 FileManager &FMgr = CXXUnit->getFileManager();
4148 return const_cast<FileEntry *>(FMgr.getFile(file_name));
4149}
4150
Dmitri Gribenko256454f2014-02-11 14:34:14 +00004151unsigned clang_isFileMultipleIncludeGuarded(CXTranslationUnit TU,
4152 CXFile file) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00004153 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00004154 LOG_BAD_TU(TU);
4155 return 0;
4156 }
4157
4158 if (!file)
Guy Benyei11169dd2012-12-18 14:30:41 +00004159 return 0;
4160
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00004161 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00004162 FileEntry *FEnt = static_cast<FileEntry *>(file);
4163 return CXXUnit->getPreprocessor().getHeaderSearchInfo()
4164 .isFileMultipleIncludeGuarded(FEnt);
4165}
4166
Argyrios Kyrtzidisac08b262013-01-26 04:52:52 +00004167int clang_getFileUniqueID(CXFile file, CXFileUniqueID *outID) {
4168 if (!file || !outID)
4169 return 1;
4170
Argyrios Kyrtzidisac08b262013-01-26 04:52:52 +00004171 FileEntry *FEnt = static_cast<FileEntry *>(file);
Rafael Espindolaf8f91b82013-08-01 21:42:11 +00004172 const llvm::sys::fs::UniqueID &ID = FEnt->getUniqueID();
4173 outID->data[0] = ID.getDevice();
4174 outID->data[1] = ID.getFile();
Argyrios Kyrtzidisac08b262013-01-26 04:52:52 +00004175 outID->data[2] = FEnt->getModificationTime();
4176 return 0;
Argyrios Kyrtzidisac08b262013-01-26 04:52:52 +00004177}
4178
Argyrios Kyrtzidisac3997e2014-08-16 00:26:19 +00004179int clang_File_isEqual(CXFile file1, CXFile file2) {
4180 if (file1 == file2)
4181 return true;
4182
4183 if (!file1 || !file2)
4184 return false;
4185
4186 FileEntry *FEnt1 = static_cast<FileEntry *>(file1);
4187 FileEntry *FEnt2 = static_cast<FileEntry *>(file2);
4188 return FEnt1->getUniqueID() == FEnt2->getUniqueID();
4189}
4190
Guy Benyei11169dd2012-12-18 14:30:41 +00004191//===----------------------------------------------------------------------===//
4192// CXCursor Operations.
4193//===----------------------------------------------------------------------===//
4194
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004195static const Decl *getDeclFromExpr(const Stmt *E) {
4196 if (const ImplicitCastExpr *CE = dyn_cast<ImplicitCastExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00004197 return getDeclFromExpr(CE->getSubExpr());
4198
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004199 if (const DeclRefExpr *RefExpr = dyn_cast<DeclRefExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00004200 return RefExpr->getDecl();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004201 if (const MemberExpr *ME = dyn_cast<MemberExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00004202 return ME->getMemberDecl();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004203 if (const ObjCIvarRefExpr *RE = dyn_cast<ObjCIvarRefExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00004204 return RE->getDecl();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004205 if (const ObjCPropertyRefExpr *PRE = dyn_cast<ObjCPropertyRefExpr>(E)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004206 if (PRE->isExplicitProperty())
4207 return PRE->getExplicitProperty();
4208 // It could be messaging both getter and setter as in:
4209 // ++myobj.myprop;
4210 // in which case prefer to associate the setter since it is less obvious
4211 // from inspecting the source that the setter is going to get called.
4212 if (PRE->isMessagingSetter())
4213 return PRE->getImplicitPropertySetter();
4214 return PRE->getImplicitPropertyGetter();
4215 }
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004216 if (const PseudoObjectExpr *POE = dyn_cast<PseudoObjectExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00004217 return getDeclFromExpr(POE->getSyntacticForm());
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004218 if (const OpaqueValueExpr *OVE = dyn_cast<OpaqueValueExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00004219 if (Expr *Src = OVE->getSourceExpr())
4220 return getDeclFromExpr(Src);
4221
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004222 if (const CallExpr *CE = dyn_cast<CallExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00004223 return getDeclFromExpr(CE->getCallee());
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004224 if (const CXXConstructExpr *CE = dyn_cast<CXXConstructExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00004225 if (!CE->isElidable())
4226 return CE->getConstructor();
Richard Smith5179eb72016-06-28 19:03:57 +00004227 if (const CXXInheritedCtorInitExpr *CE =
4228 dyn_cast<CXXInheritedCtorInitExpr>(E))
4229 return CE->getConstructor();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004230 if (const ObjCMessageExpr *OME = dyn_cast<ObjCMessageExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00004231 return OME->getMethodDecl();
4232
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004233 if (const ObjCProtocolExpr *PE = dyn_cast<ObjCProtocolExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00004234 return PE->getProtocol();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004235 if (const SubstNonTypeTemplateParmPackExpr *NTTP
Guy Benyei11169dd2012-12-18 14:30:41 +00004236 = dyn_cast<SubstNonTypeTemplateParmPackExpr>(E))
4237 return NTTP->getParameterPack();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004238 if (const SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00004239 if (isa<NonTypeTemplateParmDecl>(SizeOfPack->getPack()) ||
4240 isa<ParmVarDecl>(SizeOfPack->getPack()))
4241 return SizeOfPack->getPack();
Craig Topper69186e72014-06-08 08:38:04 +00004242
4243 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00004244}
4245
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00004246static SourceLocation getLocationFromExpr(const Expr *E) {
4247 if (const ImplicitCastExpr *CE = dyn_cast<ImplicitCastExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00004248 return getLocationFromExpr(CE->getSubExpr());
4249
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00004250 if (const ObjCMessageExpr *Msg = dyn_cast<ObjCMessageExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00004251 return /*FIXME:*/Msg->getLeftLoc();
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00004252 if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00004253 return DRE->getLocation();
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00004254 if (const MemberExpr *Member = dyn_cast<MemberExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00004255 return Member->getMemberLoc();
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00004256 if (const ObjCIvarRefExpr *Ivar = dyn_cast<ObjCIvarRefExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00004257 return Ivar->getLocation();
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00004258 if (const SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00004259 return SizeOfPack->getPackLoc();
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00004260 if (const ObjCPropertyRefExpr *PropRef = dyn_cast<ObjCPropertyRefExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00004261 return PropRef->getLocation();
4262
4263 return E->getLocStart();
4264}
4265
NAKAMURA Takumia01f4c32016-12-19 16:50:43 +00004266extern "C" {
4267
Guy Benyei11169dd2012-12-18 14:30:41 +00004268unsigned clang_visitChildren(CXCursor parent,
4269 CXCursorVisitor visitor,
4270 CXClientData client_data) {
4271 CursorVisitor CursorVis(getCursorTU(parent), visitor, client_data,
4272 /*VisitPreprocessorLast=*/false);
4273 return CursorVis.VisitChildren(parent);
4274}
4275
4276#ifndef __has_feature
4277#define __has_feature(x) 0
4278#endif
4279#if __has_feature(blocks)
4280typedef enum CXChildVisitResult
4281 (^CXCursorVisitorBlock)(CXCursor cursor, CXCursor parent);
4282
4283static enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
4284 CXClientData client_data) {
4285 CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
4286 return block(cursor, parent);
4287}
4288#else
4289// If we are compiled with a compiler that doesn't have native blocks support,
4290// define and call the block manually, so the
4291typedef struct _CXChildVisitResult
4292{
4293 void *isa;
4294 int flags;
4295 int reserved;
4296 enum CXChildVisitResult(*invoke)(struct _CXChildVisitResult*, CXCursor,
4297 CXCursor);
4298} *CXCursorVisitorBlock;
4299
4300static enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
4301 CXClientData client_data) {
4302 CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
4303 return block->invoke(block, cursor, parent);
4304}
4305#endif
4306
4307
4308unsigned clang_visitChildrenWithBlock(CXCursor parent,
4309 CXCursorVisitorBlock block) {
4310 return clang_visitChildren(parent, visitWithBlock, block);
4311}
4312
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004313static CXString getDeclSpelling(const Decl *D) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004314 if (!D)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004315 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004316
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004317 const NamedDecl *ND = dyn_cast<NamedDecl>(D);
Guy Benyei11169dd2012-12-18 14:30:41 +00004318 if (!ND) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004319 if (const ObjCPropertyImplDecl *PropImpl =
4320 dyn_cast<ObjCPropertyImplDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00004321 if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004322 return cxstring::createDup(Property->getIdentifier()->getName());
Guy Benyei11169dd2012-12-18 14:30:41 +00004323
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004324 if (const ImportDecl *ImportD = dyn_cast<ImportDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00004325 if (Module *Mod = ImportD->getImportedModule())
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004326 return cxstring::createDup(Mod->getFullModuleName());
Guy Benyei11169dd2012-12-18 14:30:41 +00004327
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004328 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004329 }
4330
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004331 if (const ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004332 return cxstring::createDup(OMD->getSelector().getAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004333
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004334 if (const ObjCCategoryImplDecl *CIMP = dyn_cast<ObjCCategoryImplDecl>(ND))
Guy Benyei11169dd2012-12-18 14:30:41 +00004335 // No, this isn't the same as the code below. getIdentifier() is non-virtual
4336 // and returns different names. NamedDecl returns the class name and
4337 // ObjCCategoryImplDecl returns the category name.
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004338 return cxstring::createRef(CIMP->getIdentifier()->getNameStart());
Guy Benyei11169dd2012-12-18 14:30:41 +00004339
4340 if (isa<UsingDirectiveDecl>(D))
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004341 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004342
4343 SmallString<1024> S;
4344 llvm::raw_svector_ostream os(S);
4345 ND->printName(os);
4346
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004347 return cxstring::createDup(os.str());
Guy Benyei11169dd2012-12-18 14:30:41 +00004348}
4349
4350CXString clang_getCursorSpelling(CXCursor C) {
4351 if (clang_isTranslationUnit(C.kind))
Dmitri Gribenko2c173b42013-01-11 19:28:44 +00004352 return clang_getTranslationUnitSpelling(getCursorTU(C));
Guy Benyei11169dd2012-12-18 14:30:41 +00004353
4354 if (clang_isReference(C.kind)) {
4355 switch (C.kind) {
4356 case CXCursor_ObjCSuperClassRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004357 const ObjCInterfaceDecl *Super = getCursorObjCSuperClassRef(C).first;
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004358 return cxstring::createRef(Super->getIdentifier()->getNameStart());
Guy Benyei11169dd2012-12-18 14:30:41 +00004359 }
4360 case CXCursor_ObjCClassRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004361 const ObjCInterfaceDecl *Class = getCursorObjCClassRef(C).first;
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004362 return cxstring::createRef(Class->getIdentifier()->getNameStart());
Guy Benyei11169dd2012-12-18 14:30:41 +00004363 }
4364 case CXCursor_ObjCProtocolRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004365 const ObjCProtocolDecl *OID = getCursorObjCProtocolRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004366 assert(OID && "getCursorSpelling(): Missing protocol decl");
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004367 return cxstring::createRef(OID->getIdentifier()->getNameStart());
Guy Benyei11169dd2012-12-18 14:30:41 +00004368 }
4369 case CXCursor_CXXBaseSpecifier: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004370 const CXXBaseSpecifier *B = getCursorCXXBaseSpecifier(C);
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004371 return cxstring::createDup(B->getType().getAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004372 }
4373 case CXCursor_TypeRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004374 const TypeDecl *Type = getCursorTypeRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004375 assert(Type && "Missing type decl");
4376
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004377 return cxstring::createDup(getCursorContext(C).getTypeDeclType(Type).
Guy Benyei11169dd2012-12-18 14:30:41 +00004378 getAsString());
4379 }
4380 case CXCursor_TemplateRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004381 const TemplateDecl *Template = getCursorTemplateRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004382 assert(Template && "Missing template decl");
4383
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004384 return cxstring::createDup(Template->getNameAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004385 }
4386
4387 case CXCursor_NamespaceRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004388 const NamedDecl *NS = getCursorNamespaceRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004389 assert(NS && "Missing namespace decl");
4390
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004391 return cxstring::createDup(NS->getNameAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004392 }
4393
4394 case CXCursor_MemberRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004395 const FieldDecl *Field = getCursorMemberRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004396 assert(Field && "Missing member decl");
4397
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004398 return cxstring::createDup(Field->getNameAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004399 }
4400
4401 case CXCursor_LabelRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004402 const LabelStmt *Label = getCursorLabelRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004403 assert(Label && "Missing label");
4404
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004405 return cxstring::createRef(Label->getName());
Guy Benyei11169dd2012-12-18 14:30:41 +00004406 }
4407
4408 case CXCursor_OverloadedDeclRef: {
4409 OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004410 if (const Decl *D = Storage.dyn_cast<const Decl *>()) {
4411 if (const NamedDecl *ND = dyn_cast<NamedDecl>(D))
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004412 return cxstring::createDup(ND->getNameAsString());
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004413 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004414 }
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004415 if (const OverloadExpr *E = Storage.dyn_cast<const OverloadExpr *>())
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004416 return cxstring::createDup(E->getName().getAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004417 OverloadedTemplateStorage *Ovl
4418 = Storage.get<OverloadedTemplateStorage*>();
4419 if (Ovl->size() == 0)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004420 return cxstring::createEmpty();
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004421 return cxstring::createDup((*Ovl->begin())->getNameAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004422 }
4423
4424 case CXCursor_VariableRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004425 const VarDecl *Var = getCursorVariableRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004426 assert(Var && "Missing variable decl");
4427
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004428 return cxstring::createDup(Var->getNameAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004429 }
4430
4431 default:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004432 return cxstring::createRef("<not implemented>");
Guy Benyei11169dd2012-12-18 14:30:41 +00004433 }
4434 }
4435
4436 if (clang_isExpression(C.kind)) {
Argyrios Kyrtzidis3227d862014-03-03 19:40:52 +00004437 const Expr *E = getCursorExpr(C);
4438
4439 if (C.kind == CXCursor_ObjCStringLiteral ||
4440 C.kind == CXCursor_StringLiteral) {
4441 const StringLiteral *SLit;
4442 if (const ObjCStringLiteral *OSL = dyn_cast<ObjCStringLiteral>(E)) {
4443 SLit = OSL->getString();
4444 } else {
4445 SLit = cast<StringLiteral>(E);
4446 }
4447 SmallString<256> Buf;
4448 llvm::raw_svector_ostream OS(Buf);
4449 SLit->outputString(OS);
4450 return cxstring::createDup(OS.str());
4451 }
4452
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004453 const Decl *D = getDeclFromExpr(getCursorExpr(C));
Guy Benyei11169dd2012-12-18 14:30:41 +00004454 if (D)
4455 return getDeclSpelling(D);
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004456 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004457 }
4458
4459 if (clang_isStatement(C.kind)) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00004460 const Stmt *S = getCursorStmt(C);
4461 if (const LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S))
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004462 return cxstring::createRef(Label->getName());
Guy Benyei11169dd2012-12-18 14:30:41 +00004463
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004464 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004465 }
4466
4467 if (C.kind == CXCursor_MacroExpansion)
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004468 return cxstring::createRef(getCursorMacroExpansion(C).getName()
Guy Benyei11169dd2012-12-18 14:30:41 +00004469 ->getNameStart());
4470
4471 if (C.kind == CXCursor_MacroDefinition)
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004472 return cxstring::createRef(getCursorMacroDefinition(C)->getName()
Guy Benyei11169dd2012-12-18 14:30:41 +00004473 ->getNameStart());
4474
4475 if (C.kind == CXCursor_InclusionDirective)
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004476 return cxstring::createDup(getCursorInclusionDirective(C)->getFileName());
Guy Benyei11169dd2012-12-18 14:30:41 +00004477
4478 if (clang_isDeclaration(C.kind))
4479 return getDeclSpelling(getCursorDecl(C));
4480
4481 if (C.kind == CXCursor_AnnotateAttr) {
Dmitri Gribenkoe4baea62013-01-26 18:08:08 +00004482 const AnnotateAttr *AA = cast<AnnotateAttr>(cxcursor::getCursorAttr(C));
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004483 return cxstring::createDup(AA->getAnnotation());
Guy Benyei11169dd2012-12-18 14:30:41 +00004484 }
4485
4486 if (C.kind == CXCursor_AsmLabelAttr) {
Dmitri Gribenkoe4baea62013-01-26 18:08:08 +00004487 const AsmLabelAttr *AA = cast<AsmLabelAttr>(cxcursor::getCursorAttr(C));
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004488 return cxstring::createDup(AA->getLabel());
Guy Benyei11169dd2012-12-18 14:30:41 +00004489 }
4490
Argyrios Kyrtzidis16834f12013-09-25 00:14:38 +00004491 if (C.kind == CXCursor_PackedAttr) {
4492 return cxstring::createRef("packed");
4493 }
4494
Saleem Abdulrasool79c69712015-09-05 18:53:43 +00004495 if (C.kind == CXCursor_VisibilityAttr) {
4496 const VisibilityAttr *AA = cast<VisibilityAttr>(cxcursor::getCursorAttr(C));
4497 switch (AA->getVisibility()) {
4498 case VisibilityAttr::VisibilityType::Default:
4499 return cxstring::createRef("default");
4500 case VisibilityAttr::VisibilityType::Hidden:
4501 return cxstring::createRef("hidden");
4502 case VisibilityAttr::VisibilityType::Protected:
4503 return cxstring::createRef("protected");
4504 }
4505 llvm_unreachable("unknown visibility type");
4506 }
4507
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004508 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004509}
4510
4511CXSourceRange clang_Cursor_getSpellingNameRange(CXCursor C,
4512 unsigned pieceIndex,
4513 unsigned options) {
4514 if (clang_Cursor_isNull(C))
4515 return clang_getNullRange();
4516
4517 ASTContext &Ctx = getCursorContext(C);
4518
4519 if (clang_isStatement(C.kind)) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00004520 const Stmt *S = getCursorStmt(C);
4521 if (const LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004522 if (pieceIndex > 0)
4523 return clang_getNullRange();
4524 return cxloc::translateSourceRange(Ctx, Label->getIdentLoc());
4525 }
4526
4527 return clang_getNullRange();
4528 }
4529
4530 if (C.kind == CXCursor_ObjCMessageExpr) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00004531 if (const ObjCMessageExpr *
Guy Benyei11169dd2012-12-18 14:30:41 +00004532 ME = dyn_cast_or_null<ObjCMessageExpr>(getCursorExpr(C))) {
4533 if (pieceIndex >= ME->getNumSelectorLocs())
4534 return clang_getNullRange();
4535 return cxloc::translateSourceRange(Ctx, ME->getSelectorLoc(pieceIndex));
4536 }
4537 }
4538
4539 if (C.kind == CXCursor_ObjCInstanceMethodDecl ||
4540 C.kind == CXCursor_ObjCClassMethodDecl) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004541 if (const ObjCMethodDecl *
Guy Benyei11169dd2012-12-18 14:30:41 +00004542 MD = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(C))) {
4543 if (pieceIndex >= MD->getNumSelectorLocs())
4544 return clang_getNullRange();
4545 return cxloc::translateSourceRange(Ctx, MD->getSelectorLoc(pieceIndex));
4546 }
4547 }
4548
4549 if (C.kind == CXCursor_ObjCCategoryDecl ||
4550 C.kind == CXCursor_ObjCCategoryImplDecl) {
4551 if (pieceIndex > 0)
4552 return clang_getNullRange();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004553 if (const ObjCCategoryDecl *
Guy Benyei11169dd2012-12-18 14:30:41 +00004554 CD = dyn_cast_or_null<ObjCCategoryDecl>(getCursorDecl(C)))
4555 return cxloc::translateSourceRange(Ctx, CD->getCategoryNameLoc());
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004556 if (const ObjCCategoryImplDecl *
Guy Benyei11169dd2012-12-18 14:30:41 +00004557 CID = dyn_cast_or_null<ObjCCategoryImplDecl>(getCursorDecl(C)))
4558 return cxloc::translateSourceRange(Ctx, CID->getCategoryNameLoc());
4559 }
4560
4561 if (C.kind == CXCursor_ModuleImportDecl) {
4562 if (pieceIndex > 0)
4563 return clang_getNullRange();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004564 if (const ImportDecl *ImportD =
4565 dyn_cast_or_null<ImportDecl>(getCursorDecl(C))) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004566 ArrayRef<SourceLocation> Locs = ImportD->getIdentifierLocs();
4567 if (!Locs.empty())
4568 return cxloc::translateSourceRange(Ctx,
4569 SourceRange(Locs.front(), Locs.back()));
4570 }
4571 return clang_getNullRange();
4572 }
4573
Argyrios Kyrtzidisa2a1e532014-08-26 20:23:26 +00004574 if (C.kind == CXCursor_CXXMethod || C.kind == CXCursor_Destructor ||
Kevin Funk4be5d672016-12-20 09:56:56 +00004575 C.kind == CXCursor_ConversionFunction ||
4576 C.kind == CXCursor_FunctionDecl) {
Argyrios Kyrtzidisa2a1e532014-08-26 20:23:26 +00004577 if (pieceIndex > 0)
4578 return clang_getNullRange();
4579 if (const FunctionDecl *FD =
4580 dyn_cast_or_null<FunctionDecl>(getCursorDecl(C))) {
4581 DeclarationNameInfo FunctionName = FD->getNameInfo();
4582 return cxloc::translateSourceRange(Ctx, FunctionName.getSourceRange());
4583 }
4584 return clang_getNullRange();
4585 }
4586
Guy Benyei11169dd2012-12-18 14:30:41 +00004587 // FIXME: A CXCursor_InclusionDirective should give the location of the
4588 // filename, but we don't keep track of this.
4589
4590 // FIXME: A CXCursor_AnnotateAttr should give the location of the annotation
4591 // but we don't keep track of this.
4592
4593 // FIXME: A CXCursor_AsmLabelAttr should give the location of the label
4594 // but we don't keep track of this.
4595
4596 // Default handling, give the location of the cursor.
4597
4598 if (pieceIndex > 0)
4599 return clang_getNullRange();
4600
4601 CXSourceLocation CXLoc = clang_getCursorLocation(C);
4602 SourceLocation Loc = cxloc::translateSourceLocation(CXLoc);
4603 return cxloc::translateSourceRange(Ctx, Loc);
4604}
4605
Eli Bendersky44a206f2014-07-31 18:04:56 +00004606CXString clang_Cursor_getMangling(CXCursor C) {
4607 if (clang_isInvalid(C.kind) || !clang_isDeclaration(C.kind))
4608 return cxstring::createEmpty();
4609
Eli Bendersky44a206f2014-07-31 18:04:56 +00004610 // Mangling only works for functions and variables.
Eli Bendersky79759592014-08-01 15:01:10 +00004611 const Decl *D = getCursorDecl(C);
Eli Bendersky44a206f2014-07-31 18:04:56 +00004612 if (!D || !(isa<FunctionDecl>(D) || isa<VarDecl>(D)))
4613 return cxstring::createEmpty();
4614
Argyrios Kyrtzidisca741ce2016-02-14 22:30:14 +00004615 ASTContext &Ctx = D->getASTContext();
4616 index::CodegenNameGenerator CGNameGen(Ctx);
4617 return cxstring::createDup(CGNameGen.getName(D));
Eli Bendersky44a206f2014-07-31 18:04:56 +00004618}
4619
Saleem Abdulrasool60034432015-11-12 03:57:22 +00004620CXStringSet *clang_Cursor_getCXXManglings(CXCursor C) {
4621 if (clang_isInvalid(C.kind) || !clang_isDeclaration(C.kind))
4622 return nullptr;
4623
4624 const Decl *D = getCursorDecl(C);
4625 if (!(isa<CXXRecordDecl>(D) || isa<CXXMethodDecl>(D)))
4626 return nullptr;
4627
Argyrios Kyrtzidisca741ce2016-02-14 22:30:14 +00004628 ASTContext &Ctx = D->getASTContext();
4629 index::CodegenNameGenerator CGNameGen(Ctx);
4630 std::vector<std::string> Manglings = CGNameGen.getAllManglings(D);
Saleem Abdulrasool60034432015-11-12 03:57:22 +00004631 return cxstring::createSet(Manglings);
4632}
4633
Guy Benyei11169dd2012-12-18 14:30:41 +00004634CXString clang_getCursorDisplayName(CXCursor C) {
4635 if (!clang_isDeclaration(C.kind))
4636 return clang_getCursorSpelling(C);
4637
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004638 const Decl *D = getCursorDecl(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00004639 if (!D)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004640 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004641
4642 PrintingPolicy Policy = getCursorContext(C).getPrintingPolicy();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004643 if (const FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00004644 D = FunTmpl->getTemplatedDecl();
4645
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004646 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004647 SmallString<64> Str;
4648 llvm::raw_svector_ostream OS(Str);
4649 OS << *Function;
4650 if (Function->getPrimaryTemplate())
4651 OS << "<>";
4652 OS << "(";
4653 for (unsigned I = 0, N = Function->getNumParams(); I != N; ++I) {
4654 if (I)
4655 OS << ", ";
4656 OS << Function->getParamDecl(I)->getType().getAsString(Policy);
4657 }
4658
4659 if (Function->isVariadic()) {
4660 if (Function->getNumParams())
4661 OS << ", ";
4662 OS << "...";
4663 }
4664 OS << ")";
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004665 return cxstring::createDup(OS.str());
Guy Benyei11169dd2012-12-18 14:30:41 +00004666 }
4667
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004668 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004669 SmallString<64> Str;
4670 llvm::raw_svector_ostream OS(Str);
4671 OS << *ClassTemplate;
4672 OS << "<";
4673 TemplateParameterList *Params = ClassTemplate->getTemplateParameters();
4674 for (unsigned I = 0, N = Params->size(); I != N; ++I) {
4675 if (I)
4676 OS << ", ";
4677
4678 NamedDecl *Param = Params->getParam(I);
4679 if (Param->getIdentifier()) {
4680 OS << Param->getIdentifier()->getName();
4681 continue;
4682 }
4683
4684 // There is no parameter name, which makes this tricky. Try to come up
4685 // with something useful that isn't too long.
4686 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
4687 OS << (TTP->wasDeclaredWithTypename()? "typename" : "class");
4688 else if (NonTypeTemplateParmDecl *NTTP
4689 = dyn_cast<NonTypeTemplateParmDecl>(Param))
4690 OS << NTTP->getType().getAsString(Policy);
4691 else
4692 OS << "template<...> class";
4693 }
4694
4695 OS << ">";
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004696 return cxstring::createDup(OS.str());
Guy Benyei11169dd2012-12-18 14:30:41 +00004697 }
4698
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004699 if (const ClassTemplateSpecializationDecl *ClassSpec
Guy Benyei11169dd2012-12-18 14:30:41 +00004700 = dyn_cast<ClassTemplateSpecializationDecl>(D)) {
4701 // If the type was explicitly written, use that.
4702 if (TypeSourceInfo *TSInfo = ClassSpec->getTypeAsWritten())
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004703 return cxstring::createDup(TSInfo->getType().getAsString(Policy));
Guy Benyei11169dd2012-12-18 14:30:41 +00004704
Benjamin Kramer9170e912013-02-22 15:46:01 +00004705 SmallString<128> Str;
Guy Benyei11169dd2012-12-18 14:30:41 +00004706 llvm::raw_svector_ostream OS(Str);
4707 OS << *ClassSpec;
David Majnemer6fbeee32016-07-07 04:43:07 +00004708 TemplateSpecializationType::PrintTemplateArgumentList(
4709 OS, ClassSpec->getTemplateArgs().asArray(), Policy);
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004710 return cxstring::createDup(OS.str());
Guy Benyei11169dd2012-12-18 14:30:41 +00004711 }
4712
4713 return clang_getCursorSpelling(C);
4714}
4715
4716CXString clang_getCursorKindSpelling(enum CXCursorKind Kind) {
4717 switch (Kind) {
4718 case CXCursor_FunctionDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004719 return cxstring::createRef("FunctionDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004720 case CXCursor_TypedefDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004721 return cxstring::createRef("TypedefDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004722 case CXCursor_EnumDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004723 return cxstring::createRef("EnumDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004724 case CXCursor_EnumConstantDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004725 return cxstring::createRef("EnumConstantDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004726 case CXCursor_StructDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004727 return cxstring::createRef("StructDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004728 case CXCursor_UnionDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004729 return cxstring::createRef("UnionDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004730 case CXCursor_ClassDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004731 return cxstring::createRef("ClassDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004732 case CXCursor_FieldDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004733 return cxstring::createRef("FieldDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004734 case CXCursor_VarDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004735 return cxstring::createRef("VarDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004736 case CXCursor_ParmDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004737 return cxstring::createRef("ParmDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004738 case CXCursor_ObjCInterfaceDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004739 return cxstring::createRef("ObjCInterfaceDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004740 case CXCursor_ObjCCategoryDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004741 return cxstring::createRef("ObjCCategoryDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004742 case CXCursor_ObjCProtocolDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004743 return cxstring::createRef("ObjCProtocolDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004744 case CXCursor_ObjCPropertyDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004745 return cxstring::createRef("ObjCPropertyDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004746 case CXCursor_ObjCIvarDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004747 return cxstring::createRef("ObjCIvarDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004748 case CXCursor_ObjCInstanceMethodDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004749 return cxstring::createRef("ObjCInstanceMethodDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004750 case CXCursor_ObjCClassMethodDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004751 return cxstring::createRef("ObjCClassMethodDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004752 case CXCursor_ObjCImplementationDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004753 return cxstring::createRef("ObjCImplementationDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004754 case CXCursor_ObjCCategoryImplDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004755 return cxstring::createRef("ObjCCategoryImplDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004756 case CXCursor_CXXMethod:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004757 return cxstring::createRef("CXXMethod");
Guy Benyei11169dd2012-12-18 14:30:41 +00004758 case CXCursor_UnexposedDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004759 return cxstring::createRef("UnexposedDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004760 case CXCursor_ObjCSuperClassRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004761 return cxstring::createRef("ObjCSuperClassRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004762 case CXCursor_ObjCProtocolRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004763 return cxstring::createRef("ObjCProtocolRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004764 case CXCursor_ObjCClassRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004765 return cxstring::createRef("ObjCClassRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004766 case CXCursor_TypeRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004767 return cxstring::createRef("TypeRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004768 case CXCursor_TemplateRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004769 return cxstring::createRef("TemplateRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004770 case CXCursor_NamespaceRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004771 return cxstring::createRef("NamespaceRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004772 case CXCursor_MemberRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004773 return cxstring::createRef("MemberRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004774 case CXCursor_LabelRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004775 return cxstring::createRef("LabelRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004776 case CXCursor_OverloadedDeclRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004777 return cxstring::createRef("OverloadedDeclRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004778 case CXCursor_VariableRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004779 return cxstring::createRef("VariableRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004780 case CXCursor_IntegerLiteral:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004781 return cxstring::createRef("IntegerLiteral");
Guy Benyei11169dd2012-12-18 14:30:41 +00004782 case CXCursor_FloatingLiteral:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004783 return cxstring::createRef("FloatingLiteral");
Guy Benyei11169dd2012-12-18 14:30:41 +00004784 case CXCursor_ImaginaryLiteral:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004785 return cxstring::createRef("ImaginaryLiteral");
Guy Benyei11169dd2012-12-18 14:30:41 +00004786 case CXCursor_StringLiteral:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004787 return cxstring::createRef("StringLiteral");
Guy Benyei11169dd2012-12-18 14:30:41 +00004788 case CXCursor_CharacterLiteral:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004789 return cxstring::createRef("CharacterLiteral");
Guy Benyei11169dd2012-12-18 14:30:41 +00004790 case CXCursor_ParenExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004791 return cxstring::createRef("ParenExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004792 case CXCursor_UnaryOperator:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004793 return cxstring::createRef("UnaryOperator");
Guy Benyei11169dd2012-12-18 14:30:41 +00004794 case CXCursor_ArraySubscriptExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004795 return cxstring::createRef("ArraySubscriptExpr");
Alexey Bataev1a3320e2015-08-25 14:24:04 +00004796 case CXCursor_OMPArraySectionExpr:
4797 return cxstring::createRef("OMPArraySectionExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004798 case CXCursor_BinaryOperator:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004799 return cxstring::createRef("BinaryOperator");
Guy Benyei11169dd2012-12-18 14:30:41 +00004800 case CXCursor_CompoundAssignOperator:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004801 return cxstring::createRef("CompoundAssignOperator");
Guy Benyei11169dd2012-12-18 14:30:41 +00004802 case CXCursor_ConditionalOperator:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004803 return cxstring::createRef("ConditionalOperator");
Guy Benyei11169dd2012-12-18 14:30:41 +00004804 case CXCursor_CStyleCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004805 return cxstring::createRef("CStyleCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004806 case CXCursor_CompoundLiteralExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004807 return cxstring::createRef("CompoundLiteralExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004808 case CXCursor_InitListExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004809 return cxstring::createRef("InitListExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004810 case CXCursor_AddrLabelExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004811 return cxstring::createRef("AddrLabelExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004812 case CXCursor_StmtExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004813 return cxstring::createRef("StmtExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004814 case CXCursor_GenericSelectionExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004815 return cxstring::createRef("GenericSelectionExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004816 case CXCursor_GNUNullExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004817 return cxstring::createRef("GNUNullExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004818 case CXCursor_CXXStaticCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004819 return cxstring::createRef("CXXStaticCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004820 case CXCursor_CXXDynamicCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004821 return cxstring::createRef("CXXDynamicCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004822 case CXCursor_CXXReinterpretCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004823 return cxstring::createRef("CXXReinterpretCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004824 case CXCursor_CXXConstCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004825 return cxstring::createRef("CXXConstCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004826 case CXCursor_CXXFunctionalCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004827 return cxstring::createRef("CXXFunctionalCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004828 case CXCursor_CXXTypeidExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004829 return cxstring::createRef("CXXTypeidExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004830 case CXCursor_CXXBoolLiteralExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004831 return cxstring::createRef("CXXBoolLiteralExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004832 case CXCursor_CXXNullPtrLiteralExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004833 return cxstring::createRef("CXXNullPtrLiteralExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004834 case CXCursor_CXXThisExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004835 return cxstring::createRef("CXXThisExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004836 case CXCursor_CXXThrowExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004837 return cxstring::createRef("CXXThrowExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004838 case CXCursor_CXXNewExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004839 return cxstring::createRef("CXXNewExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004840 case CXCursor_CXXDeleteExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004841 return cxstring::createRef("CXXDeleteExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004842 case CXCursor_UnaryExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004843 return cxstring::createRef("UnaryExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004844 case CXCursor_ObjCStringLiteral:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004845 return cxstring::createRef("ObjCStringLiteral");
Guy Benyei11169dd2012-12-18 14:30:41 +00004846 case CXCursor_ObjCBoolLiteralExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004847 return cxstring::createRef("ObjCBoolLiteralExpr");
Erik Pilkington29099de2016-07-16 00:35:23 +00004848 case CXCursor_ObjCAvailabilityCheckExpr:
4849 return cxstring::createRef("ObjCAvailabilityCheckExpr");
Argyrios Kyrtzidisc2233be2013-04-23 17:57:17 +00004850 case CXCursor_ObjCSelfExpr:
4851 return cxstring::createRef("ObjCSelfExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004852 case CXCursor_ObjCEncodeExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004853 return cxstring::createRef("ObjCEncodeExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004854 case CXCursor_ObjCSelectorExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004855 return cxstring::createRef("ObjCSelectorExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004856 case CXCursor_ObjCProtocolExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004857 return cxstring::createRef("ObjCProtocolExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004858 case CXCursor_ObjCBridgedCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004859 return cxstring::createRef("ObjCBridgedCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004860 case CXCursor_BlockExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004861 return cxstring::createRef("BlockExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004862 case CXCursor_PackExpansionExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004863 return cxstring::createRef("PackExpansionExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004864 case CXCursor_SizeOfPackExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004865 return cxstring::createRef("SizeOfPackExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004866 case CXCursor_LambdaExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004867 return cxstring::createRef("LambdaExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004868 case CXCursor_UnexposedExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004869 return cxstring::createRef("UnexposedExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004870 case CXCursor_DeclRefExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004871 return cxstring::createRef("DeclRefExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004872 case CXCursor_MemberRefExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004873 return cxstring::createRef("MemberRefExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004874 case CXCursor_CallExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004875 return cxstring::createRef("CallExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004876 case CXCursor_ObjCMessageExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004877 return cxstring::createRef("ObjCMessageExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004878 case CXCursor_UnexposedStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004879 return cxstring::createRef("UnexposedStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004880 case CXCursor_DeclStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004881 return cxstring::createRef("DeclStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004882 case CXCursor_LabelStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004883 return cxstring::createRef("LabelStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004884 case CXCursor_CompoundStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004885 return cxstring::createRef("CompoundStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004886 case CXCursor_CaseStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004887 return cxstring::createRef("CaseStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004888 case CXCursor_DefaultStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004889 return cxstring::createRef("DefaultStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004890 case CXCursor_IfStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004891 return cxstring::createRef("IfStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004892 case CXCursor_SwitchStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004893 return cxstring::createRef("SwitchStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004894 case CXCursor_WhileStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004895 return cxstring::createRef("WhileStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004896 case CXCursor_DoStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004897 return cxstring::createRef("DoStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004898 case CXCursor_ForStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004899 return cxstring::createRef("ForStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004900 case CXCursor_GotoStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004901 return cxstring::createRef("GotoStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004902 case CXCursor_IndirectGotoStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004903 return cxstring::createRef("IndirectGotoStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004904 case CXCursor_ContinueStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004905 return cxstring::createRef("ContinueStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004906 case CXCursor_BreakStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004907 return cxstring::createRef("BreakStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004908 case CXCursor_ReturnStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004909 return cxstring::createRef("ReturnStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004910 case CXCursor_GCCAsmStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004911 return cxstring::createRef("GCCAsmStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004912 case CXCursor_MSAsmStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004913 return cxstring::createRef("MSAsmStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004914 case CXCursor_ObjCAtTryStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004915 return cxstring::createRef("ObjCAtTryStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004916 case CXCursor_ObjCAtCatchStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004917 return cxstring::createRef("ObjCAtCatchStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004918 case CXCursor_ObjCAtFinallyStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004919 return cxstring::createRef("ObjCAtFinallyStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004920 case CXCursor_ObjCAtThrowStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004921 return cxstring::createRef("ObjCAtThrowStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004922 case CXCursor_ObjCAtSynchronizedStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004923 return cxstring::createRef("ObjCAtSynchronizedStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004924 case CXCursor_ObjCAutoreleasePoolStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004925 return cxstring::createRef("ObjCAutoreleasePoolStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004926 case CXCursor_ObjCForCollectionStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004927 return cxstring::createRef("ObjCForCollectionStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004928 case CXCursor_CXXCatchStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004929 return cxstring::createRef("CXXCatchStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004930 case CXCursor_CXXTryStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004931 return cxstring::createRef("CXXTryStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004932 case CXCursor_CXXForRangeStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004933 return cxstring::createRef("CXXForRangeStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004934 case CXCursor_SEHTryStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004935 return cxstring::createRef("SEHTryStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004936 case CXCursor_SEHExceptStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004937 return cxstring::createRef("SEHExceptStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004938 case CXCursor_SEHFinallyStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004939 return cxstring::createRef("SEHFinallyStmt");
Nico Weber9b982072014-07-07 00:12:30 +00004940 case CXCursor_SEHLeaveStmt:
4941 return cxstring::createRef("SEHLeaveStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004942 case CXCursor_NullStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004943 return cxstring::createRef("NullStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004944 case CXCursor_InvalidFile:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004945 return cxstring::createRef("InvalidFile");
Guy Benyei11169dd2012-12-18 14:30:41 +00004946 case CXCursor_InvalidCode:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004947 return cxstring::createRef("InvalidCode");
Guy Benyei11169dd2012-12-18 14:30:41 +00004948 case CXCursor_NoDeclFound:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004949 return cxstring::createRef("NoDeclFound");
Guy Benyei11169dd2012-12-18 14:30:41 +00004950 case CXCursor_NotImplemented:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004951 return cxstring::createRef("NotImplemented");
Guy Benyei11169dd2012-12-18 14:30:41 +00004952 case CXCursor_TranslationUnit:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004953 return cxstring::createRef("TranslationUnit");
Guy Benyei11169dd2012-12-18 14:30:41 +00004954 case CXCursor_UnexposedAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004955 return cxstring::createRef("UnexposedAttr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004956 case CXCursor_IBActionAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004957 return cxstring::createRef("attribute(ibaction)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004958 case CXCursor_IBOutletAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004959 return cxstring::createRef("attribute(iboutlet)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004960 case CXCursor_IBOutletCollectionAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004961 return cxstring::createRef("attribute(iboutletcollection)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004962 case CXCursor_CXXFinalAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004963 return cxstring::createRef("attribute(final)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004964 case CXCursor_CXXOverrideAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004965 return cxstring::createRef("attribute(override)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004966 case CXCursor_AnnotateAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004967 return cxstring::createRef("attribute(annotate)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004968 case CXCursor_AsmLabelAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004969 return cxstring::createRef("asm label");
Argyrios Kyrtzidis16834f12013-09-25 00:14:38 +00004970 case CXCursor_PackedAttr:
4971 return cxstring::createRef("attribute(packed)");
Joey Gouly81228382014-05-01 15:41:58 +00004972 case CXCursor_PureAttr:
4973 return cxstring::createRef("attribute(pure)");
4974 case CXCursor_ConstAttr:
4975 return cxstring::createRef("attribute(const)");
4976 case CXCursor_NoDuplicateAttr:
4977 return cxstring::createRef("attribute(noduplicate)");
Eli Bendersky2581e662014-05-28 19:29:58 +00004978 case CXCursor_CUDAConstantAttr:
4979 return cxstring::createRef("attribute(constant)");
4980 case CXCursor_CUDADeviceAttr:
4981 return cxstring::createRef("attribute(device)");
4982 case CXCursor_CUDAGlobalAttr:
4983 return cxstring::createRef("attribute(global)");
4984 case CXCursor_CUDAHostAttr:
4985 return cxstring::createRef("attribute(host)");
Eli Bendersky9b071472014-08-08 14:59:00 +00004986 case CXCursor_CUDASharedAttr:
4987 return cxstring::createRef("attribute(shared)");
Saleem Abdulrasool79c69712015-09-05 18:53:43 +00004988 case CXCursor_VisibilityAttr:
4989 return cxstring::createRef("attribute(visibility)");
Saleem Abdulrasool8aa0b802015-12-10 18:45:18 +00004990 case CXCursor_DLLExport:
4991 return cxstring::createRef("attribute(dllexport)");
4992 case CXCursor_DLLImport:
4993 return cxstring::createRef("attribute(dllimport)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004994 case CXCursor_PreprocessingDirective:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004995 return cxstring::createRef("preprocessing directive");
Guy Benyei11169dd2012-12-18 14:30:41 +00004996 case CXCursor_MacroDefinition:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004997 return cxstring::createRef("macro definition");
Guy Benyei11169dd2012-12-18 14:30:41 +00004998 case CXCursor_MacroExpansion:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004999 return cxstring::createRef("macro expansion");
Guy Benyei11169dd2012-12-18 14:30:41 +00005000 case CXCursor_InclusionDirective:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005001 return cxstring::createRef("inclusion directive");
Guy Benyei11169dd2012-12-18 14:30:41 +00005002 case CXCursor_Namespace:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005003 return cxstring::createRef("Namespace");
Guy Benyei11169dd2012-12-18 14:30:41 +00005004 case CXCursor_LinkageSpec:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005005 return cxstring::createRef("LinkageSpec");
Guy Benyei11169dd2012-12-18 14:30:41 +00005006 case CXCursor_CXXBaseSpecifier:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005007 return cxstring::createRef("C++ base class specifier");
Guy Benyei11169dd2012-12-18 14:30:41 +00005008 case CXCursor_Constructor:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005009 return cxstring::createRef("CXXConstructor");
Guy Benyei11169dd2012-12-18 14:30:41 +00005010 case CXCursor_Destructor:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005011 return cxstring::createRef("CXXDestructor");
Guy Benyei11169dd2012-12-18 14:30:41 +00005012 case CXCursor_ConversionFunction:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005013 return cxstring::createRef("CXXConversion");
Guy Benyei11169dd2012-12-18 14:30:41 +00005014 case CXCursor_TemplateTypeParameter:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005015 return cxstring::createRef("TemplateTypeParameter");
Guy Benyei11169dd2012-12-18 14:30:41 +00005016 case CXCursor_NonTypeTemplateParameter:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005017 return cxstring::createRef("NonTypeTemplateParameter");
Guy Benyei11169dd2012-12-18 14:30:41 +00005018 case CXCursor_TemplateTemplateParameter:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005019 return cxstring::createRef("TemplateTemplateParameter");
Guy Benyei11169dd2012-12-18 14:30:41 +00005020 case CXCursor_FunctionTemplate:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005021 return cxstring::createRef("FunctionTemplate");
Guy Benyei11169dd2012-12-18 14:30:41 +00005022 case CXCursor_ClassTemplate:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005023 return cxstring::createRef("ClassTemplate");
Guy Benyei11169dd2012-12-18 14:30:41 +00005024 case CXCursor_ClassTemplatePartialSpecialization:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005025 return cxstring::createRef("ClassTemplatePartialSpecialization");
Guy Benyei11169dd2012-12-18 14:30:41 +00005026 case CXCursor_NamespaceAlias:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005027 return cxstring::createRef("NamespaceAlias");
Guy Benyei11169dd2012-12-18 14:30:41 +00005028 case CXCursor_UsingDirective:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005029 return cxstring::createRef("UsingDirective");
Guy Benyei11169dd2012-12-18 14:30:41 +00005030 case CXCursor_UsingDeclaration:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005031 return cxstring::createRef("UsingDeclaration");
Guy Benyei11169dd2012-12-18 14:30:41 +00005032 case CXCursor_TypeAliasDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005033 return cxstring::createRef("TypeAliasDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00005034 case CXCursor_ObjCSynthesizeDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005035 return cxstring::createRef("ObjCSynthesizeDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00005036 case CXCursor_ObjCDynamicDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005037 return cxstring::createRef("ObjCDynamicDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00005038 case CXCursor_CXXAccessSpecifier:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005039 return cxstring::createRef("CXXAccessSpecifier");
Guy Benyei11169dd2012-12-18 14:30:41 +00005040 case CXCursor_ModuleImportDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005041 return cxstring::createRef("ModuleImport");
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00005042 case CXCursor_OMPParallelDirective:
Alexey Bataev1b59ab52014-02-27 08:29:12 +00005043 return cxstring::createRef("OMPParallelDirective");
5044 case CXCursor_OMPSimdDirective:
5045 return cxstring::createRef("OMPSimdDirective");
Alexey Bataevf29276e2014-06-18 04:14:57 +00005046 case CXCursor_OMPForDirective:
5047 return cxstring::createRef("OMPForDirective");
Alexander Musmanf82886e2014-09-18 05:12:34 +00005048 case CXCursor_OMPForSimdDirective:
5049 return cxstring::createRef("OMPForSimdDirective");
Alexey Bataevd3f8dd22014-06-25 11:44:49 +00005050 case CXCursor_OMPSectionsDirective:
5051 return cxstring::createRef("OMPSectionsDirective");
Alexey Bataev1e0498a2014-06-26 08:21:58 +00005052 case CXCursor_OMPSectionDirective:
5053 return cxstring::createRef("OMPSectionDirective");
Alexey Bataevd1e40fb2014-06-26 12:05:45 +00005054 case CXCursor_OMPSingleDirective:
5055 return cxstring::createRef("OMPSingleDirective");
Alexander Musman80c22892014-07-17 08:54:58 +00005056 case CXCursor_OMPMasterDirective:
5057 return cxstring::createRef("OMPMasterDirective");
Alexander Musmand9ed09f2014-07-21 09:42:05 +00005058 case CXCursor_OMPCriticalDirective:
5059 return cxstring::createRef("OMPCriticalDirective");
Alexey Bataev4acb8592014-07-07 13:01:15 +00005060 case CXCursor_OMPParallelForDirective:
5061 return cxstring::createRef("OMPParallelForDirective");
Alexander Musmane4e893b2014-09-23 09:33:00 +00005062 case CXCursor_OMPParallelForSimdDirective:
5063 return cxstring::createRef("OMPParallelForSimdDirective");
Alexey Bataev84d0b3e2014-07-08 08:12:03 +00005064 case CXCursor_OMPParallelSectionsDirective:
5065 return cxstring::createRef("OMPParallelSectionsDirective");
Alexey Bataev9c2e8ee2014-07-11 11:25:16 +00005066 case CXCursor_OMPTaskDirective:
5067 return cxstring::createRef("OMPTaskDirective");
Alexey Bataev68446b72014-07-18 07:47:19 +00005068 case CXCursor_OMPTaskyieldDirective:
5069 return cxstring::createRef("OMPTaskyieldDirective");
Alexey Bataev4d1dfea2014-07-18 09:11:51 +00005070 case CXCursor_OMPBarrierDirective:
5071 return cxstring::createRef("OMPBarrierDirective");
Alexey Bataev2df347a2014-07-18 10:17:07 +00005072 case CXCursor_OMPTaskwaitDirective:
5073 return cxstring::createRef("OMPTaskwaitDirective");
Alexey Bataevc30dd2d2015-06-18 12:14:09 +00005074 case CXCursor_OMPTaskgroupDirective:
5075 return cxstring::createRef("OMPTaskgroupDirective");
Alexey Bataev6125da92014-07-21 11:26:11 +00005076 case CXCursor_OMPFlushDirective:
5077 return cxstring::createRef("OMPFlushDirective");
Alexey Bataev9fb6e642014-07-22 06:45:04 +00005078 case CXCursor_OMPOrderedDirective:
5079 return cxstring::createRef("OMPOrderedDirective");
Alexey Bataev0162e452014-07-22 10:10:35 +00005080 case CXCursor_OMPAtomicDirective:
5081 return cxstring::createRef("OMPAtomicDirective");
Alexey Bataev0bd520b2014-09-19 08:19:49 +00005082 case CXCursor_OMPTargetDirective:
5083 return cxstring::createRef("OMPTargetDirective");
Michael Wong65f367f2015-07-21 13:44:28 +00005084 case CXCursor_OMPTargetDataDirective:
5085 return cxstring::createRef("OMPTargetDataDirective");
Samuel Antaodf67fc42016-01-19 19:15:56 +00005086 case CXCursor_OMPTargetEnterDataDirective:
5087 return cxstring::createRef("OMPTargetEnterDataDirective");
Samuel Antao72590762016-01-19 20:04:50 +00005088 case CXCursor_OMPTargetExitDataDirective:
5089 return cxstring::createRef("OMPTargetExitDataDirective");
Arpith Chacko Jacobe955b3d2016-01-26 18:48:41 +00005090 case CXCursor_OMPTargetParallelDirective:
5091 return cxstring::createRef("OMPTargetParallelDirective");
Arpith Chacko Jacob05bebb52016-02-03 15:46:42 +00005092 case CXCursor_OMPTargetParallelForDirective:
5093 return cxstring::createRef("OMPTargetParallelForDirective");
Samuel Antao686c70c2016-05-26 17:30:50 +00005094 case CXCursor_OMPTargetUpdateDirective:
5095 return cxstring::createRef("OMPTargetUpdateDirective");
Alexey Bataev13314bf2014-10-09 04:18:56 +00005096 case CXCursor_OMPTeamsDirective:
5097 return cxstring::createRef("OMPTeamsDirective");
Alexey Bataev6d4ed052015-07-01 06:57:41 +00005098 case CXCursor_OMPCancellationPointDirective:
5099 return cxstring::createRef("OMPCancellationPointDirective");
Alexey Bataev80909872015-07-02 11:25:17 +00005100 case CXCursor_OMPCancelDirective:
5101 return cxstring::createRef("OMPCancelDirective");
Alexey Bataev49f6e782015-12-01 04:18:41 +00005102 case CXCursor_OMPTaskLoopDirective:
5103 return cxstring::createRef("OMPTaskLoopDirective");
Alexey Bataev0a6ed842015-12-03 09:40:15 +00005104 case CXCursor_OMPTaskLoopSimdDirective:
5105 return cxstring::createRef("OMPTaskLoopSimdDirective");
Carlo Bertolli6200a3d2015-12-14 14:51:25 +00005106 case CXCursor_OMPDistributeDirective:
5107 return cxstring::createRef("OMPDistributeDirective");
Carlo Bertolli9925f152016-06-27 14:55:37 +00005108 case CXCursor_OMPDistributeParallelForDirective:
5109 return cxstring::createRef("OMPDistributeParallelForDirective");
Kelvin Li4a39add2016-07-05 05:00:15 +00005110 case CXCursor_OMPDistributeParallelForSimdDirective:
5111 return cxstring::createRef("OMPDistributeParallelForSimdDirective");
Kelvin Li787f3fc2016-07-06 04:45:38 +00005112 case CXCursor_OMPDistributeSimdDirective:
5113 return cxstring::createRef("OMPDistributeSimdDirective");
Kelvin Lia579b912016-07-14 02:54:56 +00005114 case CXCursor_OMPTargetParallelForSimdDirective:
5115 return cxstring::createRef("OMPTargetParallelForSimdDirective");
Kelvin Li986330c2016-07-20 22:57:10 +00005116 case CXCursor_OMPTargetSimdDirective:
5117 return cxstring::createRef("OMPTargetSimdDirective");
Kelvin Li02532872016-08-05 14:37:37 +00005118 case CXCursor_OMPTeamsDistributeDirective:
5119 return cxstring::createRef("OMPTeamsDistributeDirective");
Kelvin Li4e325f72016-10-25 12:50:55 +00005120 case CXCursor_OMPTeamsDistributeSimdDirective:
5121 return cxstring::createRef("OMPTeamsDistributeSimdDirective");
Kelvin Li579e41c2016-11-30 23:51:03 +00005122 case CXCursor_OMPTeamsDistributeParallelForSimdDirective:
5123 return cxstring::createRef("OMPTeamsDistributeParallelForSimdDirective");
Kelvin Li7ade93f2016-12-09 03:24:30 +00005124 case CXCursor_OMPTeamsDistributeParallelForDirective:
5125 return cxstring::createRef("OMPTeamsDistributeParallelForDirective");
Kelvin Libf594a52016-12-17 05:48:59 +00005126 case CXCursor_OMPTargetTeamsDirective:
5127 return cxstring::createRef("OMPTargetTeamsDirective");
Kelvin Li83c451e2016-12-25 04:52:54 +00005128 case CXCursor_OMPTargetTeamsDistributeDirective:
5129 return cxstring::createRef("OMPTargetTeamsDistributeDirective");
Kelvin Li80e8f562016-12-29 22:16:30 +00005130 case CXCursor_OMPTargetTeamsDistributeParallelForDirective:
5131 return cxstring::createRef("OMPTargetTeamsDistributeParallelForDirective");
Kelvin Li1851df52017-01-03 05:23:48 +00005132 case CXCursor_OMPTargetTeamsDistributeParallelForSimdDirective:
5133 return cxstring::createRef(
5134 "OMPTargetTeamsDistributeParallelForSimdDirective");
Kelvin Lida681182017-01-10 18:08:18 +00005135 case CXCursor_OMPTargetTeamsDistributeSimdDirective:
5136 return cxstring::createRef("OMPTargetTeamsDistributeSimdDirective");
Francisco Lopes da Silva975a9f62015-01-21 16:24:11 +00005137 case CXCursor_OverloadCandidate:
5138 return cxstring::createRef("OverloadCandidate");
Sergey Kalinichev8f3b1872015-11-15 13:48:32 +00005139 case CXCursor_TypeAliasTemplateDecl:
5140 return cxstring::createRef("TypeAliasTemplateDecl");
Olivier Goffart81978012016-06-09 16:15:55 +00005141 case CXCursor_StaticAssert:
5142 return cxstring::createRef("StaticAssert");
Olivier Goffartd211c642016-11-04 06:29:27 +00005143 case CXCursor_FriendDecl:
5144 return cxstring::createRef("FriendDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00005145 }
5146
5147 llvm_unreachable("Unhandled CXCursorKind");
5148}
5149
5150struct GetCursorData {
5151 SourceLocation TokenBeginLoc;
5152 bool PointsAtMacroArgExpansion;
5153 bool VisitedObjCPropertyImplDecl;
5154 SourceLocation VisitedDeclaratorDeclStartLoc;
5155 CXCursor &BestCursor;
5156
5157 GetCursorData(SourceManager &SM,
5158 SourceLocation tokenBegin, CXCursor &outputCursor)
5159 : TokenBeginLoc(tokenBegin), BestCursor(outputCursor) {
5160 PointsAtMacroArgExpansion = SM.isMacroArgExpansion(tokenBegin);
5161 VisitedObjCPropertyImplDecl = false;
5162 }
5163};
5164
5165static enum CXChildVisitResult GetCursorVisitor(CXCursor cursor,
5166 CXCursor parent,
5167 CXClientData client_data) {
5168 GetCursorData *Data = static_cast<GetCursorData *>(client_data);
5169 CXCursor *BestCursor = &Data->BestCursor;
5170
5171 // If we point inside a macro argument we should provide info of what the
5172 // token is so use the actual cursor, don't replace it with a macro expansion
5173 // cursor.
5174 if (cursor.kind == CXCursor_MacroExpansion && Data->PointsAtMacroArgExpansion)
5175 return CXChildVisit_Recurse;
5176
5177 if (clang_isDeclaration(cursor.kind)) {
5178 // Avoid having the implicit methods override the property decls.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005179 if (const ObjCMethodDecl *MD
Guy Benyei11169dd2012-12-18 14:30:41 +00005180 = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(cursor))) {
5181 if (MD->isImplicit())
5182 return CXChildVisit_Break;
5183
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005184 } else if (const ObjCInterfaceDecl *ID
Guy Benyei11169dd2012-12-18 14:30:41 +00005185 = dyn_cast_or_null<ObjCInterfaceDecl>(getCursorDecl(cursor))) {
5186 // Check that when we have multiple @class references in the same line,
5187 // that later ones do not override the previous ones.
5188 // If we have:
5189 // @class Foo, Bar;
5190 // source ranges for both start at '@', so 'Bar' will end up overriding
5191 // 'Foo' even though the cursor location was at 'Foo'.
5192 if (BestCursor->kind == CXCursor_ObjCInterfaceDecl ||
5193 BestCursor->kind == CXCursor_ObjCClassRef)
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005194 if (const ObjCInterfaceDecl *PrevID
Guy Benyei11169dd2012-12-18 14:30:41 +00005195 = dyn_cast_or_null<ObjCInterfaceDecl>(getCursorDecl(*BestCursor))){
5196 if (PrevID != ID &&
5197 !PrevID->isThisDeclarationADefinition() &&
5198 !ID->isThisDeclarationADefinition())
5199 return CXChildVisit_Break;
5200 }
5201
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005202 } else if (const DeclaratorDecl *DD
Guy Benyei11169dd2012-12-18 14:30:41 +00005203 = dyn_cast_or_null<DeclaratorDecl>(getCursorDecl(cursor))) {
5204 SourceLocation StartLoc = DD->getSourceRange().getBegin();
5205 // Check that when we have multiple declarators in the same line,
5206 // that later ones do not override the previous ones.
5207 // If we have:
5208 // int Foo, Bar;
5209 // source ranges for both start at 'int', so 'Bar' will end up overriding
5210 // 'Foo' even though the cursor location was at 'Foo'.
5211 if (Data->VisitedDeclaratorDeclStartLoc == StartLoc)
5212 return CXChildVisit_Break;
5213 Data->VisitedDeclaratorDeclStartLoc = StartLoc;
5214
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005215 } else if (const ObjCPropertyImplDecl *PropImp
Guy Benyei11169dd2012-12-18 14:30:41 +00005216 = dyn_cast_or_null<ObjCPropertyImplDecl>(getCursorDecl(cursor))) {
5217 (void)PropImp;
5218 // Check that when we have multiple @synthesize in the same line,
5219 // that later ones do not override the previous ones.
5220 // If we have:
5221 // @synthesize Foo, Bar;
5222 // source ranges for both start at '@', so 'Bar' will end up overriding
5223 // 'Foo' even though the cursor location was at 'Foo'.
5224 if (Data->VisitedObjCPropertyImplDecl)
5225 return CXChildVisit_Break;
5226 Data->VisitedObjCPropertyImplDecl = true;
5227 }
5228 }
5229
5230 if (clang_isExpression(cursor.kind) &&
5231 clang_isDeclaration(BestCursor->kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005232 if (const Decl *D = getCursorDecl(*BestCursor)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005233 // Avoid having the cursor of an expression replace the declaration cursor
5234 // when the expression source range overlaps the declaration range.
5235 // This can happen for C++ constructor expressions whose range generally
5236 // include the variable declaration, e.g.:
5237 // MyCXXClass foo; // Make sure pointing at 'foo' returns a VarDecl cursor.
5238 if (D->getLocation().isValid() && Data->TokenBeginLoc.isValid() &&
5239 D->getLocation() == Data->TokenBeginLoc)
5240 return CXChildVisit_Break;
5241 }
5242 }
5243
5244 // If our current best cursor is the construction of a temporary object,
5245 // don't replace that cursor with a type reference, because we want
5246 // clang_getCursor() to point at the constructor.
5247 if (clang_isExpression(BestCursor->kind) &&
5248 isa<CXXTemporaryObjectExpr>(getCursorExpr(*BestCursor)) &&
5249 cursor.kind == CXCursor_TypeRef) {
5250 // Keep the cursor pointing at CXXTemporaryObjectExpr but also mark it
5251 // as having the actual point on the type reference.
5252 *BestCursor = getTypeRefedCallExprCursor(*BestCursor);
5253 return CXChildVisit_Recurse;
5254 }
Douglas Gregore9d95f12015-07-07 03:57:35 +00005255
5256 // If we already have an Objective-C superclass reference, don't
5257 // update it further.
5258 if (BestCursor->kind == CXCursor_ObjCSuperClassRef)
5259 return CXChildVisit_Break;
5260
Guy Benyei11169dd2012-12-18 14:30:41 +00005261 *BestCursor = cursor;
5262 return CXChildVisit_Recurse;
5263}
5264
5265CXCursor clang_getCursor(CXTranslationUnit TU, CXSourceLocation Loc) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00005266 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00005267 LOG_BAD_TU(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00005268 return clang_getNullCursor();
Dmitri Gribenko256454f2014-02-11 14:34:14 +00005269 }
Guy Benyei11169dd2012-12-18 14:30:41 +00005270
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00005271 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00005272 ASTUnit::ConcurrencyCheck Check(*CXXUnit);
5273
5274 SourceLocation SLoc = cxloc::translateSourceLocation(Loc);
5275 CXCursor Result = cxcursor::getCursor(TU, SLoc);
5276
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00005277 LOG_FUNC_SECTION {
Guy Benyei11169dd2012-12-18 14:30:41 +00005278 CXFile SearchFile;
5279 unsigned SearchLine, SearchColumn;
5280 CXFile ResultFile;
5281 unsigned ResultLine, ResultColumn;
5282 CXString SearchFileName, ResultFileName, KindSpelling, USR;
5283 const char *IsDef = clang_isCursorDefinition(Result)? " (Definition)" : "";
5284 CXSourceLocation ResultLoc = clang_getCursorLocation(Result);
Craig Topper69186e72014-06-08 08:38:04 +00005285
5286 clang_getFileLocation(Loc, &SearchFile, &SearchLine, &SearchColumn,
5287 nullptr);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00005288 clang_getFileLocation(ResultLoc, &ResultFile, &ResultLine,
Craig Topper69186e72014-06-08 08:38:04 +00005289 &ResultColumn, nullptr);
Guy Benyei11169dd2012-12-18 14:30:41 +00005290 SearchFileName = clang_getFileName(SearchFile);
5291 ResultFileName = clang_getFileName(ResultFile);
5292 KindSpelling = clang_getCursorKindSpelling(Result.kind);
5293 USR = clang_getCursorUSR(Result);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00005294 *Log << llvm::format("(%s:%d:%d) = %s",
5295 clang_getCString(SearchFileName), SearchLine, SearchColumn,
5296 clang_getCString(KindSpelling))
5297 << llvm::format("(%s:%d:%d):%s%s",
5298 clang_getCString(ResultFileName), ResultLine, ResultColumn,
5299 clang_getCString(USR), IsDef);
Guy Benyei11169dd2012-12-18 14:30:41 +00005300 clang_disposeString(SearchFileName);
5301 clang_disposeString(ResultFileName);
5302 clang_disposeString(KindSpelling);
5303 clang_disposeString(USR);
5304
5305 CXCursor Definition = clang_getCursorDefinition(Result);
5306 if (!clang_equalCursors(Definition, clang_getNullCursor())) {
5307 CXSourceLocation DefinitionLoc = clang_getCursorLocation(Definition);
5308 CXString DefinitionKindSpelling
5309 = clang_getCursorKindSpelling(Definition.kind);
5310 CXFile DefinitionFile;
5311 unsigned DefinitionLine, DefinitionColumn;
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00005312 clang_getFileLocation(DefinitionLoc, &DefinitionFile,
Craig Topper69186e72014-06-08 08:38:04 +00005313 &DefinitionLine, &DefinitionColumn, nullptr);
Guy Benyei11169dd2012-12-18 14:30:41 +00005314 CXString DefinitionFileName = clang_getFileName(DefinitionFile);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00005315 *Log << llvm::format(" -> %s(%s:%d:%d)",
5316 clang_getCString(DefinitionKindSpelling),
5317 clang_getCString(DefinitionFileName),
5318 DefinitionLine, DefinitionColumn);
Guy Benyei11169dd2012-12-18 14:30:41 +00005319 clang_disposeString(DefinitionFileName);
5320 clang_disposeString(DefinitionKindSpelling);
5321 }
5322 }
5323
5324 return Result;
5325}
5326
5327CXCursor clang_getNullCursor(void) {
5328 return MakeCXCursorInvalid(CXCursor_InvalidFile);
5329}
5330
5331unsigned clang_equalCursors(CXCursor X, CXCursor Y) {
Argyrios Kyrtzidisbf1be592013-01-08 18:23:28 +00005332 // Clear out the "FirstInDeclGroup" part in a declaration cursor, since we
5333 // can't set consistently. For example, when visiting a DeclStmt we will set
5334 // it but we don't set it on the result of clang_getCursorDefinition for
5335 // a reference of the same declaration.
5336 // FIXME: Setting "FirstInDeclGroup" in CXCursors is a hack that only works
5337 // when visiting a DeclStmt currently, the AST should be enhanced to be able
5338 // to provide that kind of info.
5339 if (clang_isDeclaration(X.kind))
Craig Topper69186e72014-06-08 08:38:04 +00005340 X.data[1] = nullptr;
Argyrios Kyrtzidisbf1be592013-01-08 18:23:28 +00005341 if (clang_isDeclaration(Y.kind))
Craig Topper69186e72014-06-08 08:38:04 +00005342 Y.data[1] = nullptr;
Argyrios Kyrtzidisbf1be592013-01-08 18:23:28 +00005343
Guy Benyei11169dd2012-12-18 14:30:41 +00005344 return X == Y;
5345}
5346
5347unsigned clang_hashCursor(CXCursor C) {
5348 unsigned Index = 0;
5349 if (clang_isExpression(C.kind) || clang_isStatement(C.kind))
5350 Index = 1;
5351
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005352 return llvm::DenseMapInfo<std::pair<unsigned, const void*> >::getHashValue(
Guy Benyei11169dd2012-12-18 14:30:41 +00005353 std::make_pair(C.kind, C.data[Index]));
5354}
5355
5356unsigned clang_isInvalid(enum CXCursorKind K) {
5357 return K >= CXCursor_FirstInvalid && K <= CXCursor_LastInvalid;
5358}
5359
5360unsigned clang_isDeclaration(enum CXCursorKind K) {
5361 return (K >= CXCursor_FirstDecl && K <= CXCursor_LastDecl) ||
5362 (K >= CXCursor_FirstExtraDecl && K <= CXCursor_LastExtraDecl);
5363}
5364
5365unsigned clang_isReference(enum CXCursorKind K) {
5366 return K >= CXCursor_FirstRef && K <= CXCursor_LastRef;
5367}
5368
5369unsigned clang_isExpression(enum CXCursorKind K) {
5370 return K >= CXCursor_FirstExpr && K <= CXCursor_LastExpr;
5371}
5372
5373unsigned clang_isStatement(enum CXCursorKind K) {
5374 return K >= CXCursor_FirstStmt && K <= CXCursor_LastStmt;
5375}
5376
5377unsigned clang_isAttribute(enum CXCursorKind K) {
5378 return K >= CXCursor_FirstAttr && K <= CXCursor_LastAttr;
5379}
5380
5381unsigned clang_isTranslationUnit(enum CXCursorKind K) {
5382 return K == CXCursor_TranslationUnit;
5383}
5384
5385unsigned clang_isPreprocessing(enum CXCursorKind K) {
5386 return K >= CXCursor_FirstPreprocessing && K <= CXCursor_LastPreprocessing;
5387}
5388
5389unsigned clang_isUnexposed(enum CXCursorKind K) {
5390 switch (K) {
5391 case CXCursor_UnexposedDecl:
5392 case CXCursor_UnexposedExpr:
5393 case CXCursor_UnexposedStmt:
5394 case CXCursor_UnexposedAttr:
5395 return true;
5396 default:
5397 return false;
5398 }
5399}
5400
5401CXCursorKind clang_getCursorKind(CXCursor C) {
5402 return C.kind;
5403}
5404
5405CXSourceLocation clang_getCursorLocation(CXCursor C) {
5406 if (clang_isReference(C.kind)) {
5407 switch (C.kind) {
5408 case CXCursor_ObjCSuperClassRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005409 std::pair<const ObjCInterfaceDecl *, SourceLocation> P
Guy Benyei11169dd2012-12-18 14:30:41 +00005410 = getCursorObjCSuperClassRef(C);
5411 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5412 }
5413
5414 case CXCursor_ObjCProtocolRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005415 std::pair<const ObjCProtocolDecl *, SourceLocation> P
Guy Benyei11169dd2012-12-18 14:30:41 +00005416 = getCursorObjCProtocolRef(C);
5417 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5418 }
5419
5420 case CXCursor_ObjCClassRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005421 std::pair<const ObjCInterfaceDecl *, SourceLocation> P
Guy Benyei11169dd2012-12-18 14:30:41 +00005422 = getCursorObjCClassRef(C);
5423 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5424 }
5425
5426 case CXCursor_TypeRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005427 std::pair<const TypeDecl *, SourceLocation> P = getCursorTypeRef(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005428 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5429 }
5430
5431 case CXCursor_TemplateRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005432 std::pair<const TemplateDecl *, SourceLocation> P =
5433 getCursorTemplateRef(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005434 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5435 }
5436
5437 case CXCursor_NamespaceRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005438 std::pair<const NamedDecl *, SourceLocation> P = getCursorNamespaceRef(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005439 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5440 }
5441
5442 case CXCursor_MemberRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005443 std::pair<const FieldDecl *, SourceLocation> P = getCursorMemberRef(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005444 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5445 }
5446
5447 case CXCursor_VariableRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005448 std::pair<const VarDecl *, SourceLocation> P = getCursorVariableRef(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005449 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5450 }
5451
5452 case CXCursor_CXXBaseSpecifier: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005453 const CXXBaseSpecifier *BaseSpec = getCursorCXXBaseSpecifier(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005454 if (!BaseSpec)
5455 return clang_getNullLocation();
5456
5457 if (TypeSourceInfo *TSInfo = BaseSpec->getTypeSourceInfo())
5458 return cxloc::translateSourceLocation(getCursorContext(C),
5459 TSInfo->getTypeLoc().getBeginLoc());
5460
5461 return cxloc::translateSourceLocation(getCursorContext(C),
5462 BaseSpec->getLocStart());
5463 }
5464
5465 case CXCursor_LabelRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005466 std::pair<const LabelStmt *, SourceLocation> P = getCursorLabelRef(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005467 return cxloc::translateSourceLocation(getCursorContext(C), P.second);
5468 }
5469
5470 case CXCursor_OverloadedDeclRef:
5471 return cxloc::translateSourceLocation(getCursorContext(C),
5472 getCursorOverloadedDeclRef(C).second);
5473
5474 default:
5475 // FIXME: Need a way to enumerate all non-reference cases.
5476 llvm_unreachable("Missed a reference kind");
5477 }
5478 }
5479
5480 if (clang_isExpression(C.kind))
5481 return cxloc::translateSourceLocation(getCursorContext(C),
5482 getLocationFromExpr(getCursorExpr(C)));
5483
5484 if (clang_isStatement(C.kind))
5485 return cxloc::translateSourceLocation(getCursorContext(C),
5486 getCursorStmt(C)->getLocStart());
5487
5488 if (C.kind == CXCursor_PreprocessingDirective) {
5489 SourceLocation L = cxcursor::getCursorPreprocessingDirective(C).getBegin();
5490 return cxloc::translateSourceLocation(getCursorContext(C), L);
5491 }
5492
5493 if (C.kind == CXCursor_MacroExpansion) {
5494 SourceLocation L
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00005495 = cxcursor::getCursorMacroExpansion(C).getSourceRange().getBegin();
Guy Benyei11169dd2012-12-18 14:30:41 +00005496 return cxloc::translateSourceLocation(getCursorContext(C), L);
5497 }
5498
5499 if (C.kind == CXCursor_MacroDefinition) {
5500 SourceLocation L = cxcursor::getCursorMacroDefinition(C)->getLocation();
5501 return cxloc::translateSourceLocation(getCursorContext(C), L);
5502 }
5503
5504 if (C.kind == CXCursor_InclusionDirective) {
5505 SourceLocation L
5506 = cxcursor::getCursorInclusionDirective(C)->getSourceRange().getBegin();
5507 return cxloc::translateSourceLocation(getCursorContext(C), L);
5508 }
5509
Argyrios Kyrtzidis16834f12013-09-25 00:14:38 +00005510 if (clang_isAttribute(C.kind)) {
5511 SourceLocation L
5512 = cxcursor::getCursorAttr(C)->getLocation();
5513 return cxloc::translateSourceLocation(getCursorContext(C), L);
5514 }
5515
Guy Benyei11169dd2012-12-18 14:30:41 +00005516 if (!clang_isDeclaration(C.kind))
5517 return clang_getNullLocation();
5518
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005519 const Decl *D = getCursorDecl(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005520 if (!D)
5521 return clang_getNullLocation();
5522
5523 SourceLocation Loc = D->getLocation();
5524 // FIXME: Multiple variables declared in a single declaration
5525 // currently lack the information needed to correctly determine their
5526 // ranges when accounting for the type-specifier. We use context
5527 // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
5528 // and if so, whether it is the first decl.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005529 if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005530 if (!cxcursor::isFirstInDeclGroup(C))
5531 Loc = VD->getLocation();
5532 }
5533
5534 // For ObjC methods, give the start location of the method name.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005535 if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00005536 Loc = MD->getSelectorStartLoc();
5537
5538 return cxloc::translateSourceLocation(getCursorContext(C), Loc);
5539}
5540
NAKAMURA Takumia01f4c32016-12-19 16:50:43 +00005541} // end extern "C"
5542
Guy Benyei11169dd2012-12-18 14:30:41 +00005543CXCursor cxcursor::getCursor(CXTranslationUnit TU, SourceLocation SLoc) {
5544 assert(TU);
5545
5546 // Guard against an invalid SourceLocation, or we may assert in one
5547 // of the following calls.
5548 if (SLoc.isInvalid())
5549 return clang_getNullCursor();
5550
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00005551 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00005552
5553 // Translate the given source location to make it point at the beginning of
5554 // the token under the cursor.
5555 SLoc = Lexer::GetBeginningOfToken(SLoc, CXXUnit->getSourceManager(),
5556 CXXUnit->getASTContext().getLangOpts());
5557
5558 CXCursor Result = MakeCXCursorInvalid(CXCursor_NoDeclFound);
5559 if (SLoc.isValid()) {
5560 GetCursorData ResultData(CXXUnit->getSourceManager(), SLoc, Result);
5561 CursorVisitor CursorVis(TU, GetCursorVisitor, &ResultData,
5562 /*VisitPreprocessorLast=*/true,
5563 /*VisitIncludedEntities=*/false,
5564 SourceLocation(SLoc));
5565 CursorVis.visitFileRegion();
5566 }
5567
5568 return Result;
5569}
5570
5571static SourceRange getRawCursorExtent(CXCursor C) {
5572 if (clang_isReference(C.kind)) {
5573 switch (C.kind) {
5574 case CXCursor_ObjCSuperClassRef:
5575 return getCursorObjCSuperClassRef(C).second;
5576
5577 case CXCursor_ObjCProtocolRef:
5578 return getCursorObjCProtocolRef(C).second;
5579
5580 case CXCursor_ObjCClassRef:
5581 return getCursorObjCClassRef(C).second;
5582
5583 case CXCursor_TypeRef:
5584 return getCursorTypeRef(C).second;
5585
5586 case CXCursor_TemplateRef:
5587 return getCursorTemplateRef(C).second;
5588
5589 case CXCursor_NamespaceRef:
5590 return getCursorNamespaceRef(C).second;
5591
5592 case CXCursor_MemberRef:
5593 return getCursorMemberRef(C).second;
5594
5595 case CXCursor_CXXBaseSpecifier:
5596 return getCursorCXXBaseSpecifier(C)->getSourceRange();
5597
5598 case CXCursor_LabelRef:
5599 return getCursorLabelRef(C).second;
5600
5601 case CXCursor_OverloadedDeclRef:
5602 return getCursorOverloadedDeclRef(C).second;
5603
5604 case CXCursor_VariableRef:
5605 return getCursorVariableRef(C).second;
5606
5607 default:
5608 // FIXME: Need a way to enumerate all non-reference cases.
5609 llvm_unreachable("Missed a reference kind");
5610 }
5611 }
5612
5613 if (clang_isExpression(C.kind))
5614 return getCursorExpr(C)->getSourceRange();
5615
5616 if (clang_isStatement(C.kind))
5617 return getCursorStmt(C)->getSourceRange();
5618
5619 if (clang_isAttribute(C.kind))
5620 return getCursorAttr(C)->getRange();
5621
5622 if (C.kind == CXCursor_PreprocessingDirective)
5623 return cxcursor::getCursorPreprocessingDirective(C);
5624
5625 if (C.kind == CXCursor_MacroExpansion) {
5626 ASTUnit *TU = getCursorASTUnit(C);
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00005627 SourceRange Range = cxcursor::getCursorMacroExpansion(C).getSourceRange();
Guy Benyei11169dd2012-12-18 14:30:41 +00005628 return TU->mapRangeFromPreamble(Range);
5629 }
5630
5631 if (C.kind == CXCursor_MacroDefinition) {
5632 ASTUnit *TU = getCursorASTUnit(C);
5633 SourceRange Range = cxcursor::getCursorMacroDefinition(C)->getSourceRange();
5634 return TU->mapRangeFromPreamble(Range);
5635 }
5636
5637 if (C.kind == CXCursor_InclusionDirective) {
5638 ASTUnit *TU = getCursorASTUnit(C);
5639 SourceRange Range = cxcursor::getCursorInclusionDirective(C)->getSourceRange();
5640 return TU->mapRangeFromPreamble(Range);
5641 }
5642
5643 if (C.kind == CXCursor_TranslationUnit) {
5644 ASTUnit *TU = getCursorASTUnit(C);
5645 FileID MainID = TU->getSourceManager().getMainFileID();
5646 SourceLocation Start = TU->getSourceManager().getLocForStartOfFile(MainID);
5647 SourceLocation End = TU->getSourceManager().getLocForEndOfFile(MainID);
5648 return SourceRange(Start, End);
5649 }
5650
5651 if (clang_isDeclaration(C.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005652 const Decl *D = cxcursor::getCursorDecl(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005653 if (!D)
5654 return SourceRange();
5655
5656 SourceRange R = D->getSourceRange();
5657 // FIXME: Multiple variables declared in a single declaration
5658 // currently lack the information needed to correctly determine their
5659 // ranges when accounting for the type-specifier. We use context
5660 // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
5661 // and if so, whether it is the first decl.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005662 if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005663 if (!cxcursor::isFirstInDeclGroup(C))
5664 R.setBegin(VD->getLocation());
5665 }
5666 return R;
5667 }
5668 return SourceRange();
5669}
5670
5671/// \brief Retrieves the "raw" cursor extent, which is then extended to include
5672/// the decl-specifier-seq for declarations.
5673static SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr) {
5674 if (clang_isDeclaration(C.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005675 const Decl *D = cxcursor::getCursorDecl(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005676 if (!D)
5677 return SourceRange();
5678
5679 SourceRange R = D->getSourceRange();
5680
5681 // Adjust the start of the location for declarations preceded by
5682 // declaration specifiers.
5683 SourceLocation StartLoc;
5684 if (const DeclaratorDecl *DD = dyn_cast<DeclaratorDecl>(D)) {
5685 if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
5686 StartLoc = TI->getTypeLoc().getLocStart();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005687 } else if (const TypedefDecl *Typedef = dyn_cast<TypedefDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005688 if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
5689 StartLoc = TI->getTypeLoc().getLocStart();
5690 }
5691
5692 if (StartLoc.isValid() && R.getBegin().isValid() &&
5693 SrcMgr.isBeforeInTranslationUnit(StartLoc, R.getBegin()))
5694 R.setBegin(StartLoc);
5695
5696 // FIXME: Multiple variables declared in a single declaration
5697 // currently lack the information needed to correctly determine their
5698 // ranges when accounting for the type-specifier. We use context
5699 // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
5700 // and if so, whether it is the first decl.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005701 if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005702 if (!cxcursor::isFirstInDeclGroup(C))
5703 R.setBegin(VD->getLocation());
5704 }
5705
5706 return R;
5707 }
5708
5709 return getRawCursorExtent(C);
5710}
5711
Guy Benyei11169dd2012-12-18 14:30:41 +00005712CXSourceRange clang_getCursorExtent(CXCursor C) {
5713 SourceRange R = getRawCursorExtent(C);
5714 if (R.isInvalid())
5715 return clang_getNullRange();
5716
5717 return cxloc::translateSourceRange(getCursorContext(C), R);
5718}
5719
5720CXCursor clang_getCursorReferenced(CXCursor C) {
5721 if (clang_isInvalid(C.kind))
5722 return clang_getNullCursor();
5723
5724 CXTranslationUnit tu = getCursorTU(C);
5725 if (clang_isDeclaration(C.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005726 const Decl *D = getCursorDecl(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005727 if (!D)
5728 return clang_getNullCursor();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005729 if (const UsingDecl *Using = dyn_cast<UsingDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00005730 return MakeCursorOverloadedDeclRef(Using, D->getLocation(), tu);
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005731 if (const ObjCPropertyImplDecl *PropImpl =
5732 dyn_cast<ObjCPropertyImplDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00005733 if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
5734 return MakeCXCursor(Property, tu);
5735
5736 return C;
5737 }
5738
5739 if (clang_isExpression(C.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005740 const Expr *E = getCursorExpr(C);
5741 const Decl *D = getDeclFromExpr(E);
Guy Benyei11169dd2012-12-18 14:30:41 +00005742 if (D) {
5743 CXCursor declCursor = MakeCXCursor(D, tu);
5744 declCursor = getSelectorIdentifierCursor(getSelectorIdentifierIndex(C),
5745 declCursor);
5746 return declCursor;
5747 }
5748
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005749 if (const OverloadExpr *Ovl = dyn_cast_or_null<OverloadExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00005750 return MakeCursorOverloadedDeclRef(Ovl, tu);
5751
5752 return clang_getNullCursor();
5753 }
5754
5755 if (clang_isStatement(C.kind)) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00005756 const Stmt *S = getCursorStmt(C);
5757 if (const GotoStmt *Goto = dyn_cast_or_null<GotoStmt>(S))
Guy Benyei11169dd2012-12-18 14:30:41 +00005758 if (LabelDecl *label = Goto->getLabel())
5759 if (LabelStmt *labelS = label->getStmt())
5760 return MakeCXCursor(labelS, getCursorDecl(C), tu);
5761
5762 return clang_getNullCursor();
5763 }
Richard Smith66a81862015-05-04 02:25:31 +00005764
Guy Benyei11169dd2012-12-18 14:30:41 +00005765 if (C.kind == CXCursor_MacroExpansion) {
Richard Smith66a81862015-05-04 02:25:31 +00005766 if (const MacroDefinitionRecord *Def =
5767 getCursorMacroExpansion(C).getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00005768 return MakeMacroDefinitionCursor(Def, tu);
5769 }
5770
5771 if (!clang_isReference(C.kind))
5772 return clang_getNullCursor();
5773
5774 switch (C.kind) {
5775 case CXCursor_ObjCSuperClassRef:
5776 return MakeCXCursor(getCursorObjCSuperClassRef(C).first, tu);
5777
5778 case CXCursor_ObjCProtocolRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005779 const ObjCProtocolDecl *Prot = getCursorObjCProtocolRef(C).first;
5780 if (const ObjCProtocolDecl *Def = Prot->getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00005781 return MakeCXCursor(Def, tu);
5782
5783 return MakeCXCursor(Prot, tu);
5784 }
5785
5786 case CXCursor_ObjCClassRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005787 const ObjCInterfaceDecl *Class = getCursorObjCClassRef(C).first;
5788 if (const ObjCInterfaceDecl *Def = Class->getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00005789 return MakeCXCursor(Def, tu);
5790
5791 return MakeCXCursor(Class, tu);
5792 }
5793
5794 case CXCursor_TypeRef:
5795 return MakeCXCursor(getCursorTypeRef(C).first, tu );
5796
5797 case CXCursor_TemplateRef:
5798 return MakeCXCursor(getCursorTemplateRef(C).first, tu );
5799
5800 case CXCursor_NamespaceRef:
5801 return MakeCXCursor(getCursorNamespaceRef(C).first, tu );
5802
5803 case CXCursor_MemberRef:
5804 return MakeCXCursor(getCursorMemberRef(C).first, tu );
5805
5806 case CXCursor_CXXBaseSpecifier: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005807 const CXXBaseSpecifier *B = cxcursor::getCursorCXXBaseSpecifier(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005808 return clang_getTypeDeclaration(cxtype::MakeCXType(B->getType(),
5809 tu ));
5810 }
5811
5812 case CXCursor_LabelRef:
5813 // FIXME: We end up faking the "parent" declaration here because we
5814 // don't want to make CXCursor larger.
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00005815 return MakeCXCursor(getCursorLabelRef(C).first,
5816 cxtu::getASTUnit(tu)->getASTContext()
5817 .getTranslationUnitDecl(),
Guy Benyei11169dd2012-12-18 14:30:41 +00005818 tu);
5819
5820 case CXCursor_OverloadedDeclRef:
5821 return C;
5822
5823 case CXCursor_VariableRef:
5824 return MakeCXCursor(getCursorVariableRef(C).first, tu);
5825
5826 default:
5827 // We would prefer to enumerate all non-reference cursor kinds here.
5828 llvm_unreachable("Unhandled reference cursor kind");
5829 }
5830}
5831
5832CXCursor clang_getCursorDefinition(CXCursor C) {
5833 if (clang_isInvalid(C.kind))
5834 return clang_getNullCursor();
5835
5836 CXTranslationUnit TU = getCursorTU(C);
5837
5838 bool WasReference = false;
5839 if (clang_isReference(C.kind) || clang_isExpression(C.kind)) {
5840 C = clang_getCursorReferenced(C);
5841 WasReference = true;
5842 }
5843
5844 if (C.kind == CXCursor_MacroExpansion)
5845 return clang_getCursorReferenced(C);
5846
5847 if (!clang_isDeclaration(C.kind))
5848 return clang_getNullCursor();
5849
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005850 const Decl *D = getCursorDecl(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005851 if (!D)
5852 return clang_getNullCursor();
5853
5854 switch (D->getKind()) {
5855 // Declaration kinds that don't really separate the notions of
5856 // declaration and definition.
5857 case Decl::Namespace:
5858 case Decl::Typedef:
5859 case Decl::TypeAlias:
5860 case Decl::TypeAliasTemplate:
5861 case Decl::TemplateTypeParm:
5862 case Decl::EnumConstant:
5863 case Decl::Field:
Richard Smithbdb84f32016-07-22 23:36:59 +00005864 case Decl::Binding:
John McCall5e77d762013-04-16 07:28:30 +00005865 case Decl::MSProperty:
Guy Benyei11169dd2012-12-18 14:30:41 +00005866 case Decl::IndirectField:
5867 case Decl::ObjCIvar:
5868 case Decl::ObjCAtDefsField:
5869 case Decl::ImplicitParam:
5870 case Decl::ParmVar:
5871 case Decl::NonTypeTemplateParm:
5872 case Decl::TemplateTemplateParm:
5873 case Decl::ObjCCategoryImpl:
5874 case Decl::ObjCImplementation:
5875 case Decl::AccessSpec:
5876 case Decl::LinkageSpec:
Richard Smith8df390f2016-09-08 23:14:54 +00005877 case Decl::Export:
Guy Benyei11169dd2012-12-18 14:30:41 +00005878 case Decl::ObjCPropertyImpl:
5879 case Decl::FileScopeAsm:
5880 case Decl::StaticAssert:
5881 case Decl::Block:
Tareq A. Siraj6dfa25a2013-04-16 19:37:38 +00005882 case Decl::Captured:
Alexey Bataev4244be22016-02-11 05:35:55 +00005883 case Decl::OMPCapturedExpr:
Guy Benyei11169dd2012-12-18 14:30:41 +00005884 case Decl::Label: // FIXME: Is this right??
5885 case Decl::ClassScopeFunctionSpecialization:
Richard Smithbc491202017-02-17 20:05:37 +00005886 case Decl::CXXDeductionGuide:
Guy Benyei11169dd2012-12-18 14:30:41 +00005887 case Decl::Import:
Alexey Bataeva769e072013-03-22 06:34:35 +00005888 case Decl::OMPThreadPrivate:
Alexey Bataev94a4f0c2016-03-03 05:21:39 +00005889 case Decl::OMPDeclareReduction:
Douglas Gregor85f3f952015-07-07 03:57:15 +00005890 case Decl::ObjCTypeParam:
David Majnemerd9b1a4f2015-11-04 03:40:30 +00005891 case Decl::BuiltinTemplate:
Nico Weber66220292016-03-02 17:28:48 +00005892 case Decl::PragmaComment:
Nico Webercbbaeb12016-03-02 19:28:54 +00005893 case Decl::PragmaDetectMismatch:
Richard Smith151c4562016-12-20 21:35:28 +00005894 case Decl::UsingPack:
Guy Benyei11169dd2012-12-18 14:30:41 +00005895 return C;
5896
5897 // Declaration kinds that don't make any sense here, but are
5898 // nonetheless harmless.
David Blaikief005d3c2013-02-22 17:44:58 +00005899 case Decl::Empty:
Guy Benyei11169dd2012-12-18 14:30:41 +00005900 case Decl::TranslationUnit:
Richard Smithf19e1272015-03-07 00:04:49 +00005901 case Decl::ExternCContext:
Guy Benyei11169dd2012-12-18 14:30:41 +00005902 break;
5903
5904 // Declaration kinds for which the definition is not resolvable.
5905 case Decl::UnresolvedUsingTypename:
5906 case Decl::UnresolvedUsingValue:
5907 break;
5908
5909 case Decl::UsingDirective:
5910 return MakeCXCursor(cast<UsingDirectiveDecl>(D)->getNominatedNamespace(),
5911 TU);
5912
5913 case Decl::NamespaceAlias:
5914 return MakeCXCursor(cast<NamespaceAliasDecl>(D)->getNamespace(), TU);
5915
5916 case Decl::Enum:
5917 case Decl::Record:
5918 case Decl::CXXRecord:
5919 case Decl::ClassTemplateSpecialization:
5920 case Decl::ClassTemplatePartialSpecialization:
5921 if (TagDecl *Def = cast<TagDecl>(D)->getDefinition())
5922 return MakeCXCursor(Def, TU);
5923 return clang_getNullCursor();
5924
5925 case Decl::Function:
5926 case Decl::CXXMethod:
5927 case Decl::CXXConstructor:
5928 case Decl::CXXDestructor:
5929 case Decl::CXXConversion: {
Craig Topper69186e72014-06-08 08:38:04 +00005930 const FunctionDecl *Def = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00005931 if (cast<FunctionDecl>(D)->getBody(Def))
Dmitri Gribenko9c256e32013-01-14 00:46:27 +00005932 return MakeCXCursor(Def, TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00005933 return clang_getNullCursor();
5934 }
5935
Larisse Voufo39a1e502013-08-06 01:03:05 +00005936 case Decl::Var:
5937 case Decl::VarTemplateSpecialization:
Richard Smithbdb84f32016-07-22 23:36:59 +00005938 case Decl::VarTemplatePartialSpecialization:
5939 case Decl::Decomposition: {
Guy Benyei11169dd2012-12-18 14:30:41 +00005940 // Ask the variable if it has a definition.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005941 if (const VarDecl *Def = cast<VarDecl>(D)->getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00005942 return MakeCXCursor(Def, TU);
5943 return clang_getNullCursor();
5944 }
5945
5946 case Decl::FunctionTemplate: {
Craig Topper69186e72014-06-08 08:38:04 +00005947 const FunctionDecl *Def = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00005948 if (cast<FunctionTemplateDecl>(D)->getTemplatedDecl()->getBody(Def))
5949 return MakeCXCursor(Def->getDescribedFunctionTemplate(), TU);
5950 return clang_getNullCursor();
5951 }
5952
5953 case Decl::ClassTemplate: {
5954 if (RecordDecl *Def = cast<ClassTemplateDecl>(D)->getTemplatedDecl()
5955 ->getDefinition())
5956 return MakeCXCursor(cast<CXXRecordDecl>(Def)->getDescribedClassTemplate(),
5957 TU);
5958 return clang_getNullCursor();
5959 }
5960
Larisse Voufo39a1e502013-08-06 01:03:05 +00005961 case Decl::VarTemplate: {
5962 if (VarDecl *Def =
5963 cast<VarTemplateDecl>(D)->getTemplatedDecl()->getDefinition())
5964 return MakeCXCursor(cast<VarDecl>(Def)->getDescribedVarTemplate(), TU);
5965 return clang_getNullCursor();
5966 }
5967
Guy Benyei11169dd2012-12-18 14:30:41 +00005968 case Decl::Using:
5969 return MakeCursorOverloadedDeclRef(cast<UsingDecl>(D),
5970 D->getLocation(), TU);
5971
5972 case Decl::UsingShadow:
Richard Smith5179eb72016-06-28 19:03:57 +00005973 case Decl::ConstructorUsingShadow:
Guy Benyei11169dd2012-12-18 14:30:41 +00005974 return clang_getCursorDefinition(
5975 MakeCXCursor(cast<UsingShadowDecl>(D)->getTargetDecl(),
5976 TU));
5977
5978 case Decl::ObjCMethod: {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005979 const ObjCMethodDecl *Method = cast<ObjCMethodDecl>(D);
Guy Benyei11169dd2012-12-18 14:30:41 +00005980 if (Method->isThisDeclarationADefinition())
5981 return C;
5982
5983 // Dig out the method definition in the associated
5984 // @implementation, if we have it.
5985 // FIXME: The ASTs should make finding the definition easier.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005986 if (const ObjCInterfaceDecl *Class
Guy Benyei11169dd2012-12-18 14:30:41 +00005987 = dyn_cast<ObjCInterfaceDecl>(Method->getDeclContext()))
5988 if (ObjCImplementationDecl *ClassImpl = Class->getImplementation())
5989 if (ObjCMethodDecl *Def = ClassImpl->getMethod(Method->getSelector(),
5990 Method->isInstanceMethod()))
5991 if (Def->isThisDeclarationADefinition())
5992 return MakeCXCursor(Def, TU);
5993
5994 return clang_getNullCursor();
5995 }
5996
5997 case Decl::ObjCCategory:
5998 if (ObjCCategoryImplDecl *Impl
5999 = cast<ObjCCategoryDecl>(D)->getImplementation())
6000 return MakeCXCursor(Impl, TU);
6001 return clang_getNullCursor();
6002
6003 case Decl::ObjCProtocol:
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006004 if (const ObjCProtocolDecl *Def = cast<ObjCProtocolDecl>(D)->getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00006005 return MakeCXCursor(Def, TU);
6006 return clang_getNullCursor();
6007
6008 case Decl::ObjCInterface: {
6009 // There are two notions of a "definition" for an Objective-C
6010 // class: the interface and its implementation. When we resolved a
6011 // reference to an Objective-C class, produce the @interface as
6012 // the definition; when we were provided with the interface,
6013 // produce the @implementation as the definition.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006014 const ObjCInterfaceDecl *IFace = cast<ObjCInterfaceDecl>(D);
Guy Benyei11169dd2012-12-18 14:30:41 +00006015 if (WasReference) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006016 if (const ObjCInterfaceDecl *Def = IFace->getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00006017 return MakeCXCursor(Def, TU);
6018 } else if (ObjCImplementationDecl *Impl = IFace->getImplementation())
6019 return MakeCXCursor(Impl, TU);
6020 return clang_getNullCursor();
6021 }
6022
6023 case Decl::ObjCProperty:
6024 // FIXME: We don't really know where to find the
6025 // ObjCPropertyImplDecls that implement this property.
6026 return clang_getNullCursor();
6027
6028 case Decl::ObjCCompatibleAlias:
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006029 if (const ObjCInterfaceDecl *Class
Guy Benyei11169dd2012-12-18 14:30:41 +00006030 = cast<ObjCCompatibleAliasDecl>(D)->getClassInterface())
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006031 if (const ObjCInterfaceDecl *Def = Class->getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00006032 return MakeCXCursor(Def, TU);
6033
6034 return clang_getNullCursor();
6035
6036 case Decl::Friend:
6037 if (NamedDecl *Friend = cast<FriendDecl>(D)->getFriendDecl())
6038 return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
6039 return clang_getNullCursor();
6040
6041 case Decl::FriendTemplate:
6042 if (NamedDecl *Friend = cast<FriendTemplateDecl>(D)->getFriendDecl())
6043 return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
6044 return clang_getNullCursor();
6045 }
6046
6047 return clang_getNullCursor();
6048}
6049
6050unsigned clang_isCursorDefinition(CXCursor C) {
6051 if (!clang_isDeclaration(C.kind))
6052 return 0;
6053
6054 return clang_getCursorDefinition(C) == C;
6055}
6056
6057CXCursor clang_getCanonicalCursor(CXCursor C) {
6058 if (!clang_isDeclaration(C.kind))
6059 return C;
6060
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006061 if (const Decl *D = getCursorDecl(C)) {
6062 if (const ObjCCategoryImplDecl *CatImplD = dyn_cast<ObjCCategoryImplDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00006063 if (ObjCCategoryDecl *CatD = CatImplD->getCategoryDecl())
6064 return MakeCXCursor(CatD, getCursorTU(C));
6065
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006066 if (const ObjCImplDecl *ImplD = dyn_cast<ObjCImplDecl>(D))
6067 if (const ObjCInterfaceDecl *IFD = ImplD->getClassInterface())
Guy Benyei11169dd2012-12-18 14:30:41 +00006068 return MakeCXCursor(IFD, getCursorTU(C));
6069
6070 return MakeCXCursor(D->getCanonicalDecl(), getCursorTU(C));
6071 }
6072
6073 return C;
6074}
6075
6076int clang_Cursor_getObjCSelectorIndex(CXCursor cursor) {
6077 return cxcursor::getSelectorIdentifierIndexAndLoc(cursor).first;
6078}
6079
6080unsigned clang_getNumOverloadedDecls(CXCursor C) {
6081 if (C.kind != CXCursor_OverloadedDeclRef)
6082 return 0;
6083
6084 OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006085 if (const OverloadExpr *E = Storage.dyn_cast<const OverloadExpr *>())
Guy Benyei11169dd2012-12-18 14:30:41 +00006086 return E->getNumDecls();
6087
6088 if (OverloadedTemplateStorage *S
6089 = Storage.dyn_cast<OverloadedTemplateStorage*>())
6090 return S->size();
6091
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006092 const Decl *D = Storage.get<const Decl *>();
6093 if (const UsingDecl *Using = dyn_cast<UsingDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00006094 return Using->shadow_size();
6095
6096 return 0;
6097}
6098
6099CXCursor clang_getOverloadedDecl(CXCursor cursor, unsigned index) {
6100 if (cursor.kind != CXCursor_OverloadedDeclRef)
6101 return clang_getNullCursor();
6102
6103 if (index >= clang_getNumOverloadedDecls(cursor))
6104 return clang_getNullCursor();
6105
6106 CXTranslationUnit TU = getCursorTU(cursor);
6107 OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(cursor).first;
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006108 if (const OverloadExpr *E = Storage.dyn_cast<const OverloadExpr *>())
Guy Benyei11169dd2012-12-18 14:30:41 +00006109 return MakeCXCursor(E->decls_begin()[index], TU);
6110
6111 if (OverloadedTemplateStorage *S
6112 = Storage.dyn_cast<OverloadedTemplateStorage*>())
6113 return MakeCXCursor(S->begin()[index], TU);
6114
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006115 const Decl *D = Storage.get<const Decl *>();
6116 if (const UsingDecl *Using = dyn_cast<UsingDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00006117 // FIXME: This is, unfortunately, linear time.
6118 UsingDecl::shadow_iterator Pos = Using->shadow_begin();
6119 std::advance(Pos, index);
6120 return MakeCXCursor(cast<UsingShadowDecl>(*Pos)->getTargetDecl(), TU);
6121 }
6122
6123 return clang_getNullCursor();
6124}
6125
6126void clang_getDefinitionSpellingAndExtent(CXCursor C,
6127 const char **startBuf,
6128 const char **endBuf,
6129 unsigned *startLine,
6130 unsigned *startColumn,
6131 unsigned *endLine,
6132 unsigned *endColumn) {
6133 assert(getCursorDecl(C) && "CXCursor has null decl");
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006134 const FunctionDecl *FD = dyn_cast<FunctionDecl>(getCursorDecl(C));
Guy Benyei11169dd2012-12-18 14:30:41 +00006135 CompoundStmt *Body = dyn_cast<CompoundStmt>(FD->getBody());
6136
6137 SourceManager &SM = FD->getASTContext().getSourceManager();
6138 *startBuf = SM.getCharacterData(Body->getLBracLoc());
6139 *endBuf = SM.getCharacterData(Body->getRBracLoc());
6140 *startLine = SM.getSpellingLineNumber(Body->getLBracLoc());
6141 *startColumn = SM.getSpellingColumnNumber(Body->getLBracLoc());
6142 *endLine = SM.getSpellingLineNumber(Body->getRBracLoc());
6143 *endColumn = SM.getSpellingColumnNumber(Body->getRBracLoc());
6144}
6145
6146
6147CXSourceRange clang_getCursorReferenceNameRange(CXCursor C, unsigned NameFlags,
6148 unsigned PieceIndex) {
6149 RefNamePieces Pieces;
6150
6151 switch (C.kind) {
6152 case CXCursor_MemberRefExpr:
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00006153 if (const MemberExpr *E = dyn_cast<MemberExpr>(getCursorExpr(C)))
Guy Benyei11169dd2012-12-18 14:30:41 +00006154 Pieces = buildPieces(NameFlags, true, E->getMemberNameInfo(),
6155 E->getQualifierLoc().getSourceRange());
6156 break;
6157
6158 case CXCursor_DeclRefExpr:
James Y Knight04ec5bf2015-12-24 02:59:37 +00006159 if (const DeclRefExpr *E = dyn_cast<DeclRefExpr>(getCursorExpr(C))) {
6160 SourceRange TemplateArgLoc(E->getLAngleLoc(), E->getRAngleLoc());
6161 Pieces =
6162 buildPieces(NameFlags, false, E->getNameInfo(),
6163 E->getQualifierLoc().getSourceRange(), &TemplateArgLoc);
6164 }
Guy Benyei11169dd2012-12-18 14:30:41 +00006165 break;
6166
6167 case CXCursor_CallExpr:
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00006168 if (const CXXOperatorCallExpr *OCE =
Guy Benyei11169dd2012-12-18 14:30:41 +00006169 dyn_cast<CXXOperatorCallExpr>(getCursorExpr(C))) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00006170 const Expr *Callee = OCE->getCallee();
6171 if (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Callee))
Guy Benyei11169dd2012-12-18 14:30:41 +00006172 Callee = ICE->getSubExpr();
6173
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00006174 if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Callee))
Guy Benyei11169dd2012-12-18 14:30:41 +00006175 Pieces = buildPieces(NameFlags, false, DRE->getNameInfo(),
6176 DRE->getQualifierLoc().getSourceRange());
6177 }
6178 break;
6179
6180 default:
6181 break;
6182 }
6183
6184 if (Pieces.empty()) {
6185 if (PieceIndex == 0)
6186 return clang_getCursorExtent(C);
6187 } else if (PieceIndex < Pieces.size()) {
6188 SourceRange R = Pieces[PieceIndex];
6189 if (R.isValid())
6190 return cxloc::translateSourceRange(getCursorContext(C), R);
6191 }
6192
6193 return clang_getNullRange();
6194}
6195
6196void clang_enableStackTraces(void) {
Richard Smithdfed58a2016-06-09 00:53:41 +00006197 // FIXME: Provide an argv0 here so we can find llvm-symbolizer.
6198 llvm::sys::PrintStackTraceOnErrorSignal(StringRef());
Guy Benyei11169dd2012-12-18 14:30:41 +00006199}
6200
6201void clang_executeOnThread(void (*fn)(void*), void *user_data,
6202 unsigned stack_size) {
6203 llvm::llvm_execute_on_thread(fn, user_data, stack_size);
6204}
6205
Guy Benyei11169dd2012-12-18 14:30:41 +00006206//===----------------------------------------------------------------------===//
6207// Token-based Operations.
6208//===----------------------------------------------------------------------===//
6209
6210/* CXToken layout:
6211 * int_data[0]: a CXTokenKind
6212 * int_data[1]: starting token location
6213 * int_data[2]: token length
6214 * int_data[3]: reserved
6215 * ptr_data: for identifiers and keywords, an IdentifierInfo*.
6216 * otherwise unused.
6217 */
Guy Benyei11169dd2012-12-18 14:30:41 +00006218CXTokenKind clang_getTokenKind(CXToken CXTok) {
6219 return static_cast<CXTokenKind>(CXTok.int_data[0]);
6220}
6221
6222CXString clang_getTokenSpelling(CXTranslationUnit TU, CXToken CXTok) {
6223 switch (clang_getTokenKind(CXTok)) {
6224 case CXToken_Identifier:
6225 case CXToken_Keyword:
6226 // We know we have an IdentifierInfo*, so use that.
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00006227 return cxstring::createRef(static_cast<IdentifierInfo *>(CXTok.ptr_data)
Guy Benyei11169dd2012-12-18 14:30:41 +00006228 ->getNameStart());
6229
6230 case CXToken_Literal: {
6231 // We have stashed the starting pointer in the ptr_data field. Use it.
6232 const char *Text = static_cast<const char *>(CXTok.ptr_data);
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00006233 return cxstring::createDup(StringRef(Text, CXTok.int_data[2]));
Guy Benyei11169dd2012-12-18 14:30:41 +00006234 }
6235
6236 case CXToken_Punctuation:
6237 case CXToken_Comment:
6238 break;
6239 }
6240
Dmitri Gribenko852d6222014-02-11 15:02:48 +00006241 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00006242 LOG_BAD_TU(TU);
6243 return cxstring::createEmpty();
6244 }
6245
Guy Benyei11169dd2012-12-18 14:30:41 +00006246 // We have to find the starting buffer pointer the hard way, by
6247 // deconstructing the source location.
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006248 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00006249 if (!CXXUnit)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00006250 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00006251
6252 SourceLocation Loc = SourceLocation::getFromRawEncoding(CXTok.int_data[1]);
6253 std::pair<FileID, unsigned> LocInfo
6254 = CXXUnit->getSourceManager().getDecomposedSpellingLoc(Loc);
6255 bool Invalid = false;
6256 StringRef Buffer
6257 = CXXUnit->getSourceManager().getBufferData(LocInfo.first, &Invalid);
6258 if (Invalid)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00006259 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00006260
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00006261 return cxstring::createDup(Buffer.substr(LocInfo.second, CXTok.int_data[2]));
Guy Benyei11169dd2012-12-18 14:30:41 +00006262}
6263
6264CXSourceLocation clang_getTokenLocation(CXTranslationUnit TU, CXToken CXTok) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00006265 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00006266 LOG_BAD_TU(TU);
6267 return clang_getNullLocation();
6268 }
6269
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006270 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00006271 if (!CXXUnit)
6272 return clang_getNullLocation();
6273
6274 return cxloc::translateSourceLocation(CXXUnit->getASTContext(),
6275 SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
6276}
6277
6278CXSourceRange clang_getTokenExtent(CXTranslationUnit TU, CXToken CXTok) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00006279 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00006280 LOG_BAD_TU(TU);
6281 return clang_getNullRange();
6282 }
6283
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006284 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00006285 if (!CXXUnit)
6286 return clang_getNullRange();
6287
6288 return cxloc::translateSourceRange(CXXUnit->getASTContext(),
6289 SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
6290}
6291
6292static void getTokens(ASTUnit *CXXUnit, SourceRange Range,
6293 SmallVectorImpl<CXToken> &CXTokens) {
6294 SourceManager &SourceMgr = CXXUnit->getSourceManager();
6295 std::pair<FileID, unsigned> BeginLocInfo
Argyrios Kyrtzidis5d47a9b2013-02-13 18:33:28 +00006296 = SourceMgr.getDecomposedSpellingLoc(Range.getBegin());
Guy Benyei11169dd2012-12-18 14:30:41 +00006297 std::pair<FileID, unsigned> EndLocInfo
Argyrios Kyrtzidis5d47a9b2013-02-13 18:33:28 +00006298 = SourceMgr.getDecomposedSpellingLoc(Range.getEnd());
Guy Benyei11169dd2012-12-18 14:30:41 +00006299
6300 // Cannot tokenize across files.
6301 if (BeginLocInfo.first != EndLocInfo.first)
6302 return;
6303
6304 // Create a lexer
6305 bool Invalid = false;
6306 StringRef Buffer
6307 = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
6308 if (Invalid)
6309 return;
6310
6311 Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
6312 CXXUnit->getASTContext().getLangOpts(),
6313 Buffer.begin(), Buffer.data() + BeginLocInfo.second, Buffer.end());
6314 Lex.SetCommentRetentionState(true);
6315
6316 // Lex tokens until we hit the end of the range.
6317 const char *EffectiveBufferEnd = Buffer.data() + EndLocInfo.second;
6318 Token Tok;
6319 bool previousWasAt = false;
6320 do {
6321 // Lex the next token
6322 Lex.LexFromRawLexer(Tok);
6323 if (Tok.is(tok::eof))
6324 break;
6325
6326 // Initialize the CXToken.
6327 CXToken CXTok;
6328
6329 // - Common fields
6330 CXTok.int_data[1] = Tok.getLocation().getRawEncoding();
6331 CXTok.int_data[2] = Tok.getLength();
6332 CXTok.int_data[3] = 0;
6333
6334 // - Kind-specific fields
6335 if (Tok.isLiteral()) {
6336 CXTok.int_data[0] = CXToken_Literal;
Dmitri Gribenkof9304482013-01-23 15:56:07 +00006337 CXTok.ptr_data = const_cast<char *>(Tok.getLiteralData());
Guy Benyei11169dd2012-12-18 14:30:41 +00006338 } else if (Tok.is(tok::raw_identifier)) {
6339 // Lookup the identifier to determine whether we have a keyword.
6340 IdentifierInfo *II
6341 = CXXUnit->getPreprocessor().LookUpIdentifierInfo(Tok);
6342
6343 if ((II->getObjCKeywordID() != tok::objc_not_keyword) && previousWasAt) {
6344 CXTok.int_data[0] = CXToken_Keyword;
6345 }
6346 else {
6347 CXTok.int_data[0] = Tok.is(tok::identifier)
6348 ? CXToken_Identifier
6349 : CXToken_Keyword;
6350 }
6351 CXTok.ptr_data = II;
6352 } else if (Tok.is(tok::comment)) {
6353 CXTok.int_data[0] = CXToken_Comment;
Craig Topper69186e72014-06-08 08:38:04 +00006354 CXTok.ptr_data = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00006355 } else {
6356 CXTok.int_data[0] = CXToken_Punctuation;
Craig Topper69186e72014-06-08 08:38:04 +00006357 CXTok.ptr_data = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00006358 }
6359 CXTokens.push_back(CXTok);
6360 previousWasAt = Tok.is(tok::at);
Argyrios Kyrtzidisc7c6a072016-11-09 23:58:39 +00006361 } while (Lex.getBufferLocation() < EffectiveBufferEnd);
Guy Benyei11169dd2012-12-18 14:30:41 +00006362}
6363
6364void clang_tokenize(CXTranslationUnit TU, CXSourceRange Range,
6365 CXToken **Tokens, unsigned *NumTokens) {
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00006366 LOG_FUNC_SECTION {
6367 *Log << TU << ' ' << Range;
6368 }
6369
Guy Benyei11169dd2012-12-18 14:30:41 +00006370 if (Tokens)
Craig Topper69186e72014-06-08 08:38:04 +00006371 *Tokens = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00006372 if (NumTokens)
6373 *NumTokens = 0;
6374
Dmitri Gribenko852d6222014-02-11 15:02:48 +00006375 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00006376 LOG_BAD_TU(TU);
Argyrios Kyrtzidis0e95fca2013-04-04 22:40:59 +00006377 return;
Dmitri Gribenko256454f2014-02-11 14:34:14 +00006378 }
Argyrios Kyrtzidis0e95fca2013-04-04 22:40:59 +00006379
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006380 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00006381 if (!CXXUnit || !Tokens || !NumTokens)
6382 return;
6383
6384 ASTUnit::ConcurrencyCheck Check(*CXXUnit);
6385
6386 SourceRange R = cxloc::translateCXSourceRange(Range);
6387 if (R.isInvalid())
6388 return;
6389
6390 SmallVector<CXToken, 32> CXTokens;
6391 getTokens(CXXUnit, R, CXTokens);
6392
6393 if (CXTokens.empty())
6394 return;
6395
6396 *Tokens = (CXToken *)malloc(sizeof(CXToken) * CXTokens.size());
6397 memmove(*Tokens, CXTokens.data(), sizeof(CXToken) * CXTokens.size());
6398 *NumTokens = CXTokens.size();
6399}
6400
6401void clang_disposeTokens(CXTranslationUnit TU,
6402 CXToken *Tokens, unsigned NumTokens) {
6403 free(Tokens);
6404}
6405
Guy Benyei11169dd2012-12-18 14:30:41 +00006406//===----------------------------------------------------------------------===//
6407// Token annotation APIs.
6408//===----------------------------------------------------------------------===//
6409
Guy Benyei11169dd2012-12-18 14:30:41 +00006410static enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
6411 CXCursor parent,
6412 CXClientData client_data);
6413static bool AnnotateTokensPostChildrenVisitor(CXCursor cursor,
6414 CXClientData client_data);
6415
6416namespace {
6417class AnnotateTokensWorker {
Guy Benyei11169dd2012-12-18 14:30:41 +00006418 CXToken *Tokens;
6419 CXCursor *Cursors;
6420 unsigned NumTokens;
6421 unsigned TokIdx;
6422 unsigned PreprocessingTokIdx;
6423 CursorVisitor AnnotateVis;
6424 SourceManager &SrcMgr;
6425 bool HasContextSensitiveKeywords;
6426
6427 struct PostChildrenInfo {
6428 CXCursor Cursor;
6429 SourceRange CursorRange;
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006430 unsigned BeforeReachingCursorIdx;
Guy Benyei11169dd2012-12-18 14:30:41 +00006431 unsigned BeforeChildrenTokenIdx;
6432 };
Dmitri Gribenkof8579502013-01-12 19:30:44 +00006433 SmallVector<PostChildrenInfo, 8> PostChildrenInfos;
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006434
6435 CXToken &getTok(unsigned Idx) {
6436 assert(Idx < NumTokens);
6437 return Tokens[Idx];
6438 }
6439 const CXToken &getTok(unsigned Idx) const {
6440 assert(Idx < NumTokens);
6441 return Tokens[Idx];
6442 }
Guy Benyei11169dd2012-12-18 14:30:41 +00006443 bool MoreTokens() const { return TokIdx < NumTokens; }
6444 unsigned NextToken() const { return TokIdx; }
6445 void AdvanceToken() { ++TokIdx; }
6446 SourceLocation GetTokenLoc(unsigned tokI) {
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006447 return SourceLocation::getFromRawEncoding(getTok(tokI).int_data[1]);
Guy Benyei11169dd2012-12-18 14:30:41 +00006448 }
6449 bool isFunctionMacroToken(unsigned tokI) const {
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006450 return getTok(tokI).int_data[3] != 0;
Guy Benyei11169dd2012-12-18 14:30:41 +00006451 }
6452 SourceLocation getFunctionMacroTokenLoc(unsigned tokI) const {
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006453 return SourceLocation::getFromRawEncoding(getTok(tokI).int_data[3]);
Guy Benyei11169dd2012-12-18 14:30:41 +00006454 }
6455
6456 void annotateAndAdvanceTokens(CXCursor, RangeComparisonResult, SourceRange);
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006457 bool annotateAndAdvanceFunctionMacroTokens(CXCursor, RangeComparisonResult,
Guy Benyei11169dd2012-12-18 14:30:41 +00006458 SourceRange);
6459
6460public:
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006461 AnnotateTokensWorker(CXToken *tokens, CXCursor *cursors, unsigned numTokens,
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006462 CXTranslationUnit TU, SourceRange RegionOfInterest)
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006463 : Tokens(tokens), Cursors(cursors),
Guy Benyei11169dd2012-12-18 14:30:41 +00006464 NumTokens(numTokens), TokIdx(0), PreprocessingTokIdx(0),
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006465 AnnotateVis(TU,
Guy Benyei11169dd2012-12-18 14:30:41 +00006466 AnnotateTokensVisitor, this,
6467 /*VisitPreprocessorLast=*/true,
6468 /*VisitIncludedEntities=*/false,
6469 RegionOfInterest,
6470 /*VisitDeclsOnly=*/false,
6471 AnnotateTokensPostChildrenVisitor),
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006472 SrcMgr(cxtu::getASTUnit(TU)->getSourceManager()),
Guy Benyei11169dd2012-12-18 14:30:41 +00006473 HasContextSensitiveKeywords(false) { }
6474
6475 void VisitChildren(CXCursor C) { AnnotateVis.VisitChildren(C); }
6476 enum CXChildVisitResult Visit(CXCursor cursor, CXCursor parent);
6477 bool postVisitChildren(CXCursor cursor);
6478 void AnnotateTokens();
6479
6480 /// \brief Determine whether the annotator saw any cursors that have
6481 /// context-sensitive keywords.
6482 bool hasContextSensitiveKeywords() const {
6483 return HasContextSensitiveKeywords;
6484 }
6485
6486 ~AnnotateTokensWorker() {
6487 assert(PostChildrenInfos.empty());
6488 }
6489};
Alexander Kornienkoab9db512015-06-22 23:07:51 +00006490}
Guy Benyei11169dd2012-12-18 14:30:41 +00006491
6492void AnnotateTokensWorker::AnnotateTokens() {
6493 // Walk the AST within the region of interest, annotating tokens
6494 // along the way.
6495 AnnotateVis.visitFileRegion();
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006496}
Guy Benyei11169dd2012-12-18 14:30:41 +00006497
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006498static inline void updateCursorAnnotation(CXCursor &Cursor,
6499 const CXCursor &updateC) {
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006500 if (clang_isInvalid(updateC.kind) || !clang_isInvalid(Cursor.kind))
Guy Benyei11169dd2012-12-18 14:30:41 +00006501 return;
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006502 Cursor = updateC;
Guy Benyei11169dd2012-12-18 14:30:41 +00006503}
6504
6505/// \brief It annotates and advances tokens with a cursor until the comparison
6506//// between the cursor location and the source range is the same as
6507/// \arg compResult.
6508///
6509/// Pass RangeBefore to annotate tokens with a cursor until a range is reached.
6510/// Pass RangeOverlap to annotate tokens inside a range.
6511void AnnotateTokensWorker::annotateAndAdvanceTokens(CXCursor updateC,
6512 RangeComparisonResult compResult,
6513 SourceRange range) {
6514 while (MoreTokens()) {
6515 const unsigned I = NextToken();
6516 if (isFunctionMacroToken(I))
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006517 if (!annotateAndAdvanceFunctionMacroTokens(updateC, compResult, range))
6518 return;
Guy Benyei11169dd2012-12-18 14:30:41 +00006519
6520 SourceLocation TokLoc = GetTokenLoc(I);
6521 if (LocationCompare(SrcMgr, TokLoc, range) == compResult) {
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006522 updateCursorAnnotation(Cursors[I], updateC);
Guy Benyei11169dd2012-12-18 14:30:41 +00006523 AdvanceToken();
6524 continue;
6525 }
6526 break;
6527 }
6528}
6529
6530/// \brief Special annotation handling for macro argument tokens.
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006531/// \returns true if it advanced beyond all macro tokens, false otherwise.
6532bool AnnotateTokensWorker::annotateAndAdvanceFunctionMacroTokens(
Guy Benyei11169dd2012-12-18 14:30:41 +00006533 CXCursor updateC,
6534 RangeComparisonResult compResult,
6535 SourceRange range) {
6536 assert(MoreTokens());
6537 assert(isFunctionMacroToken(NextToken()) &&
6538 "Should be called only for macro arg tokens");
6539
6540 // This works differently than annotateAndAdvanceTokens; because expanded
6541 // macro arguments can have arbitrary translation-unit source order, we do not
6542 // advance the token index one by one until a token fails the range test.
6543 // We only advance once past all of the macro arg tokens if all of them
6544 // pass the range test. If one of them fails we keep the token index pointing
6545 // at the start of the macro arg tokens so that the failing token will be
6546 // annotated by a subsequent annotation try.
6547
6548 bool atLeastOneCompFail = false;
6549
6550 unsigned I = NextToken();
6551 for (; I < NumTokens && isFunctionMacroToken(I); ++I) {
6552 SourceLocation TokLoc = getFunctionMacroTokenLoc(I);
6553 if (TokLoc.isFileID())
6554 continue; // not macro arg token, it's parens or comma.
6555 if (LocationCompare(SrcMgr, TokLoc, range) == compResult) {
6556 if (clang_isInvalid(clang_getCursorKind(Cursors[I])))
6557 Cursors[I] = updateC;
6558 } else
6559 atLeastOneCompFail = true;
6560 }
6561
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006562 if (atLeastOneCompFail)
6563 return false;
6564
6565 TokIdx = I; // All of the tokens were handled, advance beyond all of them.
6566 return true;
Guy Benyei11169dd2012-12-18 14:30:41 +00006567}
6568
6569enum CXChildVisitResult
6570AnnotateTokensWorker::Visit(CXCursor cursor, CXCursor parent) {
Guy Benyei11169dd2012-12-18 14:30:41 +00006571 SourceRange cursorRange = getRawCursorExtent(cursor);
6572 if (cursorRange.isInvalid())
6573 return CXChildVisit_Recurse;
6574
6575 if (!HasContextSensitiveKeywords) {
6576 // Objective-C properties can have context-sensitive keywords.
6577 if (cursor.kind == CXCursor_ObjCPropertyDecl) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006578 if (const ObjCPropertyDecl *Property
Guy Benyei11169dd2012-12-18 14:30:41 +00006579 = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(cursor)))
6580 HasContextSensitiveKeywords = Property->getPropertyAttributesAsWritten() != 0;
6581 }
6582 // Objective-C methods can have context-sensitive keywords.
6583 else if (cursor.kind == CXCursor_ObjCInstanceMethodDecl ||
6584 cursor.kind == CXCursor_ObjCClassMethodDecl) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006585 if (const ObjCMethodDecl *Method
Guy Benyei11169dd2012-12-18 14:30:41 +00006586 = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(cursor))) {
6587 if (Method->getObjCDeclQualifier())
6588 HasContextSensitiveKeywords = true;
6589 else {
David Majnemer59f77922016-06-24 04:05:48 +00006590 for (const auto *P : Method->parameters()) {
Aaron Ballman43b68be2014-03-07 17:50:17 +00006591 if (P->getObjCDeclQualifier()) {
Guy Benyei11169dd2012-12-18 14:30:41 +00006592 HasContextSensitiveKeywords = true;
6593 break;
6594 }
6595 }
6596 }
6597 }
6598 }
6599 // C++ methods can have context-sensitive keywords.
6600 else if (cursor.kind == CXCursor_CXXMethod) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006601 if (const CXXMethodDecl *Method
Guy Benyei11169dd2012-12-18 14:30:41 +00006602 = dyn_cast_or_null<CXXMethodDecl>(getCursorDecl(cursor))) {
6603 if (Method->hasAttr<FinalAttr>() || Method->hasAttr<OverrideAttr>())
6604 HasContextSensitiveKeywords = true;
6605 }
6606 }
6607 // C++ classes can have context-sensitive keywords.
6608 else if (cursor.kind == CXCursor_StructDecl ||
6609 cursor.kind == CXCursor_ClassDecl ||
6610 cursor.kind == CXCursor_ClassTemplate ||
6611 cursor.kind == CXCursor_ClassTemplatePartialSpecialization) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006612 if (const Decl *D = getCursorDecl(cursor))
Guy Benyei11169dd2012-12-18 14:30:41 +00006613 if (D->hasAttr<FinalAttr>())
6614 HasContextSensitiveKeywords = true;
6615 }
6616 }
Argyrios Kyrtzidis990b3862013-06-04 18:24:30 +00006617
6618 // Don't override a property annotation with its getter/setter method.
6619 if (cursor.kind == CXCursor_ObjCInstanceMethodDecl &&
6620 parent.kind == CXCursor_ObjCPropertyDecl)
6621 return CXChildVisit_Continue;
Guy Benyei11169dd2012-12-18 14:30:41 +00006622
6623 if (clang_isPreprocessing(cursor.kind)) {
6624 // Items in the preprocessing record are kept separate from items in
6625 // declarations, so we keep a separate token index.
6626 unsigned SavedTokIdx = TokIdx;
6627 TokIdx = PreprocessingTokIdx;
6628
6629 // Skip tokens up until we catch up to the beginning of the preprocessing
6630 // entry.
6631 while (MoreTokens()) {
6632 const unsigned I = NextToken();
6633 SourceLocation TokLoc = GetTokenLoc(I);
6634 switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
6635 case RangeBefore:
6636 AdvanceToken();
6637 continue;
6638 case RangeAfter:
6639 case RangeOverlap:
6640 break;
6641 }
6642 break;
6643 }
6644
6645 // Look at all of the tokens within this range.
6646 while (MoreTokens()) {
6647 const unsigned I = NextToken();
6648 SourceLocation TokLoc = GetTokenLoc(I);
6649 switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
6650 case RangeBefore:
6651 llvm_unreachable("Infeasible");
6652 case RangeAfter:
6653 break;
6654 case RangeOverlap:
Argyrios Kyrtzidis5d47a9b2013-02-13 18:33:28 +00006655 // For macro expansions, just note where the beginning of the macro
6656 // expansion occurs.
6657 if (cursor.kind == CXCursor_MacroExpansion) {
6658 if (TokLoc == cursorRange.getBegin())
6659 Cursors[I] = cursor;
6660 AdvanceToken();
6661 break;
6662 }
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006663 // We may have already annotated macro names inside macro definitions.
6664 if (Cursors[I].kind != CXCursor_MacroExpansion)
6665 Cursors[I] = cursor;
Guy Benyei11169dd2012-12-18 14:30:41 +00006666 AdvanceToken();
Guy Benyei11169dd2012-12-18 14:30:41 +00006667 continue;
6668 }
6669 break;
6670 }
6671
6672 // Save the preprocessing token index; restore the non-preprocessing
6673 // token index.
6674 PreprocessingTokIdx = TokIdx;
6675 TokIdx = SavedTokIdx;
6676 return CXChildVisit_Recurse;
6677 }
6678
6679 if (cursorRange.isInvalid())
6680 return CXChildVisit_Continue;
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006681
6682 unsigned BeforeReachingCursorIdx = NextToken();
Guy Benyei11169dd2012-12-18 14:30:41 +00006683 const enum CXCursorKind cursorK = clang_getCursorKind(cursor);
Guy Benyei11169dd2012-12-18 14:30:41 +00006684 const enum CXCursorKind K = clang_getCursorKind(parent);
6685 const CXCursor updateC =
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006686 (clang_isInvalid(K) || K == CXCursor_TranslationUnit ||
6687 // Attributes are annotated out-of-order, skip tokens until we reach it.
6688 clang_isAttribute(cursor.kind))
Guy Benyei11169dd2012-12-18 14:30:41 +00006689 ? clang_getNullCursor() : parent;
6690
6691 annotateAndAdvanceTokens(updateC, RangeBefore, cursorRange);
6692
6693 // Avoid having the cursor of an expression "overwrite" the annotation of the
6694 // variable declaration that it belongs to.
6695 // This can happen for C++ constructor expressions whose range generally
6696 // include the variable declaration, e.g.:
6697 // MyCXXClass foo; // Make sure we don't annotate 'foo' as a CallExpr cursor.
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006698 if (clang_isExpression(cursorK) && MoreTokens()) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00006699 const Expr *E = getCursorExpr(cursor);
Dmitri Gribenkoa1691182013-01-26 18:12:08 +00006700 if (const Decl *D = getCursorParentDecl(cursor)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00006701 const unsigned I = NextToken();
6702 if (E->getLocStart().isValid() && D->getLocation().isValid() &&
6703 E->getLocStart() == D->getLocation() &&
6704 E->getLocStart() == GetTokenLoc(I)) {
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006705 updateCursorAnnotation(Cursors[I], updateC);
Guy Benyei11169dd2012-12-18 14:30:41 +00006706 AdvanceToken();
6707 }
6708 }
6709 }
6710
6711 // Before recursing into the children keep some state that we are going
6712 // to use in the AnnotateTokensWorker::postVisitChildren callback to do some
6713 // extra work after the child nodes are visited.
6714 // Note that we don't call VisitChildren here to avoid traversing statements
6715 // code-recursively which can blow the stack.
6716
6717 PostChildrenInfo Info;
6718 Info.Cursor = cursor;
6719 Info.CursorRange = cursorRange;
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006720 Info.BeforeReachingCursorIdx = BeforeReachingCursorIdx;
Guy Benyei11169dd2012-12-18 14:30:41 +00006721 Info.BeforeChildrenTokenIdx = NextToken();
6722 PostChildrenInfos.push_back(Info);
6723
6724 return CXChildVisit_Recurse;
6725}
6726
6727bool AnnotateTokensWorker::postVisitChildren(CXCursor cursor) {
6728 if (PostChildrenInfos.empty())
6729 return false;
6730 const PostChildrenInfo &Info = PostChildrenInfos.back();
6731 if (!clang_equalCursors(Info.Cursor, cursor))
6732 return false;
6733
6734 const unsigned BeforeChildren = Info.BeforeChildrenTokenIdx;
6735 const unsigned AfterChildren = NextToken();
6736 SourceRange cursorRange = Info.CursorRange;
6737
6738 // Scan the tokens that are at the end of the cursor, but are not captured
6739 // but the child cursors.
6740 annotateAndAdvanceTokens(cursor, RangeOverlap, cursorRange);
6741
6742 // Scan the tokens that are at the beginning of the cursor, but are not
6743 // capture by the child cursors.
6744 for (unsigned I = BeforeChildren; I != AfterChildren; ++I) {
6745 if (!clang_isInvalid(clang_getCursorKind(Cursors[I])))
6746 break;
6747
6748 Cursors[I] = cursor;
6749 }
6750
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006751 // Attributes are annotated out-of-order, rewind TokIdx to when we first
6752 // encountered the attribute cursor.
6753 if (clang_isAttribute(cursor.kind))
6754 TokIdx = Info.BeforeReachingCursorIdx;
6755
Guy Benyei11169dd2012-12-18 14:30:41 +00006756 PostChildrenInfos.pop_back();
6757 return false;
6758}
6759
6760static enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
6761 CXCursor parent,
6762 CXClientData client_data) {
6763 return static_cast<AnnotateTokensWorker*>(client_data)->Visit(cursor, parent);
6764}
6765
6766static bool AnnotateTokensPostChildrenVisitor(CXCursor cursor,
6767 CXClientData client_data) {
6768 return static_cast<AnnotateTokensWorker*>(client_data)->
6769 postVisitChildren(cursor);
6770}
6771
6772namespace {
6773
6774/// \brief Uses the macro expansions in the preprocessing record to find
6775/// and mark tokens that are macro arguments. This info is used by the
6776/// AnnotateTokensWorker.
6777class MarkMacroArgTokensVisitor {
6778 SourceManager &SM;
6779 CXToken *Tokens;
6780 unsigned NumTokens;
6781 unsigned CurIdx;
6782
6783public:
6784 MarkMacroArgTokensVisitor(SourceManager &SM,
6785 CXToken *tokens, unsigned numTokens)
6786 : SM(SM), Tokens(tokens), NumTokens(numTokens), CurIdx(0) { }
6787
6788 CXChildVisitResult visit(CXCursor cursor, CXCursor parent) {
6789 if (cursor.kind != CXCursor_MacroExpansion)
6790 return CXChildVisit_Continue;
6791
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00006792 SourceRange macroRange = getCursorMacroExpansion(cursor).getSourceRange();
Guy Benyei11169dd2012-12-18 14:30:41 +00006793 if (macroRange.getBegin() == macroRange.getEnd())
6794 return CXChildVisit_Continue; // it's not a function macro.
6795
6796 for (; CurIdx < NumTokens; ++CurIdx) {
6797 if (!SM.isBeforeInTranslationUnit(getTokenLoc(CurIdx),
6798 macroRange.getBegin()))
6799 break;
6800 }
6801
6802 if (CurIdx == NumTokens)
6803 return CXChildVisit_Break;
6804
6805 for (; CurIdx < NumTokens; ++CurIdx) {
6806 SourceLocation tokLoc = getTokenLoc(CurIdx);
6807 if (!SM.isBeforeInTranslationUnit(tokLoc, macroRange.getEnd()))
6808 break;
6809
6810 setFunctionMacroTokenLoc(CurIdx, SM.getMacroArgExpandedLocation(tokLoc));
6811 }
6812
6813 if (CurIdx == NumTokens)
6814 return CXChildVisit_Break;
6815
6816 return CXChildVisit_Continue;
6817 }
6818
6819private:
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006820 CXToken &getTok(unsigned Idx) {
6821 assert(Idx < NumTokens);
6822 return Tokens[Idx];
6823 }
6824 const CXToken &getTok(unsigned Idx) const {
6825 assert(Idx < NumTokens);
6826 return Tokens[Idx];
6827 }
6828
Guy Benyei11169dd2012-12-18 14:30:41 +00006829 SourceLocation getTokenLoc(unsigned tokI) {
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006830 return SourceLocation::getFromRawEncoding(getTok(tokI).int_data[1]);
Guy Benyei11169dd2012-12-18 14:30:41 +00006831 }
6832
6833 void setFunctionMacroTokenLoc(unsigned tokI, SourceLocation loc) {
6834 // The third field is reserved and currently not used. Use it here
6835 // to mark macro arg expanded tokens with their expanded locations.
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006836 getTok(tokI).int_data[3] = loc.getRawEncoding();
Guy Benyei11169dd2012-12-18 14:30:41 +00006837 }
6838};
6839
6840} // end anonymous namespace
6841
6842static CXChildVisitResult
6843MarkMacroArgTokensVisitorDelegate(CXCursor cursor, CXCursor parent,
6844 CXClientData client_data) {
6845 return static_cast<MarkMacroArgTokensVisitor*>(client_data)->visit(cursor,
6846 parent);
6847}
6848
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006849/// \brief Used by \c annotatePreprocessorTokens.
6850/// \returns true if lexing was finished, false otherwise.
6851static bool lexNext(Lexer &Lex, Token &Tok,
6852 unsigned &NextIdx, unsigned NumTokens) {
6853 if (NextIdx >= NumTokens)
6854 return true;
6855
6856 ++NextIdx;
6857 Lex.LexFromRawLexer(Tok);
Alexander Kornienko1a9f1842015-12-28 15:24:08 +00006858 return Tok.is(tok::eof);
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006859}
6860
Guy Benyei11169dd2012-12-18 14:30:41 +00006861static void annotatePreprocessorTokens(CXTranslationUnit TU,
6862 SourceRange RegionOfInterest,
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006863 CXCursor *Cursors,
6864 CXToken *Tokens,
6865 unsigned NumTokens) {
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006866 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00006867
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006868 Preprocessor &PP = CXXUnit->getPreprocessor();
Guy Benyei11169dd2012-12-18 14:30:41 +00006869 SourceManager &SourceMgr = CXXUnit->getSourceManager();
6870 std::pair<FileID, unsigned> BeginLocInfo
Argyrios Kyrtzidis5d47a9b2013-02-13 18:33:28 +00006871 = SourceMgr.getDecomposedSpellingLoc(RegionOfInterest.getBegin());
Guy Benyei11169dd2012-12-18 14:30:41 +00006872 std::pair<FileID, unsigned> EndLocInfo
Argyrios Kyrtzidis5d47a9b2013-02-13 18:33:28 +00006873 = SourceMgr.getDecomposedSpellingLoc(RegionOfInterest.getEnd());
Guy Benyei11169dd2012-12-18 14:30:41 +00006874
6875 if (BeginLocInfo.first != EndLocInfo.first)
6876 return;
6877
6878 StringRef Buffer;
6879 bool Invalid = false;
6880 Buffer = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
6881 if (Buffer.empty() || Invalid)
6882 return;
6883
6884 Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
6885 CXXUnit->getASTContext().getLangOpts(),
6886 Buffer.begin(), Buffer.data() + BeginLocInfo.second,
6887 Buffer.end());
6888 Lex.SetCommentRetentionState(true);
6889
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006890 unsigned NextIdx = 0;
Guy Benyei11169dd2012-12-18 14:30:41 +00006891 // Lex tokens in raw mode until we hit the end of the range, to avoid
6892 // entering #includes or expanding macros.
6893 while (true) {
6894 Token Tok;
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006895 if (lexNext(Lex, Tok, NextIdx, NumTokens))
6896 break;
6897 unsigned TokIdx = NextIdx-1;
6898 assert(Tok.getLocation() ==
6899 SourceLocation::getFromRawEncoding(Tokens[TokIdx].int_data[1]));
Guy Benyei11169dd2012-12-18 14:30:41 +00006900
6901 reprocess:
6902 if (Tok.is(tok::hash) && Tok.isAtStartOfLine()) {
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006903 // We have found a preprocessing directive. Annotate the tokens
6904 // appropriately.
Guy Benyei11169dd2012-12-18 14:30:41 +00006905 //
6906 // FIXME: Some simple tests here could identify macro definitions and
6907 // #undefs, to provide specific cursor kinds for those.
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006908
6909 SourceLocation BeginLoc = Tok.getLocation();
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006910 if (lexNext(Lex, Tok, NextIdx, NumTokens))
6911 break;
6912
Craig Topper69186e72014-06-08 08:38:04 +00006913 MacroInfo *MI = nullptr;
Alp Toker2d57cea2014-05-17 04:53:25 +00006914 if (Tok.is(tok::raw_identifier) && Tok.getRawIdentifier() == "define") {
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006915 if (lexNext(Lex, Tok, NextIdx, NumTokens))
6916 break;
6917
6918 if (Tok.is(tok::raw_identifier)) {
Alp Toker2d57cea2014-05-17 04:53:25 +00006919 IdentifierInfo &II =
6920 PP.getIdentifierTable().get(Tok.getRawIdentifier());
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006921 SourceLocation MappedTokLoc =
6922 CXXUnit->mapLocationToPreamble(Tok.getLocation());
6923 MI = getMacroInfo(II, MappedTokLoc, TU);
6924 }
6925 }
6926
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006927 bool finished = false;
Guy Benyei11169dd2012-12-18 14:30:41 +00006928 do {
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006929 if (lexNext(Lex, Tok, NextIdx, NumTokens)) {
6930 finished = true;
6931 break;
6932 }
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006933 // If we are in a macro definition, check if the token was ever a
6934 // macro name and annotate it if that's the case.
6935 if (MI) {
6936 SourceLocation SaveLoc = Tok.getLocation();
6937 Tok.setLocation(CXXUnit->mapLocationToPreamble(SaveLoc));
Richard Smith66a81862015-05-04 02:25:31 +00006938 MacroDefinitionRecord *MacroDef =
6939 checkForMacroInMacroDefinition(MI, Tok, TU);
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006940 Tok.setLocation(SaveLoc);
6941 if (MacroDef)
Richard Smith66a81862015-05-04 02:25:31 +00006942 Cursors[NextIdx - 1] =
6943 MakeMacroExpansionCursor(MacroDef, Tok.getLocation(), TU);
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006944 }
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006945 } while (!Tok.isAtStartOfLine());
6946
6947 unsigned LastIdx = finished ? NextIdx-1 : NextIdx-2;
6948 assert(TokIdx <= LastIdx);
6949 SourceLocation EndLoc =
6950 SourceLocation::getFromRawEncoding(Tokens[LastIdx].int_data[1]);
6951 CXCursor Cursor =
6952 MakePreprocessingDirectiveCursor(SourceRange(BeginLoc, EndLoc), TU);
6953
6954 for (; TokIdx <= LastIdx; ++TokIdx)
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006955 updateCursorAnnotation(Cursors[TokIdx], Cursor);
Guy Benyei11169dd2012-12-18 14:30:41 +00006956
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006957 if (finished)
6958 break;
6959 goto reprocess;
Guy Benyei11169dd2012-12-18 14:30:41 +00006960 }
Guy Benyei11169dd2012-12-18 14:30:41 +00006961 }
6962}
6963
6964// This gets run a separate thread to avoid stack blowout.
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00006965static void clang_annotateTokensImpl(CXTranslationUnit TU, ASTUnit *CXXUnit,
6966 CXToken *Tokens, unsigned NumTokens,
6967 CXCursor *Cursors) {
Dmitri Gribenko183436e2013-01-26 21:49:50 +00006968 CIndexer *CXXIdx = TU->CIdx;
Guy Benyei11169dd2012-12-18 14:30:41 +00006969 if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForEditing))
6970 setThreadBackgroundPriority();
6971
6972 // Determine the region of interest, which contains all of the tokens.
6973 SourceRange RegionOfInterest;
6974 RegionOfInterest.setBegin(
6975 cxloc::translateSourceLocation(clang_getTokenLocation(TU, Tokens[0])));
6976 RegionOfInterest.setEnd(
6977 cxloc::translateSourceLocation(clang_getTokenLocation(TU,
6978 Tokens[NumTokens-1])));
6979
Guy Benyei11169dd2012-12-18 14:30:41 +00006980 // Relex the tokens within the source range to look for preprocessing
6981 // directives.
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006982 annotatePreprocessorTokens(TU, RegionOfInterest, Cursors, Tokens, NumTokens);
Argyrios Kyrtzidis5d47a9b2013-02-13 18:33:28 +00006983
6984 // If begin location points inside a macro argument, set it to the expansion
6985 // location so we can have the full context when annotating semantically.
6986 {
6987 SourceManager &SM = CXXUnit->getSourceManager();
6988 SourceLocation Loc =
6989 SM.getMacroArgExpandedLocation(RegionOfInterest.getBegin());
6990 if (Loc.isMacroID())
6991 RegionOfInterest.setBegin(SM.getExpansionLoc(Loc));
6992 }
6993
Guy Benyei11169dd2012-12-18 14:30:41 +00006994 if (CXXUnit->getPreprocessor().getPreprocessingRecord()) {
6995 // Search and mark tokens that are macro argument expansions.
6996 MarkMacroArgTokensVisitor Visitor(CXXUnit->getSourceManager(),
6997 Tokens, NumTokens);
6998 CursorVisitor MacroArgMarker(TU,
6999 MarkMacroArgTokensVisitorDelegate, &Visitor,
7000 /*VisitPreprocessorLast=*/true,
7001 /*VisitIncludedEntities=*/false,
7002 RegionOfInterest);
7003 MacroArgMarker.visitPreprocessedEntitiesInRegion();
7004 }
7005
7006 // Annotate all of the source locations in the region of interest that map to
7007 // a specific cursor.
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00007008 AnnotateTokensWorker W(Tokens, Cursors, NumTokens, TU, RegionOfInterest);
Guy Benyei11169dd2012-12-18 14:30:41 +00007009
7010 // FIXME: We use a ridiculous stack size here because the data-recursion
7011 // algorithm uses a large stack frame than the non-data recursive version,
7012 // and AnnotationTokensWorker currently transforms the data-recursion
7013 // algorithm back into a traditional recursion by explicitly calling
7014 // VisitChildren(). We will need to remove this explicit recursive call.
7015 W.AnnotateTokens();
7016
7017 // If we ran into any entities that involve context-sensitive keywords,
7018 // take another pass through the tokens to mark them as such.
7019 if (W.hasContextSensitiveKeywords()) {
7020 for (unsigned I = 0; I != NumTokens; ++I) {
7021 if (clang_getTokenKind(Tokens[I]) != CXToken_Identifier)
7022 continue;
7023
7024 if (Cursors[I].kind == CXCursor_ObjCPropertyDecl) {
7025 IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007026 if (const ObjCPropertyDecl *Property
Guy Benyei11169dd2012-12-18 14:30:41 +00007027 = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(Cursors[I]))) {
7028 if (Property->getPropertyAttributesAsWritten() != 0 &&
7029 llvm::StringSwitch<bool>(II->getName())
7030 .Case("readonly", true)
7031 .Case("assign", true)
7032 .Case("unsafe_unretained", true)
7033 .Case("readwrite", true)
7034 .Case("retain", true)
7035 .Case("copy", true)
7036 .Case("nonatomic", true)
7037 .Case("atomic", true)
7038 .Case("getter", true)
7039 .Case("setter", true)
7040 .Case("strong", true)
7041 .Case("weak", true)
Manman Ren04fd4d82016-05-31 23:22:04 +00007042 .Case("class", true)
Guy Benyei11169dd2012-12-18 14:30:41 +00007043 .Default(false))
7044 Tokens[I].int_data[0] = CXToken_Keyword;
7045 }
7046 continue;
7047 }
7048
7049 if (Cursors[I].kind == CXCursor_ObjCInstanceMethodDecl ||
7050 Cursors[I].kind == CXCursor_ObjCClassMethodDecl) {
7051 IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
7052 if (llvm::StringSwitch<bool>(II->getName())
7053 .Case("in", true)
7054 .Case("out", true)
7055 .Case("inout", true)
7056 .Case("oneway", true)
7057 .Case("bycopy", true)
7058 .Case("byref", true)
7059 .Default(false))
7060 Tokens[I].int_data[0] = CXToken_Keyword;
7061 continue;
7062 }
7063
7064 if (Cursors[I].kind == CXCursor_CXXFinalAttr ||
7065 Cursors[I].kind == CXCursor_CXXOverrideAttr) {
7066 Tokens[I].int_data[0] = CXToken_Keyword;
7067 continue;
7068 }
7069 }
7070 }
7071}
7072
Guy Benyei11169dd2012-12-18 14:30:41 +00007073void clang_annotateTokens(CXTranslationUnit TU,
7074 CXToken *Tokens, unsigned NumTokens,
7075 CXCursor *Cursors) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00007076 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00007077 LOG_BAD_TU(TU);
7078 return;
7079 }
7080 if (NumTokens == 0 || !Tokens || !Cursors) {
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007081 LOG_FUNC_SECTION { *Log << "<null input>"; }
Guy Benyei11169dd2012-12-18 14:30:41 +00007082 return;
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007083 }
7084
7085 LOG_FUNC_SECTION {
7086 *Log << TU << ' ';
7087 CXSourceLocation bloc = clang_getTokenLocation(TU, Tokens[0]);
7088 CXSourceLocation eloc = clang_getTokenLocation(TU, Tokens[NumTokens-1]);
7089 *Log << clang_getRange(bloc, eloc);
7090 }
Guy Benyei11169dd2012-12-18 14:30:41 +00007091
7092 // Any token we don't specifically annotate will have a NULL cursor.
7093 CXCursor C = clang_getNullCursor();
7094 for (unsigned I = 0; I != NumTokens; ++I)
7095 Cursors[I] = C;
7096
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00007097 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00007098 if (!CXXUnit)
7099 return;
7100
7101 ASTUnit::ConcurrencyCheck Check(*CXXUnit);
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00007102
7103 auto AnnotateTokensImpl = [=]() {
7104 clang_annotateTokensImpl(TU, CXXUnit, Tokens, NumTokens, Cursors);
7105 };
Guy Benyei11169dd2012-12-18 14:30:41 +00007106 llvm::CrashRecoveryContext CRC;
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00007107 if (!RunSafely(CRC, AnnotateTokensImpl, GetSafetyThreadStackSize() * 2)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00007108 fprintf(stderr, "libclang: crash detected while annotating tokens\n");
7109 }
7110}
7111
Guy Benyei11169dd2012-12-18 14:30:41 +00007112//===----------------------------------------------------------------------===//
7113// Operations for querying linkage of a cursor.
7114//===----------------------------------------------------------------------===//
7115
Guy Benyei11169dd2012-12-18 14:30:41 +00007116CXLinkageKind clang_getCursorLinkage(CXCursor cursor) {
7117 if (!clang_isDeclaration(cursor.kind))
7118 return CXLinkage_Invalid;
7119
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007120 const Decl *D = cxcursor::getCursorDecl(cursor);
7121 if (const NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D))
Rafael Espindola3ae00052013-05-13 00:12:11 +00007122 switch (ND->getLinkageInternal()) {
Rafael Espindola50df3a02013-05-25 17:16:20 +00007123 case NoLinkage:
7124 case VisibleNoLinkage: return CXLinkage_NoLinkage;
Richard Smithaf10ea22017-07-08 00:37:59 +00007125 case ModuleInternalLinkage:
Guy Benyei11169dd2012-12-18 14:30:41 +00007126 case InternalLinkage: return CXLinkage_Internal;
7127 case UniqueExternalLinkage: return CXLinkage_UniqueExternal;
Richard Smithaf10ea22017-07-08 00:37:59 +00007128 case ModuleLinkage:
Guy Benyei11169dd2012-12-18 14:30:41 +00007129 case ExternalLinkage: return CXLinkage_External;
7130 };
7131
7132 return CXLinkage_Invalid;
7133}
Guy Benyei11169dd2012-12-18 14:30:41 +00007134
7135//===----------------------------------------------------------------------===//
Ehsan Akhgarib743de72016-05-31 15:55:51 +00007136// Operations for querying visibility of a cursor.
7137//===----------------------------------------------------------------------===//
7138
Ehsan Akhgarib743de72016-05-31 15:55:51 +00007139CXVisibilityKind clang_getCursorVisibility(CXCursor cursor) {
7140 if (!clang_isDeclaration(cursor.kind))
7141 return CXVisibility_Invalid;
7142
7143 const Decl *D = cxcursor::getCursorDecl(cursor);
7144 if (const NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D))
7145 switch (ND->getVisibility()) {
7146 case HiddenVisibility: return CXVisibility_Hidden;
7147 case ProtectedVisibility: return CXVisibility_Protected;
7148 case DefaultVisibility: return CXVisibility_Default;
7149 };
7150
7151 return CXVisibility_Invalid;
7152}
Ehsan Akhgarib743de72016-05-31 15:55:51 +00007153
7154//===----------------------------------------------------------------------===//
Guy Benyei11169dd2012-12-18 14:30:41 +00007155// Operations for querying language of a cursor.
7156//===----------------------------------------------------------------------===//
7157
7158static CXLanguageKind getDeclLanguage(const Decl *D) {
7159 if (!D)
7160 return CXLanguage_C;
7161
7162 switch (D->getKind()) {
7163 default:
7164 break;
7165 case Decl::ImplicitParam:
7166 case Decl::ObjCAtDefsField:
7167 case Decl::ObjCCategory:
7168 case Decl::ObjCCategoryImpl:
7169 case Decl::ObjCCompatibleAlias:
7170 case Decl::ObjCImplementation:
7171 case Decl::ObjCInterface:
7172 case Decl::ObjCIvar:
7173 case Decl::ObjCMethod:
7174 case Decl::ObjCProperty:
7175 case Decl::ObjCPropertyImpl:
7176 case Decl::ObjCProtocol:
Douglas Gregor85f3f952015-07-07 03:57:15 +00007177 case Decl::ObjCTypeParam:
Guy Benyei11169dd2012-12-18 14:30:41 +00007178 return CXLanguage_ObjC;
7179 case Decl::CXXConstructor:
7180 case Decl::CXXConversion:
7181 case Decl::CXXDestructor:
7182 case Decl::CXXMethod:
7183 case Decl::CXXRecord:
7184 case Decl::ClassTemplate:
7185 case Decl::ClassTemplatePartialSpecialization:
7186 case Decl::ClassTemplateSpecialization:
7187 case Decl::Friend:
7188 case Decl::FriendTemplate:
7189 case Decl::FunctionTemplate:
7190 case Decl::LinkageSpec:
7191 case Decl::Namespace:
7192 case Decl::NamespaceAlias:
7193 case Decl::NonTypeTemplateParm:
7194 case Decl::StaticAssert:
7195 case Decl::TemplateTemplateParm:
7196 case Decl::TemplateTypeParm:
7197 case Decl::UnresolvedUsingTypename:
7198 case Decl::UnresolvedUsingValue:
7199 case Decl::Using:
7200 case Decl::UsingDirective:
7201 case Decl::UsingShadow:
7202 return CXLanguage_CPlusPlus;
7203 }
7204
7205 return CXLanguage_C;
7206}
7207
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007208static CXAvailabilityKind getCursorAvailabilityForDecl(const Decl *D) {
7209 if (isa<FunctionDecl>(D) && cast<FunctionDecl>(D)->isDeleted())
Manuel Klimek8e3a7ed2015-09-25 17:53:16 +00007210 return CXAvailability_NotAvailable;
Guy Benyei11169dd2012-12-18 14:30:41 +00007211
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007212 switch (D->getAvailability()) {
7213 case AR_Available:
7214 case AR_NotYetIntroduced:
7215 if (const EnumConstantDecl *EnumConst = dyn_cast<EnumConstantDecl>(D))
Benjamin Kramer656363d2013-10-15 18:53:18 +00007216 return getCursorAvailabilityForDecl(
7217 cast<Decl>(EnumConst->getDeclContext()));
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007218 return CXAvailability_Available;
7219
7220 case AR_Deprecated:
7221 return CXAvailability_Deprecated;
7222
7223 case AR_Unavailable:
7224 return CXAvailability_NotAvailable;
7225 }
Benjamin Kramer656363d2013-10-15 18:53:18 +00007226
7227 llvm_unreachable("Unknown availability kind!");
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007228}
7229
Guy Benyei11169dd2012-12-18 14:30:41 +00007230enum CXAvailabilityKind clang_getCursorAvailability(CXCursor cursor) {
7231 if (clang_isDeclaration(cursor.kind))
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007232 if (const Decl *D = cxcursor::getCursorDecl(cursor))
7233 return getCursorAvailabilityForDecl(D);
Guy Benyei11169dd2012-12-18 14:30:41 +00007234
7235 return CXAvailability_Available;
7236}
7237
7238static CXVersion convertVersion(VersionTuple In) {
7239 CXVersion Out = { -1, -1, -1 };
7240 if (In.empty())
7241 return Out;
7242
7243 Out.Major = In.getMajor();
7244
NAKAMURA Takumic2b5d1f2013-02-21 02:32:34 +00007245 Optional<unsigned> Minor = In.getMinor();
7246 if (Minor.hasValue())
Guy Benyei11169dd2012-12-18 14:30:41 +00007247 Out.Minor = *Minor;
7248 else
7249 return Out;
7250
NAKAMURA Takumic2b5d1f2013-02-21 02:32:34 +00007251 Optional<unsigned> Subminor = In.getSubminor();
7252 if (Subminor.hasValue())
Guy Benyei11169dd2012-12-18 14:30:41 +00007253 Out.Subminor = *Subminor;
7254
7255 return Out;
7256}
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007257
Alex Lorenz1345ea22017-06-12 19:06:30 +00007258static void getCursorPlatformAvailabilityForDecl(
7259 const Decl *D, int *always_deprecated, CXString *deprecated_message,
7260 int *always_unavailable, CXString *unavailable_message,
7261 SmallVectorImpl<AvailabilityAttr *> &AvailabilityAttrs) {
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007262 bool HadAvailAttr = false;
Aaron Ballmanb97112e2014-03-08 22:19:01 +00007263 for (auto A : D->attrs()) {
7264 if (DeprecatedAttr *Deprecated = dyn_cast<DeprecatedAttr>(A)) {
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007265 HadAvailAttr = true;
7266 if (always_deprecated)
7267 *always_deprecated = 1;
Nico Weberaacf0312014-04-24 05:16:45 +00007268 if (deprecated_message) {
Argyrios Kyrtzidisedfe07f2014-04-24 06:05:40 +00007269 clang_disposeString(*deprecated_message);
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007270 *deprecated_message = cxstring::createDup(Deprecated->getMessage());
Nico Weberaacf0312014-04-24 05:16:45 +00007271 }
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007272 continue;
7273 }
Alex Lorenz1345ea22017-06-12 19:06:30 +00007274
Aaron Ballmanb97112e2014-03-08 22:19:01 +00007275 if (UnavailableAttr *Unavailable = dyn_cast<UnavailableAttr>(A)) {
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007276 HadAvailAttr = true;
7277 if (always_unavailable)
7278 *always_unavailable = 1;
7279 if (unavailable_message) {
Argyrios Kyrtzidisedfe07f2014-04-24 06:05:40 +00007280 clang_disposeString(*unavailable_message);
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007281 *unavailable_message = cxstring::createDup(Unavailable->getMessage());
7282 }
7283 continue;
7284 }
Alex Lorenz1345ea22017-06-12 19:06:30 +00007285
Aaron Ballmanb97112e2014-03-08 22:19:01 +00007286 if (AvailabilityAttr *Avail = dyn_cast<AvailabilityAttr>(A)) {
Alex Lorenz1345ea22017-06-12 19:06:30 +00007287 AvailabilityAttrs.push_back(Avail);
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007288 HadAvailAttr = true;
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007289 }
7290 }
7291
7292 if (!HadAvailAttr)
7293 if (const EnumConstantDecl *EnumConst = dyn_cast<EnumConstantDecl>(D))
7294 return getCursorPlatformAvailabilityForDecl(
Alex Lorenz1345ea22017-06-12 19:06:30 +00007295 cast<Decl>(EnumConst->getDeclContext()), always_deprecated,
7296 deprecated_message, always_unavailable, unavailable_message,
7297 AvailabilityAttrs);
7298
7299 if (AvailabilityAttrs.empty())
7300 return;
7301
7302 std::sort(AvailabilityAttrs.begin(), AvailabilityAttrs.end(),
7303 [](AvailabilityAttr *LHS, AvailabilityAttr *RHS) {
7304 return LHS->getPlatform() > RHS->getPlatform();
7305 });
7306 ASTContext &Ctx = D->getASTContext();
7307 auto It = std::unique(
7308 AvailabilityAttrs.begin(), AvailabilityAttrs.end(),
7309 [&Ctx](AvailabilityAttr *LHS, AvailabilityAttr *RHS) {
7310 if (LHS->getPlatform() != RHS->getPlatform())
7311 return false;
7312
7313 if (LHS->getIntroduced() == RHS->getIntroduced() &&
7314 LHS->getDeprecated() == RHS->getDeprecated() &&
7315 LHS->getObsoleted() == RHS->getObsoleted() &&
7316 LHS->getMessage() == RHS->getMessage() &&
7317 LHS->getReplacement() == RHS->getReplacement())
7318 return true;
7319
7320 if ((!LHS->getIntroduced().empty() && !RHS->getIntroduced().empty()) ||
7321 (!LHS->getDeprecated().empty() && !RHS->getDeprecated().empty()) ||
7322 (!LHS->getObsoleted().empty() && !RHS->getObsoleted().empty()))
7323 return false;
7324
7325 if (LHS->getIntroduced().empty() && !RHS->getIntroduced().empty())
7326 LHS->setIntroduced(Ctx, RHS->getIntroduced());
7327
7328 if (LHS->getDeprecated().empty() && !RHS->getDeprecated().empty()) {
7329 LHS->setDeprecated(Ctx, RHS->getDeprecated());
7330 if (LHS->getMessage().empty())
7331 LHS->setMessage(Ctx, RHS->getMessage());
7332 if (LHS->getReplacement().empty())
7333 LHS->setReplacement(Ctx, RHS->getReplacement());
7334 }
7335
7336 if (LHS->getObsoleted().empty() && !RHS->getObsoleted().empty()) {
7337 LHS->setObsoleted(Ctx, RHS->getObsoleted());
7338 if (LHS->getMessage().empty())
7339 LHS->setMessage(Ctx, RHS->getMessage());
7340 if (LHS->getReplacement().empty())
7341 LHS->setReplacement(Ctx, RHS->getReplacement());
7342 }
7343
7344 return true;
7345 });
7346 AvailabilityAttrs.erase(It, AvailabilityAttrs.end());
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007347}
7348
Alex Lorenz1345ea22017-06-12 19:06:30 +00007349int clang_getCursorPlatformAvailability(CXCursor cursor, int *always_deprecated,
Guy Benyei11169dd2012-12-18 14:30:41 +00007350 CXString *deprecated_message,
7351 int *always_unavailable,
7352 CXString *unavailable_message,
7353 CXPlatformAvailability *availability,
7354 int availability_size) {
7355 if (always_deprecated)
7356 *always_deprecated = 0;
7357 if (deprecated_message)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00007358 *deprecated_message = cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00007359 if (always_unavailable)
7360 *always_unavailable = 0;
7361 if (unavailable_message)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00007362 *unavailable_message = cxstring::createEmpty();
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007363
Guy Benyei11169dd2012-12-18 14:30:41 +00007364 if (!clang_isDeclaration(cursor.kind))
7365 return 0;
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007366
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007367 const Decl *D = cxcursor::getCursorDecl(cursor);
Guy Benyei11169dd2012-12-18 14:30:41 +00007368 if (!D)
7369 return 0;
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007370
Alex Lorenz1345ea22017-06-12 19:06:30 +00007371 SmallVector<AvailabilityAttr *, 8> AvailabilityAttrs;
7372 getCursorPlatformAvailabilityForDecl(D, always_deprecated, deprecated_message,
7373 always_unavailable, unavailable_message,
7374 AvailabilityAttrs);
7375 for (const auto &Avail :
7376 llvm::enumerate(llvm::makeArrayRef(AvailabilityAttrs)
7377 .take_front(availability_size))) {
7378 availability[Avail.index()].Platform =
7379 cxstring::createDup(Avail.value()->getPlatform()->getName());
7380 availability[Avail.index()].Introduced =
7381 convertVersion(Avail.value()->getIntroduced());
7382 availability[Avail.index()].Deprecated =
7383 convertVersion(Avail.value()->getDeprecated());
7384 availability[Avail.index()].Obsoleted =
7385 convertVersion(Avail.value()->getObsoleted());
7386 availability[Avail.index()].Unavailable = Avail.value()->getUnavailable();
7387 availability[Avail.index()].Message =
7388 cxstring::createDup(Avail.value()->getMessage());
7389 }
7390
7391 return AvailabilityAttrs.size();
Guy Benyei11169dd2012-12-18 14:30:41 +00007392}
Alex Lorenz1345ea22017-06-12 19:06:30 +00007393
Guy Benyei11169dd2012-12-18 14:30:41 +00007394void clang_disposeCXPlatformAvailability(CXPlatformAvailability *availability) {
7395 clang_disposeString(availability->Platform);
7396 clang_disposeString(availability->Message);
7397}
7398
7399CXLanguageKind clang_getCursorLanguage(CXCursor cursor) {
7400 if (clang_isDeclaration(cursor.kind))
7401 return getDeclLanguage(cxcursor::getCursorDecl(cursor));
7402
7403 return CXLanguage_Invalid;
7404}
7405
7406 /// \brief If the given cursor is the "templated" declaration
7407 /// descibing a class or function template, return the class or
7408 /// function template.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007409static const Decl *maybeGetTemplateCursor(const Decl *D) {
Guy Benyei11169dd2012-12-18 14:30:41 +00007410 if (!D)
Craig Topper69186e72014-06-08 08:38:04 +00007411 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007412
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007413 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00007414 if (FunctionTemplateDecl *FunTmpl = FD->getDescribedFunctionTemplate())
7415 return FunTmpl;
7416
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007417 if (const CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00007418 if (ClassTemplateDecl *ClassTmpl = RD->getDescribedClassTemplate())
7419 return ClassTmpl;
7420
7421 return D;
7422}
7423
Argyrios Kyrtzidis4e0854f2014-10-15 17:05:31 +00007424
7425enum CX_StorageClass clang_Cursor_getStorageClass(CXCursor C) {
7426 StorageClass sc = SC_None;
7427 const Decl *D = getCursorDecl(C);
7428 if (D) {
7429 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) {
7430 sc = FD->getStorageClass();
7431 } else if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
7432 sc = VD->getStorageClass();
7433 } else {
7434 return CX_SC_Invalid;
7435 }
7436 } else {
7437 return CX_SC_Invalid;
7438 }
7439 switch (sc) {
7440 case SC_None:
7441 return CX_SC_None;
7442 case SC_Extern:
7443 return CX_SC_Extern;
7444 case SC_Static:
7445 return CX_SC_Static;
7446 case SC_PrivateExtern:
7447 return CX_SC_PrivateExtern;
Argyrios Kyrtzidis4e0854f2014-10-15 17:05:31 +00007448 case SC_Auto:
7449 return CX_SC_Auto;
7450 case SC_Register:
7451 return CX_SC_Register;
Argyrios Kyrtzidis4e0854f2014-10-15 17:05:31 +00007452 }
Kaelyn Takataab61e702014-10-15 18:03:26 +00007453 llvm_unreachable("Unhandled storage class!");
Argyrios Kyrtzidis4e0854f2014-10-15 17:05:31 +00007454}
7455
Guy Benyei11169dd2012-12-18 14:30:41 +00007456CXCursor clang_getCursorSemanticParent(CXCursor cursor) {
7457 if (clang_isDeclaration(cursor.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007458 if (const Decl *D = getCursorDecl(cursor)) {
7459 const DeclContext *DC = D->getDeclContext();
Guy Benyei11169dd2012-12-18 14:30:41 +00007460 if (!DC)
7461 return clang_getNullCursor();
7462
7463 return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
7464 getCursorTU(cursor));
7465 }
7466 }
7467
7468 if (clang_isStatement(cursor.kind) || clang_isExpression(cursor.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007469 if (const Decl *D = getCursorDecl(cursor))
Guy Benyei11169dd2012-12-18 14:30:41 +00007470 return MakeCXCursor(D, getCursorTU(cursor));
7471 }
7472
7473 return clang_getNullCursor();
7474}
7475
7476CXCursor clang_getCursorLexicalParent(CXCursor cursor) {
7477 if (clang_isDeclaration(cursor.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007478 if (const Decl *D = getCursorDecl(cursor)) {
7479 const DeclContext *DC = D->getLexicalDeclContext();
Guy Benyei11169dd2012-12-18 14:30:41 +00007480 if (!DC)
7481 return clang_getNullCursor();
7482
7483 return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
7484 getCursorTU(cursor));
7485 }
7486 }
7487
7488 // FIXME: Note that we can't easily compute the lexical context of a
7489 // statement or expression, so we return nothing.
7490 return clang_getNullCursor();
7491}
7492
7493CXFile clang_getIncludedFile(CXCursor cursor) {
7494 if (cursor.kind != CXCursor_InclusionDirective)
Craig Topper69186e72014-06-08 08:38:04 +00007495 return nullptr;
7496
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00007497 const InclusionDirective *ID = getCursorInclusionDirective(cursor);
Dmitri Gribenkof9304482013-01-23 15:56:07 +00007498 return const_cast<FileEntry *>(ID->getFile());
Guy Benyei11169dd2012-12-18 14:30:41 +00007499}
7500
Argyrios Kyrtzidis9adfd8a2013-04-18 22:15:49 +00007501unsigned clang_Cursor_getObjCPropertyAttributes(CXCursor C, unsigned reserved) {
7502 if (C.kind != CXCursor_ObjCPropertyDecl)
7503 return CXObjCPropertyAttr_noattr;
7504
7505 unsigned Result = CXObjCPropertyAttr_noattr;
7506 const ObjCPropertyDecl *PD = dyn_cast<ObjCPropertyDecl>(getCursorDecl(C));
7507 ObjCPropertyDecl::PropertyAttributeKind Attr =
7508 PD->getPropertyAttributesAsWritten();
7509
7510#define SET_CXOBJCPROP_ATTR(A) \
7511 if (Attr & ObjCPropertyDecl::OBJC_PR_##A) \
7512 Result |= CXObjCPropertyAttr_##A
7513 SET_CXOBJCPROP_ATTR(readonly);
7514 SET_CXOBJCPROP_ATTR(getter);
7515 SET_CXOBJCPROP_ATTR(assign);
7516 SET_CXOBJCPROP_ATTR(readwrite);
7517 SET_CXOBJCPROP_ATTR(retain);
7518 SET_CXOBJCPROP_ATTR(copy);
7519 SET_CXOBJCPROP_ATTR(nonatomic);
7520 SET_CXOBJCPROP_ATTR(setter);
7521 SET_CXOBJCPROP_ATTR(atomic);
7522 SET_CXOBJCPROP_ATTR(weak);
7523 SET_CXOBJCPROP_ATTR(strong);
7524 SET_CXOBJCPROP_ATTR(unsafe_unretained);
Manman Ren04fd4d82016-05-31 23:22:04 +00007525 SET_CXOBJCPROP_ATTR(class);
Argyrios Kyrtzidis9adfd8a2013-04-18 22:15:49 +00007526#undef SET_CXOBJCPROP_ATTR
7527
7528 return Result;
7529}
7530
Argyrios Kyrtzidis9d9bc012013-04-18 23:29:12 +00007531unsigned clang_Cursor_getObjCDeclQualifiers(CXCursor C) {
7532 if (!clang_isDeclaration(C.kind))
7533 return CXObjCDeclQualifier_None;
7534
7535 Decl::ObjCDeclQualifier QT = Decl::OBJC_TQ_None;
7536 const Decl *D = getCursorDecl(C);
7537 if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D))
7538 QT = MD->getObjCDeclQualifier();
7539 else if (const ParmVarDecl *PD = dyn_cast<ParmVarDecl>(D))
7540 QT = PD->getObjCDeclQualifier();
7541 if (QT == Decl::OBJC_TQ_None)
7542 return CXObjCDeclQualifier_None;
7543
7544 unsigned Result = CXObjCDeclQualifier_None;
7545 if (QT & Decl::OBJC_TQ_In) Result |= CXObjCDeclQualifier_In;
7546 if (QT & Decl::OBJC_TQ_Inout) Result |= CXObjCDeclQualifier_Inout;
7547 if (QT & Decl::OBJC_TQ_Out) Result |= CXObjCDeclQualifier_Out;
7548 if (QT & Decl::OBJC_TQ_Bycopy) Result |= CXObjCDeclQualifier_Bycopy;
7549 if (QT & Decl::OBJC_TQ_Byref) Result |= CXObjCDeclQualifier_Byref;
7550 if (QT & Decl::OBJC_TQ_Oneway) Result |= CXObjCDeclQualifier_Oneway;
7551
7552 return Result;
7553}
7554
Argyrios Kyrtzidis7b50fc52013-07-05 20:44:37 +00007555unsigned clang_Cursor_isObjCOptional(CXCursor C) {
7556 if (!clang_isDeclaration(C.kind))
7557 return 0;
7558
7559 const Decl *D = getCursorDecl(C);
7560 if (const ObjCPropertyDecl *PD = dyn_cast<ObjCPropertyDecl>(D))
7561 return PD->getPropertyImplementation() == ObjCPropertyDecl::Optional;
7562 if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D))
7563 return MD->getImplementationControl() == ObjCMethodDecl::Optional;
7564
7565 return 0;
7566}
7567
Argyrios Kyrtzidis23814e42013-04-18 23:53:05 +00007568unsigned clang_Cursor_isVariadic(CXCursor C) {
7569 if (!clang_isDeclaration(C.kind))
7570 return 0;
7571
7572 const Decl *D = getCursorDecl(C);
7573 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
7574 return FD->isVariadic();
7575 if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D))
7576 return MD->isVariadic();
7577
7578 return 0;
7579}
7580
Argyrios Kyrtzidis0381cc72017-05-10 15:10:36 +00007581unsigned clang_Cursor_isExternalSymbol(CXCursor C,
7582 CXString *language, CXString *definedIn,
7583 unsigned *isGenerated) {
7584 if (!clang_isDeclaration(C.kind))
7585 return 0;
7586
7587 const Decl *D = getCursorDecl(C);
7588
Argyrios Kyrtzidis11d70482017-05-20 04:11:33 +00007589 if (auto *attr = D->getExternalSourceSymbolAttr()) {
Argyrios Kyrtzidis0381cc72017-05-10 15:10:36 +00007590 if (language)
7591 *language = cxstring::createDup(attr->getLanguage());
7592 if (definedIn)
7593 *definedIn = cxstring::createDup(attr->getDefinedIn());
7594 if (isGenerated)
7595 *isGenerated = attr->getGeneratedDeclaration();
7596 return 1;
7597 }
7598 return 0;
7599}
7600
Guy Benyei11169dd2012-12-18 14:30:41 +00007601CXSourceRange clang_Cursor_getCommentRange(CXCursor C) {
7602 if (!clang_isDeclaration(C.kind))
7603 return clang_getNullRange();
7604
7605 const Decl *D = getCursorDecl(C);
7606 ASTContext &Context = getCursorContext(C);
7607 const RawComment *RC = Context.getRawCommentForAnyRedecl(D);
7608 if (!RC)
7609 return clang_getNullRange();
7610
7611 return cxloc::translateSourceRange(Context, RC->getSourceRange());
7612}
7613
7614CXString clang_Cursor_getRawCommentText(CXCursor C) {
7615 if (!clang_isDeclaration(C.kind))
Dmitri Gribenkof98dfba2013-02-01 14:13:32 +00007616 return cxstring::createNull();
Guy Benyei11169dd2012-12-18 14:30:41 +00007617
7618 const Decl *D = getCursorDecl(C);
7619 ASTContext &Context = getCursorContext(C);
7620 const RawComment *RC = Context.getRawCommentForAnyRedecl(D);
7621 StringRef RawText = RC ? RC->getRawText(Context.getSourceManager()) :
7622 StringRef();
7623
7624 // Don't duplicate the string because RawText points directly into source
7625 // code.
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00007626 return cxstring::createRef(RawText);
Guy Benyei11169dd2012-12-18 14:30:41 +00007627}
7628
7629CXString clang_Cursor_getBriefCommentText(CXCursor C) {
7630 if (!clang_isDeclaration(C.kind))
Dmitri Gribenkof98dfba2013-02-01 14:13:32 +00007631 return cxstring::createNull();
Guy Benyei11169dd2012-12-18 14:30:41 +00007632
7633 const Decl *D = getCursorDecl(C);
7634 const ASTContext &Context = getCursorContext(C);
7635 const RawComment *RC = Context.getRawCommentForAnyRedecl(D);
7636
7637 if (RC) {
7638 StringRef BriefText = RC->getBriefText(Context);
7639
7640 // Don't duplicate the string because RawComment ensures that this memory
7641 // will not go away.
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00007642 return cxstring::createRef(BriefText);
Guy Benyei11169dd2012-12-18 14:30:41 +00007643 }
7644
Dmitri Gribenkof98dfba2013-02-01 14:13:32 +00007645 return cxstring::createNull();
Guy Benyei11169dd2012-12-18 14:30:41 +00007646}
7647
Guy Benyei11169dd2012-12-18 14:30:41 +00007648CXModule clang_Cursor_getModule(CXCursor C) {
7649 if (C.kind == CXCursor_ModuleImportDecl) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007650 if (const ImportDecl *ImportD =
7651 dyn_cast_or_null<ImportDecl>(getCursorDecl(C)))
Guy Benyei11169dd2012-12-18 14:30:41 +00007652 return ImportD->getImportedModule();
7653 }
7654
Craig Topper69186e72014-06-08 08:38:04 +00007655 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007656}
7657
Argyrios Kyrtzidisf6d49c32014-05-14 23:14:37 +00007658CXModule clang_getModuleForFile(CXTranslationUnit TU, CXFile File) {
7659 if (isNotUsableTU(TU)) {
7660 LOG_BAD_TU(TU);
7661 return nullptr;
7662 }
7663 if (!File)
7664 return nullptr;
7665 FileEntry *FE = static_cast<FileEntry *>(File);
7666
7667 ASTUnit &Unit = *cxtu::getASTUnit(TU);
7668 HeaderSearch &HS = Unit.getPreprocessor().getHeaderSearchInfo();
7669 ModuleMap::KnownHeader Header = HS.findModuleForHeader(FE);
7670
Richard Smithfeb54b62014-10-23 02:01:19 +00007671 return Header.getModule();
Argyrios Kyrtzidisf6d49c32014-05-14 23:14:37 +00007672}
7673
Argyrios Kyrtzidis12fdb9e2013-04-26 22:47:49 +00007674CXFile clang_Module_getASTFile(CXModule CXMod) {
7675 if (!CXMod)
Craig Topper69186e72014-06-08 08:38:04 +00007676 return nullptr;
Argyrios Kyrtzidis12fdb9e2013-04-26 22:47:49 +00007677 Module *Mod = static_cast<Module*>(CXMod);
7678 return const_cast<FileEntry *>(Mod->getASTFile());
7679}
7680
Guy Benyei11169dd2012-12-18 14:30:41 +00007681CXModule clang_Module_getParent(CXModule CXMod) {
7682 if (!CXMod)
Craig Topper69186e72014-06-08 08:38:04 +00007683 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007684 Module *Mod = static_cast<Module*>(CXMod);
7685 return Mod->Parent;
7686}
7687
7688CXString clang_Module_getName(CXModule CXMod) {
7689 if (!CXMod)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00007690 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00007691 Module *Mod = static_cast<Module*>(CXMod);
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00007692 return cxstring::createDup(Mod->Name);
Guy Benyei11169dd2012-12-18 14:30:41 +00007693}
7694
7695CXString clang_Module_getFullName(CXModule CXMod) {
7696 if (!CXMod)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00007697 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00007698 Module *Mod = static_cast<Module*>(CXMod);
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00007699 return cxstring::createDup(Mod->getFullModuleName());
Guy Benyei11169dd2012-12-18 14:30:41 +00007700}
7701
Argyrios Kyrtzidis884337f2014-05-15 04:44:25 +00007702int clang_Module_isSystem(CXModule CXMod) {
7703 if (!CXMod)
7704 return 0;
7705 Module *Mod = static_cast<Module*>(CXMod);
7706 return Mod->IsSystem;
7707}
7708
Argyrios Kyrtzidis3c5305c2013-03-13 21:13:43 +00007709unsigned clang_Module_getNumTopLevelHeaders(CXTranslationUnit TU,
7710 CXModule CXMod) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00007711 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00007712 LOG_BAD_TU(TU);
7713 return 0;
7714 }
7715 if (!CXMod)
Guy Benyei11169dd2012-12-18 14:30:41 +00007716 return 0;
7717 Module *Mod = static_cast<Module*>(CXMod);
Argyrios Kyrtzidis3c5305c2013-03-13 21:13:43 +00007718 FileManager &FileMgr = cxtu::getASTUnit(TU)->getFileManager();
7719 ArrayRef<const FileEntry *> TopHeaders = Mod->getTopHeaders(FileMgr);
7720 return TopHeaders.size();
Guy Benyei11169dd2012-12-18 14:30:41 +00007721}
7722
Argyrios Kyrtzidis3c5305c2013-03-13 21:13:43 +00007723CXFile clang_Module_getTopLevelHeader(CXTranslationUnit TU,
7724 CXModule CXMod, unsigned Index) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00007725 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00007726 LOG_BAD_TU(TU);
Craig Topper69186e72014-06-08 08:38:04 +00007727 return nullptr;
Dmitri Gribenko256454f2014-02-11 14:34:14 +00007728 }
7729 if (!CXMod)
Craig Topper69186e72014-06-08 08:38:04 +00007730 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007731 Module *Mod = static_cast<Module*>(CXMod);
Argyrios Kyrtzidis3c5305c2013-03-13 21:13:43 +00007732 FileManager &FileMgr = cxtu::getASTUnit(TU)->getFileManager();
Guy Benyei11169dd2012-12-18 14:30:41 +00007733
Argyrios Kyrtzidis3c5305c2013-03-13 21:13:43 +00007734 ArrayRef<const FileEntry *> TopHeaders = Mod->getTopHeaders(FileMgr);
7735 if (Index < TopHeaders.size())
7736 return const_cast<FileEntry *>(TopHeaders[Index]);
Guy Benyei11169dd2012-12-18 14:30:41 +00007737
Craig Topper69186e72014-06-08 08:38:04 +00007738 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007739}
7740
Guy Benyei11169dd2012-12-18 14:30:41 +00007741//===----------------------------------------------------------------------===//
7742// C++ AST instrospection.
7743//===----------------------------------------------------------------------===//
7744
Jonathan Coe29565352016-04-27 12:48:25 +00007745unsigned clang_CXXConstructor_isDefaultConstructor(CXCursor C) {
7746 if (!clang_isDeclaration(C.kind))
7747 return 0;
7748
7749 const Decl *D = cxcursor::getCursorDecl(C);
7750 const CXXConstructorDecl *Constructor =
7751 D ? dyn_cast_or_null<CXXConstructorDecl>(D->getAsFunction()) : nullptr;
7752 return (Constructor && Constructor->isDefaultConstructor()) ? 1 : 0;
7753}
7754
7755unsigned clang_CXXConstructor_isCopyConstructor(CXCursor C) {
7756 if (!clang_isDeclaration(C.kind))
7757 return 0;
7758
7759 const Decl *D = cxcursor::getCursorDecl(C);
7760 const CXXConstructorDecl *Constructor =
7761 D ? dyn_cast_or_null<CXXConstructorDecl>(D->getAsFunction()) : nullptr;
7762 return (Constructor && Constructor->isCopyConstructor()) ? 1 : 0;
7763}
7764
7765unsigned clang_CXXConstructor_isMoveConstructor(CXCursor C) {
7766 if (!clang_isDeclaration(C.kind))
7767 return 0;
7768
7769 const Decl *D = cxcursor::getCursorDecl(C);
7770 const CXXConstructorDecl *Constructor =
7771 D ? dyn_cast_or_null<CXXConstructorDecl>(D->getAsFunction()) : nullptr;
7772 return (Constructor && Constructor->isMoveConstructor()) ? 1 : 0;
7773}
7774
7775unsigned clang_CXXConstructor_isConvertingConstructor(CXCursor C) {
7776 if (!clang_isDeclaration(C.kind))
7777 return 0;
7778
7779 const Decl *D = cxcursor::getCursorDecl(C);
7780 const CXXConstructorDecl *Constructor =
7781 D ? dyn_cast_or_null<CXXConstructorDecl>(D->getAsFunction()) : nullptr;
7782 // Passing 'false' excludes constructors marked 'explicit'.
7783 return (Constructor && Constructor->isConvertingConstructor(false)) ? 1 : 0;
7784}
7785
Saleem Abdulrasool6ea75db2015-10-27 15:50:22 +00007786unsigned clang_CXXField_isMutable(CXCursor C) {
7787 if (!clang_isDeclaration(C.kind))
7788 return 0;
7789
7790 if (const auto D = cxcursor::getCursorDecl(C))
7791 if (const auto FD = dyn_cast_or_null<FieldDecl>(D))
7792 return FD->isMutable() ? 1 : 0;
7793 return 0;
7794}
7795
Dmitri Gribenko62770be2013-05-17 18:38:35 +00007796unsigned clang_CXXMethod_isPureVirtual(CXCursor C) {
7797 if (!clang_isDeclaration(C.kind))
7798 return 0;
7799
Dmitri Gribenko62770be2013-05-17 18:38:35 +00007800 const Decl *D = cxcursor::getCursorDecl(C);
Alp Tokera2794f92014-01-22 07:29:52 +00007801 const CXXMethodDecl *Method =
Craig Topper69186e72014-06-08 08:38:04 +00007802 D ? dyn_cast_or_null<CXXMethodDecl>(D->getAsFunction()) : nullptr;
Dmitri Gribenko62770be2013-05-17 18:38:35 +00007803 return (Method && Method->isVirtual() && Method->isPure()) ? 1 : 0;
7804}
7805
Dmitri Gribenkoe570ede2014-04-07 14:59:13 +00007806unsigned clang_CXXMethod_isConst(CXCursor C) {
7807 if (!clang_isDeclaration(C.kind))
7808 return 0;
7809
7810 const Decl *D = cxcursor::getCursorDecl(C);
7811 const CXXMethodDecl *Method =
Craig Topper69186e72014-06-08 08:38:04 +00007812 D ? dyn_cast_or_null<CXXMethodDecl>(D->getAsFunction()) : nullptr;
Dmitri Gribenkoe570ede2014-04-07 14:59:13 +00007813 return (Method && (Method->getTypeQualifiers() & Qualifiers::Const)) ? 1 : 0;
7814}
7815
Jonathan Coe29565352016-04-27 12:48:25 +00007816unsigned clang_CXXMethod_isDefaulted(CXCursor C) {
7817 if (!clang_isDeclaration(C.kind))
7818 return 0;
7819
7820 const Decl *D = cxcursor::getCursorDecl(C);
7821 const CXXMethodDecl *Method =
7822 D ? dyn_cast_or_null<CXXMethodDecl>(D->getAsFunction()) : nullptr;
7823 return (Method && Method->isDefaulted()) ? 1 : 0;
7824}
7825
Guy Benyei11169dd2012-12-18 14:30:41 +00007826unsigned clang_CXXMethod_isStatic(CXCursor C) {
7827 if (!clang_isDeclaration(C.kind))
7828 return 0;
7829
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007830 const Decl *D = cxcursor::getCursorDecl(C);
Alp Tokera2794f92014-01-22 07:29:52 +00007831 const CXXMethodDecl *Method =
Craig Topper69186e72014-06-08 08:38:04 +00007832 D ? dyn_cast_or_null<CXXMethodDecl>(D->getAsFunction()) : nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007833 return (Method && Method->isStatic()) ? 1 : 0;
7834}
7835
7836unsigned clang_CXXMethod_isVirtual(CXCursor C) {
7837 if (!clang_isDeclaration(C.kind))
7838 return 0;
7839
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007840 const Decl *D = cxcursor::getCursorDecl(C);
Alp Tokera2794f92014-01-22 07:29:52 +00007841 const CXXMethodDecl *Method =
Craig Topper69186e72014-06-08 08:38:04 +00007842 D ? dyn_cast_or_null<CXXMethodDecl>(D->getAsFunction()) : nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007843 return (Method && Method->isVirtual()) ? 1 : 0;
7844}
Guy Benyei11169dd2012-12-18 14:30:41 +00007845
Alex Lorenzff7f42e2017-07-12 11:35:11 +00007846unsigned clang_EnumDecl_isScoped(CXCursor C) {
7847 if (!clang_isDeclaration(C.kind))
7848 return 0;
7849
7850 const Decl *D = cxcursor::getCursorDecl(C);
7851 auto *Enum = dyn_cast_or_null<EnumDecl>(D);
7852 return (Enum && Enum->isScoped()) ? 1 : 0;
7853}
7854
Guy Benyei11169dd2012-12-18 14:30:41 +00007855//===----------------------------------------------------------------------===//
7856// Attribute introspection.
7857//===----------------------------------------------------------------------===//
7858
Guy Benyei11169dd2012-12-18 14:30:41 +00007859CXType clang_getIBOutletCollectionType(CXCursor C) {
7860 if (C.kind != CXCursor_IBOutletCollectionAttr)
7861 return cxtype::MakeCXType(QualType(), cxcursor::getCursorTU(C));
7862
Dmitri Gribenkoe4baea62013-01-26 18:08:08 +00007863 const IBOutletCollectionAttr *A =
Guy Benyei11169dd2012-12-18 14:30:41 +00007864 cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(C));
7865
7866 return cxtype::MakeCXType(A->getInterface(), cxcursor::getCursorTU(C));
7867}
Guy Benyei11169dd2012-12-18 14:30:41 +00007868
7869//===----------------------------------------------------------------------===//
7870// Inspecting memory usage.
7871//===----------------------------------------------------------------------===//
7872
7873typedef std::vector<CXTUResourceUsageEntry> MemUsageEntries;
7874
7875static inline void createCXTUResourceUsageEntry(MemUsageEntries &entries,
7876 enum CXTUResourceUsageKind k,
7877 unsigned long amount) {
7878 CXTUResourceUsageEntry entry = { k, amount };
7879 entries.push_back(entry);
7880}
7881
Guy Benyei11169dd2012-12-18 14:30:41 +00007882const char *clang_getTUResourceUsageName(CXTUResourceUsageKind kind) {
7883 const char *str = "";
7884 switch (kind) {
7885 case CXTUResourceUsage_AST:
7886 str = "ASTContext: expressions, declarations, and types";
7887 break;
7888 case CXTUResourceUsage_Identifiers:
7889 str = "ASTContext: identifiers";
7890 break;
7891 case CXTUResourceUsage_Selectors:
7892 str = "ASTContext: selectors";
7893 break;
7894 case CXTUResourceUsage_GlobalCompletionResults:
7895 str = "Code completion: cached global results";
7896 break;
7897 case CXTUResourceUsage_SourceManagerContentCache:
7898 str = "SourceManager: content cache allocator";
7899 break;
7900 case CXTUResourceUsage_AST_SideTables:
7901 str = "ASTContext: side tables";
7902 break;
7903 case CXTUResourceUsage_SourceManager_Membuffer_Malloc:
7904 str = "SourceManager: malloc'ed memory buffers";
7905 break;
7906 case CXTUResourceUsage_SourceManager_Membuffer_MMap:
7907 str = "SourceManager: mmap'ed memory buffers";
7908 break;
7909 case CXTUResourceUsage_ExternalASTSource_Membuffer_Malloc:
7910 str = "ExternalASTSource: malloc'ed memory buffers";
7911 break;
7912 case CXTUResourceUsage_ExternalASTSource_Membuffer_MMap:
7913 str = "ExternalASTSource: mmap'ed memory buffers";
7914 break;
7915 case CXTUResourceUsage_Preprocessor:
7916 str = "Preprocessor: malloc'ed memory";
7917 break;
7918 case CXTUResourceUsage_PreprocessingRecord:
7919 str = "Preprocessor: PreprocessingRecord";
7920 break;
7921 case CXTUResourceUsage_SourceManager_DataStructures:
7922 str = "SourceManager: data structures and tables";
7923 break;
7924 case CXTUResourceUsage_Preprocessor_HeaderSearch:
7925 str = "Preprocessor: header search tables";
7926 break;
7927 }
7928 return str;
7929}
7930
7931CXTUResourceUsage clang_getCXTUResourceUsage(CXTranslationUnit TU) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00007932 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00007933 LOG_BAD_TU(TU);
Craig Topper69186e72014-06-08 08:38:04 +00007934 CXTUResourceUsage usage = { (void*) nullptr, 0, nullptr };
Guy Benyei11169dd2012-12-18 14:30:41 +00007935 return usage;
7936 }
7937
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00007938 ASTUnit *astUnit = cxtu::getASTUnit(TU);
Ahmed Charlesb8984322014-03-07 20:03:18 +00007939 std::unique_ptr<MemUsageEntries> entries(new MemUsageEntries());
Guy Benyei11169dd2012-12-18 14:30:41 +00007940 ASTContext &astContext = astUnit->getASTContext();
7941
7942 // How much memory is used by AST nodes and types?
7943 createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_AST,
7944 (unsigned long) astContext.getASTAllocatedMemory());
7945
7946 // How much memory is used by identifiers?
7947 createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_Identifiers,
7948 (unsigned long) astContext.Idents.getAllocator().getTotalMemory());
7949
7950 // How much memory is used for selectors?
7951 createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_Selectors,
7952 (unsigned long) astContext.Selectors.getTotalMemory());
7953
7954 // How much memory is used by ASTContext's side tables?
7955 createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_AST_SideTables,
7956 (unsigned long) astContext.getSideTableAllocatedMemory());
7957
7958 // How much memory is used for caching global code completion results?
7959 unsigned long completionBytes = 0;
7960 if (GlobalCodeCompletionAllocator *completionAllocator =
Alp Tokerf994cef2014-07-05 03:08:06 +00007961 astUnit->getCachedCompletionAllocator().get()) {
Guy Benyei11169dd2012-12-18 14:30:41 +00007962 completionBytes = completionAllocator->getTotalMemory();
7963 }
7964 createCXTUResourceUsageEntry(*entries,
7965 CXTUResourceUsage_GlobalCompletionResults,
7966 completionBytes);
7967
7968 // How much memory is being used by SourceManager's content cache?
7969 createCXTUResourceUsageEntry(*entries,
7970 CXTUResourceUsage_SourceManagerContentCache,
7971 (unsigned long) astContext.getSourceManager().getContentCacheSize());
7972
7973 // How much memory is being used by the MemoryBuffer's in SourceManager?
7974 const SourceManager::MemoryBufferSizes &srcBufs =
7975 astUnit->getSourceManager().getMemoryBufferSizes();
7976
7977 createCXTUResourceUsageEntry(*entries,
7978 CXTUResourceUsage_SourceManager_Membuffer_Malloc,
7979 (unsigned long) srcBufs.malloc_bytes);
7980 createCXTUResourceUsageEntry(*entries,
7981 CXTUResourceUsage_SourceManager_Membuffer_MMap,
7982 (unsigned long) srcBufs.mmap_bytes);
7983 createCXTUResourceUsageEntry(*entries,
7984 CXTUResourceUsage_SourceManager_DataStructures,
7985 (unsigned long) astContext.getSourceManager()
7986 .getDataStructureSizes());
7987
7988 // How much memory is being used by the ExternalASTSource?
7989 if (ExternalASTSource *esrc = astContext.getExternalSource()) {
7990 const ExternalASTSource::MemoryBufferSizes &sizes =
7991 esrc->getMemoryBufferSizes();
7992
7993 createCXTUResourceUsageEntry(*entries,
7994 CXTUResourceUsage_ExternalASTSource_Membuffer_Malloc,
7995 (unsigned long) sizes.malloc_bytes);
7996 createCXTUResourceUsageEntry(*entries,
7997 CXTUResourceUsage_ExternalASTSource_Membuffer_MMap,
7998 (unsigned long) sizes.mmap_bytes);
7999 }
8000
8001 // How much memory is being used by the Preprocessor?
8002 Preprocessor &pp = astUnit->getPreprocessor();
8003 createCXTUResourceUsageEntry(*entries,
8004 CXTUResourceUsage_Preprocessor,
8005 pp.getTotalMemory());
8006
8007 if (PreprocessingRecord *pRec = pp.getPreprocessingRecord()) {
8008 createCXTUResourceUsageEntry(*entries,
8009 CXTUResourceUsage_PreprocessingRecord,
8010 pRec->getTotalMemory());
8011 }
8012
8013 createCXTUResourceUsageEntry(*entries,
8014 CXTUResourceUsage_Preprocessor_HeaderSearch,
8015 pp.getHeaderSearchInfo().getTotalMemory());
Craig Topper69186e72014-06-08 08:38:04 +00008016
Guy Benyei11169dd2012-12-18 14:30:41 +00008017 CXTUResourceUsage usage = { (void*) entries.get(),
8018 (unsigned) entries->size(),
Alexander Kornienko6ee521c2015-01-23 15:36:10 +00008019 !entries->empty() ? &(*entries)[0] : nullptr };
Eric Fiseliere95fc442016-11-14 07:03:50 +00008020 (void)entries.release();
Guy Benyei11169dd2012-12-18 14:30:41 +00008021 return usage;
8022}
8023
8024void clang_disposeCXTUResourceUsage(CXTUResourceUsage usage) {
8025 if (usage.data)
8026 delete (MemUsageEntries*) usage.data;
8027}
8028
Argyrios Kyrtzidis0e282ef2013-12-06 18:55:45 +00008029CXSourceRangeList *clang_getSkippedRanges(CXTranslationUnit TU, CXFile file) {
8030 CXSourceRangeList *skipped = new CXSourceRangeList;
Argyrios Kyrtzidis9ef57752013-12-05 08:19:32 +00008031 skipped->count = 0;
Craig Topper69186e72014-06-08 08:38:04 +00008032 skipped->ranges = nullptr;
Argyrios Kyrtzidis9ef57752013-12-05 08:19:32 +00008033
Dmitri Gribenko852d6222014-02-11 15:02:48 +00008034 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00008035 LOG_BAD_TU(TU);
8036 return skipped;
8037 }
8038
Argyrios Kyrtzidis9ef57752013-12-05 08:19:32 +00008039 if (!file)
8040 return skipped;
8041
8042 ASTUnit *astUnit = cxtu::getASTUnit(TU);
8043 PreprocessingRecord *ppRec = astUnit->getPreprocessor().getPreprocessingRecord();
8044 if (!ppRec)
8045 return skipped;
8046
8047 ASTContext &Ctx = astUnit->getASTContext();
8048 SourceManager &sm = Ctx.getSourceManager();
8049 FileEntry *fileEntry = static_cast<FileEntry *>(file);
8050 FileID wantedFileID = sm.translateFile(fileEntry);
8051
8052 const std::vector<SourceRange> &SkippedRanges = ppRec->getSkippedRanges();
8053 std::vector<SourceRange> wantedRanges;
8054 for (std::vector<SourceRange>::const_iterator i = SkippedRanges.begin(), ei = SkippedRanges.end();
8055 i != ei; ++i) {
8056 if (sm.getFileID(i->getBegin()) == wantedFileID || sm.getFileID(i->getEnd()) == wantedFileID)
8057 wantedRanges.push_back(*i);
8058 }
8059
8060 skipped->count = wantedRanges.size();
8061 skipped->ranges = new CXSourceRange[skipped->count];
8062 for (unsigned i = 0, ei = skipped->count; i != ei; ++i)
8063 skipped->ranges[i] = cxloc::translateSourceRange(Ctx, wantedRanges[i]);
8064
8065 return skipped;
8066}
8067
Cameron Desrochersd8091282016-08-18 15:43:55 +00008068CXSourceRangeList *clang_getAllSkippedRanges(CXTranslationUnit TU) {
8069 CXSourceRangeList *skipped = new CXSourceRangeList;
8070 skipped->count = 0;
8071 skipped->ranges = nullptr;
8072
8073 if (isNotUsableTU(TU)) {
8074 LOG_BAD_TU(TU);
8075 return skipped;
8076 }
8077
8078 ASTUnit *astUnit = cxtu::getASTUnit(TU);
8079 PreprocessingRecord *ppRec = astUnit->getPreprocessor().getPreprocessingRecord();
8080 if (!ppRec)
8081 return skipped;
8082
8083 ASTContext &Ctx = astUnit->getASTContext();
8084
8085 const std::vector<SourceRange> &SkippedRanges = ppRec->getSkippedRanges();
8086
8087 skipped->count = SkippedRanges.size();
8088 skipped->ranges = new CXSourceRange[skipped->count];
8089 for (unsigned i = 0, ei = skipped->count; i != ei; ++i)
8090 skipped->ranges[i] = cxloc::translateSourceRange(Ctx, SkippedRanges[i]);
8091
8092 return skipped;
8093}
8094
Argyrios Kyrtzidis0e282ef2013-12-06 18:55:45 +00008095void clang_disposeSourceRangeList(CXSourceRangeList *ranges) {
8096 if (ranges) {
8097 delete[] ranges->ranges;
8098 delete ranges;
Argyrios Kyrtzidis9ef57752013-12-05 08:19:32 +00008099 }
8100}
8101
Guy Benyei11169dd2012-12-18 14:30:41 +00008102void clang::PrintLibclangResourceUsage(CXTranslationUnit TU) {
8103 CXTUResourceUsage Usage = clang_getCXTUResourceUsage(TU);
8104 for (unsigned I = 0; I != Usage.numEntries; ++I)
8105 fprintf(stderr, " %s: %lu\n",
8106 clang_getTUResourceUsageName(Usage.entries[I].kind),
8107 Usage.entries[I].amount);
8108
8109 clang_disposeCXTUResourceUsage(Usage);
8110}
8111
8112//===----------------------------------------------------------------------===//
8113// Misc. utility functions.
8114//===----------------------------------------------------------------------===//
8115
8116/// Default to using an 8 MB stack size on "safety" threads.
8117static unsigned SafetyStackThreadSize = 8 << 20;
8118
8119namespace clang {
8120
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00008121bool RunSafely(llvm::CrashRecoveryContext &CRC, llvm::function_ref<void()> Fn,
Guy Benyei11169dd2012-12-18 14:30:41 +00008122 unsigned Size) {
8123 if (!Size)
8124 Size = GetSafetyThreadStackSize();
8125 if (Size)
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00008126 return CRC.RunSafelyOnThread(Fn, Size);
8127 return CRC.RunSafely(Fn);
Guy Benyei11169dd2012-12-18 14:30:41 +00008128}
8129
8130unsigned GetSafetyThreadStackSize() {
8131 return SafetyStackThreadSize;
8132}
8133
8134void SetSafetyThreadStackSize(unsigned Value) {
8135 SafetyStackThreadSize = Value;
8136}
8137
Alexander Kornienkoab9db512015-06-22 23:07:51 +00008138}
Guy Benyei11169dd2012-12-18 14:30:41 +00008139
8140void clang::setThreadBackgroundPriority() {
8141 if (getenv("LIBCLANG_BGPRIO_DISABLE"))
8142 return;
8143
Alp Toker1a86ad22014-07-06 06:24:00 +00008144#ifdef USE_DARWIN_THREADS
Guy Benyei11169dd2012-12-18 14:30:41 +00008145 setpriority(PRIO_DARWIN_THREAD, 0, PRIO_DARWIN_BG);
8146#endif
8147}
8148
8149void cxindex::printDiagsToStderr(ASTUnit *Unit) {
8150 if (!Unit)
8151 return;
8152
8153 for (ASTUnit::stored_diag_iterator D = Unit->stored_diag_begin(),
8154 DEnd = Unit->stored_diag_end();
8155 D != DEnd; ++D) {
Ben Langmuir749323f2014-04-22 17:40:12 +00008156 CXStoredDiagnostic Diag(*D, Unit->getLangOpts());
Guy Benyei11169dd2012-12-18 14:30:41 +00008157 CXString Msg = clang_formatDiagnostic(&Diag,
8158 clang_defaultDiagnosticDisplayOptions());
8159 fprintf(stderr, "%s\n", clang_getCString(Msg));
8160 clang_disposeString(Msg);
8161 }
8162#ifdef LLVM_ON_WIN32
8163 // On Windows, force a flush, since there may be multiple copies of
8164 // stderr and stdout in the file system, all with different buffers
8165 // but writing to the same device.
8166 fflush(stderr);
8167#endif
8168}
8169
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008170MacroInfo *cxindex::getMacroInfo(const IdentifierInfo &II,
8171 SourceLocation MacroDefLoc,
8172 CXTranslationUnit TU){
8173 if (MacroDefLoc.isInvalid() || !TU)
Craig Topper69186e72014-06-08 08:38:04 +00008174 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008175 if (!II.hadMacroDefinition())
Craig Topper69186e72014-06-08 08:38:04 +00008176 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008177
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00008178 ASTUnit *Unit = cxtu::getASTUnit(TU);
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00008179 Preprocessor &PP = Unit->getPreprocessor();
Richard Smith20e883e2015-04-29 23:20:19 +00008180 MacroDirective *MD = PP.getLocalMacroDirectiveHistory(&II);
Argyrios Kyrtzidisb6210df2013-03-26 17:17:01 +00008181 if (MD) {
8182 for (MacroDirective::DefInfo
8183 Def = MD->getDefinition(); Def; Def = Def.getPreviousDefinition()) {
8184 if (MacroDefLoc == Def.getMacroInfo()->getDefinitionLoc())
8185 return Def.getMacroInfo();
8186 }
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008187 }
8188
Craig Topper69186e72014-06-08 08:38:04 +00008189 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008190}
8191
Richard Smith66a81862015-05-04 02:25:31 +00008192const MacroInfo *cxindex::getMacroInfo(const MacroDefinitionRecord *MacroDef,
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00008193 CXTranslationUnit TU) {
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008194 if (!MacroDef || !TU)
Craig Topper69186e72014-06-08 08:38:04 +00008195 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008196 const IdentifierInfo *II = MacroDef->getName();
8197 if (!II)
Craig Topper69186e72014-06-08 08:38:04 +00008198 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008199
8200 return getMacroInfo(*II, MacroDef->getLocation(), TU);
8201}
8202
Richard Smith66a81862015-05-04 02:25:31 +00008203MacroDefinitionRecord *
8204cxindex::checkForMacroInMacroDefinition(const MacroInfo *MI, const Token &Tok,
8205 CXTranslationUnit TU) {
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008206 if (!MI || !TU)
Craig Topper69186e72014-06-08 08:38:04 +00008207 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008208 if (Tok.isNot(tok::raw_identifier))
Craig Topper69186e72014-06-08 08:38:04 +00008209 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008210
8211 if (MI->getNumTokens() == 0)
Craig Topper69186e72014-06-08 08:38:04 +00008212 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008213 SourceRange DefRange(MI->getReplacementToken(0).getLocation(),
8214 MI->getDefinitionEndLoc());
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00008215 ASTUnit *Unit = cxtu::getASTUnit(TU);
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008216
8217 // Check that the token is inside the definition and not its argument list.
8218 SourceManager &SM = Unit->getSourceManager();
8219 if (SM.isBeforeInTranslationUnit(Tok.getLocation(), DefRange.getBegin()))
Craig Topper69186e72014-06-08 08:38:04 +00008220 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008221 if (SM.isBeforeInTranslationUnit(DefRange.getEnd(), Tok.getLocation()))
Craig Topper69186e72014-06-08 08:38:04 +00008222 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008223
8224 Preprocessor &PP = Unit->getPreprocessor();
8225 PreprocessingRecord *PPRec = PP.getPreprocessingRecord();
8226 if (!PPRec)
Craig Topper69186e72014-06-08 08:38:04 +00008227 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008228
Alp Toker2d57cea2014-05-17 04:53:25 +00008229 IdentifierInfo &II = PP.getIdentifierTable().get(Tok.getRawIdentifier());
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008230 if (!II.hadMacroDefinition())
Craig Topper69186e72014-06-08 08:38:04 +00008231 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008232
8233 // Check that the identifier is not one of the macro arguments.
Faisal Valiac506d72017-07-17 17:18:43 +00008234 if (std::find(MI->param_begin(), MI->param_end(), &II) != MI->param_end())
Craig Topper69186e72014-06-08 08:38:04 +00008235 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008236
Richard Smith20e883e2015-04-29 23:20:19 +00008237 MacroDirective *InnerMD = PP.getLocalMacroDirectiveHistory(&II);
Argyrios Kyrtzidis09c9e812013-02-20 00:54:57 +00008238 if (!InnerMD)
Craig Topper69186e72014-06-08 08:38:04 +00008239 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008240
Argyrios Kyrtzidisb6210df2013-03-26 17:17:01 +00008241 return PPRec->findMacroDefinition(InnerMD->getMacroInfo());
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008242}
8243
Richard Smith66a81862015-05-04 02:25:31 +00008244MacroDefinitionRecord *
8245cxindex::checkForMacroInMacroDefinition(const MacroInfo *MI, SourceLocation Loc,
8246 CXTranslationUnit TU) {
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008247 if (Loc.isInvalid() || !MI || !TU)
Craig Topper69186e72014-06-08 08:38:04 +00008248 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008249
8250 if (MI->getNumTokens() == 0)
Craig Topper69186e72014-06-08 08:38:04 +00008251 return nullptr;
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00008252 ASTUnit *Unit = cxtu::getASTUnit(TU);
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008253 Preprocessor &PP = Unit->getPreprocessor();
8254 if (!PP.getPreprocessingRecord())
Craig Topper69186e72014-06-08 08:38:04 +00008255 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008256 Loc = Unit->getSourceManager().getSpellingLoc(Loc);
8257 Token Tok;
8258 if (PP.getRawToken(Loc, Tok))
Craig Topper69186e72014-06-08 08:38:04 +00008259 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00008260
8261 return checkForMacroInMacroDefinition(MI, Tok, TU);
8262}
8263
Guy Benyei11169dd2012-12-18 14:30:41 +00008264CXString clang_getClangVersion() {
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00008265 return cxstring::createDup(getClangFullVersion());
Guy Benyei11169dd2012-12-18 14:30:41 +00008266}
8267
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00008268Logger &cxindex::Logger::operator<<(CXTranslationUnit TU) {
8269 if (TU) {
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00008270 if (ASTUnit *Unit = cxtu::getASTUnit(TU)) {
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00008271 LogOS << '<' << Unit->getMainFileName() << '>';
Argyrios Kyrtzidis37f2ab42013-03-05 20:21:14 +00008272 if (Unit->isMainFileAST())
8273 LogOS << " (" << Unit->getASTFileName() << ')';
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00008274 return *this;
8275 }
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00008276 } else {
8277 LogOS << "<NULL TU>";
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00008278 }
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00008279 return *this;
8280}
8281
Argyrios Kyrtzidisba4b5f82013-03-08 02:32:26 +00008282Logger &cxindex::Logger::operator<<(const FileEntry *FE) {
8283 *this << FE->getName();
8284 return *this;
8285}
8286
8287Logger &cxindex::Logger::operator<<(CXCursor cursor) {
8288 CXString cursorName = clang_getCursorDisplayName(cursor);
8289 *this << cursorName << "@" << clang_getCursorLocation(cursor);
8290 clang_disposeString(cursorName);
8291 return *this;
8292}
8293
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00008294Logger &cxindex::Logger::operator<<(CXSourceLocation Loc) {
8295 CXFile File;
8296 unsigned Line, Column;
Craig Topper69186e72014-06-08 08:38:04 +00008297 clang_getFileLocation(Loc, &File, &Line, &Column, nullptr);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00008298 CXString FileName = clang_getFileName(File);
8299 *this << llvm::format("(%s:%d:%d)", clang_getCString(FileName), Line, Column);
8300 clang_disposeString(FileName);
8301 return *this;
8302}
8303
8304Logger &cxindex::Logger::operator<<(CXSourceRange range) {
8305 CXSourceLocation BLoc = clang_getRangeStart(range);
8306 CXSourceLocation ELoc = clang_getRangeEnd(range);
8307
8308 CXFile BFile;
8309 unsigned BLine, BColumn;
Craig Topper69186e72014-06-08 08:38:04 +00008310 clang_getFileLocation(BLoc, &BFile, &BLine, &BColumn, nullptr);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00008311
8312 CXFile EFile;
8313 unsigned ELine, EColumn;
Craig Topper69186e72014-06-08 08:38:04 +00008314 clang_getFileLocation(ELoc, &EFile, &ELine, &EColumn, nullptr);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00008315
8316 CXString BFileName = clang_getFileName(BFile);
8317 if (BFile == EFile) {
8318 *this << llvm::format("[%s %d:%d-%d:%d]", clang_getCString(BFileName),
8319 BLine, BColumn, ELine, EColumn);
8320 } else {
8321 CXString EFileName = clang_getFileName(EFile);
8322 *this << llvm::format("[%s:%d:%d - ", clang_getCString(BFileName),
8323 BLine, BColumn)
8324 << llvm::format("%s:%d:%d]", clang_getCString(EFileName),
8325 ELine, EColumn);
8326 clang_disposeString(EFileName);
8327 }
8328 clang_disposeString(BFileName);
8329 return *this;
8330}
8331
8332Logger &cxindex::Logger::operator<<(CXString Str) {
8333 *this << clang_getCString(Str);
8334 return *this;
8335}
8336
8337Logger &cxindex::Logger::operator<<(const llvm::format_object_base &Fmt) {
8338 LogOS << Fmt;
8339 return *this;
8340}
8341
Chandler Carruth37ad2582014-06-27 15:14:39 +00008342static llvm::ManagedStatic<llvm::sys::Mutex> LoggingMutex;
8343
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00008344cxindex::Logger::~Logger() {
Chandler Carruth37ad2582014-06-27 15:14:39 +00008345 llvm::sys::ScopedLock L(*LoggingMutex);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00008346
8347 static llvm::TimeRecord sBeginTR = llvm::TimeRecord::getCurrentTime();
8348
Dmitri Gribenkof8579502013-01-12 19:30:44 +00008349 raw_ostream &OS = llvm::errs();
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00008350 OS << "[libclang:" << Name << ':';
8351
Alp Toker1a86ad22014-07-06 06:24:00 +00008352#ifdef USE_DARWIN_THREADS
8353 // TODO: Portability.
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00008354 mach_port_t tid = pthread_mach_thread_np(pthread_self());
8355 OS << tid << ':';
8356#endif
8357
8358 llvm::TimeRecord TR = llvm::TimeRecord::getCurrentTime();
8359 OS << llvm::format("%7.4f] ", TR.getWallTime() - sBeginTR.getWallTime());
Yaron Keren09fb7c62015-03-10 07:33:23 +00008360 OS << Msg << '\n';
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00008361
8362 if (Trace) {
Zachary Turner1fe2a8d2015-03-05 19:15:09 +00008363 llvm::sys::PrintStackTrace(OS);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00008364 OS << "--------------------------------------------------\n";
8365 }
8366}
Benjamin Kramerc1ffdab2016-03-03 08:58:18 +00008367
8368#ifdef CLANG_TOOL_EXTRA_BUILD
8369// This anchor is used to force the linker to link the clang-tidy plugin.
8370extern volatile int ClangTidyPluginAnchorSource;
8371static int LLVM_ATTRIBUTE_UNUSED ClangTidyPluginAnchorDestination =
8372 ClangTidyPluginAnchorSource;
Benjamin Kramer9eba7352016-11-17 15:22:36 +00008373
8374// This anchor is used to force the linker to link the clang-include-fixer
8375// plugin.
8376extern volatile int ClangIncludeFixerPluginAnchorSource;
8377static int LLVM_ATTRIBUTE_UNUSED ClangIncludeFixerPluginAnchorDestination =
8378 ClangIncludeFixerPluginAnchorSource;
Benjamin Kramerc1ffdab2016-03-03 08:58:18 +00008379#endif