blob: 4d1cfb11658ce1c98adcd89c82a4cfa098366517 [file] [log] [blame]
Guy Benyei11169dd2012-12-18 14:30:41 +00001//===- CIndex.cpp - Clang-C Source Indexing Library -----------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the main API hooks in the Clang-C Source Indexing
11// library.
12//
13//===----------------------------------------------------------------------===//
14
15#include "CIndexer.h"
16#include "CIndexDiagnostic.h"
Chandler Carruth4b417452013-01-19 08:09:44 +000017#include "CLog.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000018#include "CXCursor.h"
19#include "CXSourceLocation.h"
20#include "CXString.h"
21#include "CXTranslationUnit.h"
22#include "CXType.h"
23#include "CursorVisitor.h"
David Blaikie0a4e61f2013-09-13 18:32:52 +000024#include "clang/AST/Attr.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000025#include "clang/AST/StmtVisitor.h"
26#include "clang/Basic/Diagnostic.h"
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +000027#include "clang/Basic/DiagnosticCategories.h"
28#include "clang/Basic/DiagnosticIDs.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000029#include "clang/Basic/Version.h"
30#include "clang/Frontend/ASTUnit.h"
31#include "clang/Frontend/CompilerInstance.h"
32#include "clang/Frontend/FrontendDiagnostic.h"
Argyrios Kyrtzidisca741ce2016-02-14 22:30:14 +000033#include "clang/Index/CodegenNameGenerator.h"
Dmitri Gribenko9e605112013-11-13 22:16:51 +000034#include "clang/Index/CommentToXML.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000035#include "clang/Lex/HeaderSearch.h"
36#include "clang/Lex/Lexer.h"
37#include "clang/Lex/PreprocessingRecord.h"
38#include "clang/Lex/Preprocessor.h"
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +000039#include "clang/Serialization/SerializationDiagnostic.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000040#include "llvm/ADT/Optional.h"
41#include "llvm/ADT/STLExtras.h"
42#include "llvm/ADT/StringSwitch.h"
Alp Toker1d257e12014-06-04 03:28:55 +000043#include "llvm/Config/llvm-config.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000044#include "llvm/Support/Compiler.h"
45#include "llvm/Support/CrashRecoveryContext.h"
Chandler Carruth4b417452013-01-19 08:09:44 +000046#include "llvm/Support/Format.h"
Chandler Carruth37ad2582014-06-27 15:14:39 +000047#include "llvm/Support/ManagedStatic.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000048#include "llvm/Support/MemoryBuffer.h"
49#include "llvm/Support/Mutex.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000050#include "llvm/Support/Program.h"
51#include "llvm/Support/SaveAndRestore.h"
52#include "llvm/Support/Signals.h"
Adrian Prantlbc068582015-07-08 01:00:30 +000053#include "llvm/Support/TargetSelect.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000054#include "llvm/Support/Threading.h"
55#include "llvm/Support/Timer.h"
56#include "llvm/Support/raw_ostream.h"
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +000057
Alp Toker1a86ad22014-07-06 06:24:00 +000058#if LLVM_ENABLE_THREADS != 0 && defined(__APPLE__)
59#define USE_DARWIN_THREADS
60#endif
61
62#ifdef USE_DARWIN_THREADS
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +000063#include <pthread.h>
64#endif
Guy Benyei11169dd2012-12-18 14:30:41 +000065
66using namespace clang;
67using namespace clang::cxcursor;
Guy Benyei11169dd2012-12-18 14:30:41 +000068using namespace clang::cxtu;
69using namespace clang::cxindex;
70
Dmitri Gribenkod36209e2013-01-26 21:32:42 +000071CXTranslationUnit cxtu::MakeCXTranslationUnit(CIndexer *CIdx, ASTUnit *AU) {
72 if (!AU)
Craig Topper69186e72014-06-08 08:38:04 +000073 return nullptr;
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +000074 assert(CIdx);
Guy Benyei11169dd2012-12-18 14:30:41 +000075 CXTranslationUnit D = new CXTranslationUnitImpl();
76 D->CIdx = CIdx;
Dmitri Gribenkod36209e2013-01-26 21:32:42 +000077 D->TheASTUnit = AU;
Dmitri Gribenko74895212013-02-03 13:52:47 +000078 D->StringPool = new cxstring::CXStringPool();
Craig Topper69186e72014-06-08 08:38:04 +000079 D->Diagnostics = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +000080 D->OverridenCursorsPool = createOverridenCXCursorsPool();
Craig Topper69186e72014-06-08 08:38:04 +000081 D->CommentToXML = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +000082 return D;
83}
84
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +000085bool cxtu::isASTReadError(ASTUnit *AU) {
86 for (ASTUnit::stored_diag_iterator D = AU->stored_diag_begin(),
87 DEnd = AU->stored_diag_end();
88 D != DEnd; ++D) {
89 if (D->getLevel() >= DiagnosticsEngine::Error &&
90 DiagnosticIDs::getCategoryNumberForDiag(D->getID()) ==
91 diag::DiagCat_AST_Deserialization_Issue)
92 return true;
93 }
94 return false;
95}
96
Guy Benyei11169dd2012-12-18 14:30:41 +000097cxtu::CXTUOwner::~CXTUOwner() {
98 if (TU)
99 clang_disposeTranslationUnit(TU);
100}
101
102/// \brief Compare two source ranges to determine their relative position in
103/// the translation unit.
104static RangeComparisonResult RangeCompare(SourceManager &SM,
105 SourceRange R1,
106 SourceRange R2) {
107 assert(R1.isValid() && "First range is invalid?");
108 assert(R2.isValid() && "Second range is invalid?");
109 if (R1.getEnd() != R2.getBegin() &&
110 SM.isBeforeInTranslationUnit(R1.getEnd(), R2.getBegin()))
111 return RangeBefore;
112 if (R2.getEnd() != R1.getBegin() &&
113 SM.isBeforeInTranslationUnit(R2.getEnd(), R1.getBegin()))
114 return RangeAfter;
115 return RangeOverlap;
116}
117
118/// \brief Determine if a source location falls within, before, or after a
119/// a given source range.
120static RangeComparisonResult LocationCompare(SourceManager &SM,
121 SourceLocation L, SourceRange R) {
122 assert(R.isValid() && "First range is invalid?");
123 assert(L.isValid() && "Second range is invalid?");
124 if (L == R.getBegin() || L == R.getEnd())
125 return RangeOverlap;
126 if (SM.isBeforeInTranslationUnit(L, R.getBegin()))
127 return RangeBefore;
128 if (SM.isBeforeInTranslationUnit(R.getEnd(), L))
129 return RangeAfter;
130 return RangeOverlap;
131}
132
133/// \brief Translate a Clang source range into a CIndex source range.
134///
135/// Clang internally represents ranges where the end location points to the
136/// start of the token at the end. However, for external clients it is more
137/// useful to have a CXSourceRange be a proper half-open interval. This routine
138/// does the appropriate translation.
139CXSourceRange cxloc::translateSourceRange(const SourceManager &SM,
140 const LangOptions &LangOpts,
141 const CharSourceRange &R) {
142 // We want the last character in this location, so we will adjust the
143 // location accordingly.
144 SourceLocation EndLoc = R.getEnd();
145 if (EndLoc.isValid() && EndLoc.isMacroID() && !SM.isMacroArgExpansion(EndLoc))
146 EndLoc = SM.getExpansionRange(EndLoc).second;
Yaron Keren8b563662015-10-03 10:46:20 +0000147 if (R.isTokenRange() && EndLoc.isValid()) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000148 unsigned Length = Lexer::MeasureTokenLength(SM.getSpellingLoc(EndLoc),
149 SM, LangOpts);
150 EndLoc = EndLoc.getLocWithOffset(Length);
151 }
152
Bill Wendlingeade3622013-01-23 08:25:41 +0000153 CXSourceRange Result = {
Dmitri Gribenkof9304482013-01-23 15:56:07 +0000154 { &SM, &LangOpts },
Bill Wendlingeade3622013-01-23 08:25:41 +0000155 R.getBegin().getRawEncoding(),
156 EndLoc.getRawEncoding()
157 };
Guy Benyei11169dd2012-12-18 14:30:41 +0000158 return Result;
159}
160
161//===----------------------------------------------------------------------===//
162// Cursor visitor.
163//===----------------------------------------------------------------------===//
164
165static SourceRange getRawCursorExtent(CXCursor C);
166static SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr);
167
168
169RangeComparisonResult CursorVisitor::CompareRegionOfInterest(SourceRange R) {
170 return RangeCompare(AU->getSourceManager(), R, RegionOfInterest);
171}
172
173/// \brief Visit the given cursor and, if requested by the visitor,
174/// its children.
175///
176/// \param Cursor the cursor to visit.
177///
178/// \param CheckedRegionOfInterest if true, then the caller already checked
179/// that this cursor is within the region of interest.
180///
181/// \returns true if the visitation should be aborted, false if it
182/// should continue.
183bool CursorVisitor::Visit(CXCursor Cursor, bool CheckedRegionOfInterest) {
184 if (clang_isInvalid(Cursor.kind))
185 return false;
186
187 if (clang_isDeclaration(Cursor.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +0000188 const Decl *D = getCursorDecl(Cursor);
Guy Benyei11169dd2012-12-18 14:30:41 +0000189 if (!D) {
190 assert(0 && "Invalid declaration cursor");
191 return true; // abort.
192 }
193
194 // Ignore implicit declarations, unless it's an objc method because
195 // currently we should report implicit methods for properties when indexing.
196 if (D->isImplicit() && !isa<ObjCMethodDecl>(D))
197 return false;
198 }
199
200 // If we have a range of interest, and this cursor doesn't intersect with it,
201 // we're done.
202 if (RegionOfInterest.isValid() && !CheckedRegionOfInterest) {
203 SourceRange Range = getRawCursorExtent(Cursor);
204 if (Range.isInvalid() || CompareRegionOfInterest(Range))
205 return false;
206 }
207
208 switch (Visitor(Cursor, Parent, ClientData)) {
209 case CXChildVisit_Break:
210 return true;
211
212 case CXChildVisit_Continue:
213 return false;
214
215 case CXChildVisit_Recurse: {
216 bool ret = VisitChildren(Cursor);
217 if (PostChildrenVisitor)
218 if (PostChildrenVisitor(Cursor, ClientData))
219 return true;
220 return ret;
221 }
222 }
223
224 llvm_unreachable("Invalid CXChildVisitResult!");
225}
226
227static bool visitPreprocessedEntitiesInRange(SourceRange R,
228 PreprocessingRecord &PPRec,
229 CursorVisitor &Visitor) {
230 SourceManager &SM = Visitor.getASTUnit()->getSourceManager();
231 FileID FID;
232
233 if (!Visitor.shouldVisitIncludedEntities()) {
234 // If the begin/end of the range lie in the same FileID, do the optimization
235 // where we skip preprocessed entities that do not come from the same FileID.
236 FID = SM.getFileID(SM.getFileLoc(R.getBegin()));
237 if (FID != SM.getFileID(SM.getFileLoc(R.getEnd())))
238 FID = FileID();
239 }
240
Benjamin Kramerb4ef6682015-02-06 17:25:10 +0000241 const auto &Entities = PPRec.getPreprocessedEntitiesInRange(R);
242 return Visitor.visitPreprocessedEntities(Entities.begin(), Entities.end(),
Guy Benyei11169dd2012-12-18 14:30:41 +0000243 PPRec, FID);
244}
245
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000246bool CursorVisitor::visitFileRegion() {
Guy Benyei11169dd2012-12-18 14:30:41 +0000247 if (RegionOfInterest.isInvalid())
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000248 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000249
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +0000250 ASTUnit *Unit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +0000251 SourceManager &SM = Unit->getSourceManager();
252
253 std::pair<FileID, unsigned>
254 Begin = SM.getDecomposedLoc(SM.getFileLoc(RegionOfInterest.getBegin())),
255 End = SM.getDecomposedLoc(SM.getFileLoc(RegionOfInterest.getEnd()));
256
257 if (End.first != Begin.first) {
258 // If the end does not reside in the same file, try to recover by
259 // picking the end of the file of begin location.
260 End.first = Begin.first;
261 End.second = SM.getFileIDSize(Begin.first);
262 }
263
264 assert(Begin.first == End.first);
265 if (Begin.second > End.second)
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000266 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000267
268 FileID File = Begin.first;
269 unsigned Offset = Begin.second;
270 unsigned Length = End.second - Begin.second;
271
272 if (!VisitDeclsOnly && !VisitPreprocessorLast)
273 if (visitPreprocessedEntitiesInRegion())
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000274 return true; // visitation break.
Guy Benyei11169dd2012-12-18 14:30:41 +0000275
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000276 if (visitDeclsFromFileRegion(File, Offset, Length))
277 return true; // visitation break.
Guy Benyei11169dd2012-12-18 14:30:41 +0000278
279 if (!VisitDeclsOnly && VisitPreprocessorLast)
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000280 return visitPreprocessedEntitiesInRegion();
281
282 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000283}
284
285static bool isInLexicalContext(Decl *D, DeclContext *DC) {
286 if (!DC)
287 return false;
288
289 for (DeclContext *DeclDC = D->getLexicalDeclContext();
290 DeclDC; DeclDC = DeclDC->getLexicalParent()) {
291 if (DeclDC == DC)
292 return true;
293 }
294 return false;
295}
296
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000297bool CursorVisitor::visitDeclsFromFileRegion(FileID File,
Guy Benyei11169dd2012-12-18 14:30:41 +0000298 unsigned Offset, unsigned Length) {
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +0000299 ASTUnit *Unit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +0000300 SourceManager &SM = Unit->getSourceManager();
301 SourceRange Range = RegionOfInterest;
302
303 SmallVector<Decl *, 16> Decls;
304 Unit->findFileRegionDecls(File, Offset, Length, Decls);
305
306 // If we didn't find any file level decls for the file, try looking at the
307 // file that it was included from.
308 while (Decls.empty() || Decls.front()->isTopLevelDeclInObjCContainer()) {
309 bool Invalid = false;
310 const SrcMgr::SLocEntry &SLEntry = SM.getSLocEntry(File, &Invalid);
311 if (Invalid)
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000312 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000313
314 SourceLocation Outer;
315 if (SLEntry.isFile())
316 Outer = SLEntry.getFile().getIncludeLoc();
317 else
318 Outer = SLEntry.getExpansion().getExpansionLocStart();
319 if (Outer.isInvalid())
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000320 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000321
Benjamin Kramer867ea1d2014-03-02 13:01:17 +0000322 std::tie(File, Offset) = SM.getDecomposedExpansionLoc(Outer);
Guy Benyei11169dd2012-12-18 14:30:41 +0000323 Length = 0;
324 Unit->findFileRegionDecls(File, Offset, Length, Decls);
325 }
326
327 assert(!Decls.empty());
328
329 bool VisitedAtLeastOnce = false;
Craig Topper69186e72014-06-08 08:38:04 +0000330 DeclContext *CurDC = nullptr;
Craig Topper2341c0d2013-07-04 03:08:24 +0000331 SmallVectorImpl<Decl *>::iterator DIt = Decls.begin();
332 for (SmallVectorImpl<Decl *>::iterator DE = Decls.end(); DIt != DE; ++DIt) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000333 Decl *D = *DIt;
334 if (D->getSourceRange().isInvalid())
335 continue;
336
337 if (isInLexicalContext(D, CurDC))
338 continue;
339
340 CurDC = dyn_cast<DeclContext>(D);
341
342 if (TagDecl *TD = dyn_cast<TagDecl>(D))
343 if (!TD->isFreeStanding())
344 continue;
345
346 RangeComparisonResult CompRes = RangeCompare(SM, D->getSourceRange(),Range);
347 if (CompRes == RangeBefore)
348 continue;
349 if (CompRes == RangeAfter)
350 break;
351
352 assert(CompRes == RangeOverlap);
353 VisitedAtLeastOnce = true;
354
355 if (isa<ObjCContainerDecl>(D)) {
356 FileDI_current = &DIt;
357 FileDE_current = DE;
358 } else {
Craig Topper69186e72014-06-08 08:38:04 +0000359 FileDI_current = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +0000360 }
361
362 if (Visit(MakeCXCursor(D, TU, Range), /*CheckedRegionOfInterest=*/true))
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000363 return true; // visitation break.
Guy Benyei11169dd2012-12-18 14:30:41 +0000364 }
365
366 if (VisitedAtLeastOnce)
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000367 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000368
369 // No Decls overlapped with the range. Move up the lexical context until there
370 // is a context that contains the range or we reach the translation unit
371 // level.
372 DeclContext *DC = DIt == Decls.begin() ? (*DIt)->getLexicalDeclContext()
373 : (*(DIt-1))->getLexicalDeclContext();
374
375 while (DC && !DC->isTranslationUnit()) {
376 Decl *D = cast<Decl>(DC);
377 SourceRange CurDeclRange = D->getSourceRange();
378 if (CurDeclRange.isInvalid())
379 break;
380
381 if (RangeCompare(SM, CurDeclRange, Range) == RangeOverlap) {
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000382 if (Visit(MakeCXCursor(D, TU, Range), /*CheckedRegionOfInterest=*/true))
383 return true; // visitation break.
Guy Benyei11169dd2012-12-18 14:30:41 +0000384 }
385
386 DC = D->getLexicalDeclContext();
387 }
Argyrios Kyrtzidis951f61f2013-03-08 20:42:33 +0000388
389 return false;
Guy Benyei11169dd2012-12-18 14:30:41 +0000390}
391
392bool CursorVisitor::visitPreprocessedEntitiesInRegion() {
393 if (!AU->getPreprocessor().getPreprocessingRecord())
394 return false;
395
396 PreprocessingRecord &PPRec
397 = *AU->getPreprocessor().getPreprocessingRecord();
398 SourceManager &SM = AU->getSourceManager();
399
400 if (RegionOfInterest.isValid()) {
401 SourceRange MappedRange = AU->mapRangeToPreamble(RegionOfInterest);
402 SourceLocation B = MappedRange.getBegin();
403 SourceLocation E = MappedRange.getEnd();
404
405 if (AU->isInPreambleFileID(B)) {
406 if (SM.isLoadedSourceLocation(E))
407 return visitPreprocessedEntitiesInRange(SourceRange(B, E),
408 PPRec, *this);
409
410 // Beginning of range lies in the preamble but it also extends beyond
411 // it into the main file. Split the range into 2 parts, one covering
412 // the preamble and another covering the main file. This allows subsequent
413 // calls to visitPreprocessedEntitiesInRange to accept a source range that
414 // lies in the same FileID, allowing it to skip preprocessed entities that
415 // do not come from the same FileID.
416 bool breaked =
417 visitPreprocessedEntitiesInRange(
418 SourceRange(B, AU->getEndOfPreambleFileID()),
419 PPRec, *this);
420 if (breaked) return true;
421 return visitPreprocessedEntitiesInRange(
422 SourceRange(AU->getStartOfMainFileID(), E),
423 PPRec, *this);
424 }
425
426 return visitPreprocessedEntitiesInRange(SourceRange(B, E), PPRec, *this);
427 }
428
429 bool OnlyLocalDecls
430 = !AU->isMainFileAST() && AU->getOnlyLocalDecls();
431
432 if (OnlyLocalDecls)
433 return visitPreprocessedEntities(PPRec.local_begin(), PPRec.local_end(),
434 PPRec);
435
436 return visitPreprocessedEntities(PPRec.begin(), PPRec.end(), PPRec);
437}
438
439template<typename InputIterator>
440bool CursorVisitor::visitPreprocessedEntities(InputIterator First,
441 InputIterator Last,
442 PreprocessingRecord &PPRec,
443 FileID FID) {
444 for (; First != Last; ++First) {
445 if (!FID.isInvalid() && !PPRec.isEntityInFileID(First, FID))
446 continue;
447
448 PreprocessedEntity *PPE = *First;
Argyrios Kyrtzidis1030f262013-05-07 20:37:17 +0000449 if (!PPE)
450 continue;
451
Guy Benyei11169dd2012-12-18 14:30:41 +0000452 if (MacroExpansion *ME = dyn_cast<MacroExpansion>(PPE)) {
453 if (Visit(MakeMacroExpansionCursor(ME, TU)))
454 return true;
Richard Smith66a81862015-05-04 02:25:31 +0000455
Guy Benyei11169dd2012-12-18 14:30:41 +0000456 continue;
457 }
Richard Smith66a81862015-05-04 02:25:31 +0000458
459 if (MacroDefinitionRecord *MD = dyn_cast<MacroDefinitionRecord>(PPE)) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000460 if (Visit(MakeMacroDefinitionCursor(MD, TU)))
461 return true;
Richard Smith66a81862015-05-04 02:25:31 +0000462
Guy Benyei11169dd2012-12-18 14:30:41 +0000463 continue;
464 }
465
466 if (InclusionDirective *ID = dyn_cast<InclusionDirective>(PPE)) {
467 if (Visit(MakeInclusionDirectiveCursor(ID, TU)))
468 return true;
469
470 continue;
471 }
472 }
473
474 return false;
475}
476
477/// \brief Visit the children of the given cursor.
478///
479/// \returns true if the visitation should be aborted, false if it
480/// should continue.
481bool CursorVisitor::VisitChildren(CXCursor Cursor) {
482 if (clang_isReference(Cursor.kind) &&
483 Cursor.kind != CXCursor_CXXBaseSpecifier) {
484 // By definition, references have no children.
485 return false;
486 }
487
488 // Set the Parent field to Cursor, then back to its old value once we're
489 // done.
490 SetParentRAII SetParent(Parent, StmtParent, Cursor);
491
492 if (clang_isDeclaration(Cursor.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +0000493 Decl *D = const_cast<Decl *>(getCursorDecl(Cursor));
Guy Benyei11169dd2012-12-18 14:30:41 +0000494 if (!D)
495 return false;
496
497 return VisitAttributes(D) || Visit(D);
498 }
499
500 if (clang_isStatement(Cursor.kind)) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +0000501 if (const Stmt *S = getCursorStmt(Cursor))
Guy Benyei11169dd2012-12-18 14:30:41 +0000502 return Visit(S);
503
504 return false;
505 }
506
507 if (clang_isExpression(Cursor.kind)) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +0000508 if (const Expr *E = getCursorExpr(Cursor))
Guy Benyei11169dd2012-12-18 14:30:41 +0000509 return Visit(E);
510
511 return false;
512 }
513
514 if (clang_isTranslationUnit(Cursor.kind)) {
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +0000515 CXTranslationUnit TU = getCursorTU(Cursor);
516 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +0000517
518 int VisitOrder[2] = { VisitPreprocessorLast, !VisitPreprocessorLast };
519 for (unsigned I = 0; I != 2; ++I) {
520 if (VisitOrder[I]) {
521 if (!CXXUnit->isMainFileAST() && CXXUnit->getOnlyLocalDecls() &&
522 RegionOfInterest.isInvalid()) {
523 for (ASTUnit::top_level_iterator TL = CXXUnit->top_level_begin(),
524 TLEnd = CXXUnit->top_level_end();
525 TL != TLEnd; ++TL) {
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +0000526 if (Visit(MakeCXCursor(*TL, TU, RegionOfInterest), true))
Guy Benyei11169dd2012-12-18 14:30:41 +0000527 return true;
528 }
529 } else if (VisitDeclContext(
530 CXXUnit->getASTContext().getTranslationUnitDecl()))
531 return true;
532 continue;
533 }
534
535 // Walk the preprocessing record.
536 if (CXXUnit->getPreprocessor().getPreprocessingRecord())
537 visitPreprocessedEntitiesInRegion();
538 }
539
540 return false;
541 }
542
543 if (Cursor.kind == CXCursor_CXXBaseSpecifier) {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +0000544 if (const CXXBaseSpecifier *Base = getCursorCXXBaseSpecifier(Cursor)) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000545 if (TypeSourceInfo *BaseTSInfo = Base->getTypeSourceInfo()) {
546 return Visit(BaseTSInfo->getTypeLoc());
547 }
548 }
549 }
550
551 if (Cursor.kind == CXCursor_IBOutletCollectionAttr) {
Dmitri Gribenkoe4baea62013-01-26 18:08:08 +0000552 const IBOutletCollectionAttr *A =
Guy Benyei11169dd2012-12-18 14:30:41 +0000553 cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(Cursor));
Richard Smithb1f9a282013-10-31 01:56:18 +0000554 if (const ObjCObjectType *ObjT = A->getInterface()->getAs<ObjCObjectType>())
Richard Smithb87c4652013-10-31 21:23:20 +0000555 return Visit(cxcursor::MakeCursorObjCClassRef(
556 ObjT->getInterface(),
557 A->getInterfaceLoc()->getTypeLoc().getLocStart(), TU));
Guy Benyei11169dd2012-12-18 14:30:41 +0000558 }
559
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +0000560 // If pointing inside a macro definition, check if the token is an identifier
561 // that was ever defined as a macro. In such a case, create a "pseudo" macro
562 // expansion cursor for that token.
563 SourceLocation BeginLoc = RegionOfInterest.getBegin();
564 if (Cursor.kind == CXCursor_MacroDefinition &&
565 BeginLoc == RegionOfInterest.getEnd()) {
566 SourceLocation Loc = AU->mapLocationToPreamble(BeginLoc);
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +0000567 const MacroInfo *MI =
568 getMacroInfo(cxcursor::getCursorMacroDefinition(Cursor), TU);
Richard Smith66a81862015-05-04 02:25:31 +0000569 if (MacroDefinitionRecord *MacroDef =
570 checkForMacroInMacroDefinition(MI, Loc, TU))
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +0000571 return Visit(cxcursor::MakeMacroExpansionCursor(MacroDef, BeginLoc, TU));
572 }
573
Guy Benyei11169dd2012-12-18 14:30:41 +0000574 // Nothing to visit at the moment.
575 return false;
576}
577
578bool CursorVisitor::VisitBlockDecl(BlockDecl *B) {
579 if (TypeSourceInfo *TSInfo = B->getSignatureAsWritten())
580 if (Visit(TSInfo->getTypeLoc()))
581 return true;
582
583 if (Stmt *Body = B->getBody())
584 return Visit(MakeCXCursor(Body, StmtParent, TU, RegionOfInterest));
585
586 return false;
587}
588
Ted Kremenek03325582013-02-21 01:29:01 +0000589Optional<bool> CursorVisitor::shouldVisitCursor(CXCursor Cursor) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000590 if (RegionOfInterest.isValid()) {
591 SourceRange Range = getFullCursorExtent(Cursor, AU->getSourceManager());
592 if (Range.isInvalid())
David Blaikie7a30dc52013-02-21 01:47:18 +0000593 return None;
Guy Benyei11169dd2012-12-18 14:30:41 +0000594
595 switch (CompareRegionOfInterest(Range)) {
596 case RangeBefore:
597 // This declaration comes before the region of interest; skip it.
David Blaikie7a30dc52013-02-21 01:47:18 +0000598 return None;
Guy Benyei11169dd2012-12-18 14:30:41 +0000599
600 case RangeAfter:
601 // This declaration comes after the region of interest; we're done.
602 return false;
603
604 case RangeOverlap:
605 // This declaration overlaps the region of interest; visit it.
606 break;
607 }
608 }
609 return true;
610}
611
612bool CursorVisitor::VisitDeclContext(DeclContext *DC) {
613 DeclContext::decl_iterator I = DC->decls_begin(), E = DC->decls_end();
614
615 // FIXME: Eventually remove. This part of a hack to support proper
616 // iteration over all Decls contained lexically within an ObjC container.
617 SaveAndRestore<DeclContext::decl_iterator*> DI_saved(DI_current, &I);
618 SaveAndRestore<DeclContext::decl_iterator> DE_saved(DE_current, E);
619
620 for ( ; I != E; ++I) {
621 Decl *D = *I;
622 if (D->getLexicalDeclContext() != DC)
623 continue;
624 CXCursor Cursor = MakeCXCursor(D, TU, RegionOfInterest);
625
626 // Ignore synthesized ivars here, otherwise if we have something like:
627 // @synthesize prop = _prop;
628 // and '_prop' is not declared, we will encounter a '_prop' ivar before
629 // encountering the 'prop' synthesize declaration and we will think that
630 // we passed the region-of-interest.
631 if (ObjCIvarDecl *ivarD = dyn_cast<ObjCIvarDecl>(D)) {
632 if (ivarD->getSynthesize())
633 continue;
634 }
635
636 // FIXME: ObjCClassRef/ObjCProtocolRef for forward class/protocol
637 // declarations is a mismatch with the compiler semantics.
638 if (Cursor.kind == CXCursor_ObjCInterfaceDecl) {
639 ObjCInterfaceDecl *ID = cast<ObjCInterfaceDecl>(D);
640 if (!ID->isThisDeclarationADefinition())
641 Cursor = MakeCursorObjCClassRef(ID, ID->getLocation(), TU);
642
643 } else if (Cursor.kind == CXCursor_ObjCProtocolDecl) {
644 ObjCProtocolDecl *PD = cast<ObjCProtocolDecl>(D);
645 if (!PD->isThisDeclarationADefinition())
646 Cursor = MakeCursorObjCProtocolRef(PD, PD->getLocation(), TU);
647 }
648
Ted Kremenek03325582013-02-21 01:29:01 +0000649 const Optional<bool> &V = shouldVisitCursor(Cursor);
Guy Benyei11169dd2012-12-18 14:30:41 +0000650 if (!V.hasValue())
651 continue;
652 if (!V.getValue())
653 return false;
654 if (Visit(Cursor, true))
655 return true;
656 }
657 return false;
658}
659
660bool CursorVisitor::VisitTranslationUnitDecl(TranslationUnitDecl *D) {
661 llvm_unreachable("Translation units are visited directly by Visit()");
662}
663
Sergey Kalinichev8f3b1872015-11-15 13:48:32 +0000664bool CursorVisitor::VisitTypeAliasTemplateDecl(TypeAliasTemplateDecl *D) {
665 if (VisitTemplateParameters(D->getTemplateParameters()))
666 return true;
667
668 return Visit(MakeCXCursor(D->getTemplatedDecl(), TU, RegionOfInterest));
669}
670
Guy Benyei11169dd2012-12-18 14:30:41 +0000671bool CursorVisitor::VisitTypeAliasDecl(TypeAliasDecl *D) {
672 if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
673 return Visit(TSInfo->getTypeLoc());
674
675 return false;
676}
677
678bool CursorVisitor::VisitTypedefDecl(TypedefDecl *D) {
679 if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
680 return Visit(TSInfo->getTypeLoc());
681
682 return false;
683}
684
685bool CursorVisitor::VisitTagDecl(TagDecl *D) {
686 return VisitDeclContext(D);
687}
688
689bool CursorVisitor::VisitClassTemplateSpecializationDecl(
690 ClassTemplateSpecializationDecl *D) {
691 bool ShouldVisitBody = false;
692 switch (D->getSpecializationKind()) {
693 case TSK_Undeclared:
694 case TSK_ImplicitInstantiation:
695 // Nothing to visit
696 return false;
697
698 case TSK_ExplicitInstantiationDeclaration:
699 case TSK_ExplicitInstantiationDefinition:
700 break;
701
702 case TSK_ExplicitSpecialization:
703 ShouldVisitBody = true;
704 break;
705 }
706
707 // Visit the template arguments used in the specialization.
708 if (TypeSourceInfo *SpecType = D->getTypeAsWritten()) {
709 TypeLoc TL = SpecType->getTypeLoc();
David Blaikie6adc78e2013-02-18 22:06:02 +0000710 if (TemplateSpecializationTypeLoc TSTLoc =
711 TL.getAs<TemplateSpecializationTypeLoc>()) {
712 for (unsigned I = 0, N = TSTLoc.getNumArgs(); I != N; ++I)
713 if (VisitTemplateArgumentLoc(TSTLoc.getArgLoc(I)))
Guy Benyei11169dd2012-12-18 14:30:41 +0000714 return true;
715 }
716 }
Alexander Kornienko1a9f1842015-12-28 15:24:08 +0000717
718 return ShouldVisitBody && VisitCXXRecordDecl(D);
Guy Benyei11169dd2012-12-18 14:30:41 +0000719}
720
721bool CursorVisitor::VisitClassTemplatePartialSpecializationDecl(
722 ClassTemplatePartialSpecializationDecl *D) {
723 // FIXME: Visit the "outer" template parameter lists on the TagDecl
724 // before visiting these template parameters.
725 if (VisitTemplateParameters(D->getTemplateParameters()))
726 return true;
727
728 // Visit the partial specialization arguments.
Enea Zaffanella6dbe1872013-08-10 07:24:53 +0000729 const ASTTemplateArgumentListInfo *Info = D->getTemplateArgsAsWritten();
730 const TemplateArgumentLoc *TemplateArgs = Info->getTemplateArgs();
731 for (unsigned I = 0, N = Info->NumTemplateArgs; I != N; ++I)
Guy Benyei11169dd2012-12-18 14:30:41 +0000732 if (VisitTemplateArgumentLoc(TemplateArgs[I]))
733 return true;
734
735 return VisitCXXRecordDecl(D);
736}
737
738bool CursorVisitor::VisitTemplateTypeParmDecl(TemplateTypeParmDecl *D) {
739 // Visit the default argument.
740 if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
741 if (TypeSourceInfo *DefArg = D->getDefaultArgumentInfo())
742 if (Visit(DefArg->getTypeLoc()))
743 return true;
744
745 return false;
746}
747
748bool CursorVisitor::VisitEnumConstantDecl(EnumConstantDecl *D) {
749 if (Expr *Init = D->getInitExpr())
750 return Visit(MakeCXCursor(Init, StmtParent, TU, RegionOfInterest));
751 return false;
752}
753
754bool CursorVisitor::VisitDeclaratorDecl(DeclaratorDecl *DD) {
Argyrios Kyrtzidis2ec76742013-04-05 21:04:10 +0000755 unsigned NumParamList = DD->getNumTemplateParameterLists();
756 for (unsigned i = 0; i < NumParamList; i++) {
757 TemplateParameterList* Params = DD->getTemplateParameterList(i);
758 if (VisitTemplateParameters(Params))
759 return true;
760 }
761
Guy Benyei11169dd2012-12-18 14:30:41 +0000762 if (TypeSourceInfo *TSInfo = DD->getTypeSourceInfo())
763 if (Visit(TSInfo->getTypeLoc()))
764 return true;
765
766 // Visit the nested-name-specifier, if present.
767 if (NestedNameSpecifierLoc QualifierLoc = DD->getQualifierLoc())
768 if (VisitNestedNameSpecifierLoc(QualifierLoc))
769 return true;
770
771 return false;
772}
773
Benjamin Kramer4cadf292014-03-07 21:51:58 +0000774/// \brief Compare two base or member initializers based on their source order.
775static int CompareCXXCtorInitializers(CXXCtorInitializer *const *X,
776 CXXCtorInitializer *const *Y) {
777 return (*X)->getSourceOrder() - (*Y)->getSourceOrder();
778}
779
Guy Benyei11169dd2012-12-18 14:30:41 +0000780bool CursorVisitor::VisitFunctionDecl(FunctionDecl *ND) {
Argyrios Kyrtzidis2ec76742013-04-05 21:04:10 +0000781 unsigned NumParamList = ND->getNumTemplateParameterLists();
782 for (unsigned i = 0; i < NumParamList; i++) {
783 TemplateParameterList* Params = ND->getTemplateParameterList(i);
784 if (VisitTemplateParameters(Params))
785 return true;
786 }
787
Guy Benyei11169dd2012-12-18 14:30:41 +0000788 if (TypeSourceInfo *TSInfo = ND->getTypeSourceInfo()) {
789 // Visit the function declaration's syntactic components in the order
790 // written. This requires a bit of work.
791 TypeLoc TL = TSInfo->getTypeLoc().IgnoreParens();
David Blaikie6adc78e2013-02-18 22:06:02 +0000792 FunctionTypeLoc FTL = TL.getAs<FunctionTypeLoc>();
Guy Benyei11169dd2012-12-18 14:30:41 +0000793
794 // If we have a function declared directly (without the use of a typedef),
795 // visit just the return type. Otherwise, just visit the function's type
796 // now.
Alp Toker42a16a62014-01-25 23:51:36 +0000797 if ((FTL && !isa<CXXConversionDecl>(ND) && Visit(FTL.getReturnLoc())) ||
Guy Benyei11169dd2012-12-18 14:30:41 +0000798 (!FTL && Visit(TL)))
799 return true;
800
801 // Visit the nested-name-specifier, if present.
802 if (NestedNameSpecifierLoc QualifierLoc = ND->getQualifierLoc())
803 if (VisitNestedNameSpecifierLoc(QualifierLoc))
804 return true;
805
806 // Visit the declaration name.
Argyrios Kyrtzidis4a4d2b42014-02-09 08:13:47 +0000807 if (!isa<CXXDestructorDecl>(ND))
808 if (VisitDeclarationNameInfo(ND->getNameInfo()))
809 return true;
Guy Benyei11169dd2012-12-18 14:30:41 +0000810
811 // FIXME: Visit explicitly-specified template arguments!
812
813 // Visit the function parameters, if we have a function type.
David Blaikie6adc78e2013-02-18 22:06:02 +0000814 if (FTL && VisitFunctionTypeLoc(FTL, true))
Guy Benyei11169dd2012-12-18 14:30:41 +0000815 return true;
816
Bill Wendling44426052012-12-20 19:22:21 +0000817 // FIXME: Attributes?
Guy Benyei11169dd2012-12-18 14:30:41 +0000818 }
819
820 if (ND->doesThisDeclarationHaveABody() && !ND->isLateTemplateParsed()) {
821 if (CXXConstructorDecl *Constructor = dyn_cast<CXXConstructorDecl>(ND)) {
822 // Find the initializers that were written in the source.
823 SmallVector<CXXCtorInitializer *, 4> WrittenInits;
Aaron Ballman0ad78302014-03-13 17:34:31 +0000824 for (auto *I : Constructor->inits()) {
825 if (!I->isWritten())
Guy Benyei11169dd2012-12-18 14:30:41 +0000826 continue;
827
Aaron Ballman0ad78302014-03-13 17:34:31 +0000828 WrittenInits.push_back(I);
Guy Benyei11169dd2012-12-18 14:30:41 +0000829 }
830
831 // Sort the initializers in source order
Benjamin Kramer4cadf292014-03-07 21:51:58 +0000832 llvm::array_pod_sort(WrittenInits.begin(), WrittenInits.end(),
833 &CompareCXXCtorInitializers);
834
Guy Benyei11169dd2012-12-18 14:30:41 +0000835 // Visit the initializers in source order
836 for (unsigned I = 0, N = WrittenInits.size(); I != N; ++I) {
837 CXXCtorInitializer *Init = WrittenInits[I];
838 if (Init->isAnyMemberInitializer()) {
839 if (Visit(MakeCursorMemberRef(Init->getAnyMember(),
840 Init->getMemberLocation(), TU)))
841 return true;
842 } else if (TypeSourceInfo *TInfo = Init->getTypeSourceInfo()) {
843 if (Visit(TInfo->getTypeLoc()))
844 return true;
845 }
846
847 // Visit the initializer value.
848 if (Expr *Initializer = Init->getInit())
849 if (Visit(MakeCXCursor(Initializer, ND, TU, RegionOfInterest)))
850 return true;
851 }
852 }
853
854 if (Visit(MakeCXCursor(ND->getBody(), StmtParent, TU, RegionOfInterest)))
855 return true;
856 }
857
858 return false;
859}
860
861bool CursorVisitor::VisitFieldDecl(FieldDecl *D) {
862 if (VisitDeclaratorDecl(D))
863 return true;
864
865 if (Expr *BitWidth = D->getBitWidth())
866 return Visit(MakeCXCursor(BitWidth, StmtParent, TU, RegionOfInterest));
867
868 return false;
869}
870
871bool CursorVisitor::VisitVarDecl(VarDecl *D) {
872 if (VisitDeclaratorDecl(D))
873 return true;
874
875 if (Expr *Init = D->getInit())
876 return Visit(MakeCXCursor(Init, StmtParent, TU, RegionOfInterest));
877
878 return false;
879}
880
881bool CursorVisitor::VisitNonTypeTemplateParmDecl(NonTypeTemplateParmDecl *D) {
882 if (VisitDeclaratorDecl(D))
883 return true;
884
885 if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
886 if (Expr *DefArg = D->getDefaultArgument())
887 return Visit(MakeCXCursor(DefArg, StmtParent, TU, RegionOfInterest));
888
889 return false;
890}
891
892bool CursorVisitor::VisitFunctionTemplateDecl(FunctionTemplateDecl *D) {
893 // FIXME: Visit the "outer" template parameter lists on the FunctionDecl
894 // before visiting these template parameters.
895 if (VisitTemplateParameters(D->getTemplateParameters()))
896 return true;
897
898 return VisitFunctionDecl(D->getTemplatedDecl());
899}
900
901bool CursorVisitor::VisitClassTemplateDecl(ClassTemplateDecl *D) {
902 // FIXME: Visit the "outer" template parameter lists on the TagDecl
903 // before visiting these template parameters.
904 if (VisitTemplateParameters(D->getTemplateParameters()))
905 return true;
906
907 return VisitCXXRecordDecl(D->getTemplatedDecl());
908}
909
910bool CursorVisitor::VisitTemplateTemplateParmDecl(TemplateTemplateParmDecl *D) {
911 if (VisitTemplateParameters(D->getTemplateParameters()))
912 return true;
913
914 if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited() &&
915 VisitTemplateArgumentLoc(D->getDefaultArgument()))
916 return true;
917
918 return false;
919}
920
Douglas Gregor9bda6cf2015-07-07 03:58:14 +0000921bool CursorVisitor::VisitObjCTypeParamDecl(ObjCTypeParamDecl *D) {
922 // Visit the bound, if it's explicit.
923 if (D->hasExplicitBound()) {
924 if (auto TInfo = D->getTypeSourceInfo()) {
925 if (Visit(TInfo->getTypeLoc()))
926 return true;
927 }
928 }
929
930 return false;
931}
932
Guy Benyei11169dd2012-12-18 14:30:41 +0000933bool CursorVisitor::VisitObjCMethodDecl(ObjCMethodDecl *ND) {
Alp Toker314cc812014-01-25 16:55:45 +0000934 if (TypeSourceInfo *TSInfo = ND->getReturnTypeSourceInfo())
Guy Benyei11169dd2012-12-18 14:30:41 +0000935 if (Visit(TSInfo->getTypeLoc()))
936 return true;
937
David Majnemer59f77922016-06-24 04:05:48 +0000938 for (const auto *P : ND->parameters()) {
Aaron Ballman43b68be2014-03-07 17:50:17 +0000939 if (Visit(MakeCXCursor(P, TU, RegionOfInterest)))
Guy Benyei11169dd2012-12-18 14:30:41 +0000940 return true;
941 }
942
Alexander Kornienko1a9f1842015-12-28 15:24:08 +0000943 return ND->isThisDeclarationADefinition() &&
944 Visit(MakeCXCursor(ND->getBody(), StmtParent, TU, RegionOfInterest));
Guy Benyei11169dd2012-12-18 14:30:41 +0000945}
946
947template <typename DeclIt>
948static void addRangedDeclsInContainer(DeclIt *DI_current, DeclIt DE_current,
949 SourceManager &SM, SourceLocation EndLoc,
950 SmallVectorImpl<Decl *> &Decls) {
951 DeclIt next = *DI_current;
952 while (++next != DE_current) {
953 Decl *D_next = *next;
954 if (!D_next)
955 break;
956 SourceLocation L = D_next->getLocStart();
957 if (!L.isValid())
958 break;
959 if (SM.isBeforeInTranslationUnit(L, EndLoc)) {
960 *DI_current = next;
961 Decls.push_back(D_next);
962 continue;
963 }
964 break;
965 }
966}
967
Guy Benyei11169dd2012-12-18 14:30:41 +0000968bool CursorVisitor::VisitObjCContainerDecl(ObjCContainerDecl *D) {
969 // FIXME: Eventually convert back to just 'VisitDeclContext()'. Essentially
970 // an @implementation can lexically contain Decls that are not properly
971 // nested in the AST. When we identify such cases, we need to retrofit
972 // this nesting here.
973 if (!DI_current && !FileDI_current)
974 return VisitDeclContext(D);
975
976 // Scan the Decls that immediately come after the container
977 // in the current DeclContext. If any fall within the
978 // container's lexical region, stash them into a vector
979 // for later processing.
980 SmallVector<Decl *, 24> DeclsInContainer;
981 SourceLocation EndLoc = D->getSourceRange().getEnd();
982 SourceManager &SM = AU->getSourceManager();
983 if (EndLoc.isValid()) {
984 if (DI_current) {
985 addRangedDeclsInContainer(DI_current, DE_current, SM, EndLoc,
986 DeclsInContainer);
987 } else {
988 addRangedDeclsInContainer(FileDI_current, FileDE_current, SM, EndLoc,
989 DeclsInContainer);
990 }
991 }
992
993 // The common case.
994 if (DeclsInContainer.empty())
995 return VisitDeclContext(D);
996
997 // Get all the Decls in the DeclContext, and sort them with the
998 // additional ones we've collected. Then visit them.
Aaron Ballman629afae2014-03-07 19:56:05 +0000999 for (auto *SubDecl : D->decls()) {
1000 if (!SubDecl || SubDecl->getLexicalDeclContext() != D ||
1001 SubDecl->getLocStart().isInvalid())
Guy Benyei11169dd2012-12-18 14:30:41 +00001002 continue;
Aaron Ballman629afae2014-03-07 19:56:05 +00001003 DeclsInContainer.push_back(SubDecl);
Guy Benyei11169dd2012-12-18 14:30:41 +00001004 }
1005
1006 // Now sort the Decls so that they appear in lexical order.
1007 std::sort(DeclsInContainer.begin(), DeclsInContainer.end(),
Benjamin Kramerbbdd7642014-03-01 14:48:57 +00001008 [&SM](Decl *A, Decl *B) {
1009 SourceLocation L_A = A->getLocStart();
1010 SourceLocation L_B = B->getLocStart();
1011 assert(L_A.isValid() && L_B.isValid());
1012 return SM.isBeforeInTranslationUnit(L_A, L_B);
1013 });
Guy Benyei11169dd2012-12-18 14:30:41 +00001014
1015 // Now visit the decls.
1016 for (SmallVectorImpl<Decl*>::iterator I = DeclsInContainer.begin(),
1017 E = DeclsInContainer.end(); I != E; ++I) {
1018 CXCursor Cursor = MakeCXCursor(*I, TU, RegionOfInterest);
Ted Kremenek03325582013-02-21 01:29:01 +00001019 const Optional<bool> &V = shouldVisitCursor(Cursor);
Guy Benyei11169dd2012-12-18 14:30:41 +00001020 if (!V.hasValue())
1021 continue;
1022 if (!V.getValue())
1023 return false;
1024 if (Visit(Cursor, true))
1025 return true;
1026 }
1027 return false;
1028}
1029
1030bool CursorVisitor::VisitObjCCategoryDecl(ObjCCategoryDecl *ND) {
1031 if (Visit(MakeCursorObjCClassRef(ND->getClassInterface(), ND->getLocation(),
1032 TU)))
1033 return true;
1034
Douglas Gregore9d95f12015-07-07 03:57:35 +00001035 if (VisitObjCTypeParamList(ND->getTypeParamList()))
1036 return true;
1037
Guy Benyei11169dd2012-12-18 14:30:41 +00001038 ObjCCategoryDecl::protocol_loc_iterator PL = ND->protocol_loc_begin();
1039 for (ObjCCategoryDecl::protocol_iterator I = ND->protocol_begin(),
1040 E = ND->protocol_end(); I != E; ++I, ++PL)
1041 if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1042 return true;
1043
1044 return VisitObjCContainerDecl(ND);
1045}
1046
1047bool CursorVisitor::VisitObjCProtocolDecl(ObjCProtocolDecl *PID) {
1048 if (!PID->isThisDeclarationADefinition())
1049 return Visit(MakeCursorObjCProtocolRef(PID, PID->getLocation(), TU));
1050
1051 ObjCProtocolDecl::protocol_loc_iterator PL = PID->protocol_loc_begin();
1052 for (ObjCProtocolDecl::protocol_iterator I = PID->protocol_begin(),
1053 E = PID->protocol_end(); I != E; ++I, ++PL)
1054 if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1055 return true;
1056
1057 return VisitObjCContainerDecl(PID);
1058}
1059
1060bool CursorVisitor::VisitObjCPropertyDecl(ObjCPropertyDecl *PD) {
1061 if (PD->getTypeSourceInfo() && Visit(PD->getTypeSourceInfo()->getTypeLoc()))
1062 return true;
1063
1064 // FIXME: This implements a workaround with @property declarations also being
1065 // installed in the DeclContext for the @interface. Eventually this code
1066 // should be removed.
1067 ObjCCategoryDecl *CDecl = dyn_cast<ObjCCategoryDecl>(PD->getDeclContext());
1068 if (!CDecl || !CDecl->IsClassExtension())
1069 return false;
1070
1071 ObjCInterfaceDecl *ID = CDecl->getClassInterface();
1072 if (!ID)
1073 return false;
1074
1075 IdentifierInfo *PropertyId = PD->getIdentifier();
1076 ObjCPropertyDecl *prevDecl =
Manman Ren5b786402016-01-28 18:49:28 +00001077 ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(ID), PropertyId,
1078 PD->getQueryKind());
Guy Benyei11169dd2012-12-18 14:30:41 +00001079
1080 if (!prevDecl)
1081 return false;
1082
1083 // Visit synthesized methods since they will be skipped when visiting
1084 // the @interface.
1085 if (ObjCMethodDecl *MD = prevDecl->getGetterMethodDecl())
1086 if (MD->isPropertyAccessor() && MD->getLexicalDeclContext() == CDecl)
1087 if (Visit(MakeCXCursor(MD, TU, RegionOfInterest)))
1088 return true;
1089
1090 if (ObjCMethodDecl *MD = prevDecl->getSetterMethodDecl())
1091 if (MD->isPropertyAccessor() && MD->getLexicalDeclContext() == CDecl)
1092 if (Visit(MakeCXCursor(MD, TU, RegionOfInterest)))
1093 return true;
1094
1095 return false;
1096}
1097
Douglas Gregore9d95f12015-07-07 03:57:35 +00001098bool CursorVisitor::VisitObjCTypeParamList(ObjCTypeParamList *typeParamList) {
1099 if (!typeParamList)
1100 return false;
1101
1102 for (auto *typeParam : *typeParamList) {
1103 // Visit the type parameter.
1104 if (Visit(MakeCXCursor(typeParam, TU, RegionOfInterest)))
1105 return true;
Douglas Gregore9d95f12015-07-07 03:57:35 +00001106 }
1107
1108 return false;
1109}
1110
Guy Benyei11169dd2012-12-18 14:30:41 +00001111bool CursorVisitor::VisitObjCInterfaceDecl(ObjCInterfaceDecl *D) {
1112 if (!D->isThisDeclarationADefinition()) {
1113 // Forward declaration is treated like a reference.
1114 return Visit(MakeCursorObjCClassRef(D, D->getLocation(), TU));
1115 }
1116
Douglas Gregore9d95f12015-07-07 03:57:35 +00001117 // Objective-C type parameters.
1118 if (VisitObjCTypeParamList(D->getTypeParamListAsWritten()))
1119 return true;
1120
Guy Benyei11169dd2012-12-18 14:30:41 +00001121 // Issue callbacks for super class.
1122 if (D->getSuperClass() &&
1123 Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1124 D->getSuperClassLoc(),
1125 TU)))
1126 return true;
1127
Douglas Gregore9d95f12015-07-07 03:57:35 +00001128 if (TypeSourceInfo *SuperClassTInfo = D->getSuperClassTInfo())
1129 if (Visit(SuperClassTInfo->getTypeLoc()))
1130 return true;
1131
Guy Benyei11169dd2012-12-18 14:30:41 +00001132 ObjCInterfaceDecl::protocol_loc_iterator PL = D->protocol_loc_begin();
1133 for (ObjCInterfaceDecl::protocol_iterator I = D->protocol_begin(),
1134 E = D->protocol_end(); I != E; ++I, ++PL)
1135 if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1136 return true;
1137
1138 return VisitObjCContainerDecl(D);
1139}
1140
1141bool CursorVisitor::VisitObjCImplDecl(ObjCImplDecl *D) {
1142 return VisitObjCContainerDecl(D);
1143}
1144
1145bool CursorVisitor::VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D) {
1146 // 'ID' could be null when dealing with invalid code.
1147 if (ObjCInterfaceDecl *ID = D->getClassInterface())
1148 if (Visit(MakeCursorObjCClassRef(ID, D->getLocation(), TU)))
1149 return true;
1150
1151 return VisitObjCImplDecl(D);
1152}
1153
1154bool CursorVisitor::VisitObjCImplementationDecl(ObjCImplementationDecl *D) {
1155#if 0
1156 // Issue callbacks for super class.
1157 // FIXME: No source location information!
1158 if (D->getSuperClass() &&
1159 Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1160 D->getSuperClassLoc(),
1161 TU)))
1162 return true;
1163#endif
1164
1165 return VisitObjCImplDecl(D);
1166}
1167
1168bool CursorVisitor::VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *PD) {
1169 if (ObjCIvarDecl *Ivar = PD->getPropertyIvarDecl())
1170 if (PD->isIvarNameSpecified())
1171 return Visit(MakeCursorMemberRef(Ivar, PD->getPropertyIvarDeclLoc(), TU));
1172
1173 return false;
1174}
1175
1176bool CursorVisitor::VisitNamespaceDecl(NamespaceDecl *D) {
1177 return VisitDeclContext(D);
1178}
1179
1180bool CursorVisitor::VisitNamespaceAliasDecl(NamespaceAliasDecl *D) {
1181 // Visit nested-name-specifier.
1182 if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1183 if (VisitNestedNameSpecifierLoc(QualifierLoc))
1184 return true;
1185
1186 return Visit(MakeCursorNamespaceRef(D->getAliasedNamespace(),
1187 D->getTargetNameLoc(), TU));
1188}
1189
1190bool CursorVisitor::VisitUsingDecl(UsingDecl *D) {
1191 // Visit nested-name-specifier.
1192 if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1193 if (VisitNestedNameSpecifierLoc(QualifierLoc))
1194 return true;
1195 }
1196
1197 if (Visit(MakeCursorOverloadedDeclRef(D, D->getLocation(), TU)))
1198 return true;
1199
1200 return VisitDeclarationNameInfo(D->getNameInfo());
1201}
1202
1203bool CursorVisitor::VisitUsingDirectiveDecl(UsingDirectiveDecl *D) {
1204 // Visit nested-name-specifier.
1205 if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1206 if (VisitNestedNameSpecifierLoc(QualifierLoc))
1207 return true;
1208
1209 return Visit(MakeCursorNamespaceRef(D->getNominatedNamespaceAsWritten(),
1210 D->getIdentLocation(), TU));
1211}
1212
1213bool CursorVisitor::VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D) {
1214 // Visit nested-name-specifier.
1215 if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1216 if (VisitNestedNameSpecifierLoc(QualifierLoc))
1217 return true;
1218 }
1219
1220 return VisitDeclarationNameInfo(D->getNameInfo());
1221}
1222
1223bool CursorVisitor::VisitUnresolvedUsingTypenameDecl(
1224 UnresolvedUsingTypenameDecl *D) {
1225 // Visit nested-name-specifier.
1226 if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1227 if (VisitNestedNameSpecifierLoc(QualifierLoc))
1228 return true;
1229
1230 return false;
1231}
1232
Olivier Goffart81978012016-06-09 16:15:55 +00001233bool CursorVisitor::VisitStaticAssertDecl(StaticAssertDecl *D) {
1234 if (Visit(MakeCXCursor(D->getAssertExpr(), StmtParent, TU, RegionOfInterest)))
1235 return true;
1236 if (Visit(MakeCXCursor(D->getMessage(), StmtParent, TU, RegionOfInterest)))
1237 return true;
1238 return false;
1239}
1240
Guy Benyei11169dd2012-12-18 14:30:41 +00001241bool CursorVisitor::VisitDeclarationNameInfo(DeclarationNameInfo Name) {
1242 switch (Name.getName().getNameKind()) {
1243 case clang::DeclarationName::Identifier:
1244 case clang::DeclarationName::CXXLiteralOperatorName:
1245 case clang::DeclarationName::CXXOperatorName:
1246 case clang::DeclarationName::CXXUsingDirective:
1247 return false;
1248
1249 case clang::DeclarationName::CXXConstructorName:
1250 case clang::DeclarationName::CXXDestructorName:
1251 case clang::DeclarationName::CXXConversionFunctionName:
1252 if (TypeSourceInfo *TSInfo = Name.getNamedTypeInfo())
1253 return Visit(TSInfo->getTypeLoc());
1254 return false;
1255
1256 case clang::DeclarationName::ObjCZeroArgSelector:
1257 case clang::DeclarationName::ObjCOneArgSelector:
1258 case clang::DeclarationName::ObjCMultiArgSelector:
1259 // FIXME: Per-identifier location info?
1260 return false;
1261 }
1262
1263 llvm_unreachable("Invalid DeclarationName::Kind!");
1264}
1265
1266bool CursorVisitor::VisitNestedNameSpecifier(NestedNameSpecifier *NNS,
1267 SourceRange Range) {
1268 // FIXME: This whole routine is a hack to work around the lack of proper
1269 // source information in nested-name-specifiers (PR5791). Since we do have
1270 // a beginning source location, we can visit the first component of the
1271 // nested-name-specifier, if it's a single-token component.
1272 if (!NNS)
1273 return false;
1274
1275 // Get the first component in the nested-name-specifier.
1276 while (NestedNameSpecifier *Prefix = NNS->getPrefix())
1277 NNS = Prefix;
1278
1279 switch (NNS->getKind()) {
1280 case NestedNameSpecifier::Namespace:
1281 return Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(), Range.getBegin(),
1282 TU));
1283
1284 case NestedNameSpecifier::NamespaceAlias:
1285 return Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
1286 Range.getBegin(), TU));
1287
1288 case NestedNameSpecifier::TypeSpec: {
1289 // If the type has a form where we know that the beginning of the source
1290 // range matches up with a reference cursor. Visit the appropriate reference
1291 // cursor.
1292 const Type *T = NNS->getAsType();
1293 if (const TypedefType *Typedef = dyn_cast<TypedefType>(T))
1294 return Visit(MakeCursorTypeRef(Typedef->getDecl(), Range.getBegin(), TU));
1295 if (const TagType *Tag = dyn_cast<TagType>(T))
1296 return Visit(MakeCursorTypeRef(Tag->getDecl(), Range.getBegin(), TU));
1297 if (const TemplateSpecializationType *TST
1298 = dyn_cast<TemplateSpecializationType>(T))
1299 return VisitTemplateName(TST->getTemplateName(), Range.getBegin());
1300 break;
1301 }
1302
1303 case NestedNameSpecifier::TypeSpecWithTemplate:
1304 case NestedNameSpecifier::Global:
1305 case NestedNameSpecifier::Identifier:
Nikola Smiljanic67860242014-09-26 00:28:20 +00001306 case NestedNameSpecifier::Super:
Guy Benyei11169dd2012-12-18 14:30:41 +00001307 break;
1308 }
1309
1310 return false;
1311}
1312
1313bool
1314CursorVisitor::VisitNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
1315 SmallVector<NestedNameSpecifierLoc, 4> Qualifiers;
1316 for (; Qualifier; Qualifier = Qualifier.getPrefix())
1317 Qualifiers.push_back(Qualifier);
1318
1319 while (!Qualifiers.empty()) {
1320 NestedNameSpecifierLoc Q = Qualifiers.pop_back_val();
1321 NestedNameSpecifier *NNS = Q.getNestedNameSpecifier();
1322 switch (NNS->getKind()) {
1323 case NestedNameSpecifier::Namespace:
1324 if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(),
1325 Q.getLocalBeginLoc(),
1326 TU)))
1327 return true;
1328
1329 break;
1330
1331 case NestedNameSpecifier::NamespaceAlias:
1332 if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
1333 Q.getLocalBeginLoc(),
1334 TU)))
1335 return true;
1336
1337 break;
1338
1339 case NestedNameSpecifier::TypeSpec:
1340 case NestedNameSpecifier::TypeSpecWithTemplate:
1341 if (Visit(Q.getTypeLoc()))
1342 return true;
1343
1344 break;
1345
1346 case NestedNameSpecifier::Global:
1347 case NestedNameSpecifier::Identifier:
Nikola Smiljanic67860242014-09-26 00:28:20 +00001348 case NestedNameSpecifier::Super:
Guy Benyei11169dd2012-12-18 14:30:41 +00001349 break;
1350 }
1351 }
1352
1353 return false;
1354}
1355
1356bool CursorVisitor::VisitTemplateParameters(
1357 const TemplateParameterList *Params) {
1358 if (!Params)
1359 return false;
1360
1361 for (TemplateParameterList::const_iterator P = Params->begin(),
1362 PEnd = Params->end();
1363 P != PEnd; ++P) {
1364 if (Visit(MakeCXCursor(*P, TU, RegionOfInterest)))
1365 return true;
1366 }
1367
1368 return false;
1369}
1370
1371bool CursorVisitor::VisitTemplateName(TemplateName Name, SourceLocation Loc) {
1372 switch (Name.getKind()) {
1373 case TemplateName::Template:
1374 return Visit(MakeCursorTemplateRef(Name.getAsTemplateDecl(), Loc, TU));
1375
1376 case TemplateName::OverloadedTemplate:
1377 // Visit the overloaded template set.
1378 if (Visit(MakeCursorOverloadedDeclRef(Name, Loc, TU)))
1379 return true;
1380
1381 return false;
1382
1383 case TemplateName::DependentTemplate:
1384 // FIXME: Visit nested-name-specifier.
1385 return false;
1386
1387 case TemplateName::QualifiedTemplate:
1388 // FIXME: Visit nested-name-specifier.
1389 return Visit(MakeCursorTemplateRef(
1390 Name.getAsQualifiedTemplateName()->getDecl(),
1391 Loc, TU));
1392
1393 case TemplateName::SubstTemplateTemplateParm:
1394 return Visit(MakeCursorTemplateRef(
1395 Name.getAsSubstTemplateTemplateParm()->getParameter(),
1396 Loc, TU));
1397
1398 case TemplateName::SubstTemplateTemplateParmPack:
1399 return Visit(MakeCursorTemplateRef(
1400 Name.getAsSubstTemplateTemplateParmPack()->getParameterPack(),
1401 Loc, TU));
1402 }
1403
1404 llvm_unreachable("Invalid TemplateName::Kind!");
1405}
1406
1407bool CursorVisitor::VisitTemplateArgumentLoc(const TemplateArgumentLoc &TAL) {
1408 switch (TAL.getArgument().getKind()) {
1409 case TemplateArgument::Null:
1410 case TemplateArgument::Integral:
1411 case TemplateArgument::Pack:
1412 return false;
1413
1414 case TemplateArgument::Type:
1415 if (TypeSourceInfo *TSInfo = TAL.getTypeSourceInfo())
1416 return Visit(TSInfo->getTypeLoc());
1417 return false;
1418
1419 case TemplateArgument::Declaration:
1420 if (Expr *E = TAL.getSourceDeclExpression())
1421 return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1422 return false;
1423
1424 case TemplateArgument::NullPtr:
1425 if (Expr *E = TAL.getSourceNullPtrExpression())
1426 return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1427 return false;
1428
1429 case TemplateArgument::Expression:
1430 if (Expr *E = TAL.getSourceExpression())
1431 return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1432 return false;
1433
1434 case TemplateArgument::Template:
1435 case TemplateArgument::TemplateExpansion:
1436 if (VisitNestedNameSpecifierLoc(TAL.getTemplateQualifierLoc()))
1437 return true;
1438
1439 return VisitTemplateName(TAL.getArgument().getAsTemplateOrTemplatePattern(),
1440 TAL.getTemplateNameLoc());
1441 }
1442
1443 llvm_unreachable("Invalid TemplateArgument::Kind!");
1444}
1445
1446bool CursorVisitor::VisitLinkageSpecDecl(LinkageSpecDecl *D) {
1447 return VisitDeclContext(D);
1448}
1449
1450bool CursorVisitor::VisitQualifiedTypeLoc(QualifiedTypeLoc TL) {
1451 return Visit(TL.getUnqualifiedLoc());
1452}
1453
1454bool CursorVisitor::VisitBuiltinTypeLoc(BuiltinTypeLoc TL) {
1455 ASTContext &Context = AU->getASTContext();
1456
1457 // Some builtin types (such as Objective-C's "id", "sel", and
1458 // "Class") have associated declarations. Create cursors for those.
1459 QualType VisitType;
1460 switch (TL.getTypePtr()->getKind()) {
1461
1462 case BuiltinType::Void:
1463 case BuiltinType::NullPtr:
1464 case BuiltinType::Dependent:
Alexey Bader954ba212016-04-08 13:40:33 +00001465#define IMAGE_TYPE(ImgType, Id, SingletonId, Access, Suffix) \
1466 case BuiltinType::Id:
Alexey Baderb62f1442016-04-13 08:33:41 +00001467#include "clang/Basic/OpenCLImageTypes.def"
NAKAMURA Takumi288c42e2013-02-07 12:47:42 +00001468 case BuiltinType::OCLSampler:
Guy Benyei1b4fb3e2013-01-20 12:31:11 +00001469 case BuiltinType::OCLEvent:
Alexey Bader9c8453f2015-09-15 11:18:52 +00001470 case BuiltinType::OCLClkEvent:
1471 case BuiltinType::OCLQueue:
1472 case BuiltinType::OCLNDRange:
1473 case BuiltinType::OCLReserveID:
Guy Benyei11169dd2012-12-18 14:30:41 +00001474#define BUILTIN_TYPE(Id, SingletonId)
1475#define SIGNED_TYPE(Id, SingletonId) case BuiltinType::Id:
1476#define UNSIGNED_TYPE(Id, SingletonId) case BuiltinType::Id:
1477#define FLOATING_TYPE(Id, SingletonId) case BuiltinType::Id:
1478#define PLACEHOLDER_TYPE(Id, SingletonId) case BuiltinType::Id:
1479#include "clang/AST/BuiltinTypes.def"
1480 break;
1481
1482 case BuiltinType::ObjCId:
1483 VisitType = Context.getObjCIdType();
1484 break;
1485
1486 case BuiltinType::ObjCClass:
1487 VisitType = Context.getObjCClassType();
1488 break;
1489
1490 case BuiltinType::ObjCSel:
1491 VisitType = Context.getObjCSelType();
1492 break;
1493 }
1494
1495 if (!VisitType.isNull()) {
1496 if (const TypedefType *Typedef = VisitType->getAs<TypedefType>())
1497 return Visit(MakeCursorTypeRef(Typedef->getDecl(), TL.getBuiltinLoc(),
1498 TU));
1499 }
1500
1501 return false;
1502}
1503
1504bool CursorVisitor::VisitTypedefTypeLoc(TypedefTypeLoc TL) {
1505 return Visit(MakeCursorTypeRef(TL.getTypedefNameDecl(), TL.getNameLoc(), TU));
1506}
1507
1508bool CursorVisitor::VisitUnresolvedUsingTypeLoc(UnresolvedUsingTypeLoc TL) {
1509 return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1510}
1511
1512bool CursorVisitor::VisitTagTypeLoc(TagTypeLoc TL) {
1513 if (TL.isDefinition())
1514 return Visit(MakeCXCursor(TL.getDecl(), TU, RegionOfInterest));
1515
1516 return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1517}
1518
1519bool CursorVisitor::VisitTemplateTypeParmTypeLoc(TemplateTypeParmTypeLoc TL) {
1520 return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1521}
1522
1523bool CursorVisitor::VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL) {
Hans Wennborg59dbe862015-09-29 20:56:43 +00001524 return Visit(MakeCursorObjCClassRef(TL.getIFaceDecl(), TL.getNameLoc(), TU));
Guy Benyei11169dd2012-12-18 14:30:41 +00001525}
1526
1527bool CursorVisitor::VisitObjCObjectTypeLoc(ObjCObjectTypeLoc TL) {
1528 if (TL.hasBaseTypeAsWritten() && Visit(TL.getBaseLoc()))
1529 return true;
1530
Douglas Gregore9d95f12015-07-07 03:57:35 +00001531 for (unsigned I = 0, N = TL.getNumTypeArgs(); I != N; ++I) {
1532 if (Visit(TL.getTypeArgTInfo(I)->getTypeLoc()))
1533 return true;
1534 }
1535
Guy Benyei11169dd2012-12-18 14:30:41 +00001536 for (unsigned I = 0, N = TL.getNumProtocols(); I != N; ++I) {
1537 if (Visit(MakeCursorObjCProtocolRef(TL.getProtocol(I), TL.getProtocolLoc(I),
1538 TU)))
1539 return true;
1540 }
1541
1542 return false;
1543}
1544
1545bool CursorVisitor::VisitObjCObjectPointerTypeLoc(ObjCObjectPointerTypeLoc TL) {
1546 return Visit(TL.getPointeeLoc());
1547}
1548
1549bool CursorVisitor::VisitParenTypeLoc(ParenTypeLoc TL) {
1550 return Visit(TL.getInnerLoc());
1551}
1552
1553bool CursorVisitor::VisitPointerTypeLoc(PointerTypeLoc TL) {
1554 return Visit(TL.getPointeeLoc());
1555}
1556
1557bool CursorVisitor::VisitBlockPointerTypeLoc(BlockPointerTypeLoc TL) {
1558 return Visit(TL.getPointeeLoc());
1559}
1560
1561bool CursorVisitor::VisitMemberPointerTypeLoc(MemberPointerTypeLoc TL) {
1562 return Visit(TL.getPointeeLoc());
1563}
1564
1565bool CursorVisitor::VisitLValueReferenceTypeLoc(LValueReferenceTypeLoc TL) {
1566 return Visit(TL.getPointeeLoc());
1567}
1568
1569bool CursorVisitor::VisitRValueReferenceTypeLoc(RValueReferenceTypeLoc TL) {
1570 return Visit(TL.getPointeeLoc());
1571}
1572
1573bool CursorVisitor::VisitAttributedTypeLoc(AttributedTypeLoc TL) {
1574 return Visit(TL.getModifiedLoc());
1575}
1576
1577bool CursorVisitor::VisitFunctionTypeLoc(FunctionTypeLoc TL,
1578 bool SkipResultType) {
Alp Toker42a16a62014-01-25 23:51:36 +00001579 if (!SkipResultType && Visit(TL.getReturnLoc()))
Guy Benyei11169dd2012-12-18 14:30:41 +00001580 return true;
1581
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00001582 for (unsigned I = 0, N = TL.getNumParams(); I != N; ++I)
1583 if (Decl *D = TL.getParam(I))
Guy Benyei11169dd2012-12-18 14:30:41 +00001584 if (Visit(MakeCXCursor(D, TU, RegionOfInterest)))
1585 return true;
1586
1587 return false;
1588}
1589
1590bool CursorVisitor::VisitArrayTypeLoc(ArrayTypeLoc TL) {
1591 if (Visit(TL.getElementLoc()))
1592 return true;
1593
1594 if (Expr *Size = TL.getSizeExpr())
1595 return Visit(MakeCXCursor(Size, StmtParent, TU, RegionOfInterest));
1596
1597 return false;
1598}
1599
Reid Kleckner8a365022013-06-24 17:51:48 +00001600bool CursorVisitor::VisitDecayedTypeLoc(DecayedTypeLoc TL) {
1601 return Visit(TL.getOriginalLoc());
1602}
1603
Reid Kleckner0503a872013-12-05 01:23:43 +00001604bool CursorVisitor::VisitAdjustedTypeLoc(AdjustedTypeLoc TL) {
1605 return Visit(TL.getOriginalLoc());
1606}
1607
Guy Benyei11169dd2012-12-18 14:30:41 +00001608bool CursorVisitor::VisitTemplateSpecializationTypeLoc(
1609 TemplateSpecializationTypeLoc TL) {
1610 // Visit the template name.
1611 if (VisitTemplateName(TL.getTypePtr()->getTemplateName(),
1612 TL.getTemplateNameLoc()))
1613 return true;
1614
1615 // Visit the template arguments.
1616 for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
1617 if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
1618 return true;
1619
1620 return false;
1621}
1622
1623bool CursorVisitor::VisitTypeOfExprTypeLoc(TypeOfExprTypeLoc TL) {
1624 return Visit(MakeCXCursor(TL.getUnderlyingExpr(), StmtParent, TU));
1625}
1626
1627bool CursorVisitor::VisitTypeOfTypeLoc(TypeOfTypeLoc TL) {
1628 if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
1629 return Visit(TSInfo->getTypeLoc());
1630
1631 return false;
1632}
1633
1634bool CursorVisitor::VisitUnaryTransformTypeLoc(UnaryTransformTypeLoc TL) {
1635 if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
1636 return Visit(TSInfo->getTypeLoc());
1637
1638 return false;
1639}
1640
1641bool CursorVisitor::VisitDependentNameTypeLoc(DependentNameTypeLoc TL) {
Hans Wennborg59dbe862015-09-29 20:56:43 +00001642 return VisitNestedNameSpecifierLoc(TL.getQualifierLoc());
Guy Benyei11169dd2012-12-18 14:30:41 +00001643}
1644
1645bool CursorVisitor::VisitDependentTemplateSpecializationTypeLoc(
1646 DependentTemplateSpecializationTypeLoc TL) {
1647 // Visit the nested-name-specifier, if there is one.
1648 if (TL.getQualifierLoc() &&
1649 VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
1650 return true;
1651
1652 // Visit the template arguments.
1653 for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
1654 if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
1655 return true;
1656
1657 return false;
1658}
1659
1660bool CursorVisitor::VisitElaboratedTypeLoc(ElaboratedTypeLoc TL) {
1661 if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
1662 return true;
1663
1664 return Visit(TL.getNamedTypeLoc());
1665}
1666
1667bool CursorVisitor::VisitPackExpansionTypeLoc(PackExpansionTypeLoc TL) {
1668 return Visit(TL.getPatternLoc());
1669}
1670
1671bool CursorVisitor::VisitDecltypeTypeLoc(DecltypeTypeLoc TL) {
1672 if (Expr *E = TL.getUnderlyingExpr())
1673 return Visit(MakeCXCursor(E, StmtParent, TU));
1674
1675 return false;
1676}
1677
1678bool CursorVisitor::VisitInjectedClassNameTypeLoc(InjectedClassNameTypeLoc TL) {
1679 return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1680}
1681
1682bool CursorVisitor::VisitAtomicTypeLoc(AtomicTypeLoc TL) {
1683 return Visit(TL.getValueLoc());
1684}
1685
Xiuli Pan9c14e282016-01-09 12:53:17 +00001686bool CursorVisitor::VisitPipeTypeLoc(PipeTypeLoc TL) {
1687 return Visit(TL.getValueLoc());
1688}
1689
Guy Benyei11169dd2012-12-18 14:30:41 +00001690#define DEFAULT_TYPELOC_IMPL(CLASS, PARENT) \
1691bool CursorVisitor::Visit##CLASS##TypeLoc(CLASS##TypeLoc TL) { \
1692 return Visit##PARENT##Loc(TL); \
1693}
1694
1695DEFAULT_TYPELOC_IMPL(Complex, Type)
1696DEFAULT_TYPELOC_IMPL(ConstantArray, ArrayType)
1697DEFAULT_TYPELOC_IMPL(IncompleteArray, ArrayType)
1698DEFAULT_TYPELOC_IMPL(VariableArray, ArrayType)
1699DEFAULT_TYPELOC_IMPL(DependentSizedArray, ArrayType)
1700DEFAULT_TYPELOC_IMPL(DependentSizedExtVector, Type)
1701DEFAULT_TYPELOC_IMPL(Vector, Type)
1702DEFAULT_TYPELOC_IMPL(ExtVector, VectorType)
1703DEFAULT_TYPELOC_IMPL(FunctionProto, FunctionType)
1704DEFAULT_TYPELOC_IMPL(FunctionNoProto, FunctionType)
1705DEFAULT_TYPELOC_IMPL(Record, TagType)
1706DEFAULT_TYPELOC_IMPL(Enum, TagType)
1707DEFAULT_TYPELOC_IMPL(SubstTemplateTypeParm, Type)
1708DEFAULT_TYPELOC_IMPL(SubstTemplateTypeParmPack, Type)
1709DEFAULT_TYPELOC_IMPL(Auto, Type)
1710
1711bool CursorVisitor::VisitCXXRecordDecl(CXXRecordDecl *D) {
1712 // Visit the nested-name-specifier, if present.
1713 if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1714 if (VisitNestedNameSpecifierLoc(QualifierLoc))
1715 return true;
1716
1717 if (D->isCompleteDefinition()) {
Aaron Ballman574705e2014-03-13 15:41:46 +00001718 for (const auto &I : D->bases()) {
1719 if (Visit(cxcursor::MakeCursorCXXBaseSpecifier(&I, TU)))
Guy Benyei11169dd2012-12-18 14:30:41 +00001720 return true;
1721 }
1722 }
1723
1724 return VisitTagDecl(D);
1725}
1726
1727bool CursorVisitor::VisitAttributes(Decl *D) {
Aaron Ballmanb97112e2014-03-08 22:19:01 +00001728 for (const auto *I : D->attrs())
1729 if (Visit(MakeCXCursor(I, D, TU)))
Guy Benyei11169dd2012-12-18 14:30:41 +00001730 return true;
1731
1732 return false;
1733}
1734
1735//===----------------------------------------------------------------------===//
1736// Data-recursive visitor methods.
1737//===----------------------------------------------------------------------===//
1738
1739namespace {
1740#define DEF_JOB(NAME, DATA, KIND)\
1741class NAME : public VisitorJob {\
1742public:\
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001743 NAME(const DATA *d, CXCursor parent) : \
1744 VisitorJob(parent, VisitorJob::KIND, d) {} \
Guy Benyei11169dd2012-12-18 14:30:41 +00001745 static bool classof(const VisitorJob *VJ) { return VJ->getKind() == KIND; }\
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001746 const DATA *get() const { return static_cast<const DATA*>(data[0]); }\
Guy Benyei11169dd2012-12-18 14:30:41 +00001747};
1748
1749DEF_JOB(StmtVisit, Stmt, StmtVisitKind)
1750DEF_JOB(MemberExprParts, MemberExpr, MemberExprPartsKind)
1751DEF_JOB(DeclRefExprParts, DeclRefExpr, DeclRefExprPartsKind)
1752DEF_JOB(OverloadExprParts, OverloadExpr, OverloadExprPartsKind)
Guy Benyei11169dd2012-12-18 14:30:41 +00001753DEF_JOB(SizeOfPackExprParts, SizeOfPackExpr, SizeOfPackExprPartsKind)
1754DEF_JOB(LambdaExprParts, LambdaExpr, LambdaExprPartsKind)
1755DEF_JOB(PostChildrenVisit, void, PostChildrenVisitKind)
1756#undef DEF_JOB
1757
James Y Knight04ec5bf2015-12-24 02:59:37 +00001758class ExplicitTemplateArgsVisit : public VisitorJob {
1759public:
1760 ExplicitTemplateArgsVisit(const TemplateArgumentLoc *Begin,
1761 const TemplateArgumentLoc *End, CXCursor parent)
1762 : VisitorJob(parent, VisitorJob::ExplicitTemplateArgsVisitKind, Begin,
1763 End) {}
1764 static bool classof(const VisitorJob *VJ) {
1765 return VJ->getKind() == ExplicitTemplateArgsVisitKind;
1766 }
1767 const TemplateArgumentLoc *begin() const {
1768 return static_cast<const TemplateArgumentLoc *>(data[0]);
1769 }
1770 const TemplateArgumentLoc *end() {
1771 return static_cast<const TemplateArgumentLoc *>(data[1]);
1772 }
1773};
Guy Benyei11169dd2012-12-18 14:30:41 +00001774class DeclVisit : public VisitorJob {
1775public:
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001776 DeclVisit(const Decl *D, CXCursor parent, bool isFirst) :
Guy Benyei11169dd2012-12-18 14:30:41 +00001777 VisitorJob(parent, VisitorJob::DeclVisitKind,
Craig Topper69186e72014-06-08 08:38:04 +00001778 D, isFirst ? (void*) 1 : (void*) nullptr) {}
Guy Benyei11169dd2012-12-18 14:30:41 +00001779 static bool classof(const VisitorJob *VJ) {
1780 return VJ->getKind() == DeclVisitKind;
1781 }
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001782 const Decl *get() const { return static_cast<const Decl *>(data[0]); }
Dmitri Gribenkoe5423a72015-03-23 19:23:50 +00001783 bool isFirst() const { return data[1] != nullptr; }
Guy Benyei11169dd2012-12-18 14:30:41 +00001784};
1785class TypeLocVisit : public VisitorJob {
1786public:
1787 TypeLocVisit(TypeLoc tl, CXCursor parent) :
1788 VisitorJob(parent, VisitorJob::TypeLocVisitKind,
1789 tl.getType().getAsOpaquePtr(), tl.getOpaqueData()) {}
1790
1791 static bool classof(const VisitorJob *VJ) {
1792 return VJ->getKind() == TypeLocVisitKind;
1793 }
1794
1795 TypeLoc get() const {
1796 QualType T = QualType::getFromOpaquePtr(data[0]);
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001797 return TypeLoc(T, const_cast<void *>(data[1]));
Guy Benyei11169dd2012-12-18 14:30:41 +00001798 }
1799};
1800
1801class LabelRefVisit : public VisitorJob {
1802public:
1803 LabelRefVisit(LabelDecl *LD, SourceLocation labelLoc, CXCursor parent)
1804 : VisitorJob(parent, VisitorJob::LabelRefVisitKind, LD,
1805 labelLoc.getPtrEncoding()) {}
1806
1807 static bool classof(const VisitorJob *VJ) {
1808 return VJ->getKind() == VisitorJob::LabelRefVisitKind;
1809 }
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001810 const LabelDecl *get() const {
1811 return static_cast<const LabelDecl *>(data[0]);
1812 }
Guy Benyei11169dd2012-12-18 14:30:41 +00001813 SourceLocation getLoc() const {
1814 return SourceLocation::getFromPtrEncoding(data[1]); }
1815};
1816
1817class NestedNameSpecifierLocVisit : public VisitorJob {
1818public:
1819 NestedNameSpecifierLocVisit(NestedNameSpecifierLoc Qualifier, CXCursor parent)
1820 : VisitorJob(parent, VisitorJob::NestedNameSpecifierLocVisitKind,
1821 Qualifier.getNestedNameSpecifier(),
1822 Qualifier.getOpaqueData()) { }
1823
1824 static bool classof(const VisitorJob *VJ) {
1825 return VJ->getKind() == VisitorJob::NestedNameSpecifierLocVisitKind;
1826 }
1827
1828 NestedNameSpecifierLoc get() const {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001829 return NestedNameSpecifierLoc(
1830 const_cast<NestedNameSpecifier *>(
1831 static_cast<const NestedNameSpecifier *>(data[0])),
1832 const_cast<void *>(data[1]));
Guy Benyei11169dd2012-12-18 14:30:41 +00001833 }
1834};
1835
1836class DeclarationNameInfoVisit : public VisitorJob {
1837public:
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001838 DeclarationNameInfoVisit(const Stmt *S, CXCursor parent)
Dmitri Gribenkodd7dacf2013-02-03 13:19:54 +00001839 : VisitorJob(parent, VisitorJob::DeclarationNameInfoVisitKind, S) {}
Guy Benyei11169dd2012-12-18 14:30:41 +00001840 static bool classof(const VisitorJob *VJ) {
1841 return VJ->getKind() == VisitorJob::DeclarationNameInfoVisitKind;
1842 }
1843 DeclarationNameInfo get() const {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001844 const Stmt *S = static_cast<const Stmt *>(data[0]);
Guy Benyei11169dd2012-12-18 14:30:41 +00001845 switch (S->getStmtClass()) {
1846 default:
1847 llvm_unreachable("Unhandled Stmt");
1848 case clang::Stmt::MSDependentExistsStmtClass:
1849 return cast<MSDependentExistsStmt>(S)->getNameInfo();
1850 case Stmt::CXXDependentScopeMemberExprClass:
1851 return cast<CXXDependentScopeMemberExpr>(S)->getMemberNameInfo();
1852 case Stmt::DependentScopeDeclRefExprClass:
1853 return cast<DependentScopeDeclRefExpr>(S)->getNameInfo();
Alexander Musmand9ed09f2014-07-21 09:42:05 +00001854 case Stmt::OMPCriticalDirectiveClass:
1855 return cast<OMPCriticalDirective>(S)->getDirectiveName();
Guy Benyei11169dd2012-12-18 14:30:41 +00001856 }
1857 }
1858};
1859class MemberRefVisit : public VisitorJob {
1860public:
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001861 MemberRefVisit(const FieldDecl *D, SourceLocation L, CXCursor parent)
Guy Benyei11169dd2012-12-18 14:30:41 +00001862 : VisitorJob(parent, VisitorJob::MemberRefVisitKind, D,
1863 L.getPtrEncoding()) {}
1864 static bool classof(const VisitorJob *VJ) {
1865 return VJ->getKind() == VisitorJob::MemberRefVisitKind;
1866 }
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001867 const FieldDecl *get() const {
1868 return static_cast<const FieldDecl *>(data[0]);
Guy Benyei11169dd2012-12-18 14:30:41 +00001869 }
1870 SourceLocation getLoc() const {
1871 return SourceLocation::getFromRawEncoding((unsigned)(uintptr_t) data[1]);
1872 }
1873};
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001874class EnqueueVisitor : public ConstStmtVisitor<EnqueueVisitor, void> {
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001875 friend class OMPClauseEnqueue;
Guy Benyei11169dd2012-12-18 14:30:41 +00001876 VisitorWorkList &WL;
1877 CXCursor Parent;
1878public:
1879 EnqueueVisitor(VisitorWorkList &wl, CXCursor parent)
1880 : WL(wl), Parent(parent) {}
1881
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001882 void VisitAddrLabelExpr(const AddrLabelExpr *E);
1883 void VisitBlockExpr(const BlockExpr *B);
1884 void VisitCompoundLiteralExpr(const CompoundLiteralExpr *E);
1885 void VisitCompoundStmt(const CompoundStmt *S);
1886 void VisitCXXDefaultArgExpr(const CXXDefaultArgExpr *E) { /* Do nothing. */ }
1887 void VisitMSDependentExistsStmt(const MSDependentExistsStmt *S);
1888 void VisitCXXDependentScopeMemberExpr(const CXXDependentScopeMemberExpr *E);
1889 void VisitCXXNewExpr(const CXXNewExpr *E);
1890 void VisitCXXScalarValueInitExpr(const CXXScalarValueInitExpr *E);
1891 void VisitCXXOperatorCallExpr(const CXXOperatorCallExpr *E);
1892 void VisitCXXPseudoDestructorExpr(const CXXPseudoDestructorExpr *E);
1893 void VisitCXXTemporaryObjectExpr(const CXXTemporaryObjectExpr *E);
1894 void VisitCXXTypeidExpr(const CXXTypeidExpr *E);
1895 void VisitCXXUnresolvedConstructExpr(const CXXUnresolvedConstructExpr *E);
1896 void VisitCXXUuidofExpr(const CXXUuidofExpr *E);
1897 void VisitCXXCatchStmt(const CXXCatchStmt *S);
Argyrios Kyrtzidis99891242014-11-13 09:03:21 +00001898 void VisitCXXForRangeStmt(const CXXForRangeStmt *S);
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001899 void VisitDeclRefExpr(const DeclRefExpr *D);
1900 void VisitDeclStmt(const DeclStmt *S);
1901 void VisitDependentScopeDeclRefExpr(const DependentScopeDeclRefExpr *E);
1902 void VisitDesignatedInitExpr(const DesignatedInitExpr *E);
1903 void VisitExplicitCastExpr(const ExplicitCastExpr *E);
1904 void VisitForStmt(const ForStmt *FS);
1905 void VisitGotoStmt(const GotoStmt *GS);
1906 void VisitIfStmt(const IfStmt *If);
1907 void VisitInitListExpr(const InitListExpr *IE);
1908 void VisitMemberExpr(const MemberExpr *M);
1909 void VisitOffsetOfExpr(const OffsetOfExpr *E);
1910 void VisitObjCEncodeExpr(const ObjCEncodeExpr *E);
1911 void VisitObjCMessageExpr(const ObjCMessageExpr *M);
1912 void VisitOverloadExpr(const OverloadExpr *E);
1913 void VisitUnaryExprOrTypeTraitExpr(const UnaryExprOrTypeTraitExpr *E);
1914 void VisitStmt(const Stmt *S);
1915 void VisitSwitchStmt(const SwitchStmt *S);
1916 void VisitWhileStmt(const WhileStmt *W);
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001917 void VisitTypeTraitExpr(const TypeTraitExpr *E);
1918 void VisitArrayTypeTraitExpr(const ArrayTypeTraitExpr *E);
1919 void VisitExpressionTraitExpr(const ExpressionTraitExpr *E);
1920 void VisitUnresolvedMemberExpr(const UnresolvedMemberExpr *U);
1921 void VisitVAArgExpr(const VAArgExpr *E);
1922 void VisitSizeOfPackExpr(const SizeOfPackExpr *E);
1923 void VisitPseudoObjectExpr(const PseudoObjectExpr *E);
1924 void VisitOpaqueValueExpr(const OpaqueValueExpr *E);
1925 void VisitLambdaExpr(const LambdaExpr *E);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001926 void VisitOMPExecutableDirective(const OMPExecutableDirective *D);
Alexander Musman3aaab662014-08-19 11:27:13 +00001927 void VisitOMPLoopDirective(const OMPLoopDirective *D);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001928 void VisitOMPParallelDirective(const OMPParallelDirective *D);
Alexey Bataev1b59ab52014-02-27 08:29:12 +00001929 void VisitOMPSimdDirective(const OMPSimdDirective *D);
Alexey Bataevf29276e2014-06-18 04:14:57 +00001930 void VisitOMPForDirective(const OMPForDirective *D);
Alexander Musmanf82886e2014-09-18 05:12:34 +00001931 void VisitOMPForSimdDirective(const OMPForSimdDirective *D);
Alexey Bataevd3f8dd22014-06-25 11:44:49 +00001932 void VisitOMPSectionsDirective(const OMPSectionsDirective *D);
Alexey Bataev1e0498a2014-06-26 08:21:58 +00001933 void VisitOMPSectionDirective(const OMPSectionDirective *D);
Alexey Bataevd1e40fb2014-06-26 12:05:45 +00001934 void VisitOMPSingleDirective(const OMPSingleDirective *D);
Alexander Musman80c22892014-07-17 08:54:58 +00001935 void VisitOMPMasterDirective(const OMPMasterDirective *D);
Alexander Musmand9ed09f2014-07-21 09:42:05 +00001936 void VisitOMPCriticalDirective(const OMPCriticalDirective *D);
Alexey Bataev4acb8592014-07-07 13:01:15 +00001937 void VisitOMPParallelForDirective(const OMPParallelForDirective *D);
Alexander Musmane4e893b2014-09-23 09:33:00 +00001938 void VisitOMPParallelForSimdDirective(const OMPParallelForSimdDirective *D);
Alexey Bataev84d0b3e2014-07-08 08:12:03 +00001939 void VisitOMPParallelSectionsDirective(const OMPParallelSectionsDirective *D);
Alexey Bataev9c2e8ee2014-07-11 11:25:16 +00001940 void VisitOMPTaskDirective(const OMPTaskDirective *D);
Alexey Bataev68446b72014-07-18 07:47:19 +00001941 void VisitOMPTaskyieldDirective(const OMPTaskyieldDirective *D);
Alexey Bataev4d1dfea2014-07-18 09:11:51 +00001942 void VisitOMPBarrierDirective(const OMPBarrierDirective *D);
Alexey Bataev2df347a2014-07-18 10:17:07 +00001943 void VisitOMPTaskwaitDirective(const OMPTaskwaitDirective *D);
Alexey Bataevc30dd2d2015-06-18 12:14:09 +00001944 void VisitOMPTaskgroupDirective(const OMPTaskgroupDirective *D);
Alexey Bataev6d4ed052015-07-01 06:57:41 +00001945 void
1946 VisitOMPCancellationPointDirective(const OMPCancellationPointDirective *D);
Alexey Bataev80909872015-07-02 11:25:17 +00001947 void VisitOMPCancelDirective(const OMPCancelDirective *D);
Alexey Bataev6125da92014-07-21 11:26:11 +00001948 void VisitOMPFlushDirective(const OMPFlushDirective *D);
Alexey Bataev9fb6e642014-07-22 06:45:04 +00001949 void VisitOMPOrderedDirective(const OMPOrderedDirective *D);
Alexey Bataev0162e452014-07-22 10:10:35 +00001950 void VisitOMPAtomicDirective(const OMPAtomicDirective *D);
Alexey Bataev0bd520b2014-09-19 08:19:49 +00001951 void VisitOMPTargetDirective(const OMPTargetDirective *D);
Michael Wong65f367f2015-07-21 13:44:28 +00001952 void VisitOMPTargetDataDirective(const OMPTargetDataDirective *D);
Samuel Antaodf67fc42016-01-19 19:15:56 +00001953 void VisitOMPTargetEnterDataDirective(const OMPTargetEnterDataDirective *D);
Samuel Antao72590762016-01-19 20:04:50 +00001954 void VisitOMPTargetExitDataDirective(const OMPTargetExitDataDirective *D);
Arpith Chacko Jacobe955b3d2016-01-26 18:48:41 +00001955 void VisitOMPTargetParallelDirective(const OMPTargetParallelDirective *D);
Arpith Chacko Jacob05bebb52016-02-03 15:46:42 +00001956 void
1957 VisitOMPTargetParallelForDirective(const OMPTargetParallelForDirective *D);
Alexey Bataev13314bf2014-10-09 04:18:56 +00001958 void VisitOMPTeamsDirective(const OMPTeamsDirective *D);
Alexey Bataev49f6e782015-12-01 04:18:41 +00001959 void VisitOMPTaskLoopDirective(const OMPTaskLoopDirective *D);
Alexey Bataev0a6ed842015-12-03 09:40:15 +00001960 void VisitOMPTaskLoopSimdDirective(const OMPTaskLoopSimdDirective *D);
Carlo Bertolli6200a3d2015-12-14 14:51:25 +00001961 void VisitOMPDistributeDirective(const OMPDistributeDirective *D);
Carlo Bertolli9925f152016-06-27 14:55:37 +00001962 void VisitOMPDistributeParallelForDirective(
1963 const OMPDistributeParallelForDirective *D);
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001964
Guy Benyei11169dd2012-12-18 14:30:41 +00001965private:
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001966 void AddDeclarationNameInfo(const Stmt *S);
Guy Benyei11169dd2012-12-18 14:30:41 +00001967 void AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier);
James Y Knight04ec5bf2015-12-24 02:59:37 +00001968 void AddExplicitTemplateArgs(const TemplateArgumentLoc *A,
1969 unsigned NumTemplateArgs);
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001970 void AddMemberRef(const FieldDecl *D, SourceLocation L);
1971 void AddStmt(const Stmt *S);
1972 void AddDecl(const Decl *D, bool isFirst = true);
Guy Benyei11169dd2012-12-18 14:30:41 +00001973 void AddTypeLoc(TypeSourceInfo *TI);
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001974 void EnqueueChildren(const Stmt *S);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00001975 void EnqueueChildren(const OMPClause *S);
Guy Benyei11169dd2012-12-18 14:30:41 +00001976};
Alexander Kornienkoab9db512015-06-22 23:07:51 +00001977} // end anonyous namespace
Guy Benyei11169dd2012-12-18 14:30:41 +00001978
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001979void EnqueueVisitor::AddDeclarationNameInfo(const Stmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001980 // 'S' should always be non-null, since it comes from the
1981 // statement we are visiting.
1982 WL.push_back(DeclarationNameInfoVisit(S, Parent));
1983}
1984
1985void
1986EnqueueVisitor::AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
1987 if (Qualifier)
1988 WL.push_back(NestedNameSpecifierLocVisit(Qualifier, Parent));
1989}
1990
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001991void EnqueueVisitor::AddStmt(const Stmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001992 if (S)
1993 WL.push_back(StmtVisit(S, Parent));
1994}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00001995void EnqueueVisitor::AddDecl(const Decl *D, bool isFirst) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001996 if (D)
1997 WL.push_back(DeclVisit(D, Parent, isFirst));
1998}
James Y Knight04ec5bf2015-12-24 02:59:37 +00001999void EnqueueVisitor::AddExplicitTemplateArgs(const TemplateArgumentLoc *A,
2000 unsigned NumTemplateArgs) {
2001 WL.push_back(ExplicitTemplateArgsVisit(A, A + NumTemplateArgs, Parent));
Guy Benyei11169dd2012-12-18 14:30:41 +00002002}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002003void EnqueueVisitor::AddMemberRef(const FieldDecl *D, SourceLocation L) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002004 if (D)
2005 WL.push_back(MemberRefVisit(D, L, Parent));
2006}
2007void EnqueueVisitor::AddTypeLoc(TypeSourceInfo *TI) {
2008 if (TI)
2009 WL.push_back(TypeLocVisit(TI->getTypeLoc(), Parent));
2010 }
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002011void EnqueueVisitor::EnqueueChildren(const Stmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002012 unsigned size = WL.size();
Benjamin Kramer642f1732015-07-02 21:03:14 +00002013 for (const Stmt *SubStmt : S->children()) {
2014 AddStmt(SubStmt);
Guy Benyei11169dd2012-12-18 14:30:41 +00002015 }
2016 if (size == WL.size())
2017 return;
2018 // Now reverse the entries we just added. This will match the DFS
2019 // ordering performed by the worklist.
2020 VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
2021 std::reverse(I, E);
2022}
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002023namespace {
2024class OMPClauseEnqueue : public ConstOMPClauseVisitor<OMPClauseEnqueue> {
2025 EnqueueVisitor *Visitor;
Alexey Bataev756c1962013-09-24 03:17:45 +00002026 /// \brief Process clauses with list of variables.
2027 template <typename T>
2028 void VisitOMPClauseList(T *Node);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002029public:
2030 OMPClauseEnqueue(EnqueueVisitor *Visitor) : Visitor(Visitor) { }
2031#define OPENMP_CLAUSE(Name, Class) \
2032 void Visit##Class(const Class *C);
2033#include "clang/Basic/OpenMPKinds.def"
Alexey Bataev3392d762016-02-16 11:18:12 +00002034 void VisitOMPClauseWithPreInit(const OMPClauseWithPreInit *C);
Alexey Bataev005248a2016-02-25 05:25:57 +00002035 void VisitOMPClauseWithPostUpdate(const OMPClauseWithPostUpdate *C);
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002036};
2037
Alexey Bataev3392d762016-02-16 11:18:12 +00002038void OMPClauseEnqueue::VisitOMPClauseWithPreInit(
2039 const OMPClauseWithPreInit *C) {
2040 Visitor->AddStmt(C->getPreInitStmt());
2041}
2042
Alexey Bataev005248a2016-02-25 05:25:57 +00002043void OMPClauseEnqueue::VisitOMPClauseWithPostUpdate(
2044 const OMPClauseWithPostUpdate *C) {
Alexey Bataev37e594c2016-03-04 07:21:16 +00002045 VisitOMPClauseWithPreInit(C);
Alexey Bataev005248a2016-02-25 05:25:57 +00002046 Visitor->AddStmt(C->getPostUpdateExpr());
2047}
2048
Alexey Bataevaadd52e2014-02-13 05:29:23 +00002049void OMPClauseEnqueue::VisitOMPIfClause(const OMPIfClause *C) {
2050 Visitor->AddStmt(C->getCondition());
2051}
2052
Alexey Bataev3778b602014-07-17 07:32:53 +00002053void OMPClauseEnqueue::VisitOMPFinalClause(const OMPFinalClause *C) {
2054 Visitor->AddStmt(C->getCondition());
2055}
2056
Alexey Bataev568a8332014-03-06 06:15:19 +00002057void OMPClauseEnqueue::VisitOMPNumThreadsClause(const OMPNumThreadsClause *C) {
2058 Visitor->AddStmt(C->getNumThreads());
2059}
2060
Alexey Bataev62c87d22014-03-21 04:51:18 +00002061void OMPClauseEnqueue::VisitOMPSafelenClause(const OMPSafelenClause *C) {
2062 Visitor->AddStmt(C->getSafelen());
2063}
2064
Alexey Bataev66b15b52015-08-21 11:14:16 +00002065void OMPClauseEnqueue::VisitOMPSimdlenClause(const OMPSimdlenClause *C) {
2066 Visitor->AddStmt(C->getSimdlen());
2067}
2068
Alexander Musman8bd31e62014-05-27 15:12:19 +00002069void OMPClauseEnqueue::VisitOMPCollapseClause(const OMPCollapseClause *C) {
2070 Visitor->AddStmt(C->getNumForLoops());
2071}
2072
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002073void OMPClauseEnqueue::VisitOMPDefaultClause(const OMPDefaultClause *C) { }
Alexey Bataev756c1962013-09-24 03:17:45 +00002074
Alexey Bataevbcbadb62014-05-06 06:04:14 +00002075void OMPClauseEnqueue::VisitOMPProcBindClause(const OMPProcBindClause *C) { }
2076
Alexey Bataev56dafe82014-06-20 07:16:17 +00002077void OMPClauseEnqueue::VisitOMPScheduleClause(const OMPScheduleClause *C) {
Alexey Bataev3392d762016-02-16 11:18:12 +00002078 VisitOMPClauseWithPreInit(C);
Alexey Bataev56dafe82014-06-20 07:16:17 +00002079 Visitor->AddStmt(C->getChunkSize());
2080}
2081
Alexey Bataev10e775f2015-07-30 11:36:16 +00002082void OMPClauseEnqueue::VisitOMPOrderedClause(const OMPOrderedClause *C) {
2083 Visitor->AddStmt(C->getNumForLoops());
2084}
Alexey Bataev142e1fc2014-06-20 09:44:06 +00002085
Alexey Bataev236070f2014-06-20 11:19:47 +00002086void OMPClauseEnqueue::VisitOMPNowaitClause(const OMPNowaitClause *) {}
2087
Alexey Bataev7aea99a2014-07-17 12:19:31 +00002088void OMPClauseEnqueue::VisitOMPUntiedClause(const OMPUntiedClause *) {}
2089
Alexey Bataev74ba3a52014-07-17 12:47:03 +00002090void OMPClauseEnqueue::VisitOMPMergeableClause(const OMPMergeableClause *) {}
2091
Alexey Bataevf98b00c2014-07-23 02:27:21 +00002092void OMPClauseEnqueue::VisitOMPReadClause(const OMPReadClause *) {}
2093
Alexey Bataevdea47612014-07-23 07:46:59 +00002094void OMPClauseEnqueue::VisitOMPWriteClause(const OMPWriteClause *) {}
2095
Alexey Bataev67a4f222014-07-23 10:25:33 +00002096void OMPClauseEnqueue::VisitOMPUpdateClause(const OMPUpdateClause *) {}
2097
Alexey Bataev459dec02014-07-24 06:46:57 +00002098void OMPClauseEnqueue::VisitOMPCaptureClause(const OMPCaptureClause *) {}
2099
Alexey Bataev82bad8b2014-07-24 08:55:34 +00002100void OMPClauseEnqueue::VisitOMPSeqCstClause(const OMPSeqCstClause *) {}
2101
Alexey Bataev346265e2015-09-25 10:37:12 +00002102void OMPClauseEnqueue::VisitOMPThreadsClause(const OMPThreadsClause *) {}
2103
Alexey Bataevd14d1e62015-09-28 06:39:35 +00002104void OMPClauseEnqueue::VisitOMPSIMDClause(const OMPSIMDClause *) {}
2105
Alexey Bataevb825de12015-12-07 10:51:44 +00002106void OMPClauseEnqueue::VisitOMPNogroupClause(const OMPNogroupClause *) {}
2107
Michael Wonge710d542015-08-07 16:16:36 +00002108void OMPClauseEnqueue::VisitOMPDeviceClause(const OMPDeviceClause *C) {
2109 Visitor->AddStmt(C->getDevice());
2110}
2111
Kelvin Li099bb8c2015-11-24 20:50:12 +00002112void OMPClauseEnqueue::VisitOMPNumTeamsClause(const OMPNumTeamsClause *C) {
2113 Visitor->AddStmt(C->getNumTeams());
2114}
2115
Kelvin Lia15fb1a2015-11-27 18:47:36 +00002116void OMPClauseEnqueue::VisitOMPThreadLimitClause(const OMPThreadLimitClause *C) {
2117 Visitor->AddStmt(C->getThreadLimit());
2118}
2119
Alexey Bataeva0569352015-12-01 10:17:31 +00002120void OMPClauseEnqueue::VisitOMPPriorityClause(const OMPPriorityClause *C) {
2121 Visitor->AddStmt(C->getPriority());
2122}
2123
Alexey Bataev1fd4aed2015-12-07 12:52:51 +00002124void OMPClauseEnqueue::VisitOMPGrainsizeClause(const OMPGrainsizeClause *C) {
2125 Visitor->AddStmt(C->getGrainsize());
2126}
2127
Alexey Bataev382967a2015-12-08 12:06:20 +00002128void OMPClauseEnqueue::VisitOMPNumTasksClause(const OMPNumTasksClause *C) {
2129 Visitor->AddStmt(C->getNumTasks());
2130}
2131
Alexey Bataev28c75412015-12-15 08:19:24 +00002132void OMPClauseEnqueue::VisitOMPHintClause(const OMPHintClause *C) {
2133 Visitor->AddStmt(C->getHint());
2134}
2135
Alexey Bataev756c1962013-09-24 03:17:45 +00002136template<typename T>
2137void OMPClauseEnqueue::VisitOMPClauseList(T *Node) {
Alexey Bataev03b340a2014-10-21 03:16:40 +00002138 for (const auto *I : Node->varlists()) {
Aaron Ballman2205d2a2014-03-14 15:55:35 +00002139 Visitor->AddStmt(I);
Alexey Bataev03b340a2014-10-21 03:16:40 +00002140 }
Alexey Bataev756c1962013-09-24 03:17:45 +00002141}
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002142
2143void OMPClauseEnqueue::VisitOMPPrivateClause(const OMPPrivateClause *C) {
Alexey Bataev756c1962013-09-24 03:17:45 +00002144 VisitOMPClauseList(C);
Alexey Bataev03b340a2014-10-21 03:16:40 +00002145 for (const auto *E : C->private_copies()) {
2146 Visitor->AddStmt(E);
2147 }
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002148}
Alexey Bataevd5af8e42013-10-01 05:32:34 +00002149void OMPClauseEnqueue::VisitOMPFirstprivateClause(
2150 const OMPFirstprivateClause *C) {
2151 VisitOMPClauseList(C);
Alexey Bataev417089f2016-02-17 13:19:37 +00002152 VisitOMPClauseWithPreInit(C);
2153 for (const auto *E : C->private_copies()) {
2154 Visitor->AddStmt(E);
2155 }
2156 for (const auto *E : C->inits()) {
2157 Visitor->AddStmt(E);
2158 }
Alexey Bataevd5af8e42013-10-01 05:32:34 +00002159}
Alexander Musman1bb328c2014-06-04 13:06:39 +00002160void OMPClauseEnqueue::VisitOMPLastprivateClause(
2161 const OMPLastprivateClause *C) {
2162 VisitOMPClauseList(C);
Alexey Bataev005248a2016-02-25 05:25:57 +00002163 VisitOMPClauseWithPostUpdate(C);
Alexey Bataev38e89532015-04-16 04:54:05 +00002164 for (auto *E : C->private_copies()) {
2165 Visitor->AddStmt(E);
2166 }
2167 for (auto *E : C->source_exprs()) {
2168 Visitor->AddStmt(E);
2169 }
2170 for (auto *E : C->destination_exprs()) {
2171 Visitor->AddStmt(E);
2172 }
2173 for (auto *E : C->assignment_ops()) {
2174 Visitor->AddStmt(E);
2175 }
Alexander Musman1bb328c2014-06-04 13:06:39 +00002176}
Alexey Bataev758e55e2013-09-06 18:03:48 +00002177void OMPClauseEnqueue::VisitOMPSharedClause(const OMPSharedClause *C) {
Alexey Bataev756c1962013-09-24 03:17:45 +00002178 VisitOMPClauseList(C);
Alexey Bataev758e55e2013-09-06 18:03:48 +00002179}
Alexey Bataevc5e02582014-06-16 07:08:35 +00002180void OMPClauseEnqueue::VisitOMPReductionClause(const OMPReductionClause *C) {
2181 VisitOMPClauseList(C);
Alexey Bataev61205072016-03-02 04:57:40 +00002182 VisitOMPClauseWithPostUpdate(C);
Alexey Bataevf24e7b12015-10-08 09:10:53 +00002183 for (auto *E : C->privates()) {
2184 Visitor->AddStmt(E);
2185 }
Alexey Bataev794ba0d2015-04-10 10:43:45 +00002186 for (auto *E : C->lhs_exprs()) {
2187 Visitor->AddStmt(E);
2188 }
2189 for (auto *E : C->rhs_exprs()) {
2190 Visitor->AddStmt(E);
2191 }
2192 for (auto *E : C->reduction_ops()) {
2193 Visitor->AddStmt(E);
2194 }
Alexey Bataevc5e02582014-06-16 07:08:35 +00002195}
Alexander Musman8dba6642014-04-22 13:09:42 +00002196void OMPClauseEnqueue::VisitOMPLinearClause(const OMPLinearClause *C) {
2197 VisitOMPClauseList(C);
Alexey Bataev78849fb2016-03-09 09:49:00 +00002198 VisitOMPClauseWithPostUpdate(C);
Alexey Bataevbd9fec12015-08-18 06:47:21 +00002199 for (const auto *E : C->privates()) {
2200 Visitor->AddStmt(E);
2201 }
Alexander Musman3276a272015-03-21 10:12:56 +00002202 for (const auto *E : C->inits()) {
2203 Visitor->AddStmt(E);
2204 }
2205 for (const auto *E : C->updates()) {
2206 Visitor->AddStmt(E);
2207 }
2208 for (const auto *E : C->finals()) {
2209 Visitor->AddStmt(E);
2210 }
Alexander Musman8dba6642014-04-22 13:09:42 +00002211 Visitor->AddStmt(C->getStep());
Alexander Musman3276a272015-03-21 10:12:56 +00002212 Visitor->AddStmt(C->getCalcStep());
Alexander Musman8dba6642014-04-22 13:09:42 +00002213}
Alexander Musmanf0d76e72014-05-29 14:36:25 +00002214void OMPClauseEnqueue::VisitOMPAlignedClause(const OMPAlignedClause *C) {
2215 VisitOMPClauseList(C);
2216 Visitor->AddStmt(C->getAlignment());
2217}
Alexey Bataevd48bcd82014-03-31 03:36:38 +00002218void OMPClauseEnqueue::VisitOMPCopyinClause(const OMPCopyinClause *C) {
2219 VisitOMPClauseList(C);
Alexey Bataevf56f98c2015-04-16 05:39:01 +00002220 for (auto *E : C->source_exprs()) {
2221 Visitor->AddStmt(E);
2222 }
2223 for (auto *E : C->destination_exprs()) {
2224 Visitor->AddStmt(E);
2225 }
2226 for (auto *E : C->assignment_ops()) {
2227 Visitor->AddStmt(E);
2228 }
Alexey Bataevd48bcd82014-03-31 03:36:38 +00002229}
Alexey Bataevbae9a792014-06-27 10:37:06 +00002230void
2231OMPClauseEnqueue::VisitOMPCopyprivateClause(const OMPCopyprivateClause *C) {
2232 VisitOMPClauseList(C);
Alexey Bataeva63048e2015-03-23 06:18:07 +00002233 for (auto *E : C->source_exprs()) {
2234 Visitor->AddStmt(E);
2235 }
2236 for (auto *E : C->destination_exprs()) {
2237 Visitor->AddStmt(E);
2238 }
2239 for (auto *E : C->assignment_ops()) {
2240 Visitor->AddStmt(E);
2241 }
Alexey Bataevbae9a792014-06-27 10:37:06 +00002242}
Alexey Bataev6125da92014-07-21 11:26:11 +00002243void OMPClauseEnqueue::VisitOMPFlushClause(const OMPFlushClause *C) {
2244 VisitOMPClauseList(C);
2245}
Alexey Bataev1c2cfbc2015-06-23 14:25:19 +00002246void OMPClauseEnqueue::VisitOMPDependClause(const OMPDependClause *C) {
2247 VisitOMPClauseList(C);
2248}
Kelvin Li0bff7af2015-11-23 05:32:03 +00002249void OMPClauseEnqueue::VisitOMPMapClause(const OMPMapClause *C) {
2250 VisitOMPClauseList(C);
2251}
Carlo Bertollib4adf552016-01-15 18:50:31 +00002252void OMPClauseEnqueue::VisitOMPDistScheduleClause(
2253 const OMPDistScheduleClause *C) {
Alexey Bataev3392d762016-02-16 11:18:12 +00002254 VisitOMPClauseWithPreInit(C);
Carlo Bertollib4adf552016-01-15 18:50:31 +00002255 Visitor->AddStmt(C->getChunkSize());
Carlo Bertollib4adf552016-01-15 18:50:31 +00002256}
Alexey Bataev3392d762016-02-16 11:18:12 +00002257void OMPClauseEnqueue::VisitOMPDefaultmapClause(
2258 const OMPDefaultmapClause * /*C*/) {}
Samuel Antao661c0902016-05-26 17:39:58 +00002259void OMPClauseEnqueue::VisitOMPToClause(const OMPToClause *C) {
2260 VisitOMPClauseList(C);
2261}
Samuel Antaoec172c62016-05-26 17:49:04 +00002262void OMPClauseEnqueue::VisitOMPFromClause(const OMPFromClause *C) {
2263 VisitOMPClauseList(C);
2264}
Alexander Kornienkoab9db512015-06-22 23:07:51 +00002265}
Alexey Bataev756c1962013-09-24 03:17:45 +00002266
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002267void EnqueueVisitor::EnqueueChildren(const OMPClause *S) {
2268 unsigned size = WL.size();
2269 OMPClauseEnqueue Visitor(this);
2270 Visitor.Visit(S);
2271 if (size == WL.size())
2272 return;
2273 // Now reverse the entries we just added. This will match the DFS
2274 // ordering performed by the worklist.
2275 VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
2276 std::reverse(I, E);
2277}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002278void EnqueueVisitor::VisitAddrLabelExpr(const AddrLabelExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002279 WL.push_back(LabelRefVisit(E->getLabel(), E->getLabelLoc(), Parent));
2280}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002281void EnqueueVisitor::VisitBlockExpr(const BlockExpr *B) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002282 AddDecl(B->getBlockDecl());
2283}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002284void EnqueueVisitor::VisitCompoundLiteralExpr(const CompoundLiteralExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002285 EnqueueChildren(E);
2286 AddTypeLoc(E->getTypeSourceInfo());
2287}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002288void EnqueueVisitor::VisitCompoundStmt(const CompoundStmt *S) {
Pete Cooper57d3f142015-07-30 17:22:52 +00002289 for (auto &I : llvm::reverse(S->body()))
2290 AddStmt(I);
Guy Benyei11169dd2012-12-18 14:30:41 +00002291}
2292void EnqueueVisitor::
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002293VisitMSDependentExistsStmt(const MSDependentExistsStmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002294 AddStmt(S->getSubStmt());
2295 AddDeclarationNameInfo(S);
2296 if (NestedNameSpecifierLoc QualifierLoc = S->getQualifierLoc())
2297 AddNestedNameSpecifierLoc(QualifierLoc);
2298}
2299
2300void EnqueueVisitor::
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002301VisitCXXDependentScopeMemberExpr(const CXXDependentScopeMemberExpr *E) {
James Y Knight04ec5bf2015-12-24 02:59:37 +00002302 if (E->hasExplicitTemplateArgs())
2303 AddExplicitTemplateArgs(E->getTemplateArgs(), E->getNumTemplateArgs());
Guy Benyei11169dd2012-12-18 14:30:41 +00002304 AddDeclarationNameInfo(E);
2305 if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
2306 AddNestedNameSpecifierLoc(QualifierLoc);
2307 if (!E->isImplicitAccess())
2308 AddStmt(E->getBase());
2309}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002310void EnqueueVisitor::VisitCXXNewExpr(const CXXNewExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002311 // Enqueue the initializer , if any.
2312 AddStmt(E->getInitializer());
2313 // Enqueue the array size, if any.
2314 AddStmt(E->getArraySize());
2315 // Enqueue the allocated type.
2316 AddTypeLoc(E->getAllocatedTypeSourceInfo());
2317 // Enqueue the placement arguments.
2318 for (unsigned I = E->getNumPlacementArgs(); I > 0; --I)
2319 AddStmt(E->getPlacementArg(I-1));
2320}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002321void EnqueueVisitor::VisitCXXOperatorCallExpr(const CXXOperatorCallExpr *CE) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002322 for (unsigned I = CE->getNumArgs(); I > 1 /* Yes, this is 1 */; --I)
2323 AddStmt(CE->getArg(I-1));
2324 AddStmt(CE->getCallee());
2325 AddStmt(CE->getArg(0));
2326}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002327void EnqueueVisitor::VisitCXXPseudoDestructorExpr(
2328 const CXXPseudoDestructorExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002329 // Visit the name of the type being destroyed.
2330 AddTypeLoc(E->getDestroyedTypeInfo());
2331 // Visit the scope type that looks disturbingly like the nested-name-specifier
2332 // but isn't.
2333 AddTypeLoc(E->getScopeTypeInfo());
2334 // Visit the nested-name-specifier.
2335 if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
2336 AddNestedNameSpecifierLoc(QualifierLoc);
2337 // Visit base expression.
2338 AddStmt(E->getBase());
2339}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002340void EnqueueVisitor::VisitCXXScalarValueInitExpr(
2341 const CXXScalarValueInitExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002342 AddTypeLoc(E->getTypeSourceInfo());
2343}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002344void EnqueueVisitor::VisitCXXTemporaryObjectExpr(
2345 const CXXTemporaryObjectExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002346 EnqueueChildren(E);
2347 AddTypeLoc(E->getTypeSourceInfo());
2348}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002349void EnqueueVisitor::VisitCXXTypeidExpr(const CXXTypeidExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002350 EnqueueChildren(E);
2351 if (E->isTypeOperand())
2352 AddTypeLoc(E->getTypeOperandSourceInfo());
2353}
2354
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002355void EnqueueVisitor::VisitCXXUnresolvedConstructExpr(
2356 const CXXUnresolvedConstructExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002357 EnqueueChildren(E);
2358 AddTypeLoc(E->getTypeSourceInfo());
2359}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002360void EnqueueVisitor::VisitCXXUuidofExpr(const CXXUuidofExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002361 EnqueueChildren(E);
2362 if (E->isTypeOperand())
2363 AddTypeLoc(E->getTypeOperandSourceInfo());
2364}
2365
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002366void EnqueueVisitor::VisitCXXCatchStmt(const CXXCatchStmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002367 EnqueueChildren(S);
2368 AddDecl(S->getExceptionDecl());
2369}
2370
Argyrios Kyrtzidis99891242014-11-13 09:03:21 +00002371void EnqueueVisitor::VisitCXXForRangeStmt(const CXXForRangeStmt *S) {
Argyrios Kyrtzidiscde70692014-11-13 09:50:19 +00002372 AddStmt(S->getBody());
Argyrios Kyrtzidis99891242014-11-13 09:03:21 +00002373 AddStmt(S->getRangeInit());
Argyrios Kyrtzidiscde70692014-11-13 09:50:19 +00002374 AddDecl(S->getLoopVariable());
Argyrios Kyrtzidis99891242014-11-13 09:03:21 +00002375}
2376
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002377void EnqueueVisitor::VisitDeclRefExpr(const DeclRefExpr *DR) {
James Y Knight04ec5bf2015-12-24 02:59:37 +00002378 if (DR->hasExplicitTemplateArgs())
2379 AddExplicitTemplateArgs(DR->getTemplateArgs(), DR->getNumTemplateArgs());
Guy Benyei11169dd2012-12-18 14:30:41 +00002380 WL.push_back(DeclRefExprParts(DR, Parent));
2381}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002382void EnqueueVisitor::VisitDependentScopeDeclRefExpr(
2383 const DependentScopeDeclRefExpr *E) {
James Y Knight04ec5bf2015-12-24 02:59:37 +00002384 if (E->hasExplicitTemplateArgs())
2385 AddExplicitTemplateArgs(E->getTemplateArgs(), E->getNumTemplateArgs());
Guy Benyei11169dd2012-12-18 14:30:41 +00002386 AddDeclarationNameInfo(E);
2387 AddNestedNameSpecifierLoc(E->getQualifierLoc());
2388}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002389void EnqueueVisitor::VisitDeclStmt(const DeclStmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002390 unsigned size = WL.size();
2391 bool isFirst = true;
Aaron Ballman535bbcc2014-03-14 17:01:24 +00002392 for (const auto *D : S->decls()) {
2393 AddDecl(D, isFirst);
Guy Benyei11169dd2012-12-18 14:30:41 +00002394 isFirst = false;
2395 }
2396 if (size == WL.size())
2397 return;
2398 // Now reverse the entries we just added. This will match the DFS
2399 // ordering performed by the worklist.
2400 VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
2401 std::reverse(I, E);
2402}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002403void EnqueueVisitor::VisitDesignatedInitExpr(const DesignatedInitExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002404 AddStmt(E->getInit());
David Majnemerf7e36092016-06-23 00:15:04 +00002405 for (const DesignatedInitExpr::Designator &D :
2406 llvm::reverse(E->designators())) {
2407 if (D.isFieldDesignator()) {
2408 if (FieldDecl *Field = D.getField())
2409 AddMemberRef(Field, D.getFieldLoc());
Guy Benyei11169dd2012-12-18 14:30:41 +00002410 continue;
2411 }
David Majnemerf7e36092016-06-23 00:15:04 +00002412 if (D.isArrayDesignator()) {
2413 AddStmt(E->getArrayIndex(D));
Guy Benyei11169dd2012-12-18 14:30:41 +00002414 continue;
2415 }
David Majnemerf7e36092016-06-23 00:15:04 +00002416 assert(D.isArrayRangeDesignator() && "Unknown designator kind");
2417 AddStmt(E->getArrayRangeEnd(D));
2418 AddStmt(E->getArrayRangeStart(D));
Guy Benyei11169dd2012-12-18 14:30:41 +00002419 }
2420}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002421void EnqueueVisitor::VisitExplicitCastExpr(const ExplicitCastExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002422 EnqueueChildren(E);
2423 AddTypeLoc(E->getTypeInfoAsWritten());
2424}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002425void EnqueueVisitor::VisitForStmt(const ForStmt *FS) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002426 AddStmt(FS->getBody());
2427 AddStmt(FS->getInc());
2428 AddStmt(FS->getCond());
2429 AddDecl(FS->getConditionVariable());
2430 AddStmt(FS->getInit());
2431}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002432void EnqueueVisitor::VisitGotoStmt(const GotoStmt *GS) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002433 WL.push_back(LabelRefVisit(GS->getLabel(), GS->getLabelLoc(), Parent));
2434}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002435void EnqueueVisitor::VisitIfStmt(const IfStmt *If) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002436 AddStmt(If->getElse());
2437 AddStmt(If->getThen());
2438 AddStmt(If->getCond());
2439 AddDecl(If->getConditionVariable());
2440}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002441void EnqueueVisitor::VisitInitListExpr(const InitListExpr *IE) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002442 // We care about the syntactic form of the initializer list, only.
2443 if (InitListExpr *Syntactic = IE->getSyntacticForm())
2444 IE = Syntactic;
2445 EnqueueChildren(IE);
2446}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002447void EnqueueVisitor::VisitMemberExpr(const MemberExpr *M) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002448 WL.push_back(MemberExprParts(M, Parent));
2449
2450 // If the base of the member access expression is an implicit 'this', don't
2451 // visit it.
2452 // FIXME: If we ever want to show these implicit accesses, this will be
2453 // unfortunate. However, clang_getCursor() relies on this behavior.
Argyrios Kyrtzidis58d0e7a2015-03-13 04:40:07 +00002454 if (M->isImplicitAccess())
2455 return;
2456
2457 // Ignore base anonymous struct/union fields, otherwise they will shadow the
2458 // real field that that we are interested in.
2459 if (auto *SubME = dyn_cast<MemberExpr>(M->getBase())) {
2460 if (auto *FD = dyn_cast_or_null<FieldDecl>(SubME->getMemberDecl())) {
2461 if (FD->isAnonymousStructOrUnion()) {
2462 AddStmt(SubME->getBase());
2463 return;
2464 }
2465 }
2466 }
2467
2468 AddStmt(M->getBase());
Guy Benyei11169dd2012-12-18 14:30:41 +00002469}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002470void EnqueueVisitor::VisitObjCEncodeExpr(const ObjCEncodeExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002471 AddTypeLoc(E->getEncodedTypeSourceInfo());
2472}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002473void EnqueueVisitor::VisitObjCMessageExpr(const ObjCMessageExpr *M) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002474 EnqueueChildren(M);
2475 AddTypeLoc(M->getClassReceiverTypeInfo());
2476}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002477void EnqueueVisitor::VisitOffsetOfExpr(const OffsetOfExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002478 // Visit the components of the offsetof expression.
2479 for (unsigned N = E->getNumComponents(), I = N; I > 0; --I) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002480 const OffsetOfNode &Node = E->getComponent(I-1);
2481 switch (Node.getKind()) {
2482 case OffsetOfNode::Array:
2483 AddStmt(E->getIndexExpr(Node.getArrayExprIndex()));
2484 break;
2485 case OffsetOfNode::Field:
2486 AddMemberRef(Node.getField(), Node.getSourceRange().getEnd());
2487 break;
2488 case OffsetOfNode::Identifier:
2489 case OffsetOfNode::Base:
2490 continue;
2491 }
2492 }
2493 // Visit the type into which we're computing the offset.
2494 AddTypeLoc(E->getTypeSourceInfo());
2495}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002496void EnqueueVisitor::VisitOverloadExpr(const OverloadExpr *E) {
James Y Knight04ec5bf2015-12-24 02:59:37 +00002497 if (E->hasExplicitTemplateArgs())
2498 AddExplicitTemplateArgs(E->getTemplateArgs(), E->getNumTemplateArgs());
Guy Benyei11169dd2012-12-18 14:30:41 +00002499 WL.push_back(OverloadExprParts(E, Parent));
2500}
2501void EnqueueVisitor::VisitUnaryExprOrTypeTraitExpr(
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002502 const UnaryExprOrTypeTraitExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002503 EnqueueChildren(E);
2504 if (E->isArgumentType())
2505 AddTypeLoc(E->getArgumentTypeInfo());
2506}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002507void EnqueueVisitor::VisitStmt(const Stmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002508 EnqueueChildren(S);
2509}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002510void EnqueueVisitor::VisitSwitchStmt(const SwitchStmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002511 AddStmt(S->getBody());
2512 AddStmt(S->getCond());
2513 AddDecl(S->getConditionVariable());
2514}
2515
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002516void EnqueueVisitor::VisitWhileStmt(const WhileStmt *W) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002517 AddStmt(W->getBody());
2518 AddStmt(W->getCond());
2519 AddDecl(W->getConditionVariable());
2520}
2521
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002522void EnqueueVisitor::VisitTypeTraitExpr(const TypeTraitExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002523 for (unsigned I = E->getNumArgs(); I > 0; --I)
2524 AddTypeLoc(E->getArg(I-1));
2525}
2526
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002527void EnqueueVisitor::VisitArrayTypeTraitExpr(const ArrayTypeTraitExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002528 AddTypeLoc(E->getQueriedTypeSourceInfo());
2529}
2530
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002531void EnqueueVisitor::VisitExpressionTraitExpr(const ExpressionTraitExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002532 EnqueueChildren(E);
2533}
2534
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002535void EnqueueVisitor::VisitUnresolvedMemberExpr(const UnresolvedMemberExpr *U) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002536 VisitOverloadExpr(U);
2537 if (!U->isImplicitAccess())
2538 AddStmt(U->getBase());
2539}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002540void EnqueueVisitor::VisitVAArgExpr(const VAArgExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002541 AddStmt(E->getSubExpr());
2542 AddTypeLoc(E->getWrittenTypeInfo());
2543}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002544void EnqueueVisitor::VisitSizeOfPackExpr(const SizeOfPackExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002545 WL.push_back(SizeOfPackExprParts(E, Parent));
2546}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002547void EnqueueVisitor::VisitOpaqueValueExpr(const OpaqueValueExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002548 // If the opaque value has a source expression, just transparently
2549 // visit that. This is useful for (e.g.) pseudo-object expressions.
2550 if (Expr *SourceExpr = E->getSourceExpr())
2551 return Visit(SourceExpr);
2552}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002553void EnqueueVisitor::VisitLambdaExpr(const LambdaExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002554 AddStmt(E->getBody());
2555 WL.push_back(LambdaExprParts(E, Parent));
2556}
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002557void EnqueueVisitor::VisitPseudoObjectExpr(const PseudoObjectExpr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002558 // Treat the expression like its syntactic form.
2559 Visit(E->getSyntacticForm());
2560}
2561
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002562void EnqueueVisitor::VisitOMPExecutableDirective(
2563 const OMPExecutableDirective *D) {
2564 EnqueueChildren(D);
2565 for (ArrayRef<OMPClause *>::iterator I = D->clauses().begin(),
2566 E = D->clauses().end();
2567 I != E; ++I)
2568 EnqueueChildren(*I);
2569}
2570
Alexander Musman3aaab662014-08-19 11:27:13 +00002571void EnqueueVisitor::VisitOMPLoopDirective(const OMPLoopDirective *D) {
2572 VisitOMPExecutableDirective(D);
2573}
2574
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00002575void EnqueueVisitor::VisitOMPParallelDirective(const OMPParallelDirective *D) {
2576 VisitOMPExecutableDirective(D);
2577}
2578
Alexey Bataev1b59ab52014-02-27 08:29:12 +00002579void EnqueueVisitor::VisitOMPSimdDirective(const OMPSimdDirective *D) {
Alexander Musman3aaab662014-08-19 11:27:13 +00002580 VisitOMPLoopDirective(D);
Alexey Bataev1b59ab52014-02-27 08:29:12 +00002581}
2582
Alexey Bataevf29276e2014-06-18 04:14:57 +00002583void EnqueueVisitor::VisitOMPForDirective(const OMPForDirective *D) {
Alexander Musman3aaab662014-08-19 11:27:13 +00002584 VisitOMPLoopDirective(D);
Alexey Bataevf29276e2014-06-18 04:14:57 +00002585}
2586
Alexander Musmanf82886e2014-09-18 05:12:34 +00002587void EnqueueVisitor::VisitOMPForSimdDirective(const OMPForSimdDirective *D) {
2588 VisitOMPLoopDirective(D);
2589}
2590
Alexey Bataevd3f8dd22014-06-25 11:44:49 +00002591void EnqueueVisitor::VisitOMPSectionsDirective(const OMPSectionsDirective *D) {
2592 VisitOMPExecutableDirective(D);
2593}
2594
Alexey Bataev1e0498a2014-06-26 08:21:58 +00002595void EnqueueVisitor::VisitOMPSectionDirective(const OMPSectionDirective *D) {
2596 VisitOMPExecutableDirective(D);
2597}
2598
Alexey Bataevd1e40fb2014-06-26 12:05:45 +00002599void EnqueueVisitor::VisitOMPSingleDirective(const OMPSingleDirective *D) {
2600 VisitOMPExecutableDirective(D);
2601}
2602
Alexander Musman80c22892014-07-17 08:54:58 +00002603void EnqueueVisitor::VisitOMPMasterDirective(const OMPMasterDirective *D) {
2604 VisitOMPExecutableDirective(D);
2605}
2606
Alexander Musmand9ed09f2014-07-21 09:42:05 +00002607void EnqueueVisitor::VisitOMPCriticalDirective(const OMPCriticalDirective *D) {
2608 VisitOMPExecutableDirective(D);
2609 AddDeclarationNameInfo(D);
2610}
2611
Alexey Bataev4acb8592014-07-07 13:01:15 +00002612void
2613EnqueueVisitor::VisitOMPParallelForDirective(const OMPParallelForDirective *D) {
Alexander Musman3aaab662014-08-19 11:27:13 +00002614 VisitOMPLoopDirective(D);
Alexey Bataev4acb8592014-07-07 13:01:15 +00002615}
2616
Alexander Musmane4e893b2014-09-23 09:33:00 +00002617void EnqueueVisitor::VisitOMPParallelForSimdDirective(
2618 const OMPParallelForSimdDirective *D) {
2619 VisitOMPLoopDirective(D);
2620}
2621
Alexey Bataev84d0b3e2014-07-08 08:12:03 +00002622void EnqueueVisitor::VisitOMPParallelSectionsDirective(
2623 const OMPParallelSectionsDirective *D) {
2624 VisitOMPExecutableDirective(D);
2625}
2626
Alexey Bataev9c2e8ee2014-07-11 11:25:16 +00002627void EnqueueVisitor::VisitOMPTaskDirective(const OMPTaskDirective *D) {
2628 VisitOMPExecutableDirective(D);
2629}
2630
Alexey Bataev68446b72014-07-18 07:47:19 +00002631void
2632EnqueueVisitor::VisitOMPTaskyieldDirective(const OMPTaskyieldDirective *D) {
2633 VisitOMPExecutableDirective(D);
2634}
2635
Alexey Bataev4d1dfea2014-07-18 09:11:51 +00002636void EnqueueVisitor::VisitOMPBarrierDirective(const OMPBarrierDirective *D) {
2637 VisitOMPExecutableDirective(D);
2638}
2639
Alexey Bataev2df347a2014-07-18 10:17:07 +00002640void EnqueueVisitor::VisitOMPTaskwaitDirective(const OMPTaskwaitDirective *D) {
2641 VisitOMPExecutableDirective(D);
2642}
2643
Alexey Bataevc30dd2d2015-06-18 12:14:09 +00002644void EnqueueVisitor::VisitOMPTaskgroupDirective(
2645 const OMPTaskgroupDirective *D) {
2646 VisitOMPExecutableDirective(D);
2647}
2648
Alexey Bataev6125da92014-07-21 11:26:11 +00002649void EnqueueVisitor::VisitOMPFlushDirective(const OMPFlushDirective *D) {
2650 VisitOMPExecutableDirective(D);
2651}
2652
Alexey Bataev9fb6e642014-07-22 06:45:04 +00002653void EnqueueVisitor::VisitOMPOrderedDirective(const OMPOrderedDirective *D) {
2654 VisitOMPExecutableDirective(D);
2655}
2656
Alexey Bataev0162e452014-07-22 10:10:35 +00002657void EnqueueVisitor::VisitOMPAtomicDirective(const OMPAtomicDirective *D) {
2658 VisitOMPExecutableDirective(D);
2659}
2660
Alexey Bataev0bd520b2014-09-19 08:19:49 +00002661void EnqueueVisitor::VisitOMPTargetDirective(const OMPTargetDirective *D) {
2662 VisitOMPExecutableDirective(D);
2663}
2664
Michael Wong65f367f2015-07-21 13:44:28 +00002665void EnqueueVisitor::VisitOMPTargetDataDirective(const
2666 OMPTargetDataDirective *D) {
2667 VisitOMPExecutableDirective(D);
2668}
2669
Samuel Antaodf67fc42016-01-19 19:15:56 +00002670void EnqueueVisitor::VisitOMPTargetEnterDataDirective(
2671 const OMPTargetEnterDataDirective *D) {
2672 VisitOMPExecutableDirective(D);
2673}
2674
Samuel Antao72590762016-01-19 20:04:50 +00002675void EnqueueVisitor::VisitOMPTargetExitDataDirective(
2676 const OMPTargetExitDataDirective *D) {
2677 VisitOMPExecutableDirective(D);
2678}
2679
Arpith Chacko Jacobe955b3d2016-01-26 18:48:41 +00002680void EnqueueVisitor::VisitOMPTargetParallelDirective(
2681 const OMPTargetParallelDirective *D) {
2682 VisitOMPExecutableDirective(D);
2683}
2684
Arpith Chacko Jacob05bebb52016-02-03 15:46:42 +00002685void EnqueueVisitor::VisitOMPTargetParallelForDirective(
2686 const OMPTargetParallelForDirective *D) {
2687 VisitOMPLoopDirective(D);
2688}
2689
Alexey Bataev13314bf2014-10-09 04:18:56 +00002690void EnqueueVisitor::VisitOMPTeamsDirective(const OMPTeamsDirective *D) {
2691 VisitOMPExecutableDirective(D);
2692}
2693
Alexey Bataev6d4ed052015-07-01 06:57:41 +00002694void EnqueueVisitor::VisitOMPCancellationPointDirective(
2695 const OMPCancellationPointDirective *D) {
2696 VisitOMPExecutableDirective(D);
2697}
2698
Alexey Bataev80909872015-07-02 11:25:17 +00002699void EnqueueVisitor::VisitOMPCancelDirective(const OMPCancelDirective *D) {
2700 VisitOMPExecutableDirective(D);
2701}
2702
Alexey Bataev49f6e782015-12-01 04:18:41 +00002703void EnqueueVisitor::VisitOMPTaskLoopDirective(const OMPTaskLoopDirective *D) {
2704 VisitOMPLoopDirective(D);
2705}
2706
Alexey Bataev0a6ed842015-12-03 09:40:15 +00002707void EnqueueVisitor::VisitOMPTaskLoopSimdDirective(
2708 const OMPTaskLoopSimdDirective *D) {
2709 VisitOMPLoopDirective(D);
2710}
2711
Carlo Bertolli6200a3d2015-12-14 14:51:25 +00002712void EnqueueVisitor::VisitOMPDistributeDirective(
2713 const OMPDistributeDirective *D) {
2714 VisitOMPLoopDirective(D);
2715}
2716
Carlo Bertolli9925f152016-06-27 14:55:37 +00002717void EnqueueVisitor::VisitOMPDistributeParallelForDirective(
2718 const OMPDistributeParallelForDirective *D) {
2719 VisitOMPLoopDirective(D);
2720}
2721
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002722void CursorVisitor::EnqueueWorkList(VisitorWorkList &WL, const Stmt *S) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002723 EnqueueVisitor(WL, MakeCXCursor(S, StmtParent, TU,RegionOfInterest)).Visit(S);
2724}
2725
2726bool CursorVisitor::IsInRegionOfInterest(CXCursor C) {
2727 if (RegionOfInterest.isValid()) {
2728 SourceRange Range = getRawCursorExtent(C);
2729 if (Range.isInvalid() || CompareRegionOfInterest(Range))
2730 return false;
2731 }
2732 return true;
2733}
2734
2735bool CursorVisitor::RunVisitorWorkList(VisitorWorkList &WL) {
2736 while (!WL.empty()) {
2737 // Dequeue the worklist item.
Robert Wilhelm25284cc2013-08-23 16:11:15 +00002738 VisitorJob LI = WL.pop_back_val();
Guy Benyei11169dd2012-12-18 14:30:41 +00002739
2740 // Set the Parent field, then back to its old value once we're done.
2741 SetParentRAII SetParent(Parent, StmtParent, LI.getParent());
2742
2743 switch (LI.getKind()) {
2744 case VisitorJob::DeclVisitKind: {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002745 const Decl *D = cast<DeclVisit>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002746 if (!D)
2747 continue;
2748
2749 // For now, perform default visitation for Decls.
2750 if (Visit(MakeCXCursor(D, TU, RegionOfInterest,
2751 cast<DeclVisit>(&LI)->isFirst())))
2752 return true;
2753
2754 continue;
2755 }
2756 case VisitorJob::ExplicitTemplateArgsVisitKind: {
James Y Knight04ec5bf2015-12-24 02:59:37 +00002757 for (const TemplateArgumentLoc &Arg :
2758 *cast<ExplicitTemplateArgsVisit>(&LI)) {
2759 if (VisitTemplateArgumentLoc(Arg))
Guy Benyei11169dd2012-12-18 14:30:41 +00002760 return true;
2761 }
2762 continue;
2763 }
2764 case VisitorJob::TypeLocVisitKind: {
2765 // Perform default visitation for TypeLocs.
2766 if (Visit(cast<TypeLocVisit>(&LI)->get()))
2767 return true;
2768 continue;
2769 }
2770 case VisitorJob::LabelRefVisitKind: {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002771 const LabelDecl *LS = cast<LabelRefVisit>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002772 if (LabelStmt *stmt = LS->getStmt()) {
2773 if (Visit(MakeCursorLabelRef(stmt, cast<LabelRefVisit>(&LI)->getLoc(),
2774 TU))) {
2775 return true;
2776 }
2777 }
2778 continue;
2779 }
2780
2781 case VisitorJob::NestedNameSpecifierLocVisitKind: {
2782 NestedNameSpecifierLocVisit *V = cast<NestedNameSpecifierLocVisit>(&LI);
2783 if (VisitNestedNameSpecifierLoc(V->get()))
2784 return true;
2785 continue;
2786 }
2787
2788 case VisitorJob::DeclarationNameInfoVisitKind: {
2789 if (VisitDeclarationNameInfo(cast<DeclarationNameInfoVisit>(&LI)
2790 ->get()))
2791 return true;
2792 continue;
2793 }
2794 case VisitorJob::MemberRefVisitKind: {
2795 MemberRefVisit *V = cast<MemberRefVisit>(&LI);
2796 if (Visit(MakeCursorMemberRef(V->get(), V->getLoc(), TU)))
2797 return true;
2798 continue;
2799 }
2800 case VisitorJob::StmtVisitKind: {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002801 const Stmt *S = cast<StmtVisit>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002802 if (!S)
2803 continue;
2804
2805 // Update the current cursor.
2806 CXCursor Cursor = MakeCXCursor(S, StmtParent, TU, RegionOfInterest);
2807 if (!IsInRegionOfInterest(Cursor))
2808 continue;
2809 switch (Visitor(Cursor, Parent, ClientData)) {
2810 case CXChildVisit_Break: return true;
2811 case CXChildVisit_Continue: break;
2812 case CXChildVisit_Recurse:
2813 if (PostChildrenVisitor)
Craig Topper69186e72014-06-08 08:38:04 +00002814 WL.push_back(PostChildrenVisit(nullptr, Cursor));
Guy Benyei11169dd2012-12-18 14:30:41 +00002815 EnqueueWorkList(WL, S);
2816 break;
2817 }
2818 continue;
2819 }
2820 case VisitorJob::MemberExprPartsKind: {
2821 // Handle the other pieces in the MemberExpr besides the base.
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002822 const MemberExpr *M = cast<MemberExprParts>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002823
2824 // Visit the nested-name-specifier
2825 if (NestedNameSpecifierLoc QualifierLoc = M->getQualifierLoc())
2826 if (VisitNestedNameSpecifierLoc(QualifierLoc))
2827 return true;
2828
2829 // Visit the declaration name.
2830 if (VisitDeclarationNameInfo(M->getMemberNameInfo()))
2831 return true;
2832
2833 // Visit the explicitly-specified template arguments, if any.
2834 if (M->hasExplicitTemplateArgs()) {
2835 for (const TemplateArgumentLoc *Arg = M->getTemplateArgs(),
2836 *ArgEnd = Arg + M->getNumTemplateArgs();
2837 Arg != ArgEnd; ++Arg) {
2838 if (VisitTemplateArgumentLoc(*Arg))
2839 return true;
2840 }
2841 }
2842 continue;
2843 }
2844 case VisitorJob::DeclRefExprPartsKind: {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002845 const DeclRefExpr *DR = cast<DeclRefExprParts>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002846 // Visit nested-name-specifier, if present.
2847 if (NestedNameSpecifierLoc QualifierLoc = DR->getQualifierLoc())
2848 if (VisitNestedNameSpecifierLoc(QualifierLoc))
2849 return true;
2850 // Visit declaration name.
2851 if (VisitDeclarationNameInfo(DR->getNameInfo()))
2852 return true;
2853 continue;
2854 }
2855 case VisitorJob::OverloadExprPartsKind: {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002856 const OverloadExpr *O = cast<OverloadExprParts>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002857 // Visit the nested-name-specifier.
2858 if (NestedNameSpecifierLoc QualifierLoc = O->getQualifierLoc())
2859 if (VisitNestedNameSpecifierLoc(QualifierLoc))
2860 return true;
2861 // Visit the declaration name.
2862 if (VisitDeclarationNameInfo(O->getNameInfo()))
2863 return true;
2864 // Visit the overloaded declaration reference.
2865 if (Visit(MakeCursorOverloadedDeclRef(O, TU)))
2866 return true;
2867 continue;
2868 }
2869 case VisitorJob::SizeOfPackExprPartsKind: {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002870 const SizeOfPackExpr *E = cast<SizeOfPackExprParts>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002871 NamedDecl *Pack = E->getPack();
2872 if (isa<TemplateTypeParmDecl>(Pack)) {
2873 if (Visit(MakeCursorTypeRef(cast<TemplateTypeParmDecl>(Pack),
2874 E->getPackLoc(), TU)))
2875 return true;
2876
2877 continue;
2878 }
2879
2880 if (isa<TemplateTemplateParmDecl>(Pack)) {
2881 if (Visit(MakeCursorTemplateRef(cast<TemplateTemplateParmDecl>(Pack),
2882 E->getPackLoc(), TU)))
2883 return true;
2884
2885 continue;
2886 }
2887
2888 // Non-type template parameter packs and function parameter packs are
2889 // treated like DeclRefExpr cursors.
2890 continue;
2891 }
2892
2893 case VisitorJob::LambdaExprPartsKind: {
2894 // Visit captures.
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002895 const LambdaExpr *E = cast<LambdaExprParts>(&LI)->get();
Guy Benyei11169dd2012-12-18 14:30:41 +00002896 for (LambdaExpr::capture_iterator C = E->explicit_capture_begin(),
2897 CEnd = E->explicit_capture_end();
2898 C != CEnd; ++C) {
Richard Smithba71c082013-05-16 06:20:58 +00002899 // FIXME: Lambda init-captures.
2900 if (!C->capturesVariable())
Guy Benyei11169dd2012-12-18 14:30:41 +00002901 continue;
Richard Smithba71c082013-05-16 06:20:58 +00002902
Guy Benyei11169dd2012-12-18 14:30:41 +00002903 if (Visit(MakeCursorVariableRef(C->getCapturedVar(),
2904 C->getLocation(),
2905 TU)))
2906 return true;
2907 }
2908
2909 // Visit parameters and return type, if present.
2910 if (E->hasExplicitParameters() || E->hasExplicitResultType()) {
2911 TypeLoc TL = E->getCallOperator()->getTypeSourceInfo()->getTypeLoc();
2912 if (E->hasExplicitParameters() && E->hasExplicitResultType()) {
2913 // Visit the whole type.
2914 if (Visit(TL))
2915 return true;
David Blaikie6adc78e2013-02-18 22:06:02 +00002916 } else if (FunctionProtoTypeLoc Proto =
2917 TL.getAs<FunctionProtoTypeLoc>()) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002918 if (E->hasExplicitParameters()) {
2919 // Visit parameters.
Alp Tokerb3fd5cf2014-01-21 00:32:38 +00002920 for (unsigned I = 0, N = Proto.getNumParams(); I != N; ++I)
2921 if (Visit(MakeCXCursor(Proto.getParam(I), TU)))
Guy Benyei11169dd2012-12-18 14:30:41 +00002922 return true;
2923 } else {
2924 // Visit result type.
Alp Toker42a16a62014-01-25 23:51:36 +00002925 if (Visit(Proto.getReturnLoc()))
Guy Benyei11169dd2012-12-18 14:30:41 +00002926 return true;
2927 }
2928 }
2929 }
2930 break;
2931 }
2932
2933 case VisitorJob::PostChildrenVisitKind:
2934 if (PostChildrenVisitor(Parent, ClientData))
2935 return true;
2936 break;
2937 }
2938 }
2939 return false;
2940}
2941
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00002942bool CursorVisitor::Visit(const Stmt *S) {
Craig Topper69186e72014-06-08 08:38:04 +00002943 VisitorWorkList *WL = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00002944 if (!WorkListFreeList.empty()) {
2945 WL = WorkListFreeList.back();
2946 WL->clear();
2947 WorkListFreeList.pop_back();
2948 }
2949 else {
2950 WL = new VisitorWorkList();
2951 WorkListCache.push_back(WL);
2952 }
2953 EnqueueWorkList(*WL, S);
2954 bool result = RunVisitorWorkList(*WL);
2955 WorkListFreeList.push_back(WL);
2956 return result;
2957}
2958
2959namespace {
Dmitri Gribenkof8579502013-01-12 19:30:44 +00002960typedef SmallVector<SourceRange, 4> RefNamePieces;
James Y Knight04ec5bf2015-12-24 02:59:37 +00002961RefNamePieces buildPieces(unsigned NameFlags, bool IsMemberRefExpr,
2962 const DeclarationNameInfo &NI, SourceRange QLoc,
2963 const SourceRange *TemplateArgsLoc = nullptr) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002964 const bool WantQualifier = NameFlags & CXNameRange_WantQualifier;
2965 const bool WantTemplateArgs = NameFlags & CXNameRange_WantTemplateArgs;
2966 const bool WantSinglePiece = NameFlags & CXNameRange_WantSinglePiece;
2967
2968 const DeclarationName::NameKind Kind = NI.getName().getNameKind();
2969
2970 RefNamePieces Pieces;
2971
2972 if (WantQualifier && QLoc.isValid())
2973 Pieces.push_back(QLoc);
2974
2975 if (Kind != DeclarationName::CXXOperatorName || IsMemberRefExpr)
2976 Pieces.push_back(NI.getLoc());
James Y Knight04ec5bf2015-12-24 02:59:37 +00002977
2978 if (WantTemplateArgs && TemplateArgsLoc && TemplateArgsLoc->isValid())
2979 Pieces.push_back(*TemplateArgsLoc);
2980
Guy Benyei11169dd2012-12-18 14:30:41 +00002981 if (Kind == DeclarationName::CXXOperatorName) {
2982 Pieces.push_back(SourceLocation::getFromRawEncoding(
2983 NI.getInfo().CXXOperatorName.BeginOpNameLoc));
2984 Pieces.push_back(SourceLocation::getFromRawEncoding(
2985 NI.getInfo().CXXOperatorName.EndOpNameLoc));
2986 }
2987
2988 if (WantSinglePiece) {
2989 SourceRange R(Pieces.front().getBegin(), Pieces.back().getEnd());
2990 Pieces.clear();
2991 Pieces.push_back(R);
2992 }
2993
2994 return Pieces;
2995}
Alexander Kornienkoab9db512015-06-22 23:07:51 +00002996}
Guy Benyei11169dd2012-12-18 14:30:41 +00002997
2998//===----------------------------------------------------------------------===//
2999// Misc. API hooks.
3000//===----------------------------------------------------------------------===//
3001
Chad Rosier05c71aa2013-03-27 18:28:23 +00003002static void fatal_error_handler(void *user_data, const std::string& reason,
3003 bool gen_crash_diag) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003004 // Write the result out to stderr avoiding errs() because raw_ostreams can
3005 // call report_fatal_error.
3006 fprintf(stderr, "LIBCLANG FATAL ERROR: %s\n", reason.c_str());
3007 ::abort();
3008}
3009
Chandler Carruth66660742014-06-27 16:37:27 +00003010namespace {
3011struct RegisterFatalErrorHandler {
3012 RegisterFatalErrorHandler() {
3013 llvm::install_fatal_error_handler(fatal_error_handler, nullptr);
3014 }
3015};
3016}
3017
3018static llvm::ManagedStatic<RegisterFatalErrorHandler> RegisterFatalErrorHandlerOnce;
3019
Guy Benyei11169dd2012-12-18 14:30:41 +00003020extern "C" {
3021CXIndex clang_createIndex(int excludeDeclarationsFromPCH,
3022 int displayDiagnostics) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003023 // We use crash recovery to make some of our APIs more reliable, implicitly
3024 // enable it.
Argyrios Kyrtzidis3701f542013-11-27 08:58:09 +00003025 if (!getenv("LIBCLANG_DISABLE_CRASH_RECOVERY"))
3026 llvm::CrashRecoveryContext::Enable();
Guy Benyei11169dd2012-12-18 14:30:41 +00003027
Chandler Carruth66660742014-06-27 16:37:27 +00003028 // Look through the managed static to trigger construction of the managed
3029 // static which registers our fatal error handler. This ensures it is only
3030 // registered once.
3031 (void)*RegisterFatalErrorHandlerOnce;
Guy Benyei11169dd2012-12-18 14:30:41 +00003032
Adrian Prantlbc068582015-07-08 01:00:30 +00003033 // Initialize targets for clang module support.
3034 llvm::InitializeAllTargets();
3035 llvm::InitializeAllTargetMCs();
3036 llvm::InitializeAllAsmPrinters();
3037 llvm::InitializeAllAsmParsers();
3038
Adrian Prantlfb2398d2015-07-17 01:19:54 +00003039 CIndexer *CIdxr = new CIndexer();
3040
Guy Benyei11169dd2012-12-18 14:30:41 +00003041 if (excludeDeclarationsFromPCH)
3042 CIdxr->setOnlyLocalDecls();
3043 if (displayDiagnostics)
3044 CIdxr->setDisplayDiagnostics();
3045
3046 if (getenv("LIBCLANG_BGPRIO_INDEX"))
3047 CIdxr->setCXGlobalOptFlags(CIdxr->getCXGlobalOptFlags() |
3048 CXGlobalOpt_ThreadBackgroundPriorityForIndexing);
3049 if (getenv("LIBCLANG_BGPRIO_EDIT"))
3050 CIdxr->setCXGlobalOptFlags(CIdxr->getCXGlobalOptFlags() |
3051 CXGlobalOpt_ThreadBackgroundPriorityForEditing);
3052
3053 return CIdxr;
3054}
3055
3056void clang_disposeIndex(CXIndex CIdx) {
3057 if (CIdx)
3058 delete static_cast<CIndexer *>(CIdx);
3059}
3060
3061void clang_CXIndex_setGlobalOptions(CXIndex CIdx, unsigned options) {
3062 if (CIdx)
3063 static_cast<CIndexer *>(CIdx)->setCXGlobalOptFlags(options);
3064}
3065
3066unsigned clang_CXIndex_getGlobalOptions(CXIndex CIdx) {
3067 if (CIdx)
3068 return static_cast<CIndexer *>(CIdx)->getCXGlobalOptFlags();
3069 return 0;
3070}
3071
3072void clang_toggleCrashRecovery(unsigned isEnabled) {
3073 if (isEnabled)
3074 llvm::CrashRecoveryContext::Enable();
3075 else
3076 llvm::CrashRecoveryContext::Disable();
3077}
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003078
Guy Benyei11169dd2012-12-18 14:30:41 +00003079CXTranslationUnit clang_createTranslationUnit(CXIndex CIdx,
3080 const char *ast_filename) {
Dmitri Gribenko8850cda2014-02-19 10:24:00 +00003081 CXTranslationUnit TU;
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003082 enum CXErrorCode Result =
3083 clang_createTranslationUnit2(CIdx, ast_filename, &TU);
Reid Klecknerfd48fc62014-02-12 23:56:20 +00003084 (void)Result;
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003085 assert((TU && Result == CXError_Success) ||
3086 (!TU && Result != CXError_Success));
3087 return TU;
3088}
3089
3090enum CXErrorCode clang_createTranslationUnit2(CXIndex CIdx,
3091 const char *ast_filename,
3092 CXTranslationUnit *out_TU) {
Dmitri Gribenko8850cda2014-02-19 10:24:00 +00003093 if (out_TU)
Craig Topper69186e72014-06-08 08:38:04 +00003094 *out_TU = nullptr;
Dmitri Gribenko8850cda2014-02-19 10:24:00 +00003095
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003096 if (!CIdx || !ast_filename || !out_TU)
3097 return CXError_InvalidArguments;
Guy Benyei11169dd2012-12-18 14:30:41 +00003098
Argyrios Kyrtzidis27021012013-05-24 22:24:07 +00003099 LOG_FUNC_SECTION {
3100 *Log << ast_filename;
3101 }
3102
Guy Benyei11169dd2012-12-18 14:30:41 +00003103 CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
3104 FileSystemOptions FileSystemOpts;
3105
Justin Bognerd512c1e2014-10-15 00:33:06 +00003106 IntrusiveRefCntPtr<DiagnosticsEngine> Diags =
3107 CompilerInstance::createDiagnostics(new DiagnosticOptions());
David Blaikie6f7382d2014-08-10 19:08:04 +00003108 std::unique_ptr<ASTUnit> AU = ASTUnit::LoadFromASTFile(
Adrian Prantlfb2398d2015-07-17 01:19:54 +00003109 ast_filename, CXXIdx->getPCHContainerOperations()->getRawReader(), Diags,
Adrian Prantl6b21ab22015-08-27 19:46:20 +00003110 FileSystemOpts, /*UseDebugInfo=*/false,
3111 CXXIdx->getOnlyLocalDecls(), None,
David Blaikie6f7382d2014-08-10 19:08:04 +00003112 /*CaptureDiagnostics=*/true,
3113 /*AllowPCHWithCompilerErrors=*/true,
3114 /*UserFilesAreVolatile=*/true);
3115 *out_TU = MakeCXTranslationUnit(CXXIdx, AU.release());
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003116 return *out_TU ? CXError_Success : CXError_Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00003117}
3118
3119unsigned clang_defaultEditingTranslationUnitOptions() {
3120 return CXTranslationUnit_PrecompiledPreamble |
3121 CXTranslationUnit_CacheCompletionResults;
3122}
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003123
Guy Benyei11169dd2012-12-18 14:30:41 +00003124CXTranslationUnit
3125clang_createTranslationUnitFromSourceFile(CXIndex CIdx,
3126 const char *source_filename,
3127 int num_command_line_args,
3128 const char * const *command_line_args,
3129 unsigned num_unsaved_files,
3130 struct CXUnsavedFile *unsaved_files) {
3131 unsigned Options = CXTranslationUnit_DetailedPreprocessingRecord;
3132 return clang_parseTranslationUnit(CIdx, source_filename,
3133 command_line_args, num_command_line_args,
3134 unsaved_files, num_unsaved_files,
3135 Options);
3136}
3137
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003138static CXErrorCode
3139clang_parseTranslationUnit_Impl(CXIndex CIdx, const char *source_filename,
3140 const char *const *command_line_args,
3141 int num_command_line_args,
3142 ArrayRef<CXUnsavedFile> unsaved_files,
3143 unsigned options, CXTranslationUnit *out_TU) {
Dmitri Gribenko1bf8d912014-02-18 15:20:02 +00003144 // Set up the initial return values.
3145 if (out_TU)
Craig Topper69186e72014-06-08 08:38:04 +00003146 *out_TU = nullptr;
Dmitri Gribenko1bf8d912014-02-18 15:20:02 +00003147
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003148 // Check arguments.
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003149 if (!CIdx || !out_TU)
3150 return CXError_InvalidArguments;
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003151
Guy Benyei11169dd2012-12-18 14:30:41 +00003152 CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
3153
3154 if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForIndexing))
3155 setThreadBackgroundPriority();
3156
3157 bool PrecompilePreamble = options & CXTranslationUnit_PrecompiledPreamble;
Benjamin Kramer5c248d82015-12-15 09:30:31 +00003158 bool CreatePreambleOnFirstParse =
3159 options & CXTranslationUnit_CreatePreambleOnFirstParse;
Guy Benyei11169dd2012-12-18 14:30:41 +00003160 // FIXME: Add a flag for modules.
3161 TranslationUnitKind TUKind
3162 = (options & CXTranslationUnit_Incomplete)? TU_Prefix : TU_Complete;
Alp Toker8c8a8752013-12-03 06:53:35 +00003163 bool CacheCodeCompletionResults
Guy Benyei11169dd2012-12-18 14:30:41 +00003164 = options & CXTranslationUnit_CacheCompletionResults;
3165 bool IncludeBriefCommentsInCodeCompletion
3166 = options & CXTranslationUnit_IncludeBriefCommentsInCodeCompletion;
3167 bool SkipFunctionBodies = options & CXTranslationUnit_SkipFunctionBodies;
3168 bool ForSerialization = options & CXTranslationUnit_ForSerialization;
3169
3170 // Configure the diagnostics.
3171 IntrusiveRefCntPtr<DiagnosticsEngine>
Sean Silvaf1b49e22013-01-20 01:58:28 +00003172 Diags(CompilerInstance::createDiagnostics(new DiagnosticOptions));
Guy Benyei11169dd2012-12-18 14:30:41 +00003173
Manuel Klimek016c0242016-03-01 10:56:19 +00003174 if (options & CXTranslationUnit_KeepGoing)
3175 Diags->setFatalsAsError(true);
3176
Guy Benyei11169dd2012-12-18 14:30:41 +00003177 // Recover resources if we crash before exiting this function.
3178 llvm::CrashRecoveryContextCleanupRegistrar<DiagnosticsEngine,
3179 llvm::CrashRecoveryContextReleaseRefCleanup<DiagnosticsEngine> >
Alp Tokerf994cef2014-07-05 03:08:06 +00003180 DiagCleanup(Diags.get());
Guy Benyei11169dd2012-12-18 14:30:41 +00003181
Ahmed Charlesb8984322014-03-07 20:03:18 +00003182 std::unique_ptr<std::vector<ASTUnit::RemappedFile>> RemappedFiles(
3183 new std::vector<ASTUnit::RemappedFile>());
Guy Benyei11169dd2012-12-18 14:30:41 +00003184
3185 // Recover resources if we crash before exiting this function.
3186 llvm::CrashRecoveryContextCleanupRegistrar<
3187 std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
3188
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003189 for (auto &UF : unsaved_files) {
Rafael Espindolad87f8d72014-08-27 20:03:29 +00003190 std::unique_ptr<llvm::MemoryBuffer> MB =
Alp Toker9d85b182014-07-07 01:23:14 +00003191 llvm::MemoryBuffer::getMemBufferCopy(getContents(UF), UF.Filename);
Rafael Espindolad87f8d72014-08-27 20:03:29 +00003192 RemappedFiles->push_back(std::make_pair(UF.Filename, MB.release()));
Guy Benyei11169dd2012-12-18 14:30:41 +00003193 }
3194
Ahmed Charlesb8984322014-03-07 20:03:18 +00003195 std::unique_ptr<std::vector<const char *>> Args(
3196 new std::vector<const char *>());
Guy Benyei11169dd2012-12-18 14:30:41 +00003197
3198 // Recover resources if we crash before exiting this method.
3199 llvm::CrashRecoveryContextCleanupRegistrar<std::vector<const char*> >
3200 ArgsCleanup(Args.get());
3201
3202 // Since the Clang C library is primarily used by batch tools dealing with
3203 // (often very broken) source code, where spell-checking can have a
3204 // significant negative impact on performance (particularly when
3205 // precompiled headers are involved), we disable it by default.
3206 // Only do this if we haven't found a spell-checking-related argument.
3207 bool FoundSpellCheckingArgument = false;
3208 for (int I = 0; I != num_command_line_args; ++I) {
3209 if (strcmp(command_line_args[I], "-fno-spell-checking") == 0 ||
3210 strcmp(command_line_args[I], "-fspell-checking") == 0) {
3211 FoundSpellCheckingArgument = true;
3212 break;
3213 }
3214 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003215 Args->insert(Args->end(), command_line_args,
3216 command_line_args + num_command_line_args);
3217
Benjamin Kramerc02670e2015-11-18 16:14:27 +00003218 if (!FoundSpellCheckingArgument)
3219 Args->insert(Args->begin() + 1, "-fno-spell-checking");
3220
Guy Benyei11169dd2012-12-18 14:30:41 +00003221 // The 'source_filename' argument is optional. If the caller does not
3222 // specify it then it is assumed that the source file is specified
3223 // in the actual argument list.
3224 // Put the source file after command_line_args otherwise if '-x' flag is
3225 // present it will be unused.
3226 if (source_filename)
3227 Args->push_back(source_filename);
3228
3229 // Do we need the detailed preprocessing record?
3230 if (options & CXTranslationUnit_DetailedPreprocessingRecord) {
3231 Args->push_back("-Xclang");
3232 Args->push_back("-detailed-preprocessing-record");
3233 }
3234
3235 unsigned NumErrors = Diags->getClient()->getNumErrors();
Ahmed Charlesb8984322014-03-07 20:03:18 +00003236 std::unique_ptr<ASTUnit> ErrUnit;
Benjamin Kramer5c248d82015-12-15 09:30:31 +00003237 // Unless the user specified that they want the preamble on the first parse
3238 // set it up to be created on the first reparse. This makes the first parse
3239 // faster, trading for a slower (first) reparse.
3240 unsigned PrecompilePreambleAfterNParses =
3241 !PrecompilePreamble ? 0 : 2 - CreatePreambleOnFirstParse;
Ahmed Charlesb8984322014-03-07 20:03:18 +00003242 std::unique_ptr<ASTUnit> Unit(ASTUnit::LoadFromCommandLine(
Adrian Prantlbb165fb2015-06-20 18:53:08 +00003243 Args->data(), Args->data() + Args->size(),
3244 CXXIdx->getPCHContainerOperations(), Diags,
Ahmed Charlesb8984322014-03-07 20:03:18 +00003245 CXXIdx->getClangResourcesPath(), CXXIdx->getOnlyLocalDecls(),
3246 /*CaptureDiagnostics=*/true, *RemappedFiles.get(),
Benjamin Kramer5c248d82015-12-15 09:30:31 +00003247 /*RemappedFilesKeepOriginalName=*/true, PrecompilePreambleAfterNParses,
3248 TUKind, CacheCodeCompletionResults, IncludeBriefCommentsInCodeCompletion,
Ahmed Charlesb8984322014-03-07 20:03:18 +00003249 /*AllowPCHWithCompilerErrors=*/true, SkipFunctionBodies,
Argyrios Kyrtzidisa3e2ff12015-11-20 03:36:21 +00003250 /*UserFilesAreVolatile=*/true, ForSerialization,
3251 CXXIdx->getPCHContainerOperations()->getRawReader().getFormat(),
3252 &ErrUnit));
Guy Benyei11169dd2012-12-18 14:30:41 +00003253
Artem Belevich0ff05cd2015-07-13 23:27:56 +00003254 // Early failures in LoadFromCommandLine may return with ErrUnit unset.
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003255 if (!Unit && !ErrUnit)
3256 return CXError_ASTReadError;
Artem Belevich0ff05cd2015-07-13 23:27:56 +00003257
Guy Benyei11169dd2012-12-18 14:30:41 +00003258 if (NumErrors != Diags->getClient()->getNumErrors()) {
3259 // Make sure to check that 'Unit' is non-NULL.
3260 if (CXXIdx->getDisplayDiagnostics())
3261 printDiagsToStderr(Unit ? Unit.get() : ErrUnit.get());
3262 }
3263
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003264 if (isASTReadError(Unit ? Unit.get() : ErrUnit.get()))
3265 return CXError_ASTReadError;
3266
3267 *out_TU = MakeCXTranslationUnit(CXXIdx, Unit.release());
3268 return *out_TU ? CXError_Success : CXError_Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00003269}
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003270
3271CXTranslationUnit
3272clang_parseTranslationUnit(CXIndex CIdx,
3273 const char *source_filename,
3274 const char *const *command_line_args,
3275 int num_command_line_args,
3276 struct CXUnsavedFile *unsaved_files,
3277 unsigned num_unsaved_files,
3278 unsigned options) {
3279 CXTranslationUnit TU;
3280 enum CXErrorCode Result = clang_parseTranslationUnit2(
3281 CIdx, source_filename, command_line_args, num_command_line_args,
3282 unsaved_files, num_unsaved_files, options, &TU);
Reid Kleckner6eaf05a2014-02-13 01:19:59 +00003283 (void)Result;
Dmitri Gribenko1bf8d912014-02-18 15:20:02 +00003284 assert((TU && Result == CXError_Success) ||
3285 (!TU && Result != CXError_Success));
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003286 return TU;
3287}
3288
3289enum CXErrorCode clang_parseTranslationUnit2(
Benjamin Kramerc02670e2015-11-18 16:14:27 +00003290 CXIndex CIdx, const char *source_filename,
3291 const char *const *command_line_args, int num_command_line_args,
3292 struct CXUnsavedFile *unsaved_files, unsigned num_unsaved_files,
3293 unsigned options, CXTranslationUnit *out_TU) {
3294 SmallVector<const char *, 4> Args;
3295 Args.push_back("clang");
3296 Args.append(command_line_args, command_line_args + num_command_line_args);
3297 return clang_parseTranslationUnit2FullArgv(
3298 CIdx, source_filename, Args.data(), Args.size(), unsaved_files,
3299 num_unsaved_files, options, out_TU);
3300}
3301
3302enum CXErrorCode clang_parseTranslationUnit2FullArgv(
3303 CXIndex CIdx, const char *source_filename,
3304 const char *const *command_line_args, int num_command_line_args,
3305 struct CXUnsavedFile *unsaved_files, unsigned num_unsaved_files,
3306 unsigned options, CXTranslationUnit *out_TU) {
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00003307 LOG_FUNC_SECTION {
3308 *Log << source_filename << ": ";
3309 for (int i = 0; i != num_command_line_args; ++i)
3310 *Log << command_line_args[i] << " ";
3311 }
3312
Alp Toker9d85b182014-07-07 01:23:14 +00003313 if (num_unsaved_files && !unsaved_files)
3314 return CXError_InvalidArguments;
3315
Alp Toker5c532982014-07-07 22:42:03 +00003316 CXErrorCode result = CXError_Failure;
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003317 auto ParseTranslationUnitImpl = [=, &result] {
3318 result = clang_parseTranslationUnit_Impl(
3319 CIdx, source_filename, command_line_args, num_command_line_args,
3320 llvm::makeArrayRef(unsaved_files, num_unsaved_files), options, out_TU);
3321 };
Guy Benyei11169dd2012-12-18 14:30:41 +00003322 llvm::CrashRecoveryContext CRC;
3323
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003324 if (!RunSafely(CRC, ParseTranslationUnitImpl)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003325 fprintf(stderr, "libclang: crash detected during parsing: {\n");
3326 fprintf(stderr, " 'source_filename' : '%s'\n", source_filename);
3327 fprintf(stderr, " 'command_line_args' : [");
3328 for (int i = 0; i != num_command_line_args; ++i) {
3329 if (i)
3330 fprintf(stderr, ", ");
3331 fprintf(stderr, "'%s'", command_line_args[i]);
3332 }
3333 fprintf(stderr, "],\n");
3334 fprintf(stderr, " 'unsaved_files' : [");
3335 for (unsigned i = 0; i != num_unsaved_files; ++i) {
3336 if (i)
3337 fprintf(stderr, ", ");
3338 fprintf(stderr, "('%s', '...', %ld)", unsaved_files[i].Filename,
3339 unsaved_files[i].Length);
3340 }
3341 fprintf(stderr, "],\n");
3342 fprintf(stderr, " 'options' : %d,\n", options);
3343 fprintf(stderr, "}\n");
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003344
3345 return CXError_Crashed;
Guy Benyei11169dd2012-12-18 14:30:41 +00003346 } else if (getenv("LIBCLANG_RESOURCE_USAGE")) {
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003347 if (CXTranslationUnit *TU = out_TU)
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003348 PrintLibclangResourceUsage(*TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00003349 }
Alp Toker5c532982014-07-07 22:42:03 +00003350
3351 return result;
Guy Benyei11169dd2012-12-18 14:30:41 +00003352}
3353
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003354CXString clang_Type_getObjCEncoding(CXType CT) {
3355 CXTranslationUnit tu = static_cast<CXTranslationUnit>(CT.data[1]);
3356 ASTContext &Ctx = getASTUnit(tu)->getASTContext();
3357 std::string encoding;
3358 Ctx.getObjCEncodingForType(QualType::getFromOpaquePtr(CT.data[0]),
3359 encoding);
3360
3361 return cxstring::createDup(encoding);
3362}
3363
3364static const IdentifierInfo *getMacroIdentifier(CXCursor C) {
3365 if (C.kind == CXCursor_MacroDefinition) {
3366 if (const MacroDefinitionRecord *MDR = getCursorMacroDefinition(C))
3367 return MDR->getName();
3368 } else if (C.kind == CXCursor_MacroExpansion) {
3369 MacroExpansionCursor ME = getCursorMacroExpansion(C);
3370 return ME.getName();
3371 }
3372 return nullptr;
3373}
3374
3375unsigned clang_Cursor_isMacroFunctionLike(CXCursor C) {
3376 const IdentifierInfo *II = getMacroIdentifier(C);
3377 if (!II) {
3378 return false;
3379 }
3380 ASTUnit *ASTU = getCursorASTUnit(C);
3381 Preprocessor &PP = ASTU->getPreprocessor();
3382 if (const MacroInfo *MI = PP.getMacroInfo(II))
3383 return MI->isFunctionLike();
3384 return false;
3385}
3386
3387unsigned clang_Cursor_isMacroBuiltin(CXCursor C) {
3388 const IdentifierInfo *II = getMacroIdentifier(C);
3389 if (!II) {
3390 return false;
3391 }
3392 ASTUnit *ASTU = getCursorASTUnit(C);
3393 Preprocessor &PP = ASTU->getPreprocessor();
3394 if (const MacroInfo *MI = PP.getMacroInfo(II))
3395 return MI->isBuiltinMacro();
3396 return false;
3397}
3398
3399unsigned clang_Cursor_isFunctionInlined(CXCursor C) {
3400 const Decl *D = getCursorDecl(C);
3401 const FunctionDecl *FD = dyn_cast_or_null<FunctionDecl>(D);
3402 if (!FD) {
3403 return false;
3404 }
3405 return FD->isInlined();
3406}
3407
3408static StringLiteral* getCFSTR_value(CallExpr *callExpr) {
3409 if (callExpr->getNumArgs() != 1) {
3410 return nullptr;
3411 }
3412
3413 StringLiteral *S = nullptr;
3414 auto *arg = callExpr->getArg(0);
3415 if (arg->getStmtClass() == Stmt::ImplicitCastExprClass) {
3416 ImplicitCastExpr *I = static_cast<ImplicitCastExpr *>(arg);
3417 auto *subExpr = I->getSubExprAsWritten();
3418
3419 if(subExpr->getStmtClass() != Stmt::StringLiteralClass){
3420 return nullptr;
3421 }
3422
3423 S = static_cast<StringLiteral *>(I->getSubExprAsWritten());
3424 } else if (arg->getStmtClass() == Stmt::StringLiteralClass) {
3425 S = static_cast<StringLiteral *>(callExpr->getArg(0));
3426 } else {
3427 return nullptr;
3428 }
3429 return S;
3430}
3431
David Blaikie59272572016-04-13 18:23:33 +00003432struct ExprEvalResult {
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003433 CXEvalResultKind EvalType;
3434 union {
3435 int intVal;
3436 double floatVal;
3437 char *stringVal;
3438 } EvalData;
David Blaikie59272572016-04-13 18:23:33 +00003439 ~ExprEvalResult() {
3440 if (EvalType != CXEval_UnExposed && EvalType != CXEval_Float &&
3441 EvalType != CXEval_Int) {
3442 delete EvalData.stringVal;
3443 }
3444 }
3445};
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003446
3447void clang_EvalResult_dispose(CXEvalResult E) {
David Blaikie59272572016-04-13 18:23:33 +00003448 delete static_cast<ExprEvalResult *>(E);
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003449}
3450
3451CXEvalResultKind clang_EvalResult_getKind(CXEvalResult E) {
3452 if (!E) {
3453 return CXEval_UnExposed;
3454 }
3455 return ((ExprEvalResult *)E)->EvalType;
3456}
3457
3458int clang_EvalResult_getAsInt(CXEvalResult E) {
3459 if (!E) {
3460 return 0;
3461 }
3462 return ((ExprEvalResult *)E)->EvalData.intVal;
3463}
3464
3465double clang_EvalResult_getAsDouble(CXEvalResult E) {
3466 if (!E) {
3467 return 0;
3468 }
3469 return ((ExprEvalResult *)E)->EvalData.floatVal;
3470}
3471
3472const char* clang_EvalResult_getAsStr(CXEvalResult E) {
3473 if (!E) {
3474 return nullptr;
3475 }
3476 return ((ExprEvalResult *)E)->EvalData.stringVal;
3477}
3478
3479static const ExprEvalResult* evaluateExpr(Expr *expr, CXCursor C) {
3480 Expr::EvalResult ER;
3481 ASTContext &ctx = getCursorContext(C);
David Blaikiebbc00882016-04-13 18:36:19 +00003482 if (!expr)
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003483 return nullptr;
David Blaikiebbc00882016-04-13 18:36:19 +00003484
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003485 expr = expr->IgnoreParens();
David Blaikiebbc00882016-04-13 18:36:19 +00003486 if (!expr->EvaluateAsRValue(ER, ctx))
3487 return nullptr;
3488
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003489 QualType rettype;
3490 CallExpr *callExpr;
David Blaikie59272572016-04-13 18:23:33 +00003491 auto result = llvm::make_unique<ExprEvalResult>();
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003492 result->EvalType = CXEval_UnExposed;
3493
David Blaikiebbc00882016-04-13 18:36:19 +00003494 if (ER.Val.isInt()) {
3495 result->EvalType = CXEval_Int;
3496 result->EvalData.intVal = ER.Val.getInt().getExtValue();
3497 return result.release();
3498 }
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003499
David Blaikiebbc00882016-04-13 18:36:19 +00003500 if (ER.Val.isFloat()) {
3501 llvm::SmallVector<char, 100> Buffer;
3502 ER.Val.getFloat().toString(Buffer);
3503 std::string floatStr(Buffer.data(), Buffer.size());
3504 result->EvalType = CXEval_Float;
3505 bool ignored;
3506 llvm::APFloat apFloat = ER.Val.getFloat();
3507 apFloat.convert(llvm::APFloat::IEEEdouble,
3508 llvm::APFloat::rmNearestTiesToEven, &ignored);
3509 result->EvalData.floatVal = apFloat.convertToDouble();
3510 return result.release();
3511 }
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003512
David Blaikiebbc00882016-04-13 18:36:19 +00003513 if (expr->getStmtClass() == Stmt::ImplicitCastExprClass) {
3514 const ImplicitCastExpr *I = dyn_cast<ImplicitCastExpr>(expr);
3515 auto *subExpr = I->getSubExprAsWritten();
3516 if (subExpr->getStmtClass() == Stmt::StringLiteralClass ||
3517 subExpr->getStmtClass() == Stmt::ObjCStringLiteralClass) {
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003518 const StringLiteral *StrE = nullptr;
3519 const ObjCStringLiteral *ObjCExpr;
David Blaikiebbc00882016-04-13 18:36:19 +00003520 ObjCExpr = dyn_cast<ObjCStringLiteral>(subExpr);
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003521
3522 if (ObjCExpr) {
3523 StrE = ObjCExpr->getString();
3524 result->EvalType = CXEval_ObjCStrLiteral;
3525 } else {
David Blaikiebbc00882016-04-13 18:36:19 +00003526 StrE = cast<StringLiteral>(I->getSubExprAsWritten());
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003527 result->EvalType = CXEval_StrLiteral;
3528 }
3529
3530 std::string strRef(StrE->getString().str());
David Blaikie59272572016-04-13 18:23:33 +00003531 result->EvalData.stringVal = new char[strRef.size() + 1];
David Blaikiebbc00882016-04-13 18:36:19 +00003532 strncpy((char *)result->EvalData.stringVal, strRef.c_str(),
3533 strRef.size());
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003534 result->EvalData.stringVal[strRef.size()] = '\0';
David Blaikie59272572016-04-13 18:23:33 +00003535 return result.release();
David Blaikiebbc00882016-04-13 18:36:19 +00003536 }
3537 } else if (expr->getStmtClass() == Stmt::ObjCStringLiteralClass ||
3538 expr->getStmtClass() == Stmt::StringLiteralClass) {
3539 const StringLiteral *StrE = nullptr;
3540 const ObjCStringLiteral *ObjCExpr;
3541 ObjCExpr = dyn_cast<ObjCStringLiteral>(expr);
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003542
David Blaikiebbc00882016-04-13 18:36:19 +00003543 if (ObjCExpr) {
3544 StrE = ObjCExpr->getString();
3545 result->EvalType = CXEval_ObjCStrLiteral;
3546 } else {
3547 StrE = cast<StringLiteral>(expr);
3548 result->EvalType = CXEval_StrLiteral;
3549 }
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003550
David Blaikiebbc00882016-04-13 18:36:19 +00003551 std::string strRef(StrE->getString().str());
3552 result->EvalData.stringVal = new char[strRef.size() + 1];
3553 strncpy((char *)result->EvalData.stringVal, strRef.c_str(), strRef.size());
3554 result->EvalData.stringVal[strRef.size()] = '\0';
3555 return result.release();
3556 }
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003557
David Blaikiebbc00882016-04-13 18:36:19 +00003558 if (expr->getStmtClass() == Stmt::CStyleCastExprClass) {
3559 CStyleCastExpr *CC = static_cast<CStyleCastExpr *>(expr);
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003560
David Blaikiebbc00882016-04-13 18:36:19 +00003561 rettype = CC->getType();
3562 if (rettype.getAsString() == "CFStringRef" &&
3563 CC->getSubExpr()->getStmtClass() == Stmt::CallExprClass) {
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003564
David Blaikiebbc00882016-04-13 18:36:19 +00003565 callExpr = static_cast<CallExpr *>(CC->getSubExpr());
3566 StringLiteral *S = getCFSTR_value(callExpr);
3567 if (S) {
3568 std::string strLiteral(S->getString().str());
3569 result->EvalType = CXEval_CFStr;
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003570
David Blaikiebbc00882016-04-13 18:36:19 +00003571 result->EvalData.stringVal = new char[strLiteral.size() + 1];
3572 strncpy((char *)result->EvalData.stringVal, strLiteral.c_str(),
3573 strLiteral.size());
3574 result->EvalData.stringVal[strLiteral.size()] = '\0';
David Blaikie59272572016-04-13 18:23:33 +00003575 return result.release();
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003576 }
3577 }
3578
David Blaikiebbc00882016-04-13 18:36:19 +00003579 } else if (expr->getStmtClass() == Stmt::CallExprClass) {
3580 callExpr = static_cast<CallExpr *>(expr);
3581 rettype = callExpr->getCallReturnType(ctx);
3582
3583 if (rettype->isVectorType() || callExpr->getNumArgs() > 1)
3584 return nullptr;
3585
3586 if (rettype->isIntegralType(ctx) || rettype->isRealFloatingType()) {
3587 if (callExpr->getNumArgs() == 1 &&
3588 !callExpr->getArg(0)->getType()->isIntegralType(ctx))
3589 return nullptr;
3590 } else if (rettype.getAsString() == "CFStringRef") {
3591
3592 StringLiteral *S = getCFSTR_value(callExpr);
3593 if (S) {
3594 std::string strLiteral(S->getString().str());
3595 result->EvalType = CXEval_CFStr;
3596 result->EvalData.stringVal = new char[strLiteral.size() + 1];
3597 strncpy((char *)result->EvalData.stringVal, strLiteral.c_str(),
3598 strLiteral.size());
3599 result->EvalData.stringVal[strLiteral.size()] = '\0';
3600 return result.release();
3601 }
3602 }
3603 } else if (expr->getStmtClass() == Stmt::DeclRefExprClass) {
3604 DeclRefExpr *D = static_cast<DeclRefExpr *>(expr);
3605 ValueDecl *V = D->getDecl();
3606 if (V->getKind() == Decl::Function) {
3607 std::string strName = V->getNameAsString();
3608 result->EvalType = CXEval_Other;
3609 result->EvalData.stringVal = new char[strName.size() + 1];
3610 strncpy(result->EvalData.stringVal, strName.c_str(), strName.size());
3611 result->EvalData.stringVal[strName.size()] = '\0';
3612 return result.release();
3613 }
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003614 }
3615
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003616 return nullptr;
3617}
3618
3619CXEvalResult clang_Cursor_Evaluate(CXCursor C) {
3620 const Decl *D = getCursorDecl(C);
3621 if (D) {
3622 const Expr *expr = nullptr;
3623 if (auto *Var = dyn_cast<VarDecl>(D)) {
3624 expr = Var->getInit();
3625 } else if (auto *Field = dyn_cast<FieldDecl>(D)) {
3626 expr = Field->getInClassInitializer();
3627 }
3628 if (expr)
Aaron Ballman01dc1572016-01-20 15:25:30 +00003629 return const_cast<CXEvalResult>(reinterpret_cast<const void *>(
3630 evaluateExpr(const_cast<Expr *>(expr), C)));
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003631 return nullptr;
3632 }
3633
3634 const CompoundStmt *compoundStmt = dyn_cast_or_null<CompoundStmt>(getCursorStmt(C));
3635 if (compoundStmt) {
3636 Expr *expr = nullptr;
3637 for (auto *bodyIterator : compoundStmt->body()) {
3638 if ((expr = dyn_cast<Expr>(bodyIterator))) {
3639 break;
3640 }
3641 }
3642 if (expr)
Aaron Ballman01dc1572016-01-20 15:25:30 +00003643 return const_cast<CXEvalResult>(
3644 reinterpret_cast<const void *>(evaluateExpr(expr, C)));
Argyrios Kyrtzidis785705b2016-01-16 00:20:02 +00003645 }
3646 return nullptr;
3647}
3648
3649unsigned clang_Cursor_hasAttrs(CXCursor C) {
3650 const Decl *D = getCursorDecl(C);
3651 if (!D) {
3652 return 0;
3653 }
3654
3655 if (D->hasAttrs()) {
3656 return 1;
3657 }
3658
3659 return 0;
3660}
Guy Benyei11169dd2012-12-18 14:30:41 +00003661unsigned clang_defaultSaveOptions(CXTranslationUnit TU) {
3662 return CXSaveTranslationUnit_None;
3663}
3664
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003665static CXSaveError clang_saveTranslationUnit_Impl(CXTranslationUnit TU,
3666 const char *FileName,
3667 unsigned options) {
3668 CIndexer *CXXIdx = TU->CIdx;
Guy Benyei11169dd2012-12-18 14:30:41 +00003669 if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForIndexing))
3670 setThreadBackgroundPriority();
3671
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003672 bool hadError = cxtu::getASTUnit(TU)->Save(FileName);
3673 return hadError ? CXSaveError_Unknown : CXSaveError_None;
Guy Benyei11169dd2012-12-18 14:30:41 +00003674}
3675
3676int clang_saveTranslationUnit(CXTranslationUnit TU, const char *FileName,
3677 unsigned options) {
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00003678 LOG_FUNC_SECTION {
3679 *Log << TU << ' ' << FileName;
3680 }
3681
Dmitri Gribenko852d6222014-02-11 15:02:48 +00003682 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003683 LOG_BAD_TU(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00003684 return CXSaveError_InvalidTU;
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003685 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003686
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003687 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00003688 ASTUnit::ConcurrencyCheck Check(*CXXUnit);
3689 if (!CXXUnit->hasSema())
3690 return CXSaveError_InvalidTU;
3691
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003692 CXSaveError result;
3693 auto SaveTranslationUnitImpl = [=, &result]() {
3694 result = clang_saveTranslationUnit_Impl(TU, FileName, options);
3695 };
Guy Benyei11169dd2012-12-18 14:30:41 +00003696
3697 if (!CXXUnit->getDiagnostics().hasUnrecoverableErrorOccurred() ||
3698 getenv("LIBCLANG_NOTHREADS")) {
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003699 SaveTranslationUnitImpl();
Guy Benyei11169dd2012-12-18 14:30:41 +00003700
3701 if (getenv("LIBCLANG_RESOURCE_USAGE"))
3702 PrintLibclangResourceUsage(TU);
3703
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003704 return result;
Guy Benyei11169dd2012-12-18 14:30:41 +00003705 }
3706
3707 // We have an AST that has invalid nodes due to compiler errors.
3708 // Use a crash recovery thread for protection.
3709
3710 llvm::CrashRecoveryContext CRC;
3711
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003712 if (!RunSafely(CRC, SaveTranslationUnitImpl)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003713 fprintf(stderr, "libclang: crash detected during AST saving: {\n");
3714 fprintf(stderr, " 'filename' : '%s'\n", FileName);
3715 fprintf(stderr, " 'options' : %d,\n", options);
3716 fprintf(stderr, "}\n");
3717
3718 return CXSaveError_Unknown;
3719
3720 } else if (getenv("LIBCLANG_RESOURCE_USAGE")) {
3721 PrintLibclangResourceUsage(TU);
3722 }
3723
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003724 return result;
Guy Benyei11169dd2012-12-18 14:30:41 +00003725}
3726
3727void clang_disposeTranslationUnit(CXTranslationUnit CTUnit) {
3728 if (CTUnit) {
3729 // If the translation unit has been marked as unsafe to free, just discard
3730 // it.
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003731 ASTUnit *Unit = cxtu::getASTUnit(CTUnit);
3732 if (Unit && Unit->isUnsafeToFree())
Guy Benyei11169dd2012-12-18 14:30:41 +00003733 return;
3734
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003735 delete cxtu::getASTUnit(CTUnit);
Dmitri Gribenkob95b3f12013-01-26 22:44:19 +00003736 delete CTUnit->StringPool;
Guy Benyei11169dd2012-12-18 14:30:41 +00003737 delete static_cast<CXDiagnosticSetImpl *>(CTUnit->Diagnostics);
3738 disposeOverridenCXCursorsPool(CTUnit->OverridenCursorsPool);
Dmitri Gribenko9e605112013-11-13 22:16:51 +00003739 delete CTUnit->CommentToXML;
Guy Benyei11169dd2012-12-18 14:30:41 +00003740 delete CTUnit;
3741 }
3742}
3743
3744unsigned clang_defaultReparseOptions(CXTranslationUnit TU) {
3745 return CXReparse_None;
3746}
3747
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003748static CXErrorCode
3749clang_reparseTranslationUnit_Impl(CXTranslationUnit TU,
3750 ArrayRef<CXUnsavedFile> unsaved_files,
3751 unsigned options) {
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003752 // Check arguments.
Dmitri Gribenko852d6222014-02-11 15:02:48 +00003753 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003754 LOG_BAD_TU(TU);
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003755 return CXError_InvalidArguments;
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003756 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003757
3758 // Reset the associated diagnostics.
3759 delete static_cast<CXDiagnosticSetImpl*>(TU->Diagnostics);
Craig Topper69186e72014-06-08 08:38:04 +00003760 TU->Diagnostics = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00003761
Dmitri Gribenko183436e2013-01-26 21:49:50 +00003762 CIndexer *CXXIdx = TU->CIdx;
Guy Benyei11169dd2012-12-18 14:30:41 +00003763 if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForEditing))
3764 setThreadBackgroundPriority();
3765
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003766 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00003767 ASTUnit::ConcurrencyCheck Check(*CXXUnit);
Ahmed Charlesb8984322014-03-07 20:03:18 +00003768
3769 std::unique_ptr<std::vector<ASTUnit::RemappedFile>> RemappedFiles(
3770 new std::vector<ASTUnit::RemappedFile>());
3771
Guy Benyei11169dd2012-12-18 14:30:41 +00003772 // Recover resources if we crash before exiting this function.
3773 llvm::CrashRecoveryContextCleanupRegistrar<
3774 std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
Alp Toker9d85b182014-07-07 01:23:14 +00003775
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003776 for (auto &UF : unsaved_files) {
Rafael Espindolad87f8d72014-08-27 20:03:29 +00003777 std::unique_ptr<llvm::MemoryBuffer> MB =
Alp Toker9d85b182014-07-07 01:23:14 +00003778 llvm::MemoryBuffer::getMemBufferCopy(getContents(UF), UF.Filename);
Rafael Espindolad87f8d72014-08-27 20:03:29 +00003779 RemappedFiles->push_back(std::make_pair(UF.Filename, MB.release()));
Guy Benyei11169dd2012-12-18 14:30:41 +00003780 }
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003781
Adrian Prantlbb165fb2015-06-20 18:53:08 +00003782 if (!CXXUnit->Reparse(CXXIdx->getPCHContainerOperations(),
3783 *RemappedFiles.get()))
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003784 return CXError_Success;
3785 if (isASTReadError(CXXUnit))
3786 return CXError_ASTReadError;
3787 return CXError_Failure;
Guy Benyei11169dd2012-12-18 14:30:41 +00003788}
3789
3790int clang_reparseTranslationUnit(CXTranslationUnit TU,
3791 unsigned num_unsaved_files,
3792 struct CXUnsavedFile *unsaved_files,
3793 unsigned options) {
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00003794 LOG_FUNC_SECTION {
3795 *Log << TU;
3796 }
3797
Alp Toker9d85b182014-07-07 01:23:14 +00003798 if (num_unsaved_files && !unsaved_files)
3799 return CXError_InvalidArguments;
3800
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003801 CXErrorCode result;
3802 auto ReparseTranslationUnitImpl = [=, &result]() {
3803 result = clang_reparseTranslationUnit_Impl(
3804 TU, llvm::makeArrayRef(unsaved_files, num_unsaved_files), options);
3805 };
Guy Benyei11169dd2012-12-18 14:30:41 +00003806
3807 if (getenv("LIBCLANG_NOTHREADS")) {
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003808 ReparseTranslationUnitImpl();
Alp Toker5c532982014-07-07 22:42:03 +00003809 return result;
Guy Benyei11169dd2012-12-18 14:30:41 +00003810 }
3811
3812 llvm::CrashRecoveryContext CRC;
3813
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00003814 if (!RunSafely(CRC, ReparseTranslationUnitImpl)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003815 fprintf(stderr, "libclang: crash detected during reparsing\n");
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003816 cxtu::getASTUnit(TU)->setUnsafeToFree(true);
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00003817 return CXError_Crashed;
Guy Benyei11169dd2012-12-18 14:30:41 +00003818 } else if (getenv("LIBCLANG_RESOURCE_USAGE"))
3819 PrintLibclangResourceUsage(TU);
3820
Alp Toker5c532982014-07-07 22:42:03 +00003821 return result;
Guy Benyei11169dd2012-12-18 14:30:41 +00003822}
3823
3824
3825CXString clang_getTranslationUnitSpelling(CXTranslationUnit CTUnit) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00003826 if (isNotUsableTU(CTUnit)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003827 LOG_BAD_TU(CTUnit);
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00003828 return cxstring::createEmpty();
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003829 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003830
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003831 ASTUnit *CXXUnit = cxtu::getASTUnit(CTUnit);
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00003832 return cxstring::createDup(CXXUnit->getOriginalSourceFileName());
Guy Benyei11169dd2012-12-18 14:30:41 +00003833}
3834
3835CXCursor clang_getTranslationUnitCursor(CXTranslationUnit TU) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00003836 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003837 LOG_BAD_TU(TU);
Argyrios Kyrtzidis0e95fca2013-04-04 22:40:59 +00003838 return clang_getNullCursor();
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003839 }
Argyrios Kyrtzidis0e95fca2013-04-04 22:40:59 +00003840
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003841 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00003842 return MakeCXCursor(CXXUnit->getASTContext().getTranslationUnitDecl(), TU);
3843}
3844
3845} // end: extern "C"
3846
3847//===----------------------------------------------------------------------===//
3848// CXFile Operations.
3849//===----------------------------------------------------------------------===//
3850
3851extern "C" {
3852CXString clang_getFileName(CXFile SFile) {
3853 if (!SFile)
Dmitri Gribenkof98dfba2013-02-01 14:13:32 +00003854 return cxstring::createNull();
Guy Benyei11169dd2012-12-18 14:30:41 +00003855
3856 FileEntry *FEnt = static_cast<FileEntry *>(SFile);
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00003857 return cxstring::createRef(FEnt->getName());
Guy Benyei11169dd2012-12-18 14:30:41 +00003858}
3859
3860time_t clang_getFileTime(CXFile SFile) {
3861 if (!SFile)
3862 return 0;
3863
3864 FileEntry *FEnt = static_cast<FileEntry *>(SFile);
3865 return FEnt->getModificationTime();
3866}
3867
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003868CXFile clang_getFile(CXTranslationUnit TU, const char *file_name) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00003869 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003870 LOG_BAD_TU(TU);
Craig Topper69186e72014-06-08 08:38:04 +00003871 return nullptr;
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003872 }
Guy Benyei11169dd2012-12-18 14:30:41 +00003873
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003874 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00003875
3876 FileManager &FMgr = CXXUnit->getFileManager();
3877 return const_cast<FileEntry *>(FMgr.getFile(file_name));
3878}
3879
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003880unsigned clang_isFileMultipleIncludeGuarded(CXTranslationUnit TU,
3881 CXFile file) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00003882 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00003883 LOG_BAD_TU(TU);
3884 return 0;
3885 }
3886
3887 if (!file)
Guy Benyei11169dd2012-12-18 14:30:41 +00003888 return 0;
3889
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00003890 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00003891 FileEntry *FEnt = static_cast<FileEntry *>(file);
3892 return CXXUnit->getPreprocessor().getHeaderSearchInfo()
3893 .isFileMultipleIncludeGuarded(FEnt);
3894}
3895
Argyrios Kyrtzidisac08b262013-01-26 04:52:52 +00003896int clang_getFileUniqueID(CXFile file, CXFileUniqueID *outID) {
3897 if (!file || !outID)
3898 return 1;
3899
Argyrios Kyrtzidisac08b262013-01-26 04:52:52 +00003900 FileEntry *FEnt = static_cast<FileEntry *>(file);
Rafael Espindolaf8f91b82013-08-01 21:42:11 +00003901 const llvm::sys::fs::UniqueID &ID = FEnt->getUniqueID();
3902 outID->data[0] = ID.getDevice();
3903 outID->data[1] = ID.getFile();
Argyrios Kyrtzidisac08b262013-01-26 04:52:52 +00003904 outID->data[2] = FEnt->getModificationTime();
3905 return 0;
Argyrios Kyrtzidisac08b262013-01-26 04:52:52 +00003906}
3907
Argyrios Kyrtzidisac3997e2014-08-16 00:26:19 +00003908int clang_File_isEqual(CXFile file1, CXFile file2) {
3909 if (file1 == file2)
3910 return true;
3911
3912 if (!file1 || !file2)
3913 return false;
3914
3915 FileEntry *FEnt1 = static_cast<FileEntry *>(file1);
3916 FileEntry *FEnt2 = static_cast<FileEntry *>(file2);
3917 return FEnt1->getUniqueID() == FEnt2->getUniqueID();
3918}
3919
Guy Benyei11169dd2012-12-18 14:30:41 +00003920} // end: extern "C"
3921
3922//===----------------------------------------------------------------------===//
3923// CXCursor Operations.
3924//===----------------------------------------------------------------------===//
3925
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003926static const Decl *getDeclFromExpr(const Stmt *E) {
3927 if (const ImplicitCastExpr *CE = dyn_cast<ImplicitCastExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003928 return getDeclFromExpr(CE->getSubExpr());
3929
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003930 if (const DeclRefExpr *RefExpr = dyn_cast<DeclRefExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003931 return RefExpr->getDecl();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003932 if (const MemberExpr *ME = dyn_cast<MemberExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003933 return ME->getMemberDecl();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003934 if (const ObjCIvarRefExpr *RE = dyn_cast<ObjCIvarRefExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003935 return RE->getDecl();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003936 if (const ObjCPropertyRefExpr *PRE = dyn_cast<ObjCPropertyRefExpr>(E)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00003937 if (PRE->isExplicitProperty())
3938 return PRE->getExplicitProperty();
3939 // It could be messaging both getter and setter as in:
3940 // ++myobj.myprop;
3941 // in which case prefer to associate the setter since it is less obvious
3942 // from inspecting the source that the setter is going to get called.
3943 if (PRE->isMessagingSetter())
3944 return PRE->getImplicitPropertySetter();
3945 return PRE->getImplicitPropertyGetter();
3946 }
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003947 if (const PseudoObjectExpr *POE = dyn_cast<PseudoObjectExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003948 return getDeclFromExpr(POE->getSyntacticForm());
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003949 if (const OpaqueValueExpr *OVE = dyn_cast<OpaqueValueExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003950 if (Expr *Src = OVE->getSourceExpr())
3951 return getDeclFromExpr(Src);
3952
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003953 if (const CallExpr *CE = dyn_cast<CallExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003954 return getDeclFromExpr(CE->getCallee());
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003955 if (const CXXConstructExpr *CE = dyn_cast<CXXConstructExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003956 if (!CE->isElidable())
3957 return CE->getConstructor();
Richard Smith5179eb72016-06-28 19:03:57 +00003958 if (const CXXInheritedCtorInitExpr *CE =
3959 dyn_cast<CXXInheritedCtorInitExpr>(E))
3960 return CE->getConstructor();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003961 if (const ObjCMessageExpr *OME = dyn_cast<ObjCMessageExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003962 return OME->getMethodDecl();
3963
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003964 if (const ObjCProtocolExpr *PE = dyn_cast<ObjCProtocolExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003965 return PE->getProtocol();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003966 if (const SubstNonTypeTemplateParmPackExpr *NTTP
Guy Benyei11169dd2012-12-18 14:30:41 +00003967 = dyn_cast<SubstNonTypeTemplateParmPackExpr>(E))
3968 return NTTP->getParameterPack();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00003969 if (const SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003970 if (isa<NonTypeTemplateParmDecl>(SizeOfPack->getPack()) ||
3971 isa<ParmVarDecl>(SizeOfPack->getPack()))
3972 return SizeOfPack->getPack();
Craig Topper69186e72014-06-08 08:38:04 +00003973
3974 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00003975}
3976
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003977static SourceLocation getLocationFromExpr(const Expr *E) {
3978 if (const ImplicitCastExpr *CE = dyn_cast<ImplicitCastExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003979 return getLocationFromExpr(CE->getSubExpr());
3980
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003981 if (const ObjCMessageExpr *Msg = dyn_cast<ObjCMessageExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003982 return /*FIXME:*/Msg->getLeftLoc();
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003983 if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003984 return DRE->getLocation();
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003985 if (const MemberExpr *Member = dyn_cast<MemberExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003986 return Member->getMemberLoc();
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003987 if (const ObjCIvarRefExpr *Ivar = dyn_cast<ObjCIvarRefExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003988 return Ivar->getLocation();
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003989 if (const SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003990 return SizeOfPack->getPackLoc();
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00003991 if (const ObjCPropertyRefExpr *PropRef = dyn_cast<ObjCPropertyRefExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00003992 return PropRef->getLocation();
3993
3994 return E->getLocStart();
3995}
3996
3997extern "C" {
3998
3999unsigned clang_visitChildren(CXCursor parent,
4000 CXCursorVisitor visitor,
4001 CXClientData client_data) {
4002 CursorVisitor CursorVis(getCursorTU(parent), visitor, client_data,
4003 /*VisitPreprocessorLast=*/false);
4004 return CursorVis.VisitChildren(parent);
4005}
4006
4007#ifndef __has_feature
4008#define __has_feature(x) 0
4009#endif
4010#if __has_feature(blocks)
4011typedef enum CXChildVisitResult
4012 (^CXCursorVisitorBlock)(CXCursor cursor, CXCursor parent);
4013
4014static enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
4015 CXClientData client_data) {
4016 CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
4017 return block(cursor, parent);
4018}
4019#else
4020// If we are compiled with a compiler that doesn't have native blocks support,
4021// define and call the block manually, so the
4022typedef struct _CXChildVisitResult
4023{
4024 void *isa;
4025 int flags;
4026 int reserved;
4027 enum CXChildVisitResult(*invoke)(struct _CXChildVisitResult*, CXCursor,
4028 CXCursor);
4029} *CXCursorVisitorBlock;
4030
4031static enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
4032 CXClientData client_data) {
4033 CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
4034 return block->invoke(block, cursor, parent);
4035}
4036#endif
4037
4038
4039unsigned clang_visitChildrenWithBlock(CXCursor parent,
4040 CXCursorVisitorBlock block) {
4041 return clang_visitChildren(parent, visitWithBlock, block);
4042}
4043
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004044static CXString getDeclSpelling(const Decl *D) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004045 if (!D)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004046 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004047
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004048 const NamedDecl *ND = dyn_cast<NamedDecl>(D);
Guy Benyei11169dd2012-12-18 14:30:41 +00004049 if (!ND) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004050 if (const ObjCPropertyImplDecl *PropImpl =
4051 dyn_cast<ObjCPropertyImplDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00004052 if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004053 return cxstring::createDup(Property->getIdentifier()->getName());
Guy Benyei11169dd2012-12-18 14:30:41 +00004054
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004055 if (const ImportDecl *ImportD = dyn_cast<ImportDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00004056 if (Module *Mod = ImportD->getImportedModule())
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004057 return cxstring::createDup(Mod->getFullModuleName());
Guy Benyei11169dd2012-12-18 14:30:41 +00004058
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004059 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004060 }
4061
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004062 if (const ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004063 return cxstring::createDup(OMD->getSelector().getAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004064
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004065 if (const ObjCCategoryImplDecl *CIMP = dyn_cast<ObjCCategoryImplDecl>(ND))
Guy Benyei11169dd2012-12-18 14:30:41 +00004066 // No, this isn't the same as the code below. getIdentifier() is non-virtual
4067 // and returns different names. NamedDecl returns the class name and
4068 // ObjCCategoryImplDecl returns the category name.
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004069 return cxstring::createRef(CIMP->getIdentifier()->getNameStart());
Guy Benyei11169dd2012-12-18 14:30:41 +00004070
4071 if (isa<UsingDirectiveDecl>(D))
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004072 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004073
4074 SmallString<1024> S;
4075 llvm::raw_svector_ostream os(S);
4076 ND->printName(os);
4077
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004078 return cxstring::createDup(os.str());
Guy Benyei11169dd2012-12-18 14:30:41 +00004079}
4080
4081CXString clang_getCursorSpelling(CXCursor C) {
4082 if (clang_isTranslationUnit(C.kind))
Dmitri Gribenko2c173b42013-01-11 19:28:44 +00004083 return clang_getTranslationUnitSpelling(getCursorTU(C));
Guy Benyei11169dd2012-12-18 14:30:41 +00004084
4085 if (clang_isReference(C.kind)) {
4086 switch (C.kind) {
4087 case CXCursor_ObjCSuperClassRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004088 const ObjCInterfaceDecl *Super = getCursorObjCSuperClassRef(C).first;
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004089 return cxstring::createRef(Super->getIdentifier()->getNameStart());
Guy Benyei11169dd2012-12-18 14:30:41 +00004090 }
4091 case CXCursor_ObjCClassRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004092 const ObjCInterfaceDecl *Class = getCursorObjCClassRef(C).first;
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004093 return cxstring::createRef(Class->getIdentifier()->getNameStart());
Guy Benyei11169dd2012-12-18 14:30:41 +00004094 }
4095 case CXCursor_ObjCProtocolRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004096 const ObjCProtocolDecl *OID = getCursorObjCProtocolRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004097 assert(OID && "getCursorSpelling(): Missing protocol decl");
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004098 return cxstring::createRef(OID->getIdentifier()->getNameStart());
Guy Benyei11169dd2012-12-18 14:30:41 +00004099 }
4100 case CXCursor_CXXBaseSpecifier: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004101 const CXXBaseSpecifier *B = getCursorCXXBaseSpecifier(C);
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004102 return cxstring::createDup(B->getType().getAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004103 }
4104 case CXCursor_TypeRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004105 const TypeDecl *Type = getCursorTypeRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004106 assert(Type && "Missing type decl");
4107
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004108 return cxstring::createDup(getCursorContext(C).getTypeDeclType(Type).
Guy Benyei11169dd2012-12-18 14:30:41 +00004109 getAsString());
4110 }
4111 case CXCursor_TemplateRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004112 const TemplateDecl *Template = getCursorTemplateRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004113 assert(Template && "Missing template decl");
4114
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004115 return cxstring::createDup(Template->getNameAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004116 }
4117
4118 case CXCursor_NamespaceRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004119 const NamedDecl *NS = getCursorNamespaceRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004120 assert(NS && "Missing namespace decl");
4121
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004122 return cxstring::createDup(NS->getNameAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004123 }
4124
4125 case CXCursor_MemberRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004126 const FieldDecl *Field = getCursorMemberRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004127 assert(Field && "Missing member decl");
4128
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004129 return cxstring::createDup(Field->getNameAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004130 }
4131
4132 case CXCursor_LabelRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004133 const LabelStmt *Label = getCursorLabelRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004134 assert(Label && "Missing label");
4135
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004136 return cxstring::createRef(Label->getName());
Guy Benyei11169dd2012-12-18 14:30:41 +00004137 }
4138
4139 case CXCursor_OverloadedDeclRef: {
4140 OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004141 if (const Decl *D = Storage.dyn_cast<const Decl *>()) {
4142 if (const NamedDecl *ND = dyn_cast<NamedDecl>(D))
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004143 return cxstring::createDup(ND->getNameAsString());
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004144 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004145 }
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004146 if (const OverloadExpr *E = Storage.dyn_cast<const OverloadExpr *>())
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004147 return cxstring::createDup(E->getName().getAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004148 OverloadedTemplateStorage *Ovl
4149 = Storage.get<OverloadedTemplateStorage*>();
4150 if (Ovl->size() == 0)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004151 return cxstring::createEmpty();
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004152 return cxstring::createDup((*Ovl->begin())->getNameAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004153 }
4154
4155 case CXCursor_VariableRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00004156 const VarDecl *Var = getCursorVariableRef(C).first;
Guy Benyei11169dd2012-12-18 14:30:41 +00004157 assert(Var && "Missing variable decl");
4158
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004159 return cxstring::createDup(Var->getNameAsString());
Guy Benyei11169dd2012-12-18 14:30:41 +00004160 }
4161
4162 default:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004163 return cxstring::createRef("<not implemented>");
Guy Benyei11169dd2012-12-18 14:30:41 +00004164 }
4165 }
4166
4167 if (clang_isExpression(C.kind)) {
Argyrios Kyrtzidis3227d862014-03-03 19:40:52 +00004168 const Expr *E = getCursorExpr(C);
4169
4170 if (C.kind == CXCursor_ObjCStringLiteral ||
4171 C.kind == CXCursor_StringLiteral) {
4172 const StringLiteral *SLit;
4173 if (const ObjCStringLiteral *OSL = dyn_cast<ObjCStringLiteral>(E)) {
4174 SLit = OSL->getString();
4175 } else {
4176 SLit = cast<StringLiteral>(E);
4177 }
4178 SmallString<256> Buf;
4179 llvm::raw_svector_ostream OS(Buf);
4180 SLit->outputString(OS);
4181 return cxstring::createDup(OS.str());
4182 }
4183
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004184 const Decl *D = getDeclFromExpr(getCursorExpr(C));
Guy Benyei11169dd2012-12-18 14:30:41 +00004185 if (D)
4186 return getDeclSpelling(D);
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004187 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004188 }
4189
4190 if (clang_isStatement(C.kind)) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00004191 const Stmt *S = getCursorStmt(C);
4192 if (const LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S))
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004193 return cxstring::createRef(Label->getName());
Guy Benyei11169dd2012-12-18 14:30:41 +00004194
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004195 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004196 }
4197
4198 if (C.kind == CXCursor_MacroExpansion)
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004199 return cxstring::createRef(getCursorMacroExpansion(C).getName()
Guy Benyei11169dd2012-12-18 14:30:41 +00004200 ->getNameStart());
4201
4202 if (C.kind == CXCursor_MacroDefinition)
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004203 return cxstring::createRef(getCursorMacroDefinition(C)->getName()
Guy Benyei11169dd2012-12-18 14:30:41 +00004204 ->getNameStart());
4205
4206 if (C.kind == CXCursor_InclusionDirective)
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004207 return cxstring::createDup(getCursorInclusionDirective(C)->getFileName());
Guy Benyei11169dd2012-12-18 14:30:41 +00004208
4209 if (clang_isDeclaration(C.kind))
4210 return getDeclSpelling(getCursorDecl(C));
4211
4212 if (C.kind == CXCursor_AnnotateAttr) {
Dmitri Gribenkoe4baea62013-01-26 18:08:08 +00004213 const AnnotateAttr *AA = cast<AnnotateAttr>(cxcursor::getCursorAttr(C));
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004214 return cxstring::createDup(AA->getAnnotation());
Guy Benyei11169dd2012-12-18 14:30:41 +00004215 }
4216
4217 if (C.kind == CXCursor_AsmLabelAttr) {
Dmitri Gribenkoe4baea62013-01-26 18:08:08 +00004218 const AsmLabelAttr *AA = cast<AsmLabelAttr>(cxcursor::getCursorAttr(C));
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004219 return cxstring::createDup(AA->getLabel());
Guy Benyei11169dd2012-12-18 14:30:41 +00004220 }
4221
Argyrios Kyrtzidis16834f12013-09-25 00:14:38 +00004222 if (C.kind == CXCursor_PackedAttr) {
4223 return cxstring::createRef("packed");
4224 }
4225
Saleem Abdulrasool79c69712015-09-05 18:53:43 +00004226 if (C.kind == CXCursor_VisibilityAttr) {
4227 const VisibilityAttr *AA = cast<VisibilityAttr>(cxcursor::getCursorAttr(C));
4228 switch (AA->getVisibility()) {
4229 case VisibilityAttr::VisibilityType::Default:
4230 return cxstring::createRef("default");
4231 case VisibilityAttr::VisibilityType::Hidden:
4232 return cxstring::createRef("hidden");
4233 case VisibilityAttr::VisibilityType::Protected:
4234 return cxstring::createRef("protected");
4235 }
4236 llvm_unreachable("unknown visibility type");
4237 }
4238
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004239 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004240}
4241
4242CXSourceRange clang_Cursor_getSpellingNameRange(CXCursor C,
4243 unsigned pieceIndex,
4244 unsigned options) {
4245 if (clang_Cursor_isNull(C))
4246 return clang_getNullRange();
4247
4248 ASTContext &Ctx = getCursorContext(C);
4249
4250 if (clang_isStatement(C.kind)) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00004251 const Stmt *S = getCursorStmt(C);
4252 if (const LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004253 if (pieceIndex > 0)
4254 return clang_getNullRange();
4255 return cxloc::translateSourceRange(Ctx, Label->getIdentLoc());
4256 }
4257
4258 return clang_getNullRange();
4259 }
4260
4261 if (C.kind == CXCursor_ObjCMessageExpr) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00004262 if (const ObjCMessageExpr *
Guy Benyei11169dd2012-12-18 14:30:41 +00004263 ME = dyn_cast_or_null<ObjCMessageExpr>(getCursorExpr(C))) {
4264 if (pieceIndex >= ME->getNumSelectorLocs())
4265 return clang_getNullRange();
4266 return cxloc::translateSourceRange(Ctx, ME->getSelectorLoc(pieceIndex));
4267 }
4268 }
4269
4270 if (C.kind == CXCursor_ObjCInstanceMethodDecl ||
4271 C.kind == CXCursor_ObjCClassMethodDecl) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004272 if (const ObjCMethodDecl *
Guy Benyei11169dd2012-12-18 14:30:41 +00004273 MD = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(C))) {
4274 if (pieceIndex >= MD->getNumSelectorLocs())
4275 return clang_getNullRange();
4276 return cxloc::translateSourceRange(Ctx, MD->getSelectorLoc(pieceIndex));
4277 }
4278 }
4279
4280 if (C.kind == CXCursor_ObjCCategoryDecl ||
4281 C.kind == CXCursor_ObjCCategoryImplDecl) {
4282 if (pieceIndex > 0)
4283 return clang_getNullRange();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004284 if (const ObjCCategoryDecl *
Guy Benyei11169dd2012-12-18 14:30:41 +00004285 CD = dyn_cast_or_null<ObjCCategoryDecl>(getCursorDecl(C)))
4286 return cxloc::translateSourceRange(Ctx, CD->getCategoryNameLoc());
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004287 if (const ObjCCategoryImplDecl *
Guy Benyei11169dd2012-12-18 14:30:41 +00004288 CID = dyn_cast_or_null<ObjCCategoryImplDecl>(getCursorDecl(C)))
4289 return cxloc::translateSourceRange(Ctx, CID->getCategoryNameLoc());
4290 }
4291
4292 if (C.kind == CXCursor_ModuleImportDecl) {
4293 if (pieceIndex > 0)
4294 return clang_getNullRange();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004295 if (const ImportDecl *ImportD =
4296 dyn_cast_or_null<ImportDecl>(getCursorDecl(C))) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004297 ArrayRef<SourceLocation> Locs = ImportD->getIdentifierLocs();
4298 if (!Locs.empty())
4299 return cxloc::translateSourceRange(Ctx,
4300 SourceRange(Locs.front(), Locs.back()));
4301 }
4302 return clang_getNullRange();
4303 }
4304
Argyrios Kyrtzidisa2a1e532014-08-26 20:23:26 +00004305 if (C.kind == CXCursor_CXXMethod || C.kind == CXCursor_Destructor ||
4306 C.kind == CXCursor_ConversionFunction) {
4307 if (pieceIndex > 0)
4308 return clang_getNullRange();
4309 if (const FunctionDecl *FD =
4310 dyn_cast_or_null<FunctionDecl>(getCursorDecl(C))) {
4311 DeclarationNameInfo FunctionName = FD->getNameInfo();
4312 return cxloc::translateSourceRange(Ctx, FunctionName.getSourceRange());
4313 }
4314 return clang_getNullRange();
4315 }
4316
Guy Benyei11169dd2012-12-18 14:30:41 +00004317 // FIXME: A CXCursor_InclusionDirective should give the location of the
4318 // filename, but we don't keep track of this.
4319
4320 // FIXME: A CXCursor_AnnotateAttr should give the location of the annotation
4321 // but we don't keep track of this.
4322
4323 // FIXME: A CXCursor_AsmLabelAttr should give the location of the label
4324 // but we don't keep track of this.
4325
4326 // Default handling, give the location of the cursor.
4327
4328 if (pieceIndex > 0)
4329 return clang_getNullRange();
4330
4331 CXSourceLocation CXLoc = clang_getCursorLocation(C);
4332 SourceLocation Loc = cxloc::translateSourceLocation(CXLoc);
4333 return cxloc::translateSourceRange(Ctx, Loc);
4334}
4335
Eli Bendersky44a206f2014-07-31 18:04:56 +00004336CXString clang_Cursor_getMangling(CXCursor C) {
4337 if (clang_isInvalid(C.kind) || !clang_isDeclaration(C.kind))
4338 return cxstring::createEmpty();
4339
Eli Bendersky44a206f2014-07-31 18:04:56 +00004340 // Mangling only works for functions and variables.
Eli Bendersky79759592014-08-01 15:01:10 +00004341 const Decl *D = getCursorDecl(C);
Eli Bendersky44a206f2014-07-31 18:04:56 +00004342 if (!D || !(isa<FunctionDecl>(D) || isa<VarDecl>(D)))
4343 return cxstring::createEmpty();
4344
Argyrios Kyrtzidisca741ce2016-02-14 22:30:14 +00004345 ASTContext &Ctx = D->getASTContext();
4346 index::CodegenNameGenerator CGNameGen(Ctx);
4347 return cxstring::createDup(CGNameGen.getName(D));
Eli Bendersky44a206f2014-07-31 18:04:56 +00004348}
4349
Saleem Abdulrasool60034432015-11-12 03:57:22 +00004350CXStringSet *clang_Cursor_getCXXManglings(CXCursor C) {
4351 if (clang_isInvalid(C.kind) || !clang_isDeclaration(C.kind))
4352 return nullptr;
4353
4354 const Decl *D = getCursorDecl(C);
4355 if (!(isa<CXXRecordDecl>(D) || isa<CXXMethodDecl>(D)))
4356 return nullptr;
4357
Argyrios Kyrtzidisca741ce2016-02-14 22:30:14 +00004358 ASTContext &Ctx = D->getASTContext();
4359 index::CodegenNameGenerator CGNameGen(Ctx);
4360 std::vector<std::string> Manglings = CGNameGen.getAllManglings(D);
Saleem Abdulrasool60034432015-11-12 03:57:22 +00004361 return cxstring::createSet(Manglings);
4362}
4363
Guy Benyei11169dd2012-12-18 14:30:41 +00004364CXString clang_getCursorDisplayName(CXCursor C) {
4365 if (!clang_isDeclaration(C.kind))
4366 return clang_getCursorSpelling(C);
4367
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004368 const Decl *D = getCursorDecl(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00004369 if (!D)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00004370 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00004371
4372 PrintingPolicy Policy = getCursorContext(C).getPrintingPolicy();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004373 if (const FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00004374 D = FunTmpl->getTemplatedDecl();
4375
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004376 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004377 SmallString<64> Str;
4378 llvm::raw_svector_ostream OS(Str);
4379 OS << *Function;
4380 if (Function->getPrimaryTemplate())
4381 OS << "<>";
4382 OS << "(";
4383 for (unsigned I = 0, N = Function->getNumParams(); I != N; ++I) {
4384 if (I)
4385 OS << ", ";
4386 OS << Function->getParamDecl(I)->getType().getAsString(Policy);
4387 }
4388
4389 if (Function->isVariadic()) {
4390 if (Function->getNumParams())
4391 OS << ", ";
4392 OS << "...";
4393 }
4394 OS << ")";
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004395 return cxstring::createDup(OS.str());
Guy Benyei11169dd2012-12-18 14:30:41 +00004396 }
4397
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004398 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004399 SmallString<64> Str;
4400 llvm::raw_svector_ostream OS(Str);
4401 OS << *ClassTemplate;
4402 OS << "<";
4403 TemplateParameterList *Params = ClassTemplate->getTemplateParameters();
4404 for (unsigned I = 0, N = Params->size(); I != N; ++I) {
4405 if (I)
4406 OS << ", ";
4407
4408 NamedDecl *Param = Params->getParam(I);
4409 if (Param->getIdentifier()) {
4410 OS << Param->getIdentifier()->getName();
4411 continue;
4412 }
4413
4414 // There is no parameter name, which makes this tricky. Try to come up
4415 // with something useful that isn't too long.
4416 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
4417 OS << (TTP->wasDeclaredWithTypename()? "typename" : "class");
4418 else if (NonTypeTemplateParmDecl *NTTP
4419 = dyn_cast<NonTypeTemplateParmDecl>(Param))
4420 OS << NTTP->getType().getAsString(Policy);
4421 else
4422 OS << "template<...> class";
4423 }
4424
4425 OS << ">";
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004426 return cxstring::createDup(OS.str());
Guy Benyei11169dd2012-12-18 14:30:41 +00004427 }
4428
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004429 if (const ClassTemplateSpecializationDecl *ClassSpec
Guy Benyei11169dd2012-12-18 14:30:41 +00004430 = dyn_cast<ClassTemplateSpecializationDecl>(D)) {
4431 // If the type was explicitly written, use that.
4432 if (TypeSourceInfo *TSInfo = ClassSpec->getTypeAsWritten())
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004433 return cxstring::createDup(TSInfo->getType().getAsString(Policy));
Guy Benyei11169dd2012-12-18 14:30:41 +00004434
Benjamin Kramer9170e912013-02-22 15:46:01 +00004435 SmallString<128> Str;
Guy Benyei11169dd2012-12-18 14:30:41 +00004436 llvm::raw_svector_ostream OS(Str);
4437 OS << *ClassSpec;
Benjamin Kramer9170e912013-02-22 15:46:01 +00004438 TemplateSpecializationType::PrintTemplateArgumentList(OS,
Guy Benyei11169dd2012-12-18 14:30:41 +00004439 ClassSpec->getTemplateArgs().data(),
4440 ClassSpec->getTemplateArgs().size(),
4441 Policy);
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00004442 return cxstring::createDup(OS.str());
Guy Benyei11169dd2012-12-18 14:30:41 +00004443 }
4444
4445 return clang_getCursorSpelling(C);
4446}
4447
4448CXString clang_getCursorKindSpelling(enum CXCursorKind Kind) {
4449 switch (Kind) {
4450 case CXCursor_FunctionDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004451 return cxstring::createRef("FunctionDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004452 case CXCursor_TypedefDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004453 return cxstring::createRef("TypedefDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004454 case CXCursor_EnumDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004455 return cxstring::createRef("EnumDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004456 case CXCursor_EnumConstantDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004457 return cxstring::createRef("EnumConstantDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004458 case CXCursor_StructDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004459 return cxstring::createRef("StructDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004460 case CXCursor_UnionDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004461 return cxstring::createRef("UnionDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004462 case CXCursor_ClassDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004463 return cxstring::createRef("ClassDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004464 case CXCursor_FieldDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004465 return cxstring::createRef("FieldDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004466 case CXCursor_VarDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004467 return cxstring::createRef("VarDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004468 case CXCursor_ParmDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004469 return cxstring::createRef("ParmDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004470 case CXCursor_ObjCInterfaceDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004471 return cxstring::createRef("ObjCInterfaceDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004472 case CXCursor_ObjCCategoryDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004473 return cxstring::createRef("ObjCCategoryDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004474 case CXCursor_ObjCProtocolDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004475 return cxstring::createRef("ObjCProtocolDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004476 case CXCursor_ObjCPropertyDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004477 return cxstring::createRef("ObjCPropertyDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004478 case CXCursor_ObjCIvarDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004479 return cxstring::createRef("ObjCIvarDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004480 case CXCursor_ObjCInstanceMethodDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004481 return cxstring::createRef("ObjCInstanceMethodDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004482 case CXCursor_ObjCClassMethodDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004483 return cxstring::createRef("ObjCClassMethodDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004484 case CXCursor_ObjCImplementationDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004485 return cxstring::createRef("ObjCImplementationDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004486 case CXCursor_ObjCCategoryImplDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004487 return cxstring::createRef("ObjCCategoryImplDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004488 case CXCursor_CXXMethod:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004489 return cxstring::createRef("CXXMethod");
Guy Benyei11169dd2012-12-18 14:30:41 +00004490 case CXCursor_UnexposedDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004491 return cxstring::createRef("UnexposedDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004492 case CXCursor_ObjCSuperClassRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004493 return cxstring::createRef("ObjCSuperClassRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004494 case CXCursor_ObjCProtocolRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004495 return cxstring::createRef("ObjCProtocolRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004496 case CXCursor_ObjCClassRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004497 return cxstring::createRef("ObjCClassRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004498 case CXCursor_TypeRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004499 return cxstring::createRef("TypeRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004500 case CXCursor_TemplateRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004501 return cxstring::createRef("TemplateRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004502 case CXCursor_NamespaceRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004503 return cxstring::createRef("NamespaceRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004504 case CXCursor_MemberRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004505 return cxstring::createRef("MemberRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004506 case CXCursor_LabelRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004507 return cxstring::createRef("LabelRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004508 case CXCursor_OverloadedDeclRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004509 return cxstring::createRef("OverloadedDeclRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004510 case CXCursor_VariableRef:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004511 return cxstring::createRef("VariableRef");
Guy Benyei11169dd2012-12-18 14:30:41 +00004512 case CXCursor_IntegerLiteral:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004513 return cxstring::createRef("IntegerLiteral");
Guy Benyei11169dd2012-12-18 14:30:41 +00004514 case CXCursor_FloatingLiteral:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004515 return cxstring::createRef("FloatingLiteral");
Guy Benyei11169dd2012-12-18 14:30:41 +00004516 case CXCursor_ImaginaryLiteral:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004517 return cxstring::createRef("ImaginaryLiteral");
Guy Benyei11169dd2012-12-18 14:30:41 +00004518 case CXCursor_StringLiteral:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004519 return cxstring::createRef("StringLiteral");
Guy Benyei11169dd2012-12-18 14:30:41 +00004520 case CXCursor_CharacterLiteral:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004521 return cxstring::createRef("CharacterLiteral");
Guy Benyei11169dd2012-12-18 14:30:41 +00004522 case CXCursor_ParenExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004523 return cxstring::createRef("ParenExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004524 case CXCursor_UnaryOperator:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004525 return cxstring::createRef("UnaryOperator");
Guy Benyei11169dd2012-12-18 14:30:41 +00004526 case CXCursor_ArraySubscriptExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004527 return cxstring::createRef("ArraySubscriptExpr");
Alexey Bataev1a3320e2015-08-25 14:24:04 +00004528 case CXCursor_OMPArraySectionExpr:
4529 return cxstring::createRef("OMPArraySectionExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004530 case CXCursor_BinaryOperator:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004531 return cxstring::createRef("BinaryOperator");
Guy Benyei11169dd2012-12-18 14:30:41 +00004532 case CXCursor_CompoundAssignOperator:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004533 return cxstring::createRef("CompoundAssignOperator");
Guy Benyei11169dd2012-12-18 14:30:41 +00004534 case CXCursor_ConditionalOperator:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004535 return cxstring::createRef("ConditionalOperator");
Guy Benyei11169dd2012-12-18 14:30:41 +00004536 case CXCursor_CStyleCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004537 return cxstring::createRef("CStyleCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004538 case CXCursor_CompoundLiteralExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004539 return cxstring::createRef("CompoundLiteralExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004540 case CXCursor_InitListExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004541 return cxstring::createRef("InitListExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004542 case CXCursor_AddrLabelExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004543 return cxstring::createRef("AddrLabelExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004544 case CXCursor_StmtExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004545 return cxstring::createRef("StmtExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004546 case CXCursor_GenericSelectionExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004547 return cxstring::createRef("GenericSelectionExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004548 case CXCursor_GNUNullExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004549 return cxstring::createRef("GNUNullExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004550 case CXCursor_CXXStaticCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004551 return cxstring::createRef("CXXStaticCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004552 case CXCursor_CXXDynamicCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004553 return cxstring::createRef("CXXDynamicCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004554 case CXCursor_CXXReinterpretCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004555 return cxstring::createRef("CXXReinterpretCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004556 case CXCursor_CXXConstCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004557 return cxstring::createRef("CXXConstCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004558 case CXCursor_CXXFunctionalCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004559 return cxstring::createRef("CXXFunctionalCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004560 case CXCursor_CXXTypeidExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004561 return cxstring::createRef("CXXTypeidExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004562 case CXCursor_CXXBoolLiteralExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004563 return cxstring::createRef("CXXBoolLiteralExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004564 case CXCursor_CXXNullPtrLiteralExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004565 return cxstring::createRef("CXXNullPtrLiteralExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004566 case CXCursor_CXXThisExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004567 return cxstring::createRef("CXXThisExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004568 case CXCursor_CXXThrowExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004569 return cxstring::createRef("CXXThrowExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004570 case CXCursor_CXXNewExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004571 return cxstring::createRef("CXXNewExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004572 case CXCursor_CXXDeleteExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004573 return cxstring::createRef("CXXDeleteExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004574 case CXCursor_UnaryExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004575 return cxstring::createRef("UnaryExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004576 case CXCursor_ObjCStringLiteral:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004577 return cxstring::createRef("ObjCStringLiteral");
Guy Benyei11169dd2012-12-18 14:30:41 +00004578 case CXCursor_ObjCBoolLiteralExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004579 return cxstring::createRef("ObjCBoolLiteralExpr");
Argyrios Kyrtzidisc2233be2013-04-23 17:57:17 +00004580 case CXCursor_ObjCSelfExpr:
4581 return cxstring::createRef("ObjCSelfExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004582 case CXCursor_ObjCEncodeExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004583 return cxstring::createRef("ObjCEncodeExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004584 case CXCursor_ObjCSelectorExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004585 return cxstring::createRef("ObjCSelectorExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004586 case CXCursor_ObjCProtocolExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004587 return cxstring::createRef("ObjCProtocolExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004588 case CXCursor_ObjCBridgedCastExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004589 return cxstring::createRef("ObjCBridgedCastExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004590 case CXCursor_BlockExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004591 return cxstring::createRef("BlockExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004592 case CXCursor_PackExpansionExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004593 return cxstring::createRef("PackExpansionExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004594 case CXCursor_SizeOfPackExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004595 return cxstring::createRef("SizeOfPackExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004596 case CXCursor_LambdaExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004597 return cxstring::createRef("LambdaExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004598 case CXCursor_UnexposedExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004599 return cxstring::createRef("UnexposedExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004600 case CXCursor_DeclRefExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004601 return cxstring::createRef("DeclRefExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004602 case CXCursor_MemberRefExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004603 return cxstring::createRef("MemberRefExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004604 case CXCursor_CallExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004605 return cxstring::createRef("CallExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004606 case CXCursor_ObjCMessageExpr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004607 return cxstring::createRef("ObjCMessageExpr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004608 case CXCursor_UnexposedStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004609 return cxstring::createRef("UnexposedStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004610 case CXCursor_DeclStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004611 return cxstring::createRef("DeclStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004612 case CXCursor_LabelStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004613 return cxstring::createRef("LabelStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004614 case CXCursor_CompoundStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004615 return cxstring::createRef("CompoundStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004616 case CXCursor_CaseStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004617 return cxstring::createRef("CaseStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004618 case CXCursor_DefaultStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004619 return cxstring::createRef("DefaultStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004620 case CXCursor_IfStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004621 return cxstring::createRef("IfStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004622 case CXCursor_SwitchStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004623 return cxstring::createRef("SwitchStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004624 case CXCursor_WhileStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004625 return cxstring::createRef("WhileStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004626 case CXCursor_DoStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004627 return cxstring::createRef("DoStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004628 case CXCursor_ForStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004629 return cxstring::createRef("ForStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004630 case CXCursor_GotoStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004631 return cxstring::createRef("GotoStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004632 case CXCursor_IndirectGotoStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004633 return cxstring::createRef("IndirectGotoStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004634 case CXCursor_ContinueStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004635 return cxstring::createRef("ContinueStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004636 case CXCursor_BreakStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004637 return cxstring::createRef("BreakStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004638 case CXCursor_ReturnStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004639 return cxstring::createRef("ReturnStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004640 case CXCursor_GCCAsmStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004641 return cxstring::createRef("GCCAsmStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004642 case CXCursor_MSAsmStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004643 return cxstring::createRef("MSAsmStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004644 case CXCursor_ObjCAtTryStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004645 return cxstring::createRef("ObjCAtTryStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004646 case CXCursor_ObjCAtCatchStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004647 return cxstring::createRef("ObjCAtCatchStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004648 case CXCursor_ObjCAtFinallyStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004649 return cxstring::createRef("ObjCAtFinallyStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004650 case CXCursor_ObjCAtThrowStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004651 return cxstring::createRef("ObjCAtThrowStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004652 case CXCursor_ObjCAtSynchronizedStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004653 return cxstring::createRef("ObjCAtSynchronizedStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004654 case CXCursor_ObjCAutoreleasePoolStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004655 return cxstring::createRef("ObjCAutoreleasePoolStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004656 case CXCursor_ObjCForCollectionStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004657 return cxstring::createRef("ObjCForCollectionStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004658 case CXCursor_CXXCatchStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004659 return cxstring::createRef("CXXCatchStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004660 case CXCursor_CXXTryStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004661 return cxstring::createRef("CXXTryStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004662 case CXCursor_CXXForRangeStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004663 return cxstring::createRef("CXXForRangeStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004664 case CXCursor_SEHTryStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004665 return cxstring::createRef("SEHTryStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004666 case CXCursor_SEHExceptStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004667 return cxstring::createRef("SEHExceptStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004668 case CXCursor_SEHFinallyStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004669 return cxstring::createRef("SEHFinallyStmt");
Nico Weber9b982072014-07-07 00:12:30 +00004670 case CXCursor_SEHLeaveStmt:
4671 return cxstring::createRef("SEHLeaveStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004672 case CXCursor_NullStmt:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004673 return cxstring::createRef("NullStmt");
Guy Benyei11169dd2012-12-18 14:30:41 +00004674 case CXCursor_InvalidFile:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004675 return cxstring::createRef("InvalidFile");
Guy Benyei11169dd2012-12-18 14:30:41 +00004676 case CXCursor_InvalidCode:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004677 return cxstring::createRef("InvalidCode");
Guy Benyei11169dd2012-12-18 14:30:41 +00004678 case CXCursor_NoDeclFound:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004679 return cxstring::createRef("NoDeclFound");
Guy Benyei11169dd2012-12-18 14:30:41 +00004680 case CXCursor_NotImplemented:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004681 return cxstring::createRef("NotImplemented");
Guy Benyei11169dd2012-12-18 14:30:41 +00004682 case CXCursor_TranslationUnit:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004683 return cxstring::createRef("TranslationUnit");
Guy Benyei11169dd2012-12-18 14:30:41 +00004684 case CXCursor_UnexposedAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004685 return cxstring::createRef("UnexposedAttr");
Guy Benyei11169dd2012-12-18 14:30:41 +00004686 case CXCursor_IBActionAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004687 return cxstring::createRef("attribute(ibaction)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004688 case CXCursor_IBOutletAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004689 return cxstring::createRef("attribute(iboutlet)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004690 case CXCursor_IBOutletCollectionAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004691 return cxstring::createRef("attribute(iboutletcollection)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004692 case CXCursor_CXXFinalAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004693 return cxstring::createRef("attribute(final)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004694 case CXCursor_CXXOverrideAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004695 return cxstring::createRef("attribute(override)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004696 case CXCursor_AnnotateAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004697 return cxstring::createRef("attribute(annotate)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004698 case CXCursor_AsmLabelAttr:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004699 return cxstring::createRef("asm label");
Argyrios Kyrtzidis16834f12013-09-25 00:14:38 +00004700 case CXCursor_PackedAttr:
4701 return cxstring::createRef("attribute(packed)");
Joey Gouly81228382014-05-01 15:41:58 +00004702 case CXCursor_PureAttr:
4703 return cxstring::createRef("attribute(pure)");
4704 case CXCursor_ConstAttr:
4705 return cxstring::createRef("attribute(const)");
4706 case CXCursor_NoDuplicateAttr:
4707 return cxstring::createRef("attribute(noduplicate)");
Eli Bendersky2581e662014-05-28 19:29:58 +00004708 case CXCursor_CUDAConstantAttr:
4709 return cxstring::createRef("attribute(constant)");
4710 case CXCursor_CUDADeviceAttr:
4711 return cxstring::createRef("attribute(device)");
4712 case CXCursor_CUDAGlobalAttr:
4713 return cxstring::createRef("attribute(global)");
4714 case CXCursor_CUDAHostAttr:
4715 return cxstring::createRef("attribute(host)");
Eli Bendersky9b071472014-08-08 14:59:00 +00004716 case CXCursor_CUDASharedAttr:
4717 return cxstring::createRef("attribute(shared)");
Saleem Abdulrasool79c69712015-09-05 18:53:43 +00004718 case CXCursor_VisibilityAttr:
4719 return cxstring::createRef("attribute(visibility)");
Saleem Abdulrasool8aa0b802015-12-10 18:45:18 +00004720 case CXCursor_DLLExport:
4721 return cxstring::createRef("attribute(dllexport)");
4722 case CXCursor_DLLImport:
4723 return cxstring::createRef("attribute(dllimport)");
Guy Benyei11169dd2012-12-18 14:30:41 +00004724 case CXCursor_PreprocessingDirective:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004725 return cxstring::createRef("preprocessing directive");
Guy Benyei11169dd2012-12-18 14:30:41 +00004726 case CXCursor_MacroDefinition:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004727 return cxstring::createRef("macro definition");
Guy Benyei11169dd2012-12-18 14:30:41 +00004728 case CXCursor_MacroExpansion:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004729 return cxstring::createRef("macro expansion");
Guy Benyei11169dd2012-12-18 14:30:41 +00004730 case CXCursor_InclusionDirective:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004731 return cxstring::createRef("inclusion directive");
Guy Benyei11169dd2012-12-18 14:30:41 +00004732 case CXCursor_Namespace:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004733 return cxstring::createRef("Namespace");
Guy Benyei11169dd2012-12-18 14:30:41 +00004734 case CXCursor_LinkageSpec:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004735 return cxstring::createRef("LinkageSpec");
Guy Benyei11169dd2012-12-18 14:30:41 +00004736 case CXCursor_CXXBaseSpecifier:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004737 return cxstring::createRef("C++ base class specifier");
Guy Benyei11169dd2012-12-18 14:30:41 +00004738 case CXCursor_Constructor:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004739 return cxstring::createRef("CXXConstructor");
Guy Benyei11169dd2012-12-18 14:30:41 +00004740 case CXCursor_Destructor:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004741 return cxstring::createRef("CXXDestructor");
Guy Benyei11169dd2012-12-18 14:30:41 +00004742 case CXCursor_ConversionFunction:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004743 return cxstring::createRef("CXXConversion");
Guy Benyei11169dd2012-12-18 14:30:41 +00004744 case CXCursor_TemplateTypeParameter:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004745 return cxstring::createRef("TemplateTypeParameter");
Guy Benyei11169dd2012-12-18 14:30:41 +00004746 case CXCursor_NonTypeTemplateParameter:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004747 return cxstring::createRef("NonTypeTemplateParameter");
Guy Benyei11169dd2012-12-18 14:30:41 +00004748 case CXCursor_TemplateTemplateParameter:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004749 return cxstring::createRef("TemplateTemplateParameter");
Guy Benyei11169dd2012-12-18 14:30:41 +00004750 case CXCursor_FunctionTemplate:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004751 return cxstring::createRef("FunctionTemplate");
Guy Benyei11169dd2012-12-18 14:30:41 +00004752 case CXCursor_ClassTemplate:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004753 return cxstring::createRef("ClassTemplate");
Guy Benyei11169dd2012-12-18 14:30:41 +00004754 case CXCursor_ClassTemplatePartialSpecialization:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004755 return cxstring::createRef("ClassTemplatePartialSpecialization");
Guy Benyei11169dd2012-12-18 14:30:41 +00004756 case CXCursor_NamespaceAlias:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004757 return cxstring::createRef("NamespaceAlias");
Guy Benyei11169dd2012-12-18 14:30:41 +00004758 case CXCursor_UsingDirective:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004759 return cxstring::createRef("UsingDirective");
Guy Benyei11169dd2012-12-18 14:30:41 +00004760 case CXCursor_UsingDeclaration:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004761 return cxstring::createRef("UsingDeclaration");
Guy Benyei11169dd2012-12-18 14:30:41 +00004762 case CXCursor_TypeAliasDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004763 return cxstring::createRef("TypeAliasDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004764 case CXCursor_ObjCSynthesizeDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004765 return cxstring::createRef("ObjCSynthesizeDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004766 case CXCursor_ObjCDynamicDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004767 return cxstring::createRef("ObjCDynamicDecl");
Guy Benyei11169dd2012-12-18 14:30:41 +00004768 case CXCursor_CXXAccessSpecifier:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004769 return cxstring::createRef("CXXAccessSpecifier");
Guy Benyei11169dd2012-12-18 14:30:41 +00004770 case CXCursor_ModuleImportDecl:
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00004771 return cxstring::createRef("ModuleImport");
Alexey Bataev5ec3eb12013-07-19 03:13:43 +00004772 case CXCursor_OMPParallelDirective:
Alexey Bataev1b59ab52014-02-27 08:29:12 +00004773 return cxstring::createRef("OMPParallelDirective");
4774 case CXCursor_OMPSimdDirective:
4775 return cxstring::createRef("OMPSimdDirective");
Alexey Bataevf29276e2014-06-18 04:14:57 +00004776 case CXCursor_OMPForDirective:
4777 return cxstring::createRef("OMPForDirective");
Alexander Musmanf82886e2014-09-18 05:12:34 +00004778 case CXCursor_OMPForSimdDirective:
4779 return cxstring::createRef("OMPForSimdDirective");
Alexey Bataevd3f8dd22014-06-25 11:44:49 +00004780 case CXCursor_OMPSectionsDirective:
4781 return cxstring::createRef("OMPSectionsDirective");
Alexey Bataev1e0498a2014-06-26 08:21:58 +00004782 case CXCursor_OMPSectionDirective:
4783 return cxstring::createRef("OMPSectionDirective");
Alexey Bataevd1e40fb2014-06-26 12:05:45 +00004784 case CXCursor_OMPSingleDirective:
4785 return cxstring::createRef("OMPSingleDirective");
Alexander Musman80c22892014-07-17 08:54:58 +00004786 case CXCursor_OMPMasterDirective:
4787 return cxstring::createRef("OMPMasterDirective");
Alexander Musmand9ed09f2014-07-21 09:42:05 +00004788 case CXCursor_OMPCriticalDirective:
4789 return cxstring::createRef("OMPCriticalDirective");
Alexey Bataev4acb8592014-07-07 13:01:15 +00004790 case CXCursor_OMPParallelForDirective:
4791 return cxstring::createRef("OMPParallelForDirective");
Alexander Musmane4e893b2014-09-23 09:33:00 +00004792 case CXCursor_OMPParallelForSimdDirective:
4793 return cxstring::createRef("OMPParallelForSimdDirective");
Alexey Bataev84d0b3e2014-07-08 08:12:03 +00004794 case CXCursor_OMPParallelSectionsDirective:
4795 return cxstring::createRef("OMPParallelSectionsDirective");
Alexey Bataev9c2e8ee2014-07-11 11:25:16 +00004796 case CXCursor_OMPTaskDirective:
4797 return cxstring::createRef("OMPTaskDirective");
Alexey Bataev68446b72014-07-18 07:47:19 +00004798 case CXCursor_OMPTaskyieldDirective:
4799 return cxstring::createRef("OMPTaskyieldDirective");
Alexey Bataev4d1dfea2014-07-18 09:11:51 +00004800 case CXCursor_OMPBarrierDirective:
4801 return cxstring::createRef("OMPBarrierDirective");
Alexey Bataev2df347a2014-07-18 10:17:07 +00004802 case CXCursor_OMPTaskwaitDirective:
4803 return cxstring::createRef("OMPTaskwaitDirective");
Alexey Bataevc30dd2d2015-06-18 12:14:09 +00004804 case CXCursor_OMPTaskgroupDirective:
4805 return cxstring::createRef("OMPTaskgroupDirective");
Alexey Bataev6125da92014-07-21 11:26:11 +00004806 case CXCursor_OMPFlushDirective:
4807 return cxstring::createRef("OMPFlushDirective");
Alexey Bataev9fb6e642014-07-22 06:45:04 +00004808 case CXCursor_OMPOrderedDirective:
4809 return cxstring::createRef("OMPOrderedDirective");
Alexey Bataev0162e452014-07-22 10:10:35 +00004810 case CXCursor_OMPAtomicDirective:
4811 return cxstring::createRef("OMPAtomicDirective");
Alexey Bataev0bd520b2014-09-19 08:19:49 +00004812 case CXCursor_OMPTargetDirective:
4813 return cxstring::createRef("OMPTargetDirective");
Michael Wong65f367f2015-07-21 13:44:28 +00004814 case CXCursor_OMPTargetDataDirective:
4815 return cxstring::createRef("OMPTargetDataDirective");
Samuel Antaodf67fc42016-01-19 19:15:56 +00004816 case CXCursor_OMPTargetEnterDataDirective:
4817 return cxstring::createRef("OMPTargetEnterDataDirective");
Samuel Antao72590762016-01-19 20:04:50 +00004818 case CXCursor_OMPTargetExitDataDirective:
4819 return cxstring::createRef("OMPTargetExitDataDirective");
Arpith Chacko Jacobe955b3d2016-01-26 18:48:41 +00004820 case CXCursor_OMPTargetParallelDirective:
4821 return cxstring::createRef("OMPTargetParallelDirective");
Arpith Chacko Jacob05bebb52016-02-03 15:46:42 +00004822 case CXCursor_OMPTargetParallelForDirective:
4823 return cxstring::createRef("OMPTargetParallelForDirective");
Samuel Antao686c70c2016-05-26 17:30:50 +00004824 case CXCursor_OMPTargetUpdateDirective:
4825 return cxstring::createRef("OMPTargetUpdateDirective");
Alexey Bataev13314bf2014-10-09 04:18:56 +00004826 case CXCursor_OMPTeamsDirective:
4827 return cxstring::createRef("OMPTeamsDirective");
Alexey Bataev6d4ed052015-07-01 06:57:41 +00004828 case CXCursor_OMPCancellationPointDirective:
4829 return cxstring::createRef("OMPCancellationPointDirective");
Alexey Bataev80909872015-07-02 11:25:17 +00004830 case CXCursor_OMPCancelDirective:
4831 return cxstring::createRef("OMPCancelDirective");
Alexey Bataev49f6e782015-12-01 04:18:41 +00004832 case CXCursor_OMPTaskLoopDirective:
4833 return cxstring::createRef("OMPTaskLoopDirective");
Alexey Bataev0a6ed842015-12-03 09:40:15 +00004834 case CXCursor_OMPTaskLoopSimdDirective:
4835 return cxstring::createRef("OMPTaskLoopSimdDirective");
Carlo Bertolli6200a3d2015-12-14 14:51:25 +00004836 case CXCursor_OMPDistributeDirective:
4837 return cxstring::createRef("OMPDistributeDirective");
Carlo Bertolli9925f152016-06-27 14:55:37 +00004838 case CXCursor_OMPDistributeParallelForDirective:
4839 return cxstring::createRef("OMPDistributeParallelForDirective");
Francisco Lopes da Silva975a9f62015-01-21 16:24:11 +00004840 case CXCursor_OverloadCandidate:
4841 return cxstring::createRef("OverloadCandidate");
Sergey Kalinichev8f3b1872015-11-15 13:48:32 +00004842 case CXCursor_TypeAliasTemplateDecl:
4843 return cxstring::createRef("TypeAliasTemplateDecl");
Olivier Goffart81978012016-06-09 16:15:55 +00004844 case CXCursor_StaticAssert:
4845 return cxstring::createRef("StaticAssert");
Guy Benyei11169dd2012-12-18 14:30:41 +00004846 }
4847
4848 llvm_unreachable("Unhandled CXCursorKind");
4849}
4850
4851struct GetCursorData {
4852 SourceLocation TokenBeginLoc;
4853 bool PointsAtMacroArgExpansion;
4854 bool VisitedObjCPropertyImplDecl;
4855 SourceLocation VisitedDeclaratorDeclStartLoc;
4856 CXCursor &BestCursor;
4857
4858 GetCursorData(SourceManager &SM,
4859 SourceLocation tokenBegin, CXCursor &outputCursor)
4860 : TokenBeginLoc(tokenBegin), BestCursor(outputCursor) {
4861 PointsAtMacroArgExpansion = SM.isMacroArgExpansion(tokenBegin);
4862 VisitedObjCPropertyImplDecl = false;
4863 }
4864};
4865
4866static enum CXChildVisitResult GetCursorVisitor(CXCursor cursor,
4867 CXCursor parent,
4868 CXClientData client_data) {
4869 GetCursorData *Data = static_cast<GetCursorData *>(client_data);
4870 CXCursor *BestCursor = &Data->BestCursor;
4871
4872 // If we point inside a macro argument we should provide info of what the
4873 // token is so use the actual cursor, don't replace it with a macro expansion
4874 // cursor.
4875 if (cursor.kind == CXCursor_MacroExpansion && Data->PointsAtMacroArgExpansion)
4876 return CXChildVisit_Recurse;
4877
4878 if (clang_isDeclaration(cursor.kind)) {
4879 // Avoid having the implicit methods override the property decls.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004880 if (const ObjCMethodDecl *MD
Guy Benyei11169dd2012-12-18 14:30:41 +00004881 = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(cursor))) {
4882 if (MD->isImplicit())
4883 return CXChildVisit_Break;
4884
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004885 } else if (const ObjCInterfaceDecl *ID
Guy Benyei11169dd2012-12-18 14:30:41 +00004886 = dyn_cast_or_null<ObjCInterfaceDecl>(getCursorDecl(cursor))) {
4887 // Check that when we have multiple @class references in the same line,
4888 // that later ones do not override the previous ones.
4889 // If we have:
4890 // @class Foo, Bar;
4891 // source ranges for both start at '@', so 'Bar' will end up overriding
4892 // 'Foo' even though the cursor location was at 'Foo'.
4893 if (BestCursor->kind == CXCursor_ObjCInterfaceDecl ||
4894 BestCursor->kind == CXCursor_ObjCClassRef)
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004895 if (const ObjCInterfaceDecl *PrevID
Guy Benyei11169dd2012-12-18 14:30:41 +00004896 = dyn_cast_or_null<ObjCInterfaceDecl>(getCursorDecl(*BestCursor))){
4897 if (PrevID != ID &&
4898 !PrevID->isThisDeclarationADefinition() &&
4899 !ID->isThisDeclarationADefinition())
4900 return CXChildVisit_Break;
4901 }
4902
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004903 } else if (const DeclaratorDecl *DD
Guy Benyei11169dd2012-12-18 14:30:41 +00004904 = dyn_cast_or_null<DeclaratorDecl>(getCursorDecl(cursor))) {
4905 SourceLocation StartLoc = DD->getSourceRange().getBegin();
4906 // Check that when we have multiple declarators in the same line,
4907 // that later ones do not override the previous ones.
4908 // If we have:
4909 // int Foo, Bar;
4910 // source ranges for both start at 'int', so 'Bar' will end up overriding
4911 // 'Foo' even though the cursor location was at 'Foo'.
4912 if (Data->VisitedDeclaratorDeclStartLoc == StartLoc)
4913 return CXChildVisit_Break;
4914 Data->VisitedDeclaratorDeclStartLoc = StartLoc;
4915
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004916 } else if (const ObjCPropertyImplDecl *PropImp
Guy Benyei11169dd2012-12-18 14:30:41 +00004917 = dyn_cast_or_null<ObjCPropertyImplDecl>(getCursorDecl(cursor))) {
4918 (void)PropImp;
4919 // Check that when we have multiple @synthesize in the same line,
4920 // that later ones do not override the previous ones.
4921 // If we have:
4922 // @synthesize Foo, Bar;
4923 // source ranges for both start at '@', so 'Bar' will end up overriding
4924 // 'Foo' even though the cursor location was at 'Foo'.
4925 if (Data->VisitedObjCPropertyImplDecl)
4926 return CXChildVisit_Break;
4927 Data->VisitedObjCPropertyImplDecl = true;
4928 }
4929 }
4930
4931 if (clang_isExpression(cursor.kind) &&
4932 clang_isDeclaration(BestCursor->kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00004933 if (const Decl *D = getCursorDecl(*BestCursor)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00004934 // Avoid having the cursor of an expression replace the declaration cursor
4935 // when the expression source range overlaps the declaration range.
4936 // This can happen for C++ constructor expressions whose range generally
4937 // include the variable declaration, e.g.:
4938 // MyCXXClass foo; // Make sure pointing at 'foo' returns a VarDecl cursor.
4939 if (D->getLocation().isValid() && Data->TokenBeginLoc.isValid() &&
4940 D->getLocation() == Data->TokenBeginLoc)
4941 return CXChildVisit_Break;
4942 }
4943 }
4944
4945 // If our current best cursor is the construction of a temporary object,
4946 // don't replace that cursor with a type reference, because we want
4947 // clang_getCursor() to point at the constructor.
4948 if (clang_isExpression(BestCursor->kind) &&
4949 isa<CXXTemporaryObjectExpr>(getCursorExpr(*BestCursor)) &&
4950 cursor.kind == CXCursor_TypeRef) {
4951 // Keep the cursor pointing at CXXTemporaryObjectExpr but also mark it
4952 // as having the actual point on the type reference.
4953 *BestCursor = getTypeRefedCallExprCursor(*BestCursor);
4954 return CXChildVisit_Recurse;
4955 }
Douglas Gregore9d95f12015-07-07 03:57:35 +00004956
4957 // If we already have an Objective-C superclass reference, don't
4958 // update it further.
4959 if (BestCursor->kind == CXCursor_ObjCSuperClassRef)
4960 return CXChildVisit_Break;
4961
Guy Benyei11169dd2012-12-18 14:30:41 +00004962 *BestCursor = cursor;
4963 return CXChildVisit_Recurse;
4964}
4965
4966CXCursor clang_getCursor(CXTranslationUnit TU, CXSourceLocation Loc) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00004967 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00004968 LOG_BAD_TU(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00004969 return clang_getNullCursor();
Dmitri Gribenko256454f2014-02-11 14:34:14 +00004970 }
Guy Benyei11169dd2012-12-18 14:30:41 +00004971
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00004972 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00004973 ASTUnit::ConcurrencyCheck Check(*CXXUnit);
4974
4975 SourceLocation SLoc = cxloc::translateSourceLocation(Loc);
4976 CXCursor Result = cxcursor::getCursor(TU, SLoc);
4977
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00004978 LOG_FUNC_SECTION {
Guy Benyei11169dd2012-12-18 14:30:41 +00004979 CXFile SearchFile;
4980 unsigned SearchLine, SearchColumn;
4981 CXFile ResultFile;
4982 unsigned ResultLine, ResultColumn;
4983 CXString SearchFileName, ResultFileName, KindSpelling, USR;
4984 const char *IsDef = clang_isCursorDefinition(Result)? " (Definition)" : "";
4985 CXSourceLocation ResultLoc = clang_getCursorLocation(Result);
Craig Topper69186e72014-06-08 08:38:04 +00004986
4987 clang_getFileLocation(Loc, &SearchFile, &SearchLine, &SearchColumn,
4988 nullptr);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00004989 clang_getFileLocation(ResultLoc, &ResultFile, &ResultLine,
Craig Topper69186e72014-06-08 08:38:04 +00004990 &ResultColumn, nullptr);
Guy Benyei11169dd2012-12-18 14:30:41 +00004991 SearchFileName = clang_getFileName(SearchFile);
4992 ResultFileName = clang_getFileName(ResultFile);
4993 KindSpelling = clang_getCursorKindSpelling(Result.kind);
4994 USR = clang_getCursorUSR(Result);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00004995 *Log << llvm::format("(%s:%d:%d) = %s",
4996 clang_getCString(SearchFileName), SearchLine, SearchColumn,
4997 clang_getCString(KindSpelling))
4998 << llvm::format("(%s:%d:%d):%s%s",
4999 clang_getCString(ResultFileName), ResultLine, ResultColumn,
5000 clang_getCString(USR), IsDef);
Guy Benyei11169dd2012-12-18 14:30:41 +00005001 clang_disposeString(SearchFileName);
5002 clang_disposeString(ResultFileName);
5003 clang_disposeString(KindSpelling);
5004 clang_disposeString(USR);
5005
5006 CXCursor Definition = clang_getCursorDefinition(Result);
5007 if (!clang_equalCursors(Definition, clang_getNullCursor())) {
5008 CXSourceLocation DefinitionLoc = clang_getCursorLocation(Definition);
5009 CXString DefinitionKindSpelling
5010 = clang_getCursorKindSpelling(Definition.kind);
5011 CXFile DefinitionFile;
5012 unsigned DefinitionLine, DefinitionColumn;
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00005013 clang_getFileLocation(DefinitionLoc, &DefinitionFile,
Craig Topper69186e72014-06-08 08:38:04 +00005014 &DefinitionLine, &DefinitionColumn, nullptr);
Guy Benyei11169dd2012-12-18 14:30:41 +00005015 CXString DefinitionFileName = clang_getFileName(DefinitionFile);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00005016 *Log << llvm::format(" -> %s(%s:%d:%d)",
5017 clang_getCString(DefinitionKindSpelling),
5018 clang_getCString(DefinitionFileName),
5019 DefinitionLine, DefinitionColumn);
Guy Benyei11169dd2012-12-18 14:30:41 +00005020 clang_disposeString(DefinitionFileName);
5021 clang_disposeString(DefinitionKindSpelling);
5022 }
5023 }
5024
5025 return Result;
5026}
5027
5028CXCursor clang_getNullCursor(void) {
5029 return MakeCXCursorInvalid(CXCursor_InvalidFile);
5030}
5031
5032unsigned clang_equalCursors(CXCursor X, CXCursor Y) {
Argyrios Kyrtzidisbf1be592013-01-08 18:23:28 +00005033 // Clear out the "FirstInDeclGroup" part in a declaration cursor, since we
5034 // can't set consistently. For example, when visiting a DeclStmt we will set
5035 // it but we don't set it on the result of clang_getCursorDefinition for
5036 // a reference of the same declaration.
5037 // FIXME: Setting "FirstInDeclGroup" in CXCursors is a hack that only works
5038 // when visiting a DeclStmt currently, the AST should be enhanced to be able
5039 // to provide that kind of info.
5040 if (clang_isDeclaration(X.kind))
Craig Topper69186e72014-06-08 08:38:04 +00005041 X.data[1] = nullptr;
Argyrios Kyrtzidisbf1be592013-01-08 18:23:28 +00005042 if (clang_isDeclaration(Y.kind))
Craig Topper69186e72014-06-08 08:38:04 +00005043 Y.data[1] = nullptr;
Argyrios Kyrtzidisbf1be592013-01-08 18:23:28 +00005044
Guy Benyei11169dd2012-12-18 14:30:41 +00005045 return X == Y;
5046}
5047
5048unsigned clang_hashCursor(CXCursor C) {
5049 unsigned Index = 0;
5050 if (clang_isExpression(C.kind) || clang_isStatement(C.kind))
5051 Index = 1;
5052
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005053 return llvm::DenseMapInfo<std::pair<unsigned, const void*> >::getHashValue(
Guy Benyei11169dd2012-12-18 14:30:41 +00005054 std::make_pair(C.kind, C.data[Index]));
5055}
5056
5057unsigned clang_isInvalid(enum CXCursorKind K) {
5058 return K >= CXCursor_FirstInvalid && K <= CXCursor_LastInvalid;
5059}
5060
5061unsigned clang_isDeclaration(enum CXCursorKind K) {
5062 return (K >= CXCursor_FirstDecl && K <= CXCursor_LastDecl) ||
5063 (K >= CXCursor_FirstExtraDecl && K <= CXCursor_LastExtraDecl);
5064}
5065
5066unsigned clang_isReference(enum CXCursorKind K) {
5067 return K >= CXCursor_FirstRef && K <= CXCursor_LastRef;
5068}
5069
5070unsigned clang_isExpression(enum CXCursorKind K) {
5071 return K >= CXCursor_FirstExpr && K <= CXCursor_LastExpr;
5072}
5073
5074unsigned clang_isStatement(enum CXCursorKind K) {
5075 return K >= CXCursor_FirstStmt && K <= CXCursor_LastStmt;
5076}
5077
5078unsigned clang_isAttribute(enum CXCursorKind K) {
5079 return K >= CXCursor_FirstAttr && K <= CXCursor_LastAttr;
5080}
5081
5082unsigned clang_isTranslationUnit(enum CXCursorKind K) {
5083 return K == CXCursor_TranslationUnit;
5084}
5085
5086unsigned clang_isPreprocessing(enum CXCursorKind K) {
5087 return K >= CXCursor_FirstPreprocessing && K <= CXCursor_LastPreprocessing;
5088}
5089
5090unsigned clang_isUnexposed(enum CXCursorKind K) {
5091 switch (K) {
5092 case CXCursor_UnexposedDecl:
5093 case CXCursor_UnexposedExpr:
5094 case CXCursor_UnexposedStmt:
5095 case CXCursor_UnexposedAttr:
5096 return true;
5097 default:
5098 return false;
5099 }
5100}
5101
5102CXCursorKind clang_getCursorKind(CXCursor C) {
5103 return C.kind;
5104}
5105
5106CXSourceLocation clang_getCursorLocation(CXCursor C) {
5107 if (clang_isReference(C.kind)) {
5108 switch (C.kind) {
5109 case CXCursor_ObjCSuperClassRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005110 std::pair<const ObjCInterfaceDecl *, SourceLocation> P
Guy Benyei11169dd2012-12-18 14:30:41 +00005111 = getCursorObjCSuperClassRef(C);
5112 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5113 }
5114
5115 case CXCursor_ObjCProtocolRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005116 std::pair<const ObjCProtocolDecl *, SourceLocation> P
Guy Benyei11169dd2012-12-18 14:30:41 +00005117 = getCursorObjCProtocolRef(C);
5118 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5119 }
5120
5121 case CXCursor_ObjCClassRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005122 std::pair<const ObjCInterfaceDecl *, SourceLocation> P
Guy Benyei11169dd2012-12-18 14:30:41 +00005123 = getCursorObjCClassRef(C);
5124 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5125 }
5126
5127 case CXCursor_TypeRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005128 std::pair<const TypeDecl *, SourceLocation> P = getCursorTypeRef(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005129 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5130 }
5131
5132 case CXCursor_TemplateRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005133 std::pair<const TemplateDecl *, SourceLocation> P =
5134 getCursorTemplateRef(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005135 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5136 }
5137
5138 case CXCursor_NamespaceRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005139 std::pair<const NamedDecl *, SourceLocation> P = getCursorNamespaceRef(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005140 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5141 }
5142
5143 case CXCursor_MemberRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005144 std::pair<const FieldDecl *, SourceLocation> P = getCursorMemberRef(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005145 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5146 }
5147
5148 case CXCursor_VariableRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005149 std::pair<const VarDecl *, SourceLocation> P = getCursorVariableRef(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005150 return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
5151 }
5152
5153 case CXCursor_CXXBaseSpecifier: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005154 const CXXBaseSpecifier *BaseSpec = getCursorCXXBaseSpecifier(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005155 if (!BaseSpec)
5156 return clang_getNullLocation();
5157
5158 if (TypeSourceInfo *TSInfo = BaseSpec->getTypeSourceInfo())
5159 return cxloc::translateSourceLocation(getCursorContext(C),
5160 TSInfo->getTypeLoc().getBeginLoc());
5161
5162 return cxloc::translateSourceLocation(getCursorContext(C),
5163 BaseSpec->getLocStart());
5164 }
5165
5166 case CXCursor_LabelRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005167 std::pair<const LabelStmt *, SourceLocation> P = getCursorLabelRef(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005168 return cxloc::translateSourceLocation(getCursorContext(C), P.second);
5169 }
5170
5171 case CXCursor_OverloadedDeclRef:
5172 return cxloc::translateSourceLocation(getCursorContext(C),
5173 getCursorOverloadedDeclRef(C).second);
5174
5175 default:
5176 // FIXME: Need a way to enumerate all non-reference cases.
5177 llvm_unreachable("Missed a reference kind");
5178 }
5179 }
5180
5181 if (clang_isExpression(C.kind))
5182 return cxloc::translateSourceLocation(getCursorContext(C),
5183 getLocationFromExpr(getCursorExpr(C)));
5184
5185 if (clang_isStatement(C.kind))
5186 return cxloc::translateSourceLocation(getCursorContext(C),
5187 getCursorStmt(C)->getLocStart());
5188
5189 if (C.kind == CXCursor_PreprocessingDirective) {
5190 SourceLocation L = cxcursor::getCursorPreprocessingDirective(C).getBegin();
5191 return cxloc::translateSourceLocation(getCursorContext(C), L);
5192 }
5193
5194 if (C.kind == CXCursor_MacroExpansion) {
5195 SourceLocation L
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00005196 = cxcursor::getCursorMacroExpansion(C).getSourceRange().getBegin();
Guy Benyei11169dd2012-12-18 14:30:41 +00005197 return cxloc::translateSourceLocation(getCursorContext(C), L);
5198 }
5199
5200 if (C.kind == CXCursor_MacroDefinition) {
5201 SourceLocation L = cxcursor::getCursorMacroDefinition(C)->getLocation();
5202 return cxloc::translateSourceLocation(getCursorContext(C), L);
5203 }
5204
5205 if (C.kind == CXCursor_InclusionDirective) {
5206 SourceLocation L
5207 = cxcursor::getCursorInclusionDirective(C)->getSourceRange().getBegin();
5208 return cxloc::translateSourceLocation(getCursorContext(C), L);
5209 }
5210
Argyrios Kyrtzidis16834f12013-09-25 00:14:38 +00005211 if (clang_isAttribute(C.kind)) {
5212 SourceLocation L
5213 = cxcursor::getCursorAttr(C)->getLocation();
5214 return cxloc::translateSourceLocation(getCursorContext(C), L);
5215 }
5216
Guy Benyei11169dd2012-12-18 14:30:41 +00005217 if (!clang_isDeclaration(C.kind))
5218 return clang_getNullLocation();
5219
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005220 const Decl *D = getCursorDecl(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005221 if (!D)
5222 return clang_getNullLocation();
5223
5224 SourceLocation Loc = D->getLocation();
5225 // FIXME: Multiple variables declared in a single declaration
5226 // currently lack the information needed to correctly determine their
5227 // ranges when accounting for the type-specifier. We use context
5228 // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
5229 // and if so, whether it is the first decl.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005230 if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005231 if (!cxcursor::isFirstInDeclGroup(C))
5232 Loc = VD->getLocation();
5233 }
5234
5235 // For ObjC methods, give the start location of the method name.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005236 if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00005237 Loc = MD->getSelectorStartLoc();
5238
5239 return cxloc::translateSourceLocation(getCursorContext(C), Loc);
5240}
5241
5242} // end extern "C"
5243
5244CXCursor cxcursor::getCursor(CXTranslationUnit TU, SourceLocation SLoc) {
5245 assert(TU);
5246
5247 // Guard against an invalid SourceLocation, or we may assert in one
5248 // of the following calls.
5249 if (SLoc.isInvalid())
5250 return clang_getNullCursor();
5251
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00005252 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00005253
5254 // Translate the given source location to make it point at the beginning of
5255 // the token under the cursor.
5256 SLoc = Lexer::GetBeginningOfToken(SLoc, CXXUnit->getSourceManager(),
5257 CXXUnit->getASTContext().getLangOpts());
5258
5259 CXCursor Result = MakeCXCursorInvalid(CXCursor_NoDeclFound);
5260 if (SLoc.isValid()) {
5261 GetCursorData ResultData(CXXUnit->getSourceManager(), SLoc, Result);
5262 CursorVisitor CursorVis(TU, GetCursorVisitor, &ResultData,
5263 /*VisitPreprocessorLast=*/true,
5264 /*VisitIncludedEntities=*/false,
5265 SourceLocation(SLoc));
5266 CursorVis.visitFileRegion();
5267 }
5268
5269 return Result;
5270}
5271
5272static SourceRange getRawCursorExtent(CXCursor C) {
5273 if (clang_isReference(C.kind)) {
5274 switch (C.kind) {
5275 case CXCursor_ObjCSuperClassRef:
5276 return getCursorObjCSuperClassRef(C).second;
5277
5278 case CXCursor_ObjCProtocolRef:
5279 return getCursorObjCProtocolRef(C).second;
5280
5281 case CXCursor_ObjCClassRef:
5282 return getCursorObjCClassRef(C).second;
5283
5284 case CXCursor_TypeRef:
5285 return getCursorTypeRef(C).second;
5286
5287 case CXCursor_TemplateRef:
5288 return getCursorTemplateRef(C).second;
5289
5290 case CXCursor_NamespaceRef:
5291 return getCursorNamespaceRef(C).second;
5292
5293 case CXCursor_MemberRef:
5294 return getCursorMemberRef(C).second;
5295
5296 case CXCursor_CXXBaseSpecifier:
5297 return getCursorCXXBaseSpecifier(C)->getSourceRange();
5298
5299 case CXCursor_LabelRef:
5300 return getCursorLabelRef(C).second;
5301
5302 case CXCursor_OverloadedDeclRef:
5303 return getCursorOverloadedDeclRef(C).second;
5304
5305 case CXCursor_VariableRef:
5306 return getCursorVariableRef(C).second;
5307
5308 default:
5309 // FIXME: Need a way to enumerate all non-reference cases.
5310 llvm_unreachable("Missed a reference kind");
5311 }
5312 }
5313
5314 if (clang_isExpression(C.kind))
5315 return getCursorExpr(C)->getSourceRange();
5316
5317 if (clang_isStatement(C.kind))
5318 return getCursorStmt(C)->getSourceRange();
5319
5320 if (clang_isAttribute(C.kind))
5321 return getCursorAttr(C)->getRange();
5322
5323 if (C.kind == CXCursor_PreprocessingDirective)
5324 return cxcursor::getCursorPreprocessingDirective(C);
5325
5326 if (C.kind == CXCursor_MacroExpansion) {
5327 ASTUnit *TU = getCursorASTUnit(C);
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00005328 SourceRange Range = cxcursor::getCursorMacroExpansion(C).getSourceRange();
Guy Benyei11169dd2012-12-18 14:30:41 +00005329 return TU->mapRangeFromPreamble(Range);
5330 }
5331
5332 if (C.kind == CXCursor_MacroDefinition) {
5333 ASTUnit *TU = getCursorASTUnit(C);
5334 SourceRange Range = cxcursor::getCursorMacroDefinition(C)->getSourceRange();
5335 return TU->mapRangeFromPreamble(Range);
5336 }
5337
5338 if (C.kind == CXCursor_InclusionDirective) {
5339 ASTUnit *TU = getCursorASTUnit(C);
5340 SourceRange Range = cxcursor::getCursorInclusionDirective(C)->getSourceRange();
5341 return TU->mapRangeFromPreamble(Range);
5342 }
5343
5344 if (C.kind == CXCursor_TranslationUnit) {
5345 ASTUnit *TU = getCursorASTUnit(C);
5346 FileID MainID = TU->getSourceManager().getMainFileID();
5347 SourceLocation Start = TU->getSourceManager().getLocForStartOfFile(MainID);
5348 SourceLocation End = TU->getSourceManager().getLocForEndOfFile(MainID);
5349 return SourceRange(Start, End);
5350 }
5351
5352 if (clang_isDeclaration(C.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005353 const Decl *D = cxcursor::getCursorDecl(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005354 if (!D)
5355 return SourceRange();
5356
5357 SourceRange R = D->getSourceRange();
5358 // FIXME: Multiple variables declared in a single declaration
5359 // currently lack the information needed to correctly determine their
5360 // ranges when accounting for the type-specifier. We use context
5361 // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
5362 // and if so, whether it is the first decl.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005363 if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005364 if (!cxcursor::isFirstInDeclGroup(C))
5365 R.setBegin(VD->getLocation());
5366 }
5367 return R;
5368 }
5369 return SourceRange();
5370}
5371
5372/// \brief Retrieves the "raw" cursor extent, which is then extended to include
5373/// the decl-specifier-seq for declarations.
5374static SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr) {
5375 if (clang_isDeclaration(C.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005376 const Decl *D = cxcursor::getCursorDecl(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005377 if (!D)
5378 return SourceRange();
5379
5380 SourceRange R = D->getSourceRange();
5381
5382 // Adjust the start of the location for declarations preceded by
5383 // declaration specifiers.
5384 SourceLocation StartLoc;
5385 if (const DeclaratorDecl *DD = dyn_cast<DeclaratorDecl>(D)) {
5386 if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
5387 StartLoc = TI->getTypeLoc().getLocStart();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005388 } else if (const TypedefDecl *Typedef = dyn_cast<TypedefDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005389 if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
5390 StartLoc = TI->getTypeLoc().getLocStart();
5391 }
5392
5393 if (StartLoc.isValid() && R.getBegin().isValid() &&
5394 SrcMgr.isBeforeInTranslationUnit(StartLoc, R.getBegin()))
5395 R.setBegin(StartLoc);
5396
5397 // FIXME: Multiple variables declared in a single declaration
5398 // currently lack the information needed to correctly determine their
5399 // ranges when accounting for the type-specifier. We use context
5400 // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
5401 // and if so, whether it is the first decl.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005402 if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005403 if (!cxcursor::isFirstInDeclGroup(C))
5404 R.setBegin(VD->getLocation());
5405 }
5406
5407 return R;
5408 }
5409
5410 return getRawCursorExtent(C);
5411}
5412
5413extern "C" {
5414
5415CXSourceRange clang_getCursorExtent(CXCursor C) {
5416 SourceRange R = getRawCursorExtent(C);
5417 if (R.isInvalid())
5418 return clang_getNullRange();
5419
5420 return cxloc::translateSourceRange(getCursorContext(C), R);
5421}
5422
5423CXCursor clang_getCursorReferenced(CXCursor C) {
5424 if (clang_isInvalid(C.kind))
5425 return clang_getNullCursor();
5426
5427 CXTranslationUnit tu = getCursorTU(C);
5428 if (clang_isDeclaration(C.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005429 const Decl *D = getCursorDecl(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005430 if (!D)
5431 return clang_getNullCursor();
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005432 if (const UsingDecl *Using = dyn_cast<UsingDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00005433 return MakeCursorOverloadedDeclRef(Using, D->getLocation(), tu);
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005434 if (const ObjCPropertyImplDecl *PropImpl =
5435 dyn_cast<ObjCPropertyImplDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00005436 if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
5437 return MakeCXCursor(Property, tu);
5438
5439 return C;
5440 }
5441
5442 if (clang_isExpression(C.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005443 const Expr *E = getCursorExpr(C);
5444 const Decl *D = getDeclFromExpr(E);
Guy Benyei11169dd2012-12-18 14:30:41 +00005445 if (D) {
5446 CXCursor declCursor = MakeCXCursor(D, tu);
5447 declCursor = getSelectorIdentifierCursor(getSelectorIdentifierIndex(C),
5448 declCursor);
5449 return declCursor;
5450 }
5451
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005452 if (const OverloadExpr *Ovl = dyn_cast_or_null<OverloadExpr>(E))
Guy Benyei11169dd2012-12-18 14:30:41 +00005453 return MakeCursorOverloadedDeclRef(Ovl, tu);
5454
5455 return clang_getNullCursor();
5456 }
5457
5458 if (clang_isStatement(C.kind)) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00005459 const Stmt *S = getCursorStmt(C);
5460 if (const GotoStmt *Goto = dyn_cast_or_null<GotoStmt>(S))
Guy Benyei11169dd2012-12-18 14:30:41 +00005461 if (LabelDecl *label = Goto->getLabel())
5462 if (LabelStmt *labelS = label->getStmt())
5463 return MakeCXCursor(labelS, getCursorDecl(C), tu);
5464
5465 return clang_getNullCursor();
5466 }
Richard Smith66a81862015-05-04 02:25:31 +00005467
Guy Benyei11169dd2012-12-18 14:30:41 +00005468 if (C.kind == CXCursor_MacroExpansion) {
Richard Smith66a81862015-05-04 02:25:31 +00005469 if (const MacroDefinitionRecord *Def =
5470 getCursorMacroExpansion(C).getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00005471 return MakeMacroDefinitionCursor(Def, tu);
5472 }
5473
5474 if (!clang_isReference(C.kind))
5475 return clang_getNullCursor();
5476
5477 switch (C.kind) {
5478 case CXCursor_ObjCSuperClassRef:
5479 return MakeCXCursor(getCursorObjCSuperClassRef(C).first, tu);
5480
5481 case CXCursor_ObjCProtocolRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005482 const ObjCProtocolDecl *Prot = getCursorObjCProtocolRef(C).first;
5483 if (const ObjCProtocolDecl *Def = Prot->getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00005484 return MakeCXCursor(Def, tu);
5485
5486 return MakeCXCursor(Prot, tu);
5487 }
5488
5489 case CXCursor_ObjCClassRef: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005490 const ObjCInterfaceDecl *Class = getCursorObjCClassRef(C).first;
5491 if (const ObjCInterfaceDecl *Def = Class->getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00005492 return MakeCXCursor(Def, tu);
5493
5494 return MakeCXCursor(Class, tu);
5495 }
5496
5497 case CXCursor_TypeRef:
5498 return MakeCXCursor(getCursorTypeRef(C).first, tu );
5499
5500 case CXCursor_TemplateRef:
5501 return MakeCXCursor(getCursorTemplateRef(C).first, tu );
5502
5503 case CXCursor_NamespaceRef:
5504 return MakeCXCursor(getCursorNamespaceRef(C).first, tu );
5505
5506 case CXCursor_MemberRef:
5507 return MakeCXCursor(getCursorMemberRef(C).first, tu );
5508
5509 case CXCursor_CXXBaseSpecifier: {
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00005510 const CXXBaseSpecifier *B = cxcursor::getCursorCXXBaseSpecifier(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005511 return clang_getTypeDeclaration(cxtype::MakeCXType(B->getType(),
5512 tu ));
5513 }
5514
5515 case CXCursor_LabelRef:
5516 // FIXME: We end up faking the "parent" declaration here because we
5517 // don't want to make CXCursor larger.
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00005518 return MakeCXCursor(getCursorLabelRef(C).first,
5519 cxtu::getASTUnit(tu)->getASTContext()
5520 .getTranslationUnitDecl(),
Guy Benyei11169dd2012-12-18 14:30:41 +00005521 tu);
5522
5523 case CXCursor_OverloadedDeclRef:
5524 return C;
5525
5526 case CXCursor_VariableRef:
5527 return MakeCXCursor(getCursorVariableRef(C).first, tu);
5528
5529 default:
5530 // We would prefer to enumerate all non-reference cursor kinds here.
5531 llvm_unreachable("Unhandled reference cursor kind");
5532 }
5533}
5534
5535CXCursor clang_getCursorDefinition(CXCursor C) {
5536 if (clang_isInvalid(C.kind))
5537 return clang_getNullCursor();
5538
5539 CXTranslationUnit TU = getCursorTU(C);
5540
5541 bool WasReference = false;
5542 if (clang_isReference(C.kind) || clang_isExpression(C.kind)) {
5543 C = clang_getCursorReferenced(C);
5544 WasReference = true;
5545 }
5546
5547 if (C.kind == CXCursor_MacroExpansion)
5548 return clang_getCursorReferenced(C);
5549
5550 if (!clang_isDeclaration(C.kind))
5551 return clang_getNullCursor();
5552
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005553 const Decl *D = getCursorDecl(C);
Guy Benyei11169dd2012-12-18 14:30:41 +00005554 if (!D)
5555 return clang_getNullCursor();
5556
5557 switch (D->getKind()) {
5558 // Declaration kinds that don't really separate the notions of
5559 // declaration and definition.
5560 case Decl::Namespace:
5561 case Decl::Typedef:
5562 case Decl::TypeAlias:
5563 case Decl::TypeAliasTemplate:
5564 case Decl::TemplateTypeParm:
5565 case Decl::EnumConstant:
5566 case Decl::Field:
John McCall5e77d762013-04-16 07:28:30 +00005567 case Decl::MSProperty:
Guy Benyei11169dd2012-12-18 14:30:41 +00005568 case Decl::IndirectField:
5569 case Decl::ObjCIvar:
5570 case Decl::ObjCAtDefsField:
5571 case Decl::ImplicitParam:
5572 case Decl::ParmVar:
5573 case Decl::NonTypeTemplateParm:
5574 case Decl::TemplateTemplateParm:
5575 case Decl::ObjCCategoryImpl:
5576 case Decl::ObjCImplementation:
5577 case Decl::AccessSpec:
5578 case Decl::LinkageSpec:
5579 case Decl::ObjCPropertyImpl:
5580 case Decl::FileScopeAsm:
5581 case Decl::StaticAssert:
5582 case Decl::Block:
Tareq A. Siraj6dfa25a2013-04-16 19:37:38 +00005583 case Decl::Captured:
Alexey Bataev4244be22016-02-11 05:35:55 +00005584 case Decl::OMPCapturedExpr:
Guy Benyei11169dd2012-12-18 14:30:41 +00005585 case Decl::Label: // FIXME: Is this right??
5586 case Decl::ClassScopeFunctionSpecialization:
5587 case Decl::Import:
Alexey Bataeva769e072013-03-22 06:34:35 +00005588 case Decl::OMPThreadPrivate:
Alexey Bataev94a4f0c2016-03-03 05:21:39 +00005589 case Decl::OMPDeclareReduction:
Douglas Gregor85f3f952015-07-07 03:57:15 +00005590 case Decl::ObjCTypeParam:
David Majnemerd9b1a4f2015-11-04 03:40:30 +00005591 case Decl::BuiltinTemplate:
Nico Weber66220292016-03-02 17:28:48 +00005592 case Decl::PragmaComment:
Nico Webercbbaeb12016-03-02 19:28:54 +00005593 case Decl::PragmaDetectMismatch:
Guy Benyei11169dd2012-12-18 14:30:41 +00005594 return C;
5595
5596 // Declaration kinds that don't make any sense here, but are
5597 // nonetheless harmless.
David Blaikief005d3c2013-02-22 17:44:58 +00005598 case Decl::Empty:
Guy Benyei11169dd2012-12-18 14:30:41 +00005599 case Decl::TranslationUnit:
Richard Smithf19e1272015-03-07 00:04:49 +00005600 case Decl::ExternCContext:
Guy Benyei11169dd2012-12-18 14:30:41 +00005601 break;
5602
5603 // Declaration kinds for which the definition is not resolvable.
5604 case Decl::UnresolvedUsingTypename:
5605 case Decl::UnresolvedUsingValue:
5606 break;
5607
5608 case Decl::UsingDirective:
5609 return MakeCXCursor(cast<UsingDirectiveDecl>(D)->getNominatedNamespace(),
5610 TU);
5611
5612 case Decl::NamespaceAlias:
5613 return MakeCXCursor(cast<NamespaceAliasDecl>(D)->getNamespace(), TU);
5614
5615 case Decl::Enum:
5616 case Decl::Record:
5617 case Decl::CXXRecord:
5618 case Decl::ClassTemplateSpecialization:
5619 case Decl::ClassTemplatePartialSpecialization:
5620 if (TagDecl *Def = cast<TagDecl>(D)->getDefinition())
5621 return MakeCXCursor(Def, TU);
5622 return clang_getNullCursor();
5623
5624 case Decl::Function:
5625 case Decl::CXXMethod:
5626 case Decl::CXXConstructor:
5627 case Decl::CXXDestructor:
5628 case Decl::CXXConversion: {
Craig Topper69186e72014-06-08 08:38:04 +00005629 const FunctionDecl *Def = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00005630 if (cast<FunctionDecl>(D)->getBody(Def))
Dmitri Gribenko9c256e32013-01-14 00:46:27 +00005631 return MakeCXCursor(Def, TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00005632 return clang_getNullCursor();
5633 }
5634
Larisse Voufo39a1e502013-08-06 01:03:05 +00005635 case Decl::Var:
5636 case Decl::VarTemplateSpecialization:
5637 case Decl::VarTemplatePartialSpecialization: {
Guy Benyei11169dd2012-12-18 14:30:41 +00005638 // Ask the variable if it has a definition.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005639 if (const VarDecl *Def = cast<VarDecl>(D)->getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00005640 return MakeCXCursor(Def, TU);
5641 return clang_getNullCursor();
5642 }
5643
5644 case Decl::FunctionTemplate: {
Craig Topper69186e72014-06-08 08:38:04 +00005645 const FunctionDecl *Def = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00005646 if (cast<FunctionTemplateDecl>(D)->getTemplatedDecl()->getBody(Def))
5647 return MakeCXCursor(Def->getDescribedFunctionTemplate(), TU);
5648 return clang_getNullCursor();
5649 }
5650
5651 case Decl::ClassTemplate: {
5652 if (RecordDecl *Def = cast<ClassTemplateDecl>(D)->getTemplatedDecl()
5653 ->getDefinition())
5654 return MakeCXCursor(cast<CXXRecordDecl>(Def)->getDescribedClassTemplate(),
5655 TU);
5656 return clang_getNullCursor();
5657 }
5658
Larisse Voufo39a1e502013-08-06 01:03:05 +00005659 case Decl::VarTemplate: {
5660 if (VarDecl *Def =
5661 cast<VarTemplateDecl>(D)->getTemplatedDecl()->getDefinition())
5662 return MakeCXCursor(cast<VarDecl>(Def)->getDescribedVarTemplate(), TU);
5663 return clang_getNullCursor();
5664 }
5665
Guy Benyei11169dd2012-12-18 14:30:41 +00005666 case Decl::Using:
5667 return MakeCursorOverloadedDeclRef(cast<UsingDecl>(D),
5668 D->getLocation(), TU);
5669
5670 case Decl::UsingShadow:
Richard Smith5179eb72016-06-28 19:03:57 +00005671 case Decl::ConstructorUsingShadow:
Guy Benyei11169dd2012-12-18 14:30:41 +00005672 return clang_getCursorDefinition(
5673 MakeCXCursor(cast<UsingShadowDecl>(D)->getTargetDecl(),
5674 TU));
5675
5676 case Decl::ObjCMethod: {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005677 const ObjCMethodDecl *Method = cast<ObjCMethodDecl>(D);
Guy Benyei11169dd2012-12-18 14:30:41 +00005678 if (Method->isThisDeclarationADefinition())
5679 return C;
5680
5681 // Dig out the method definition in the associated
5682 // @implementation, if we have it.
5683 // FIXME: The ASTs should make finding the definition easier.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005684 if (const ObjCInterfaceDecl *Class
Guy Benyei11169dd2012-12-18 14:30:41 +00005685 = dyn_cast<ObjCInterfaceDecl>(Method->getDeclContext()))
5686 if (ObjCImplementationDecl *ClassImpl = Class->getImplementation())
5687 if (ObjCMethodDecl *Def = ClassImpl->getMethod(Method->getSelector(),
5688 Method->isInstanceMethod()))
5689 if (Def->isThisDeclarationADefinition())
5690 return MakeCXCursor(Def, TU);
5691
5692 return clang_getNullCursor();
5693 }
5694
5695 case Decl::ObjCCategory:
5696 if (ObjCCategoryImplDecl *Impl
5697 = cast<ObjCCategoryDecl>(D)->getImplementation())
5698 return MakeCXCursor(Impl, TU);
5699 return clang_getNullCursor();
5700
5701 case Decl::ObjCProtocol:
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005702 if (const ObjCProtocolDecl *Def = cast<ObjCProtocolDecl>(D)->getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00005703 return MakeCXCursor(Def, TU);
5704 return clang_getNullCursor();
5705
5706 case Decl::ObjCInterface: {
5707 // There are two notions of a "definition" for an Objective-C
5708 // class: the interface and its implementation. When we resolved a
5709 // reference to an Objective-C class, produce the @interface as
5710 // the definition; when we were provided with the interface,
5711 // produce the @implementation as the definition.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005712 const ObjCInterfaceDecl *IFace = cast<ObjCInterfaceDecl>(D);
Guy Benyei11169dd2012-12-18 14:30:41 +00005713 if (WasReference) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005714 if (const ObjCInterfaceDecl *Def = IFace->getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00005715 return MakeCXCursor(Def, TU);
5716 } else if (ObjCImplementationDecl *Impl = IFace->getImplementation())
5717 return MakeCXCursor(Impl, TU);
5718 return clang_getNullCursor();
5719 }
5720
5721 case Decl::ObjCProperty:
5722 // FIXME: We don't really know where to find the
5723 // ObjCPropertyImplDecls that implement this property.
5724 return clang_getNullCursor();
5725
5726 case Decl::ObjCCompatibleAlias:
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005727 if (const ObjCInterfaceDecl *Class
Guy Benyei11169dd2012-12-18 14:30:41 +00005728 = cast<ObjCCompatibleAliasDecl>(D)->getClassInterface())
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005729 if (const ObjCInterfaceDecl *Def = Class->getDefinition())
Guy Benyei11169dd2012-12-18 14:30:41 +00005730 return MakeCXCursor(Def, TU);
5731
5732 return clang_getNullCursor();
5733
5734 case Decl::Friend:
5735 if (NamedDecl *Friend = cast<FriendDecl>(D)->getFriendDecl())
5736 return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
5737 return clang_getNullCursor();
5738
5739 case Decl::FriendTemplate:
5740 if (NamedDecl *Friend = cast<FriendTemplateDecl>(D)->getFriendDecl())
5741 return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
5742 return clang_getNullCursor();
5743 }
5744
5745 return clang_getNullCursor();
5746}
5747
5748unsigned clang_isCursorDefinition(CXCursor C) {
5749 if (!clang_isDeclaration(C.kind))
5750 return 0;
5751
5752 return clang_getCursorDefinition(C) == C;
5753}
5754
5755CXCursor clang_getCanonicalCursor(CXCursor C) {
5756 if (!clang_isDeclaration(C.kind))
5757 return C;
5758
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005759 if (const Decl *D = getCursorDecl(C)) {
5760 if (const ObjCCategoryImplDecl *CatImplD = dyn_cast<ObjCCategoryImplDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00005761 if (ObjCCategoryDecl *CatD = CatImplD->getCategoryDecl())
5762 return MakeCXCursor(CatD, getCursorTU(C));
5763
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005764 if (const ObjCImplDecl *ImplD = dyn_cast<ObjCImplDecl>(D))
5765 if (const ObjCInterfaceDecl *IFD = ImplD->getClassInterface())
Guy Benyei11169dd2012-12-18 14:30:41 +00005766 return MakeCXCursor(IFD, getCursorTU(C));
5767
5768 return MakeCXCursor(D->getCanonicalDecl(), getCursorTU(C));
5769 }
5770
5771 return C;
5772}
5773
5774int clang_Cursor_getObjCSelectorIndex(CXCursor cursor) {
5775 return cxcursor::getSelectorIdentifierIndexAndLoc(cursor).first;
5776}
5777
5778unsigned clang_getNumOverloadedDecls(CXCursor C) {
5779 if (C.kind != CXCursor_OverloadedDeclRef)
5780 return 0;
5781
5782 OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005783 if (const OverloadExpr *E = Storage.dyn_cast<const OverloadExpr *>())
Guy Benyei11169dd2012-12-18 14:30:41 +00005784 return E->getNumDecls();
5785
5786 if (OverloadedTemplateStorage *S
5787 = Storage.dyn_cast<OverloadedTemplateStorage*>())
5788 return S->size();
5789
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005790 const Decl *D = Storage.get<const Decl *>();
5791 if (const UsingDecl *Using = dyn_cast<UsingDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00005792 return Using->shadow_size();
5793
5794 return 0;
5795}
5796
5797CXCursor clang_getOverloadedDecl(CXCursor cursor, unsigned index) {
5798 if (cursor.kind != CXCursor_OverloadedDeclRef)
5799 return clang_getNullCursor();
5800
5801 if (index >= clang_getNumOverloadedDecls(cursor))
5802 return clang_getNullCursor();
5803
5804 CXTranslationUnit TU = getCursorTU(cursor);
5805 OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(cursor).first;
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005806 if (const OverloadExpr *E = Storage.dyn_cast<const OverloadExpr *>())
Guy Benyei11169dd2012-12-18 14:30:41 +00005807 return MakeCXCursor(E->decls_begin()[index], TU);
5808
5809 if (OverloadedTemplateStorage *S
5810 = Storage.dyn_cast<OverloadedTemplateStorage*>())
5811 return MakeCXCursor(S->begin()[index], TU);
5812
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005813 const Decl *D = Storage.get<const Decl *>();
5814 if (const UsingDecl *Using = dyn_cast<UsingDecl>(D)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00005815 // FIXME: This is, unfortunately, linear time.
5816 UsingDecl::shadow_iterator Pos = Using->shadow_begin();
5817 std::advance(Pos, index);
5818 return MakeCXCursor(cast<UsingShadowDecl>(*Pos)->getTargetDecl(), TU);
5819 }
5820
5821 return clang_getNullCursor();
5822}
5823
5824void clang_getDefinitionSpellingAndExtent(CXCursor C,
5825 const char **startBuf,
5826 const char **endBuf,
5827 unsigned *startLine,
5828 unsigned *startColumn,
5829 unsigned *endLine,
5830 unsigned *endColumn) {
5831 assert(getCursorDecl(C) && "CXCursor has null decl");
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00005832 const FunctionDecl *FD = dyn_cast<FunctionDecl>(getCursorDecl(C));
Guy Benyei11169dd2012-12-18 14:30:41 +00005833 CompoundStmt *Body = dyn_cast<CompoundStmt>(FD->getBody());
5834
5835 SourceManager &SM = FD->getASTContext().getSourceManager();
5836 *startBuf = SM.getCharacterData(Body->getLBracLoc());
5837 *endBuf = SM.getCharacterData(Body->getRBracLoc());
5838 *startLine = SM.getSpellingLineNumber(Body->getLBracLoc());
5839 *startColumn = SM.getSpellingColumnNumber(Body->getLBracLoc());
5840 *endLine = SM.getSpellingLineNumber(Body->getRBracLoc());
5841 *endColumn = SM.getSpellingColumnNumber(Body->getRBracLoc());
5842}
5843
5844
5845CXSourceRange clang_getCursorReferenceNameRange(CXCursor C, unsigned NameFlags,
5846 unsigned PieceIndex) {
5847 RefNamePieces Pieces;
5848
5849 switch (C.kind) {
5850 case CXCursor_MemberRefExpr:
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00005851 if (const MemberExpr *E = dyn_cast<MemberExpr>(getCursorExpr(C)))
Guy Benyei11169dd2012-12-18 14:30:41 +00005852 Pieces = buildPieces(NameFlags, true, E->getMemberNameInfo(),
5853 E->getQualifierLoc().getSourceRange());
5854 break;
5855
5856 case CXCursor_DeclRefExpr:
James Y Knight04ec5bf2015-12-24 02:59:37 +00005857 if (const DeclRefExpr *E = dyn_cast<DeclRefExpr>(getCursorExpr(C))) {
5858 SourceRange TemplateArgLoc(E->getLAngleLoc(), E->getRAngleLoc());
5859 Pieces =
5860 buildPieces(NameFlags, false, E->getNameInfo(),
5861 E->getQualifierLoc().getSourceRange(), &TemplateArgLoc);
5862 }
Guy Benyei11169dd2012-12-18 14:30:41 +00005863 break;
5864
5865 case CXCursor_CallExpr:
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00005866 if (const CXXOperatorCallExpr *OCE =
Guy Benyei11169dd2012-12-18 14:30:41 +00005867 dyn_cast<CXXOperatorCallExpr>(getCursorExpr(C))) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00005868 const Expr *Callee = OCE->getCallee();
5869 if (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Callee))
Guy Benyei11169dd2012-12-18 14:30:41 +00005870 Callee = ICE->getSubExpr();
5871
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00005872 if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Callee))
Guy Benyei11169dd2012-12-18 14:30:41 +00005873 Pieces = buildPieces(NameFlags, false, DRE->getNameInfo(),
5874 DRE->getQualifierLoc().getSourceRange());
5875 }
5876 break;
5877
5878 default:
5879 break;
5880 }
5881
5882 if (Pieces.empty()) {
5883 if (PieceIndex == 0)
5884 return clang_getCursorExtent(C);
5885 } else if (PieceIndex < Pieces.size()) {
5886 SourceRange R = Pieces[PieceIndex];
5887 if (R.isValid())
5888 return cxloc::translateSourceRange(getCursorContext(C), R);
5889 }
5890
5891 return clang_getNullRange();
5892}
5893
5894void clang_enableStackTraces(void) {
Richard Smithdfed58a2016-06-09 00:53:41 +00005895 // FIXME: Provide an argv0 here so we can find llvm-symbolizer.
5896 llvm::sys::PrintStackTraceOnErrorSignal(StringRef());
Guy Benyei11169dd2012-12-18 14:30:41 +00005897}
5898
5899void clang_executeOnThread(void (*fn)(void*), void *user_data,
5900 unsigned stack_size) {
5901 llvm::llvm_execute_on_thread(fn, user_data, stack_size);
5902}
5903
5904} // end: extern "C"
5905
5906//===----------------------------------------------------------------------===//
5907// Token-based Operations.
5908//===----------------------------------------------------------------------===//
5909
5910/* CXToken layout:
5911 * int_data[0]: a CXTokenKind
5912 * int_data[1]: starting token location
5913 * int_data[2]: token length
5914 * int_data[3]: reserved
5915 * ptr_data: for identifiers and keywords, an IdentifierInfo*.
5916 * otherwise unused.
5917 */
5918extern "C" {
5919
5920CXTokenKind clang_getTokenKind(CXToken CXTok) {
5921 return static_cast<CXTokenKind>(CXTok.int_data[0]);
5922}
5923
5924CXString clang_getTokenSpelling(CXTranslationUnit TU, CXToken CXTok) {
5925 switch (clang_getTokenKind(CXTok)) {
5926 case CXToken_Identifier:
5927 case CXToken_Keyword:
5928 // We know we have an IdentifierInfo*, so use that.
Dmitri Gribenko3c66b0b2013-02-02 00:02:12 +00005929 return cxstring::createRef(static_cast<IdentifierInfo *>(CXTok.ptr_data)
Guy Benyei11169dd2012-12-18 14:30:41 +00005930 ->getNameStart());
5931
5932 case CXToken_Literal: {
5933 // We have stashed the starting pointer in the ptr_data field. Use it.
5934 const char *Text = static_cast<const char *>(CXTok.ptr_data);
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00005935 return cxstring::createDup(StringRef(Text, CXTok.int_data[2]));
Guy Benyei11169dd2012-12-18 14:30:41 +00005936 }
5937
5938 case CXToken_Punctuation:
5939 case CXToken_Comment:
5940 break;
5941 }
5942
Dmitri Gribenko852d6222014-02-11 15:02:48 +00005943 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00005944 LOG_BAD_TU(TU);
5945 return cxstring::createEmpty();
5946 }
5947
Guy Benyei11169dd2012-12-18 14:30:41 +00005948 // We have to find the starting buffer pointer the hard way, by
5949 // deconstructing the source location.
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00005950 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00005951 if (!CXXUnit)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00005952 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00005953
5954 SourceLocation Loc = SourceLocation::getFromRawEncoding(CXTok.int_data[1]);
5955 std::pair<FileID, unsigned> LocInfo
5956 = CXXUnit->getSourceManager().getDecomposedSpellingLoc(Loc);
5957 bool Invalid = false;
5958 StringRef Buffer
5959 = CXXUnit->getSourceManager().getBufferData(LocInfo.first, &Invalid);
5960 if (Invalid)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00005961 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00005962
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00005963 return cxstring::createDup(Buffer.substr(LocInfo.second, CXTok.int_data[2]));
Guy Benyei11169dd2012-12-18 14:30:41 +00005964}
5965
5966CXSourceLocation clang_getTokenLocation(CXTranslationUnit TU, CXToken CXTok) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00005967 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00005968 LOG_BAD_TU(TU);
5969 return clang_getNullLocation();
5970 }
5971
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00005972 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00005973 if (!CXXUnit)
5974 return clang_getNullLocation();
5975
5976 return cxloc::translateSourceLocation(CXXUnit->getASTContext(),
5977 SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
5978}
5979
5980CXSourceRange clang_getTokenExtent(CXTranslationUnit TU, CXToken CXTok) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00005981 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00005982 LOG_BAD_TU(TU);
5983 return clang_getNullRange();
5984 }
5985
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00005986 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00005987 if (!CXXUnit)
5988 return clang_getNullRange();
5989
5990 return cxloc::translateSourceRange(CXXUnit->getASTContext(),
5991 SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
5992}
5993
5994static void getTokens(ASTUnit *CXXUnit, SourceRange Range,
5995 SmallVectorImpl<CXToken> &CXTokens) {
5996 SourceManager &SourceMgr = CXXUnit->getSourceManager();
5997 std::pair<FileID, unsigned> BeginLocInfo
Argyrios Kyrtzidis5d47a9b2013-02-13 18:33:28 +00005998 = SourceMgr.getDecomposedSpellingLoc(Range.getBegin());
Guy Benyei11169dd2012-12-18 14:30:41 +00005999 std::pair<FileID, unsigned> EndLocInfo
Argyrios Kyrtzidis5d47a9b2013-02-13 18:33:28 +00006000 = SourceMgr.getDecomposedSpellingLoc(Range.getEnd());
Guy Benyei11169dd2012-12-18 14:30:41 +00006001
6002 // Cannot tokenize across files.
6003 if (BeginLocInfo.first != EndLocInfo.first)
6004 return;
6005
6006 // Create a lexer
6007 bool Invalid = false;
6008 StringRef Buffer
6009 = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
6010 if (Invalid)
6011 return;
6012
6013 Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
6014 CXXUnit->getASTContext().getLangOpts(),
6015 Buffer.begin(), Buffer.data() + BeginLocInfo.second, Buffer.end());
6016 Lex.SetCommentRetentionState(true);
6017
6018 // Lex tokens until we hit the end of the range.
6019 const char *EffectiveBufferEnd = Buffer.data() + EndLocInfo.second;
6020 Token Tok;
6021 bool previousWasAt = false;
6022 do {
6023 // Lex the next token
6024 Lex.LexFromRawLexer(Tok);
6025 if (Tok.is(tok::eof))
6026 break;
6027
6028 // Initialize the CXToken.
6029 CXToken CXTok;
6030
6031 // - Common fields
6032 CXTok.int_data[1] = Tok.getLocation().getRawEncoding();
6033 CXTok.int_data[2] = Tok.getLength();
6034 CXTok.int_data[3] = 0;
6035
6036 // - Kind-specific fields
6037 if (Tok.isLiteral()) {
6038 CXTok.int_data[0] = CXToken_Literal;
Dmitri Gribenkof9304482013-01-23 15:56:07 +00006039 CXTok.ptr_data = const_cast<char *>(Tok.getLiteralData());
Guy Benyei11169dd2012-12-18 14:30:41 +00006040 } else if (Tok.is(tok::raw_identifier)) {
6041 // Lookup the identifier to determine whether we have a keyword.
6042 IdentifierInfo *II
6043 = CXXUnit->getPreprocessor().LookUpIdentifierInfo(Tok);
6044
6045 if ((II->getObjCKeywordID() != tok::objc_not_keyword) && previousWasAt) {
6046 CXTok.int_data[0] = CXToken_Keyword;
6047 }
6048 else {
6049 CXTok.int_data[0] = Tok.is(tok::identifier)
6050 ? CXToken_Identifier
6051 : CXToken_Keyword;
6052 }
6053 CXTok.ptr_data = II;
6054 } else if (Tok.is(tok::comment)) {
6055 CXTok.int_data[0] = CXToken_Comment;
Craig Topper69186e72014-06-08 08:38:04 +00006056 CXTok.ptr_data = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00006057 } else {
6058 CXTok.int_data[0] = CXToken_Punctuation;
Craig Topper69186e72014-06-08 08:38:04 +00006059 CXTok.ptr_data = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00006060 }
6061 CXTokens.push_back(CXTok);
6062 previousWasAt = Tok.is(tok::at);
6063 } while (Lex.getBufferLocation() <= EffectiveBufferEnd);
6064}
6065
6066void clang_tokenize(CXTranslationUnit TU, CXSourceRange Range,
6067 CXToken **Tokens, unsigned *NumTokens) {
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00006068 LOG_FUNC_SECTION {
6069 *Log << TU << ' ' << Range;
6070 }
6071
Guy Benyei11169dd2012-12-18 14:30:41 +00006072 if (Tokens)
Craig Topper69186e72014-06-08 08:38:04 +00006073 *Tokens = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00006074 if (NumTokens)
6075 *NumTokens = 0;
6076
Dmitri Gribenko852d6222014-02-11 15:02:48 +00006077 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00006078 LOG_BAD_TU(TU);
Argyrios Kyrtzidis0e95fca2013-04-04 22:40:59 +00006079 return;
Dmitri Gribenko256454f2014-02-11 14:34:14 +00006080 }
Argyrios Kyrtzidis0e95fca2013-04-04 22:40:59 +00006081
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006082 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00006083 if (!CXXUnit || !Tokens || !NumTokens)
6084 return;
6085
6086 ASTUnit::ConcurrencyCheck Check(*CXXUnit);
6087
6088 SourceRange R = cxloc::translateCXSourceRange(Range);
6089 if (R.isInvalid())
6090 return;
6091
6092 SmallVector<CXToken, 32> CXTokens;
6093 getTokens(CXXUnit, R, CXTokens);
6094
6095 if (CXTokens.empty())
6096 return;
6097
6098 *Tokens = (CXToken *)malloc(sizeof(CXToken) * CXTokens.size());
6099 memmove(*Tokens, CXTokens.data(), sizeof(CXToken) * CXTokens.size());
6100 *NumTokens = CXTokens.size();
6101}
6102
6103void clang_disposeTokens(CXTranslationUnit TU,
6104 CXToken *Tokens, unsigned NumTokens) {
6105 free(Tokens);
6106}
6107
6108} // end: extern "C"
6109
6110//===----------------------------------------------------------------------===//
6111// Token annotation APIs.
6112//===----------------------------------------------------------------------===//
6113
Guy Benyei11169dd2012-12-18 14:30:41 +00006114static enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
6115 CXCursor parent,
6116 CXClientData client_data);
6117static bool AnnotateTokensPostChildrenVisitor(CXCursor cursor,
6118 CXClientData client_data);
6119
6120namespace {
6121class AnnotateTokensWorker {
Guy Benyei11169dd2012-12-18 14:30:41 +00006122 CXToken *Tokens;
6123 CXCursor *Cursors;
6124 unsigned NumTokens;
6125 unsigned TokIdx;
6126 unsigned PreprocessingTokIdx;
6127 CursorVisitor AnnotateVis;
6128 SourceManager &SrcMgr;
6129 bool HasContextSensitiveKeywords;
6130
6131 struct PostChildrenInfo {
6132 CXCursor Cursor;
6133 SourceRange CursorRange;
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006134 unsigned BeforeReachingCursorIdx;
Guy Benyei11169dd2012-12-18 14:30:41 +00006135 unsigned BeforeChildrenTokenIdx;
6136 };
Dmitri Gribenkof8579502013-01-12 19:30:44 +00006137 SmallVector<PostChildrenInfo, 8> PostChildrenInfos;
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006138
6139 CXToken &getTok(unsigned Idx) {
6140 assert(Idx < NumTokens);
6141 return Tokens[Idx];
6142 }
6143 const CXToken &getTok(unsigned Idx) const {
6144 assert(Idx < NumTokens);
6145 return Tokens[Idx];
6146 }
Guy Benyei11169dd2012-12-18 14:30:41 +00006147 bool MoreTokens() const { return TokIdx < NumTokens; }
6148 unsigned NextToken() const { return TokIdx; }
6149 void AdvanceToken() { ++TokIdx; }
6150 SourceLocation GetTokenLoc(unsigned tokI) {
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006151 return SourceLocation::getFromRawEncoding(getTok(tokI).int_data[1]);
Guy Benyei11169dd2012-12-18 14:30:41 +00006152 }
6153 bool isFunctionMacroToken(unsigned tokI) const {
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006154 return getTok(tokI).int_data[3] != 0;
Guy Benyei11169dd2012-12-18 14:30:41 +00006155 }
6156 SourceLocation getFunctionMacroTokenLoc(unsigned tokI) const {
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006157 return SourceLocation::getFromRawEncoding(getTok(tokI).int_data[3]);
Guy Benyei11169dd2012-12-18 14:30:41 +00006158 }
6159
6160 void annotateAndAdvanceTokens(CXCursor, RangeComparisonResult, SourceRange);
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006161 bool annotateAndAdvanceFunctionMacroTokens(CXCursor, RangeComparisonResult,
Guy Benyei11169dd2012-12-18 14:30:41 +00006162 SourceRange);
6163
6164public:
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006165 AnnotateTokensWorker(CXToken *tokens, CXCursor *cursors, unsigned numTokens,
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006166 CXTranslationUnit TU, SourceRange RegionOfInterest)
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006167 : Tokens(tokens), Cursors(cursors),
Guy Benyei11169dd2012-12-18 14:30:41 +00006168 NumTokens(numTokens), TokIdx(0), PreprocessingTokIdx(0),
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006169 AnnotateVis(TU,
Guy Benyei11169dd2012-12-18 14:30:41 +00006170 AnnotateTokensVisitor, this,
6171 /*VisitPreprocessorLast=*/true,
6172 /*VisitIncludedEntities=*/false,
6173 RegionOfInterest,
6174 /*VisitDeclsOnly=*/false,
6175 AnnotateTokensPostChildrenVisitor),
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006176 SrcMgr(cxtu::getASTUnit(TU)->getSourceManager()),
Guy Benyei11169dd2012-12-18 14:30:41 +00006177 HasContextSensitiveKeywords(false) { }
6178
6179 void VisitChildren(CXCursor C) { AnnotateVis.VisitChildren(C); }
6180 enum CXChildVisitResult Visit(CXCursor cursor, CXCursor parent);
6181 bool postVisitChildren(CXCursor cursor);
6182 void AnnotateTokens();
6183
6184 /// \brief Determine whether the annotator saw any cursors that have
6185 /// context-sensitive keywords.
6186 bool hasContextSensitiveKeywords() const {
6187 return HasContextSensitiveKeywords;
6188 }
6189
6190 ~AnnotateTokensWorker() {
6191 assert(PostChildrenInfos.empty());
6192 }
6193};
Alexander Kornienkoab9db512015-06-22 23:07:51 +00006194}
Guy Benyei11169dd2012-12-18 14:30:41 +00006195
6196void AnnotateTokensWorker::AnnotateTokens() {
6197 // Walk the AST within the region of interest, annotating tokens
6198 // along the way.
6199 AnnotateVis.visitFileRegion();
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006200}
Guy Benyei11169dd2012-12-18 14:30:41 +00006201
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006202static inline void updateCursorAnnotation(CXCursor &Cursor,
6203 const CXCursor &updateC) {
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006204 if (clang_isInvalid(updateC.kind) || !clang_isInvalid(Cursor.kind))
Guy Benyei11169dd2012-12-18 14:30:41 +00006205 return;
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006206 Cursor = updateC;
Guy Benyei11169dd2012-12-18 14:30:41 +00006207}
6208
6209/// \brief It annotates and advances tokens with a cursor until the comparison
6210//// between the cursor location and the source range is the same as
6211/// \arg compResult.
6212///
6213/// Pass RangeBefore to annotate tokens with a cursor until a range is reached.
6214/// Pass RangeOverlap to annotate tokens inside a range.
6215void AnnotateTokensWorker::annotateAndAdvanceTokens(CXCursor updateC,
6216 RangeComparisonResult compResult,
6217 SourceRange range) {
6218 while (MoreTokens()) {
6219 const unsigned I = NextToken();
6220 if (isFunctionMacroToken(I))
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006221 if (!annotateAndAdvanceFunctionMacroTokens(updateC, compResult, range))
6222 return;
Guy Benyei11169dd2012-12-18 14:30:41 +00006223
6224 SourceLocation TokLoc = GetTokenLoc(I);
6225 if (LocationCompare(SrcMgr, TokLoc, range) == compResult) {
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006226 updateCursorAnnotation(Cursors[I], updateC);
Guy Benyei11169dd2012-12-18 14:30:41 +00006227 AdvanceToken();
6228 continue;
6229 }
6230 break;
6231 }
6232}
6233
6234/// \brief Special annotation handling for macro argument tokens.
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006235/// \returns true if it advanced beyond all macro tokens, false otherwise.
6236bool AnnotateTokensWorker::annotateAndAdvanceFunctionMacroTokens(
Guy Benyei11169dd2012-12-18 14:30:41 +00006237 CXCursor updateC,
6238 RangeComparisonResult compResult,
6239 SourceRange range) {
6240 assert(MoreTokens());
6241 assert(isFunctionMacroToken(NextToken()) &&
6242 "Should be called only for macro arg tokens");
6243
6244 // This works differently than annotateAndAdvanceTokens; because expanded
6245 // macro arguments can have arbitrary translation-unit source order, we do not
6246 // advance the token index one by one until a token fails the range test.
6247 // We only advance once past all of the macro arg tokens if all of them
6248 // pass the range test. If one of them fails we keep the token index pointing
6249 // at the start of the macro arg tokens so that the failing token will be
6250 // annotated by a subsequent annotation try.
6251
6252 bool atLeastOneCompFail = false;
6253
6254 unsigned I = NextToken();
6255 for (; I < NumTokens && isFunctionMacroToken(I); ++I) {
6256 SourceLocation TokLoc = getFunctionMacroTokenLoc(I);
6257 if (TokLoc.isFileID())
6258 continue; // not macro arg token, it's parens or comma.
6259 if (LocationCompare(SrcMgr, TokLoc, range) == compResult) {
6260 if (clang_isInvalid(clang_getCursorKind(Cursors[I])))
6261 Cursors[I] = updateC;
6262 } else
6263 atLeastOneCompFail = true;
6264 }
6265
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006266 if (atLeastOneCompFail)
6267 return false;
6268
6269 TokIdx = I; // All of the tokens were handled, advance beyond all of them.
6270 return true;
Guy Benyei11169dd2012-12-18 14:30:41 +00006271}
6272
6273enum CXChildVisitResult
6274AnnotateTokensWorker::Visit(CXCursor cursor, CXCursor parent) {
Guy Benyei11169dd2012-12-18 14:30:41 +00006275 SourceRange cursorRange = getRawCursorExtent(cursor);
6276 if (cursorRange.isInvalid())
6277 return CXChildVisit_Recurse;
6278
6279 if (!HasContextSensitiveKeywords) {
6280 // Objective-C properties can have context-sensitive keywords.
6281 if (cursor.kind == CXCursor_ObjCPropertyDecl) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006282 if (const ObjCPropertyDecl *Property
Guy Benyei11169dd2012-12-18 14:30:41 +00006283 = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(cursor)))
6284 HasContextSensitiveKeywords = Property->getPropertyAttributesAsWritten() != 0;
6285 }
6286 // Objective-C methods can have context-sensitive keywords.
6287 else if (cursor.kind == CXCursor_ObjCInstanceMethodDecl ||
6288 cursor.kind == CXCursor_ObjCClassMethodDecl) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006289 if (const ObjCMethodDecl *Method
Guy Benyei11169dd2012-12-18 14:30:41 +00006290 = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(cursor))) {
6291 if (Method->getObjCDeclQualifier())
6292 HasContextSensitiveKeywords = true;
6293 else {
David Majnemer59f77922016-06-24 04:05:48 +00006294 for (const auto *P : Method->parameters()) {
Aaron Ballman43b68be2014-03-07 17:50:17 +00006295 if (P->getObjCDeclQualifier()) {
Guy Benyei11169dd2012-12-18 14:30:41 +00006296 HasContextSensitiveKeywords = true;
6297 break;
6298 }
6299 }
6300 }
6301 }
6302 }
6303 // C++ methods can have context-sensitive keywords.
6304 else if (cursor.kind == CXCursor_CXXMethod) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006305 if (const CXXMethodDecl *Method
Guy Benyei11169dd2012-12-18 14:30:41 +00006306 = dyn_cast_or_null<CXXMethodDecl>(getCursorDecl(cursor))) {
6307 if (Method->hasAttr<FinalAttr>() || Method->hasAttr<OverrideAttr>())
6308 HasContextSensitiveKeywords = true;
6309 }
6310 }
6311 // C++ classes can have context-sensitive keywords.
6312 else if (cursor.kind == CXCursor_StructDecl ||
6313 cursor.kind == CXCursor_ClassDecl ||
6314 cursor.kind == CXCursor_ClassTemplate ||
6315 cursor.kind == CXCursor_ClassTemplatePartialSpecialization) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006316 if (const Decl *D = getCursorDecl(cursor))
Guy Benyei11169dd2012-12-18 14:30:41 +00006317 if (D->hasAttr<FinalAttr>())
6318 HasContextSensitiveKeywords = true;
6319 }
6320 }
Argyrios Kyrtzidis990b3862013-06-04 18:24:30 +00006321
6322 // Don't override a property annotation with its getter/setter method.
6323 if (cursor.kind == CXCursor_ObjCInstanceMethodDecl &&
6324 parent.kind == CXCursor_ObjCPropertyDecl)
6325 return CXChildVisit_Continue;
Guy Benyei11169dd2012-12-18 14:30:41 +00006326
6327 if (clang_isPreprocessing(cursor.kind)) {
6328 // Items in the preprocessing record are kept separate from items in
6329 // declarations, so we keep a separate token index.
6330 unsigned SavedTokIdx = TokIdx;
6331 TokIdx = PreprocessingTokIdx;
6332
6333 // Skip tokens up until we catch up to the beginning of the preprocessing
6334 // entry.
6335 while (MoreTokens()) {
6336 const unsigned I = NextToken();
6337 SourceLocation TokLoc = GetTokenLoc(I);
6338 switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
6339 case RangeBefore:
6340 AdvanceToken();
6341 continue;
6342 case RangeAfter:
6343 case RangeOverlap:
6344 break;
6345 }
6346 break;
6347 }
6348
6349 // Look at all of the tokens within this range.
6350 while (MoreTokens()) {
6351 const unsigned I = NextToken();
6352 SourceLocation TokLoc = GetTokenLoc(I);
6353 switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
6354 case RangeBefore:
6355 llvm_unreachable("Infeasible");
6356 case RangeAfter:
6357 break;
6358 case RangeOverlap:
Argyrios Kyrtzidis5d47a9b2013-02-13 18:33:28 +00006359 // For macro expansions, just note where the beginning of the macro
6360 // expansion occurs.
6361 if (cursor.kind == CXCursor_MacroExpansion) {
6362 if (TokLoc == cursorRange.getBegin())
6363 Cursors[I] = cursor;
6364 AdvanceToken();
6365 break;
6366 }
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006367 // We may have already annotated macro names inside macro definitions.
6368 if (Cursors[I].kind != CXCursor_MacroExpansion)
6369 Cursors[I] = cursor;
Guy Benyei11169dd2012-12-18 14:30:41 +00006370 AdvanceToken();
Guy Benyei11169dd2012-12-18 14:30:41 +00006371 continue;
6372 }
6373 break;
6374 }
6375
6376 // Save the preprocessing token index; restore the non-preprocessing
6377 // token index.
6378 PreprocessingTokIdx = TokIdx;
6379 TokIdx = SavedTokIdx;
6380 return CXChildVisit_Recurse;
6381 }
6382
6383 if (cursorRange.isInvalid())
6384 return CXChildVisit_Continue;
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006385
6386 unsigned BeforeReachingCursorIdx = NextToken();
Guy Benyei11169dd2012-12-18 14:30:41 +00006387 const enum CXCursorKind cursorK = clang_getCursorKind(cursor);
Guy Benyei11169dd2012-12-18 14:30:41 +00006388 const enum CXCursorKind K = clang_getCursorKind(parent);
6389 const CXCursor updateC =
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006390 (clang_isInvalid(K) || K == CXCursor_TranslationUnit ||
6391 // Attributes are annotated out-of-order, skip tokens until we reach it.
6392 clang_isAttribute(cursor.kind))
Guy Benyei11169dd2012-12-18 14:30:41 +00006393 ? clang_getNullCursor() : parent;
6394
6395 annotateAndAdvanceTokens(updateC, RangeBefore, cursorRange);
6396
6397 // Avoid having the cursor of an expression "overwrite" the annotation of the
6398 // variable declaration that it belongs to.
6399 // This can happen for C++ constructor expressions whose range generally
6400 // include the variable declaration, e.g.:
6401 // MyCXXClass foo; // Make sure we don't annotate 'foo' as a CallExpr cursor.
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006402 if (clang_isExpression(cursorK) && MoreTokens()) {
Dmitri Gribenkoe8354062013-01-26 15:29:08 +00006403 const Expr *E = getCursorExpr(cursor);
Dmitri Gribenkoa1691182013-01-26 18:12:08 +00006404 if (const Decl *D = getCursorParentDecl(cursor)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00006405 const unsigned I = NextToken();
6406 if (E->getLocStart().isValid() && D->getLocation().isValid() &&
6407 E->getLocStart() == D->getLocation() &&
6408 E->getLocStart() == GetTokenLoc(I)) {
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006409 updateCursorAnnotation(Cursors[I], updateC);
Guy Benyei11169dd2012-12-18 14:30:41 +00006410 AdvanceToken();
6411 }
6412 }
6413 }
6414
6415 // Before recursing into the children keep some state that we are going
6416 // to use in the AnnotateTokensWorker::postVisitChildren callback to do some
6417 // extra work after the child nodes are visited.
6418 // Note that we don't call VisitChildren here to avoid traversing statements
6419 // code-recursively which can blow the stack.
6420
6421 PostChildrenInfo Info;
6422 Info.Cursor = cursor;
6423 Info.CursorRange = cursorRange;
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006424 Info.BeforeReachingCursorIdx = BeforeReachingCursorIdx;
Guy Benyei11169dd2012-12-18 14:30:41 +00006425 Info.BeforeChildrenTokenIdx = NextToken();
6426 PostChildrenInfos.push_back(Info);
6427
6428 return CXChildVisit_Recurse;
6429}
6430
6431bool AnnotateTokensWorker::postVisitChildren(CXCursor cursor) {
6432 if (PostChildrenInfos.empty())
6433 return false;
6434 const PostChildrenInfo &Info = PostChildrenInfos.back();
6435 if (!clang_equalCursors(Info.Cursor, cursor))
6436 return false;
6437
6438 const unsigned BeforeChildren = Info.BeforeChildrenTokenIdx;
6439 const unsigned AfterChildren = NextToken();
6440 SourceRange cursorRange = Info.CursorRange;
6441
6442 // Scan the tokens that are at the end of the cursor, but are not captured
6443 // but the child cursors.
6444 annotateAndAdvanceTokens(cursor, RangeOverlap, cursorRange);
6445
6446 // Scan the tokens that are at the beginning of the cursor, but are not
6447 // capture by the child cursors.
6448 for (unsigned I = BeforeChildren; I != AfterChildren; ++I) {
6449 if (!clang_isInvalid(clang_getCursorKind(Cursors[I])))
6450 break;
6451
6452 Cursors[I] = cursor;
6453 }
6454
Argyrios Kyrtzidisa2ed8132013-02-08 01:12:25 +00006455 // Attributes are annotated out-of-order, rewind TokIdx to when we first
6456 // encountered the attribute cursor.
6457 if (clang_isAttribute(cursor.kind))
6458 TokIdx = Info.BeforeReachingCursorIdx;
6459
Guy Benyei11169dd2012-12-18 14:30:41 +00006460 PostChildrenInfos.pop_back();
6461 return false;
6462}
6463
6464static enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
6465 CXCursor parent,
6466 CXClientData client_data) {
6467 return static_cast<AnnotateTokensWorker*>(client_data)->Visit(cursor, parent);
6468}
6469
6470static bool AnnotateTokensPostChildrenVisitor(CXCursor cursor,
6471 CXClientData client_data) {
6472 return static_cast<AnnotateTokensWorker*>(client_data)->
6473 postVisitChildren(cursor);
6474}
6475
6476namespace {
6477
6478/// \brief Uses the macro expansions in the preprocessing record to find
6479/// and mark tokens that are macro arguments. This info is used by the
6480/// AnnotateTokensWorker.
6481class MarkMacroArgTokensVisitor {
6482 SourceManager &SM;
6483 CXToken *Tokens;
6484 unsigned NumTokens;
6485 unsigned CurIdx;
6486
6487public:
6488 MarkMacroArgTokensVisitor(SourceManager &SM,
6489 CXToken *tokens, unsigned numTokens)
6490 : SM(SM), Tokens(tokens), NumTokens(numTokens), CurIdx(0) { }
6491
6492 CXChildVisitResult visit(CXCursor cursor, CXCursor parent) {
6493 if (cursor.kind != CXCursor_MacroExpansion)
6494 return CXChildVisit_Continue;
6495
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00006496 SourceRange macroRange = getCursorMacroExpansion(cursor).getSourceRange();
Guy Benyei11169dd2012-12-18 14:30:41 +00006497 if (macroRange.getBegin() == macroRange.getEnd())
6498 return CXChildVisit_Continue; // it's not a function macro.
6499
6500 for (; CurIdx < NumTokens; ++CurIdx) {
6501 if (!SM.isBeforeInTranslationUnit(getTokenLoc(CurIdx),
6502 macroRange.getBegin()))
6503 break;
6504 }
6505
6506 if (CurIdx == NumTokens)
6507 return CXChildVisit_Break;
6508
6509 for (; CurIdx < NumTokens; ++CurIdx) {
6510 SourceLocation tokLoc = getTokenLoc(CurIdx);
6511 if (!SM.isBeforeInTranslationUnit(tokLoc, macroRange.getEnd()))
6512 break;
6513
6514 setFunctionMacroTokenLoc(CurIdx, SM.getMacroArgExpandedLocation(tokLoc));
6515 }
6516
6517 if (CurIdx == NumTokens)
6518 return CXChildVisit_Break;
6519
6520 return CXChildVisit_Continue;
6521 }
6522
6523private:
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006524 CXToken &getTok(unsigned Idx) {
6525 assert(Idx < NumTokens);
6526 return Tokens[Idx];
6527 }
6528 const CXToken &getTok(unsigned Idx) const {
6529 assert(Idx < NumTokens);
6530 return Tokens[Idx];
6531 }
6532
Guy Benyei11169dd2012-12-18 14:30:41 +00006533 SourceLocation getTokenLoc(unsigned tokI) {
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006534 return SourceLocation::getFromRawEncoding(getTok(tokI).int_data[1]);
Guy Benyei11169dd2012-12-18 14:30:41 +00006535 }
6536
6537 void setFunctionMacroTokenLoc(unsigned tokI, SourceLocation loc) {
6538 // The third field is reserved and currently not used. Use it here
6539 // to mark macro arg expanded tokens with their expanded locations.
Argyrios Kyrtzidis50126f12013-11-27 05:50:55 +00006540 getTok(tokI).int_data[3] = loc.getRawEncoding();
Guy Benyei11169dd2012-12-18 14:30:41 +00006541 }
6542};
6543
6544} // end anonymous namespace
6545
6546static CXChildVisitResult
6547MarkMacroArgTokensVisitorDelegate(CXCursor cursor, CXCursor parent,
6548 CXClientData client_data) {
6549 return static_cast<MarkMacroArgTokensVisitor*>(client_data)->visit(cursor,
6550 parent);
6551}
6552
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006553/// \brief Used by \c annotatePreprocessorTokens.
6554/// \returns true if lexing was finished, false otherwise.
6555static bool lexNext(Lexer &Lex, Token &Tok,
6556 unsigned &NextIdx, unsigned NumTokens) {
6557 if (NextIdx >= NumTokens)
6558 return true;
6559
6560 ++NextIdx;
6561 Lex.LexFromRawLexer(Tok);
Alexander Kornienko1a9f1842015-12-28 15:24:08 +00006562 return Tok.is(tok::eof);
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006563}
6564
Guy Benyei11169dd2012-12-18 14:30:41 +00006565static void annotatePreprocessorTokens(CXTranslationUnit TU,
6566 SourceRange RegionOfInterest,
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006567 CXCursor *Cursors,
6568 CXToken *Tokens,
6569 unsigned NumTokens) {
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006570 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00006571
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006572 Preprocessor &PP = CXXUnit->getPreprocessor();
Guy Benyei11169dd2012-12-18 14:30:41 +00006573 SourceManager &SourceMgr = CXXUnit->getSourceManager();
6574 std::pair<FileID, unsigned> BeginLocInfo
Argyrios Kyrtzidis5d47a9b2013-02-13 18:33:28 +00006575 = SourceMgr.getDecomposedSpellingLoc(RegionOfInterest.getBegin());
Guy Benyei11169dd2012-12-18 14:30:41 +00006576 std::pair<FileID, unsigned> EndLocInfo
Argyrios Kyrtzidis5d47a9b2013-02-13 18:33:28 +00006577 = SourceMgr.getDecomposedSpellingLoc(RegionOfInterest.getEnd());
Guy Benyei11169dd2012-12-18 14:30:41 +00006578
6579 if (BeginLocInfo.first != EndLocInfo.first)
6580 return;
6581
6582 StringRef Buffer;
6583 bool Invalid = false;
6584 Buffer = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
6585 if (Buffer.empty() || Invalid)
6586 return;
6587
6588 Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
6589 CXXUnit->getASTContext().getLangOpts(),
6590 Buffer.begin(), Buffer.data() + BeginLocInfo.second,
6591 Buffer.end());
6592 Lex.SetCommentRetentionState(true);
6593
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006594 unsigned NextIdx = 0;
Guy Benyei11169dd2012-12-18 14:30:41 +00006595 // Lex tokens in raw mode until we hit the end of the range, to avoid
6596 // entering #includes or expanding macros.
6597 while (true) {
6598 Token Tok;
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006599 if (lexNext(Lex, Tok, NextIdx, NumTokens))
6600 break;
6601 unsigned TokIdx = NextIdx-1;
6602 assert(Tok.getLocation() ==
6603 SourceLocation::getFromRawEncoding(Tokens[TokIdx].int_data[1]));
Guy Benyei11169dd2012-12-18 14:30:41 +00006604
6605 reprocess:
6606 if (Tok.is(tok::hash) && Tok.isAtStartOfLine()) {
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006607 // We have found a preprocessing directive. Annotate the tokens
6608 // appropriately.
Guy Benyei11169dd2012-12-18 14:30:41 +00006609 //
6610 // FIXME: Some simple tests here could identify macro definitions and
6611 // #undefs, to provide specific cursor kinds for those.
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006612
6613 SourceLocation BeginLoc = Tok.getLocation();
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006614 if (lexNext(Lex, Tok, NextIdx, NumTokens))
6615 break;
6616
Craig Topper69186e72014-06-08 08:38:04 +00006617 MacroInfo *MI = nullptr;
Alp Toker2d57cea2014-05-17 04:53:25 +00006618 if (Tok.is(tok::raw_identifier) && Tok.getRawIdentifier() == "define") {
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006619 if (lexNext(Lex, Tok, NextIdx, NumTokens))
6620 break;
6621
6622 if (Tok.is(tok::raw_identifier)) {
Alp Toker2d57cea2014-05-17 04:53:25 +00006623 IdentifierInfo &II =
6624 PP.getIdentifierTable().get(Tok.getRawIdentifier());
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006625 SourceLocation MappedTokLoc =
6626 CXXUnit->mapLocationToPreamble(Tok.getLocation());
6627 MI = getMacroInfo(II, MappedTokLoc, TU);
6628 }
6629 }
6630
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006631 bool finished = false;
Guy Benyei11169dd2012-12-18 14:30:41 +00006632 do {
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006633 if (lexNext(Lex, Tok, NextIdx, NumTokens)) {
6634 finished = true;
6635 break;
6636 }
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006637 // If we are in a macro definition, check if the token was ever a
6638 // macro name and annotate it if that's the case.
6639 if (MI) {
6640 SourceLocation SaveLoc = Tok.getLocation();
6641 Tok.setLocation(CXXUnit->mapLocationToPreamble(SaveLoc));
Richard Smith66a81862015-05-04 02:25:31 +00006642 MacroDefinitionRecord *MacroDef =
6643 checkForMacroInMacroDefinition(MI, Tok, TU);
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006644 Tok.setLocation(SaveLoc);
6645 if (MacroDef)
Richard Smith66a81862015-05-04 02:25:31 +00006646 Cursors[NextIdx - 1] =
6647 MakeMacroExpansionCursor(MacroDef, Tok.getLocation(), TU);
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006648 }
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006649 } while (!Tok.isAtStartOfLine());
6650
6651 unsigned LastIdx = finished ? NextIdx-1 : NextIdx-2;
6652 assert(TokIdx <= LastIdx);
6653 SourceLocation EndLoc =
6654 SourceLocation::getFromRawEncoding(Tokens[LastIdx].int_data[1]);
6655 CXCursor Cursor =
6656 MakePreprocessingDirectiveCursor(SourceRange(BeginLoc, EndLoc), TU);
6657
6658 for (; TokIdx <= LastIdx; ++TokIdx)
Argyrios Kyrtzidis68d31ce2013-01-07 19:16:32 +00006659 updateCursorAnnotation(Cursors[TokIdx], Cursor);
Guy Benyei11169dd2012-12-18 14:30:41 +00006660
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006661 if (finished)
6662 break;
6663 goto reprocess;
Guy Benyei11169dd2012-12-18 14:30:41 +00006664 }
Guy Benyei11169dd2012-12-18 14:30:41 +00006665 }
6666}
6667
6668// This gets run a separate thread to avoid stack blowout.
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00006669static void clang_annotateTokensImpl(CXTranslationUnit TU, ASTUnit *CXXUnit,
6670 CXToken *Tokens, unsigned NumTokens,
6671 CXCursor *Cursors) {
Dmitri Gribenko183436e2013-01-26 21:49:50 +00006672 CIndexer *CXXIdx = TU->CIdx;
Guy Benyei11169dd2012-12-18 14:30:41 +00006673 if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForEditing))
6674 setThreadBackgroundPriority();
6675
6676 // Determine the region of interest, which contains all of the tokens.
6677 SourceRange RegionOfInterest;
6678 RegionOfInterest.setBegin(
6679 cxloc::translateSourceLocation(clang_getTokenLocation(TU, Tokens[0])));
6680 RegionOfInterest.setEnd(
6681 cxloc::translateSourceLocation(clang_getTokenLocation(TU,
6682 Tokens[NumTokens-1])));
6683
Guy Benyei11169dd2012-12-18 14:30:41 +00006684 // Relex the tokens within the source range to look for preprocessing
6685 // directives.
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006686 annotatePreprocessorTokens(TU, RegionOfInterest, Cursors, Tokens, NumTokens);
Argyrios Kyrtzidis5d47a9b2013-02-13 18:33:28 +00006687
6688 // If begin location points inside a macro argument, set it to the expansion
6689 // location so we can have the full context when annotating semantically.
6690 {
6691 SourceManager &SM = CXXUnit->getSourceManager();
6692 SourceLocation Loc =
6693 SM.getMacroArgExpandedLocation(RegionOfInterest.getBegin());
6694 if (Loc.isMacroID())
6695 RegionOfInterest.setBegin(SM.getExpansionLoc(Loc));
6696 }
6697
Guy Benyei11169dd2012-12-18 14:30:41 +00006698 if (CXXUnit->getPreprocessor().getPreprocessingRecord()) {
6699 // Search and mark tokens that are macro argument expansions.
6700 MarkMacroArgTokensVisitor Visitor(CXXUnit->getSourceManager(),
6701 Tokens, NumTokens);
6702 CursorVisitor MacroArgMarker(TU,
6703 MarkMacroArgTokensVisitorDelegate, &Visitor,
6704 /*VisitPreprocessorLast=*/true,
6705 /*VisitIncludedEntities=*/false,
6706 RegionOfInterest);
6707 MacroArgMarker.visitPreprocessedEntitiesInRegion();
6708 }
6709
6710 // Annotate all of the source locations in the region of interest that map to
6711 // a specific cursor.
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00006712 AnnotateTokensWorker W(Tokens, Cursors, NumTokens, TU, RegionOfInterest);
Guy Benyei11169dd2012-12-18 14:30:41 +00006713
6714 // FIXME: We use a ridiculous stack size here because the data-recursion
6715 // algorithm uses a large stack frame than the non-data recursive version,
6716 // and AnnotationTokensWorker currently transforms the data-recursion
6717 // algorithm back into a traditional recursion by explicitly calling
6718 // VisitChildren(). We will need to remove this explicit recursive call.
6719 W.AnnotateTokens();
6720
6721 // If we ran into any entities that involve context-sensitive keywords,
6722 // take another pass through the tokens to mark them as such.
6723 if (W.hasContextSensitiveKeywords()) {
6724 for (unsigned I = 0; I != NumTokens; ++I) {
6725 if (clang_getTokenKind(Tokens[I]) != CXToken_Identifier)
6726 continue;
6727
6728 if (Cursors[I].kind == CXCursor_ObjCPropertyDecl) {
6729 IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006730 if (const ObjCPropertyDecl *Property
Guy Benyei11169dd2012-12-18 14:30:41 +00006731 = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(Cursors[I]))) {
6732 if (Property->getPropertyAttributesAsWritten() != 0 &&
6733 llvm::StringSwitch<bool>(II->getName())
6734 .Case("readonly", true)
6735 .Case("assign", true)
6736 .Case("unsafe_unretained", true)
6737 .Case("readwrite", true)
6738 .Case("retain", true)
6739 .Case("copy", true)
6740 .Case("nonatomic", true)
6741 .Case("atomic", true)
6742 .Case("getter", true)
6743 .Case("setter", true)
6744 .Case("strong", true)
6745 .Case("weak", true)
Manman Ren04fd4d82016-05-31 23:22:04 +00006746 .Case("class", true)
Guy Benyei11169dd2012-12-18 14:30:41 +00006747 .Default(false))
6748 Tokens[I].int_data[0] = CXToken_Keyword;
6749 }
6750 continue;
6751 }
6752
6753 if (Cursors[I].kind == CXCursor_ObjCInstanceMethodDecl ||
6754 Cursors[I].kind == CXCursor_ObjCClassMethodDecl) {
6755 IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
6756 if (llvm::StringSwitch<bool>(II->getName())
6757 .Case("in", true)
6758 .Case("out", true)
6759 .Case("inout", true)
6760 .Case("oneway", true)
6761 .Case("bycopy", true)
6762 .Case("byref", true)
6763 .Default(false))
6764 Tokens[I].int_data[0] = CXToken_Keyword;
6765 continue;
6766 }
6767
6768 if (Cursors[I].kind == CXCursor_CXXFinalAttr ||
6769 Cursors[I].kind == CXCursor_CXXOverrideAttr) {
6770 Tokens[I].int_data[0] = CXToken_Keyword;
6771 continue;
6772 }
6773 }
6774 }
6775}
6776
6777extern "C" {
6778
6779void clang_annotateTokens(CXTranslationUnit TU,
6780 CXToken *Tokens, unsigned NumTokens,
6781 CXCursor *Cursors) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00006782 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00006783 LOG_BAD_TU(TU);
6784 return;
6785 }
6786 if (NumTokens == 0 || !Tokens || !Cursors) {
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00006787 LOG_FUNC_SECTION { *Log << "<null input>"; }
Guy Benyei11169dd2012-12-18 14:30:41 +00006788 return;
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00006789 }
6790
6791 LOG_FUNC_SECTION {
6792 *Log << TU << ' ';
6793 CXSourceLocation bloc = clang_getTokenLocation(TU, Tokens[0]);
6794 CXSourceLocation eloc = clang_getTokenLocation(TU, Tokens[NumTokens-1]);
6795 *Log << clang_getRange(bloc, eloc);
6796 }
Guy Benyei11169dd2012-12-18 14:30:41 +00006797
6798 // Any token we don't specifically annotate will have a NULL cursor.
6799 CXCursor C = clang_getNullCursor();
6800 for (unsigned I = 0; I != NumTokens; ++I)
6801 Cursors[I] = C;
6802
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00006803 ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
Guy Benyei11169dd2012-12-18 14:30:41 +00006804 if (!CXXUnit)
6805 return;
6806
6807 ASTUnit::ConcurrencyCheck Check(*CXXUnit);
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00006808
6809 auto AnnotateTokensImpl = [=]() {
6810 clang_annotateTokensImpl(TU, CXXUnit, Tokens, NumTokens, Cursors);
6811 };
Guy Benyei11169dd2012-12-18 14:30:41 +00006812 llvm::CrashRecoveryContext CRC;
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00006813 if (!RunSafely(CRC, AnnotateTokensImpl, GetSafetyThreadStackSize() * 2)) {
Guy Benyei11169dd2012-12-18 14:30:41 +00006814 fprintf(stderr, "libclang: crash detected while annotating tokens\n");
6815 }
6816}
6817
6818} // end: extern "C"
6819
6820//===----------------------------------------------------------------------===//
6821// Operations for querying linkage of a cursor.
6822//===----------------------------------------------------------------------===//
6823
6824extern "C" {
6825CXLinkageKind clang_getCursorLinkage(CXCursor cursor) {
6826 if (!clang_isDeclaration(cursor.kind))
6827 return CXLinkage_Invalid;
6828
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00006829 const Decl *D = cxcursor::getCursorDecl(cursor);
6830 if (const NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D))
Rafael Espindola3ae00052013-05-13 00:12:11 +00006831 switch (ND->getLinkageInternal()) {
Rafael Espindola50df3a02013-05-25 17:16:20 +00006832 case NoLinkage:
6833 case VisibleNoLinkage: return CXLinkage_NoLinkage;
Guy Benyei11169dd2012-12-18 14:30:41 +00006834 case InternalLinkage: return CXLinkage_Internal;
6835 case UniqueExternalLinkage: return CXLinkage_UniqueExternal;
6836 case ExternalLinkage: return CXLinkage_External;
6837 };
6838
6839 return CXLinkage_Invalid;
6840}
6841} // end: extern "C"
6842
6843//===----------------------------------------------------------------------===//
Ehsan Akhgarib743de72016-05-31 15:55:51 +00006844// Operations for querying visibility of a cursor.
6845//===----------------------------------------------------------------------===//
6846
6847extern "C" {
6848CXVisibilityKind clang_getCursorVisibility(CXCursor cursor) {
6849 if (!clang_isDeclaration(cursor.kind))
6850 return CXVisibility_Invalid;
6851
6852 const Decl *D = cxcursor::getCursorDecl(cursor);
6853 if (const NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D))
6854 switch (ND->getVisibility()) {
6855 case HiddenVisibility: return CXVisibility_Hidden;
6856 case ProtectedVisibility: return CXVisibility_Protected;
6857 case DefaultVisibility: return CXVisibility_Default;
6858 };
6859
6860 return CXVisibility_Invalid;
6861}
6862} // end: extern "C"
6863
6864//===----------------------------------------------------------------------===//
Guy Benyei11169dd2012-12-18 14:30:41 +00006865// Operations for querying language of a cursor.
6866//===----------------------------------------------------------------------===//
6867
6868static CXLanguageKind getDeclLanguage(const Decl *D) {
6869 if (!D)
6870 return CXLanguage_C;
6871
6872 switch (D->getKind()) {
6873 default:
6874 break;
6875 case Decl::ImplicitParam:
6876 case Decl::ObjCAtDefsField:
6877 case Decl::ObjCCategory:
6878 case Decl::ObjCCategoryImpl:
6879 case Decl::ObjCCompatibleAlias:
6880 case Decl::ObjCImplementation:
6881 case Decl::ObjCInterface:
6882 case Decl::ObjCIvar:
6883 case Decl::ObjCMethod:
6884 case Decl::ObjCProperty:
6885 case Decl::ObjCPropertyImpl:
6886 case Decl::ObjCProtocol:
Douglas Gregor85f3f952015-07-07 03:57:15 +00006887 case Decl::ObjCTypeParam:
Guy Benyei11169dd2012-12-18 14:30:41 +00006888 return CXLanguage_ObjC;
6889 case Decl::CXXConstructor:
6890 case Decl::CXXConversion:
6891 case Decl::CXXDestructor:
6892 case Decl::CXXMethod:
6893 case Decl::CXXRecord:
6894 case Decl::ClassTemplate:
6895 case Decl::ClassTemplatePartialSpecialization:
6896 case Decl::ClassTemplateSpecialization:
6897 case Decl::Friend:
6898 case Decl::FriendTemplate:
6899 case Decl::FunctionTemplate:
6900 case Decl::LinkageSpec:
6901 case Decl::Namespace:
6902 case Decl::NamespaceAlias:
6903 case Decl::NonTypeTemplateParm:
6904 case Decl::StaticAssert:
6905 case Decl::TemplateTemplateParm:
6906 case Decl::TemplateTypeParm:
6907 case Decl::UnresolvedUsingTypename:
6908 case Decl::UnresolvedUsingValue:
6909 case Decl::Using:
6910 case Decl::UsingDirective:
6911 case Decl::UsingShadow:
6912 return CXLanguage_CPlusPlus;
6913 }
6914
6915 return CXLanguage_C;
6916}
6917
6918extern "C" {
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006919
6920static CXAvailabilityKind getCursorAvailabilityForDecl(const Decl *D) {
6921 if (isa<FunctionDecl>(D) && cast<FunctionDecl>(D)->isDeleted())
Manuel Klimek8e3a7ed2015-09-25 17:53:16 +00006922 return CXAvailability_NotAvailable;
Guy Benyei11169dd2012-12-18 14:30:41 +00006923
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006924 switch (D->getAvailability()) {
6925 case AR_Available:
6926 case AR_NotYetIntroduced:
6927 if (const EnumConstantDecl *EnumConst = dyn_cast<EnumConstantDecl>(D))
Benjamin Kramer656363d2013-10-15 18:53:18 +00006928 return getCursorAvailabilityForDecl(
6929 cast<Decl>(EnumConst->getDeclContext()));
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006930 return CXAvailability_Available;
6931
6932 case AR_Deprecated:
6933 return CXAvailability_Deprecated;
6934
6935 case AR_Unavailable:
6936 return CXAvailability_NotAvailable;
6937 }
Benjamin Kramer656363d2013-10-15 18:53:18 +00006938
6939 llvm_unreachable("Unknown availability kind!");
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006940}
6941
Guy Benyei11169dd2012-12-18 14:30:41 +00006942enum CXAvailabilityKind clang_getCursorAvailability(CXCursor cursor) {
6943 if (clang_isDeclaration(cursor.kind))
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006944 if (const Decl *D = cxcursor::getCursorDecl(cursor))
6945 return getCursorAvailabilityForDecl(D);
Guy Benyei11169dd2012-12-18 14:30:41 +00006946
6947 return CXAvailability_Available;
6948}
6949
6950static CXVersion convertVersion(VersionTuple In) {
6951 CXVersion Out = { -1, -1, -1 };
6952 if (In.empty())
6953 return Out;
6954
6955 Out.Major = In.getMajor();
6956
NAKAMURA Takumic2b5d1f2013-02-21 02:32:34 +00006957 Optional<unsigned> Minor = In.getMinor();
6958 if (Minor.hasValue())
Guy Benyei11169dd2012-12-18 14:30:41 +00006959 Out.Minor = *Minor;
6960 else
6961 return Out;
6962
NAKAMURA Takumic2b5d1f2013-02-21 02:32:34 +00006963 Optional<unsigned> Subminor = In.getSubminor();
6964 if (Subminor.hasValue())
Guy Benyei11169dd2012-12-18 14:30:41 +00006965 Out.Subminor = *Subminor;
6966
6967 return Out;
6968}
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006969
6970static int getCursorPlatformAvailabilityForDecl(const Decl *D,
6971 int *always_deprecated,
6972 CXString *deprecated_message,
6973 int *always_unavailable,
6974 CXString *unavailable_message,
6975 CXPlatformAvailability *availability,
6976 int availability_size) {
6977 bool HadAvailAttr = false;
6978 int N = 0;
Aaron Ballmanb97112e2014-03-08 22:19:01 +00006979 for (auto A : D->attrs()) {
6980 if (DeprecatedAttr *Deprecated = dyn_cast<DeprecatedAttr>(A)) {
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006981 HadAvailAttr = true;
6982 if (always_deprecated)
6983 *always_deprecated = 1;
Nico Weberaacf0312014-04-24 05:16:45 +00006984 if (deprecated_message) {
Argyrios Kyrtzidisedfe07f2014-04-24 06:05:40 +00006985 clang_disposeString(*deprecated_message);
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006986 *deprecated_message = cxstring::createDup(Deprecated->getMessage());
Nico Weberaacf0312014-04-24 05:16:45 +00006987 }
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006988 continue;
6989 }
6990
Aaron Ballmanb97112e2014-03-08 22:19:01 +00006991 if (UnavailableAttr *Unavailable = dyn_cast<UnavailableAttr>(A)) {
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006992 HadAvailAttr = true;
6993 if (always_unavailable)
6994 *always_unavailable = 1;
6995 if (unavailable_message) {
Argyrios Kyrtzidisedfe07f2014-04-24 06:05:40 +00006996 clang_disposeString(*unavailable_message);
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00006997 *unavailable_message = cxstring::createDup(Unavailable->getMessage());
6998 }
6999 continue;
7000 }
7001
Aaron Ballmanb97112e2014-03-08 22:19:01 +00007002 if (AvailabilityAttr *Avail = dyn_cast<AvailabilityAttr>(A)) {
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007003 HadAvailAttr = true;
7004 if (N < availability_size) {
7005 availability[N].Platform
7006 = cxstring::createDup(Avail->getPlatform()->getName());
7007 availability[N].Introduced = convertVersion(Avail->getIntroduced());
7008 availability[N].Deprecated = convertVersion(Avail->getDeprecated());
7009 availability[N].Obsoleted = convertVersion(Avail->getObsoleted());
7010 availability[N].Unavailable = Avail->getUnavailable();
7011 availability[N].Message = cxstring::createDup(Avail->getMessage());
7012 }
7013 ++N;
7014 }
7015 }
7016
7017 if (!HadAvailAttr)
7018 if (const EnumConstantDecl *EnumConst = dyn_cast<EnumConstantDecl>(D))
7019 return getCursorPlatformAvailabilityForDecl(
7020 cast<Decl>(EnumConst->getDeclContext()),
7021 always_deprecated,
7022 deprecated_message,
7023 always_unavailable,
7024 unavailable_message,
7025 availability,
7026 availability_size);
Guy Benyei11169dd2012-12-18 14:30:41 +00007027
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007028 return N;
7029}
7030
Guy Benyei11169dd2012-12-18 14:30:41 +00007031int clang_getCursorPlatformAvailability(CXCursor cursor,
7032 int *always_deprecated,
7033 CXString *deprecated_message,
7034 int *always_unavailable,
7035 CXString *unavailable_message,
7036 CXPlatformAvailability *availability,
7037 int availability_size) {
7038 if (always_deprecated)
7039 *always_deprecated = 0;
7040 if (deprecated_message)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00007041 *deprecated_message = cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00007042 if (always_unavailable)
7043 *always_unavailable = 0;
7044 if (unavailable_message)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00007045 *unavailable_message = cxstring::createEmpty();
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007046
Guy Benyei11169dd2012-12-18 14:30:41 +00007047 if (!clang_isDeclaration(cursor.kind))
7048 return 0;
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007049
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007050 const Decl *D = cxcursor::getCursorDecl(cursor);
Guy Benyei11169dd2012-12-18 14:30:41 +00007051 if (!D)
7052 return 0;
Argyrios Kyrtzidisdc2973f2013-10-15 17:00:53 +00007053
7054 return getCursorPlatformAvailabilityForDecl(D, always_deprecated,
7055 deprecated_message,
7056 always_unavailable,
7057 unavailable_message,
7058 availability,
7059 availability_size);
Guy Benyei11169dd2012-12-18 14:30:41 +00007060}
7061
7062void clang_disposeCXPlatformAvailability(CXPlatformAvailability *availability) {
7063 clang_disposeString(availability->Platform);
7064 clang_disposeString(availability->Message);
7065}
7066
7067CXLanguageKind clang_getCursorLanguage(CXCursor cursor) {
7068 if (clang_isDeclaration(cursor.kind))
7069 return getDeclLanguage(cxcursor::getCursorDecl(cursor));
7070
7071 return CXLanguage_Invalid;
7072}
7073
7074 /// \brief If the given cursor is the "templated" declaration
7075 /// descibing a class or function template, return the class or
7076 /// function template.
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007077static const Decl *maybeGetTemplateCursor(const Decl *D) {
Guy Benyei11169dd2012-12-18 14:30:41 +00007078 if (!D)
Craig Topper69186e72014-06-08 08:38:04 +00007079 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007080
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007081 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00007082 if (FunctionTemplateDecl *FunTmpl = FD->getDescribedFunctionTemplate())
7083 return FunTmpl;
7084
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007085 if (const CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(D))
Guy Benyei11169dd2012-12-18 14:30:41 +00007086 if (ClassTemplateDecl *ClassTmpl = RD->getDescribedClassTemplate())
7087 return ClassTmpl;
7088
7089 return D;
7090}
7091
Argyrios Kyrtzidis4e0854f2014-10-15 17:05:31 +00007092
7093enum CX_StorageClass clang_Cursor_getStorageClass(CXCursor C) {
7094 StorageClass sc = SC_None;
7095 const Decl *D = getCursorDecl(C);
7096 if (D) {
7097 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) {
7098 sc = FD->getStorageClass();
7099 } else if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
7100 sc = VD->getStorageClass();
7101 } else {
7102 return CX_SC_Invalid;
7103 }
7104 } else {
7105 return CX_SC_Invalid;
7106 }
7107 switch (sc) {
7108 case SC_None:
7109 return CX_SC_None;
7110 case SC_Extern:
7111 return CX_SC_Extern;
7112 case SC_Static:
7113 return CX_SC_Static;
7114 case SC_PrivateExtern:
7115 return CX_SC_PrivateExtern;
Argyrios Kyrtzidis4e0854f2014-10-15 17:05:31 +00007116 case SC_Auto:
7117 return CX_SC_Auto;
7118 case SC_Register:
7119 return CX_SC_Register;
Argyrios Kyrtzidis4e0854f2014-10-15 17:05:31 +00007120 }
Kaelyn Takataab61e702014-10-15 18:03:26 +00007121 llvm_unreachable("Unhandled storage class!");
Argyrios Kyrtzidis4e0854f2014-10-15 17:05:31 +00007122}
7123
Guy Benyei11169dd2012-12-18 14:30:41 +00007124CXCursor clang_getCursorSemanticParent(CXCursor cursor) {
7125 if (clang_isDeclaration(cursor.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007126 if (const Decl *D = getCursorDecl(cursor)) {
7127 const DeclContext *DC = D->getDeclContext();
Guy Benyei11169dd2012-12-18 14:30:41 +00007128 if (!DC)
7129 return clang_getNullCursor();
7130
7131 return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
7132 getCursorTU(cursor));
7133 }
7134 }
7135
7136 if (clang_isStatement(cursor.kind) || clang_isExpression(cursor.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007137 if (const Decl *D = getCursorDecl(cursor))
Guy Benyei11169dd2012-12-18 14:30:41 +00007138 return MakeCXCursor(D, getCursorTU(cursor));
7139 }
7140
7141 return clang_getNullCursor();
7142}
7143
7144CXCursor clang_getCursorLexicalParent(CXCursor cursor) {
7145 if (clang_isDeclaration(cursor.kind)) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007146 if (const Decl *D = getCursorDecl(cursor)) {
7147 const DeclContext *DC = D->getLexicalDeclContext();
Guy Benyei11169dd2012-12-18 14:30:41 +00007148 if (!DC)
7149 return clang_getNullCursor();
7150
7151 return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
7152 getCursorTU(cursor));
7153 }
7154 }
7155
7156 // FIXME: Note that we can't easily compute the lexical context of a
7157 // statement or expression, so we return nothing.
7158 return clang_getNullCursor();
7159}
7160
7161CXFile clang_getIncludedFile(CXCursor cursor) {
7162 if (cursor.kind != CXCursor_InclusionDirective)
Craig Topper69186e72014-06-08 08:38:04 +00007163 return nullptr;
7164
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00007165 const InclusionDirective *ID = getCursorInclusionDirective(cursor);
Dmitri Gribenkof9304482013-01-23 15:56:07 +00007166 return const_cast<FileEntry *>(ID->getFile());
Guy Benyei11169dd2012-12-18 14:30:41 +00007167}
7168
Argyrios Kyrtzidis9adfd8a2013-04-18 22:15:49 +00007169unsigned clang_Cursor_getObjCPropertyAttributes(CXCursor C, unsigned reserved) {
7170 if (C.kind != CXCursor_ObjCPropertyDecl)
7171 return CXObjCPropertyAttr_noattr;
7172
7173 unsigned Result = CXObjCPropertyAttr_noattr;
7174 const ObjCPropertyDecl *PD = dyn_cast<ObjCPropertyDecl>(getCursorDecl(C));
7175 ObjCPropertyDecl::PropertyAttributeKind Attr =
7176 PD->getPropertyAttributesAsWritten();
7177
7178#define SET_CXOBJCPROP_ATTR(A) \
7179 if (Attr & ObjCPropertyDecl::OBJC_PR_##A) \
7180 Result |= CXObjCPropertyAttr_##A
7181 SET_CXOBJCPROP_ATTR(readonly);
7182 SET_CXOBJCPROP_ATTR(getter);
7183 SET_CXOBJCPROP_ATTR(assign);
7184 SET_CXOBJCPROP_ATTR(readwrite);
7185 SET_CXOBJCPROP_ATTR(retain);
7186 SET_CXOBJCPROP_ATTR(copy);
7187 SET_CXOBJCPROP_ATTR(nonatomic);
7188 SET_CXOBJCPROP_ATTR(setter);
7189 SET_CXOBJCPROP_ATTR(atomic);
7190 SET_CXOBJCPROP_ATTR(weak);
7191 SET_CXOBJCPROP_ATTR(strong);
7192 SET_CXOBJCPROP_ATTR(unsafe_unretained);
Manman Ren04fd4d82016-05-31 23:22:04 +00007193 SET_CXOBJCPROP_ATTR(class);
Argyrios Kyrtzidis9adfd8a2013-04-18 22:15:49 +00007194#undef SET_CXOBJCPROP_ATTR
7195
7196 return Result;
7197}
7198
Argyrios Kyrtzidis9d9bc012013-04-18 23:29:12 +00007199unsigned clang_Cursor_getObjCDeclQualifiers(CXCursor C) {
7200 if (!clang_isDeclaration(C.kind))
7201 return CXObjCDeclQualifier_None;
7202
7203 Decl::ObjCDeclQualifier QT = Decl::OBJC_TQ_None;
7204 const Decl *D = getCursorDecl(C);
7205 if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D))
7206 QT = MD->getObjCDeclQualifier();
7207 else if (const ParmVarDecl *PD = dyn_cast<ParmVarDecl>(D))
7208 QT = PD->getObjCDeclQualifier();
7209 if (QT == Decl::OBJC_TQ_None)
7210 return CXObjCDeclQualifier_None;
7211
7212 unsigned Result = CXObjCDeclQualifier_None;
7213 if (QT & Decl::OBJC_TQ_In) Result |= CXObjCDeclQualifier_In;
7214 if (QT & Decl::OBJC_TQ_Inout) Result |= CXObjCDeclQualifier_Inout;
7215 if (QT & Decl::OBJC_TQ_Out) Result |= CXObjCDeclQualifier_Out;
7216 if (QT & Decl::OBJC_TQ_Bycopy) Result |= CXObjCDeclQualifier_Bycopy;
7217 if (QT & Decl::OBJC_TQ_Byref) Result |= CXObjCDeclQualifier_Byref;
7218 if (QT & Decl::OBJC_TQ_Oneway) Result |= CXObjCDeclQualifier_Oneway;
7219
7220 return Result;
7221}
7222
Argyrios Kyrtzidis7b50fc52013-07-05 20:44:37 +00007223unsigned clang_Cursor_isObjCOptional(CXCursor C) {
7224 if (!clang_isDeclaration(C.kind))
7225 return 0;
7226
7227 const Decl *D = getCursorDecl(C);
7228 if (const ObjCPropertyDecl *PD = dyn_cast<ObjCPropertyDecl>(D))
7229 return PD->getPropertyImplementation() == ObjCPropertyDecl::Optional;
7230 if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D))
7231 return MD->getImplementationControl() == ObjCMethodDecl::Optional;
7232
7233 return 0;
7234}
7235
Argyrios Kyrtzidis23814e42013-04-18 23:53:05 +00007236unsigned clang_Cursor_isVariadic(CXCursor C) {
7237 if (!clang_isDeclaration(C.kind))
7238 return 0;
7239
7240 const Decl *D = getCursorDecl(C);
7241 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
7242 return FD->isVariadic();
7243 if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D))
7244 return MD->isVariadic();
7245
7246 return 0;
7247}
7248
Guy Benyei11169dd2012-12-18 14:30:41 +00007249CXSourceRange clang_Cursor_getCommentRange(CXCursor C) {
7250 if (!clang_isDeclaration(C.kind))
7251 return clang_getNullRange();
7252
7253 const Decl *D = getCursorDecl(C);
7254 ASTContext &Context = getCursorContext(C);
7255 const RawComment *RC = Context.getRawCommentForAnyRedecl(D);
7256 if (!RC)
7257 return clang_getNullRange();
7258
7259 return cxloc::translateSourceRange(Context, RC->getSourceRange());
7260}
7261
7262CXString clang_Cursor_getRawCommentText(CXCursor C) {
7263 if (!clang_isDeclaration(C.kind))
Dmitri Gribenkof98dfba2013-02-01 14:13:32 +00007264 return cxstring::createNull();
Guy Benyei11169dd2012-12-18 14:30:41 +00007265
7266 const Decl *D = getCursorDecl(C);
7267 ASTContext &Context = getCursorContext(C);
7268 const RawComment *RC = Context.getRawCommentForAnyRedecl(D);
7269 StringRef RawText = RC ? RC->getRawText(Context.getSourceManager()) :
7270 StringRef();
7271
7272 // Don't duplicate the string because RawText points directly into source
7273 // code.
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00007274 return cxstring::createRef(RawText);
Guy Benyei11169dd2012-12-18 14:30:41 +00007275}
7276
7277CXString clang_Cursor_getBriefCommentText(CXCursor C) {
7278 if (!clang_isDeclaration(C.kind))
Dmitri Gribenkof98dfba2013-02-01 14:13:32 +00007279 return cxstring::createNull();
Guy Benyei11169dd2012-12-18 14:30:41 +00007280
7281 const Decl *D = getCursorDecl(C);
7282 const ASTContext &Context = getCursorContext(C);
7283 const RawComment *RC = Context.getRawCommentForAnyRedecl(D);
7284
7285 if (RC) {
7286 StringRef BriefText = RC->getBriefText(Context);
7287
7288 // Don't duplicate the string because RawComment ensures that this memory
7289 // will not go away.
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00007290 return cxstring::createRef(BriefText);
Guy Benyei11169dd2012-12-18 14:30:41 +00007291 }
7292
Dmitri Gribenkof98dfba2013-02-01 14:13:32 +00007293 return cxstring::createNull();
Guy Benyei11169dd2012-12-18 14:30:41 +00007294}
7295
Guy Benyei11169dd2012-12-18 14:30:41 +00007296CXModule clang_Cursor_getModule(CXCursor C) {
7297 if (C.kind == CXCursor_ModuleImportDecl) {
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007298 if (const ImportDecl *ImportD =
7299 dyn_cast_or_null<ImportDecl>(getCursorDecl(C)))
Guy Benyei11169dd2012-12-18 14:30:41 +00007300 return ImportD->getImportedModule();
7301 }
7302
Craig Topper69186e72014-06-08 08:38:04 +00007303 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007304}
7305
Argyrios Kyrtzidisf6d49c32014-05-14 23:14:37 +00007306CXModule clang_getModuleForFile(CXTranslationUnit TU, CXFile File) {
7307 if (isNotUsableTU(TU)) {
7308 LOG_BAD_TU(TU);
7309 return nullptr;
7310 }
7311 if (!File)
7312 return nullptr;
7313 FileEntry *FE = static_cast<FileEntry *>(File);
7314
7315 ASTUnit &Unit = *cxtu::getASTUnit(TU);
7316 HeaderSearch &HS = Unit.getPreprocessor().getHeaderSearchInfo();
7317 ModuleMap::KnownHeader Header = HS.findModuleForHeader(FE);
7318
Richard Smithfeb54b62014-10-23 02:01:19 +00007319 return Header.getModule();
Argyrios Kyrtzidisf6d49c32014-05-14 23:14:37 +00007320}
7321
Argyrios Kyrtzidis12fdb9e2013-04-26 22:47:49 +00007322CXFile clang_Module_getASTFile(CXModule CXMod) {
7323 if (!CXMod)
Craig Topper69186e72014-06-08 08:38:04 +00007324 return nullptr;
Argyrios Kyrtzidis12fdb9e2013-04-26 22:47:49 +00007325 Module *Mod = static_cast<Module*>(CXMod);
7326 return const_cast<FileEntry *>(Mod->getASTFile());
7327}
7328
Guy Benyei11169dd2012-12-18 14:30:41 +00007329CXModule clang_Module_getParent(CXModule CXMod) {
7330 if (!CXMod)
Craig Topper69186e72014-06-08 08:38:04 +00007331 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007332 Module *Mod = static_cast<Module*>(CXMod);
7333 return Mod->Parent;
7334}
7335
7336CXString clang_Module_getName(CXModule CXMod) {
7337 if (!CXMod)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00007338 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00007339 Module *Mod = static_cast<Module*>(CXMod);
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00007340 return cxstring::createDup(Mod->Name);
Guy Benyei11169dd2012-12-18 14:30:41 +00007341}
7342
7343CXString clang_Module_getFullName(CXModule CXMod) {
7344 if (!CXMod)
Dmitri Gribenko36a6dd02013-02-01 14:21:22 +00007345 return cxstring::createEmpty();
Guy Benyei11169dd2012-12-18 14:30:41 +00007346 Module *Mod = static_cast<Module*>(CXMod);
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00007347 return cxstring::createDup(Mod->getFullModuleName());
Guy Benyei11169dd2012-12-18 14:30:41 +00007348}
7349
Argyrios Kyrtzidis884337f2014-05-15 04:44:25 +00007350int clang_Module_isSystem(CXModule CXMod) {
7351 if (!CXMod)
7352 return 0;
7353 Module *Mod = static_cast<Module*>(CXMod);
7354 return Mod->IsSystem;
7355}
7356
Argyrios Kyrtzidis3c5305c2013-03-13 21:13:43 +00007357unsigned clang_Module_getNumTopLevelHeaders(CXTranslationUnit TU,
7358 CXModule CXMod) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00007359 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00007360 LOG_BAD_TU(TU);
7361 return 0;
7362 }
7363 if (!CXMod)
Guy Benyei11169dd2012-12-18 14:30:41 +00007364 return 0;
7365 Module *Mod = static_cast<Module*>(CXMod);
Argyrios Kyrtzidis3c5305c2013-03-13 21:13:43 +00007366 FileManager &FileMgr = cxtu::getASTUnit(TU)->getFileManager();
7367 ArrayRef<const FileEntry *> TopHeaders = Mod->getTopHeaders(FileMgr);
7368 return TopHeaders.size();
Guy Benyei11169dd2012-12-18 14:30:41 +00007369}
7370
Argyrios Kyrtzidis3c5305c2013-03-13 21:13:43 +00007371CXFile clang_Module_getTopLevelHeader(CXTranslationUnit TU,
7372 CXModule CXMod, unsigned Index) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00007373 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00007374 LOG_BAD_TU(TU);
Craig Topper69186e72014-06-08 08:38:04 +00007375 return nullptr;
Dmitri Gribenko256454f2014-02-11 14:34:14 +00007376 }
7377 if (!CXMod)
Craig Topper69186e72014-06-08 08:38:04 +00007378 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007379 Module *Mod = static_cast<Module*>(CXMod);
Argyrios Kyrtzidis3c5305c2013-03-13 21:13:43 +00007380 FileManager &FileMgr = cxtu::getASTUnit(TU)->getFileManager();
Guy Benyei11169dd2012-12-18 14:30:41 +00007381
Argyrios Kyrtzidis3c5305c2013-03-13 21:13:43 +00007382 ArrayRef<const FileEntry *> TopHeaders = Mod->getTopHeaders(FileMgr);
7383 if (Index < TopHeaders.size())
7384 return const_cast<FileEntry *>(TopHeaders[Index]);
Guy Benyei11169dd2012-12-18 14:30:41 +00007385
Craig Topper69186e72014-06-08 08:38:04 +00007386 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007387}
7388
7389} // end: extern "C"
7390
7391//===----------------------------------------------------------------------===//
7392// C++ AST instrospection.
7393//===----------------------------------------------------------------------===//
7394
7395extern "C" {
Jonathan Coe29565352016-04-27 12:48:25 +00007396
7397unsigned clang_CXXConstructor_isDefaultConstructor(CXCursor C) {
7398 if (!clang_isDeclaration(C.kind))
7399 return 0;
7400
7401 const Decl *D = cxcursor::getCursorDecl(C);
7402 const CXXConstructorDecl *Constructor =
7403 D ? dyn_cast_or_null<CXXConstructorDecl>(D->getAsFunction()) : nullptr;
7404 return (Constructor && Constructor->isDefaultConstructor()) ? 1 : 0;
7405}
7406
7407unsigned clang_CXXConstructor_isCopyConstructor(CXCursor C) {
7408 if (!clang_isDeclaration(C.kind))
7409 return 0;
7410
7411 const Decl *D = cxcursor::getCursorDecl(C);
7412 const CXXConstructorDecl *Constructor =
7413 D ? dyn_cast_or_null<CXXConstructorDecl>(D->getAsFunction()) : nullptr;
7414 return (Constructor && Constructor->isCopyConstructor()) ? 1 : 0;
7415}
7416
7417unsigned clang_CXXConstructor_isMoveConstructor(CXCursor C) {
7418 if (!clang_isDeclaration(C.kind))
7419 return 0;
7420
7421 const Decl *D = cxcursor::getCursorDecl(C);
7422 const CXXConstructorDecl *Constructor =
7423 D ? dyn_cast_or_null<CXXConstructorDecl>(D->getAsFunction()) : nullptr;
7424 return (Constructor && Constructor->isMoveConstructor()) ? 1 : 0;
7425}
7426
7427unsigned clang_CXXConstructor_isConvertingConstructor(CXCursor C) {
7428 if (!clang_isDeclaration(C.kind))
7429 return 0;
7430
7431 const Decl *D = cxcursor::getCursorDecl(C);
7432 const CXXConstructorDecl *Constructor =
7433 D ? dyn_cast_or_null<CXXConstructorDecl>(D->getAsFunction()) : nullptr;
7434 // Passing 'false' excludes constructors marked 'explicit'.
7435 return (Constructor && Constructor->isConvertingConstructor(false)) ? 1 : 0;
7436}
7437
Saleem Abdulrasool6ea75db2015-10-27 15:50:22 +00007438unsigned clang_CXXField_isMutable(CXCursor C) {
7439 if (!clang_isDeclaration(C.kind))
7440 return 0;
7441
7442 if (const auto D = cxcursor::getCursorDecl(C))
7443 if (const auto FD = dyn_cast_or_null<FieldDecl>(D))
7444 return FD->isMutable() ? 1 : 0;
7445 return 0;
7446}
7447
Dmitri Gribenko62770be2013-05-17 18:38:35 +00007448unsigned clang_CXXMethod_isPureVirtual(CXCursor C) {
7449 if (!clang_isDeclaration(C.kind))
7450 return 0;
7451
Dmitri Gribenko62770be2013-05-17 18:38:35 +00007452 const Decl *D = cxcursor::getCursorDecl(C);
Alp Tokera2794f92014-01-22 07:29:52 +00007453 const CXXMethodDecl *Method =
Craig Topper69186e72014-06-08 08:38:04 +00007454 D ? dyn_cast_or_null<CXXMethodDecl>(D->getAsFunction()) : nullptr;
Dmitri Gribenko62770be2013-05-17 18:38:35 +00007455 return (Method && Method->isVirtual() && Method->isPure()) ? 1 : 0;
7456}
7457
Dmitri Gribenkoe570ede2014-04-07 14:59:13 +00007458unsigned clang_CXXMethod_isConst(CXCursor C) {
7459 if (!clang_isDeclaration(C.kind))
7460 return 0;
7461
7462 const Decl *D = cxcursor::getCursorDecl(C);
7463 const CXXMethodDecl *Method =
Craig Topper69186e72014-06-08 08:38:04 +00007464 D ? dyn_cast_or_null<CXXMethodDecl>(D->getAsFunction()) : nullptr;
Dmitri Gribenkoe570ede2014-04-07 14:59:13 +00007465 return (Method && (Method->getTypeQualifiers() & Qualifiers::Const)) ? 1 : 0;
7466}
7467
Jonathan Coe29565352016-04-27 12:48:25 +00007468unsigned clang_CXXMethod_isDefaulted(CXCursor C) {
7469 if (!clang_isDeclaration(C.kind))
7470 return 0;
7471
7472 const Decl *D = cxcursor::getCursorDecl(C);
7473 const CXXMethodDecl *Method =
7474 D ? dyn_cast_or_null<CXXMethodDecl>(D->getAsFunction()) : nullptr;
7475 return (Method && Method->isDefaulted()) ? 1 : 0;
7476}
7477
Guy Benyei11169dd2012-12-18 14:30:41 +00007478unsigned clang_CXXMethod_isStatic(CXCursor C) {
7479 if (!clang_isDeclaration(C.kind))
7480 return 0;
7481
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007482 const Decl *D = cxcursor::getCursorDecl(C);
Alp Tokera2794f92014-01-22 07:29:52 +00007483 const CXXMethodDecl *Method =
Craig Topper69186e72014-06-08 08:38:04 +00007484 D ? dyn_cast_or_null<CXXMethodDecl>(D->getAsFunction()) : nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007485 return (Method && Method->isStatic()) ? 1 : 0;
7486}
7487
7488unsigned clang_CXXMethod_isVirtual(CXCursor C) {
7489 if (!clang_isDeclaration(C.kind))
7490 return 0;
7491
Dmitri Gribenkod15bb302013-01-23 17:25:27 +00007492 const Decl *D = cxcursor::getCursorDecl(C);
Alp Tokera2794f92014-01-22 07:29:52 +00007493 const CXXMethodDecl *Method =
Craig Topper69186e72014-06-08 08:38:04 +00007494 D ? dyn_cast_or_null<CXXMethodDecl>(D->getAsFunction()) : nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +00007495 return (Method && Method->isVirtual()) ? 1 : 0;
7496}
7497} // end: extern "C"
7498
7499//===----------------------------------------------------------------------===//
7500// Attribute introspection.
7501//===----------------------------------------------------------------------===//
7502
7503extern "C" {
7504CXType clang_getIBOutletCollectionType(CXCursor C) {
7505 if (C.kind != CXCursor_IBOutletCollectionAttr)
7506 return cxtype::MakeCXType(QualType(), cxcursor::getCursorTU(C));
7507
Dmitri Gribenkoe4baea62013-01-26 18:08:08 +00007508 const IBOutletCollectionAttr *A =
Guy Benyei11169dd2012-12-18 14:30:41 +00007509 cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(C));
7510
7511 return cxtype::MakeCXType(A->getInterface(), cxcursor::getCursorTU(C));
7512}
7513} // end: extern "C"
7514
7515//===----------------------------------------------------------------------===//
7516// Inspecting memory usage.
7517//===----------------------------------------------------------------------===//
7518
7519typedef std::vector<CXTUResourceUsageEntry> MemUsageEntries;
7520
7521static inline void createCXTUResourceUsageEntry(MemUsageEntries &entries,
7522 enum CXTUResourceUsageKind k,
7523 unsigned long amount) {
7524 CXTUResourceUsageEntry entry = { k, amount };
7525 entries.push_back(entry);
7526}
7527
7528extern "C" {
7529
7530const char *clang_getTUResourceUsageName(CXTUResourceUsageKind kind) {
7531 const char *str = "";
7532 switch (kind) {
7533 case CXTUResourceUsage_AST:
7534 str = "ASTContext: expressions, declarations, and types";
7535 break;
7536 case CXTUResourceUsage_Identifiers:
7537 str = "ASTContext: identifiers";
7538 break;
7539 case CXTUResourceUsage_Selectors:
7540 str = "ASTContext: selectors";
7541 break;
7542 case CXTUResourceUsage_GlobalCompletionResults:
7543 str = "Code completion: cached global results";
7544 break;
7545 case CXTUResourceUsage_SourceManagerContentCache:
7546 str = "SourceManager: content cache allocator";
7547 break;
7548 case CXTUResourceUsage_AST_SideTables:
7549 str = "ASTContext: side tables";
7550 break;
7551 case CXTUResourceUsage_SourceManager_Membuffer_Malloc:
7552 str = "SourceManager: malloc'ed memory buffers";
7553 break;
7554 case CXTUResourceUsage_SourceManager_Membuffer_MMap:
7555 str = "SourceManager: mmap'ed memory buffers";
7556 break;
7557 case CXTUResourceUsage_ExternalASTSource_Membuffer_Malloc:
7558 str = "ExternalASTSource: malloc'ed memory buffers";
7559 break;
7560 case CXTUResourceUsage_ExternalASTSource_Membuffer_MMap:
7561 str = "ExternalASTSource: mmap'ed memory buffers";
7562 break;
7563 case CXTUResourceUsage_Preprocessor:
7564 str = "Preprocessor: malloc'ed memory";
7565 break;
7566 case CXTUResourceUsage_PreprocessingRecord:
7567 str = "Preprocessor: PreprocessingRecord";
7568 break;
7569 case CXTUResourceUsage_SourceManager_DataStructures:
7570 str = "SourceManager: data structures and tables";
7571 break;
7572 case CXTUResourceUsage_Preprocessor_HeaderSearch:
7573 str = "Preprocessor: header search tables";
7574 break;
7575 }
7576 return str;
7577}
7578
7579CXTUResourceUsage clang_getCXTUResourceUsage(CXTranslationUnit TU) {
Dmitri Gribenko852d6222014-02-11 15:02:48 +00007580 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00007581 LOG_BAD_TU(TU);
Craig Topper69186e72014-06-08 08:38:04 +00007582 CXTUResourceUsage usage = { (void*) nullptr, 0, nullptr };
Guy Benyei11169dd2012-12-18 14:30:41 +00007583 return usage;
7584 }
7585
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00007586 ASTUnit *astUnit = cxtu::getASTUnit(TU);
Ahmed Charlesb8984322014-03-07 20:03:18 +00007587 std::unique_ptr<MemUsageEntries> entries(new MemUsageEntries());
Guy Benyei11169dd2012-12-18 14:30:41 +00007588 ASTContext &astContext = astUnit->getASTContext();
7589
7590 // How much memory is used by AST nodes and types?
7591 createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_AST,
7592 (unsigned long) astContext.getASTAllocatedMemory());
7593
7594 // How much memory is used by identifiers?
7595 createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_Identifiers,
7596 (unsigned long) astContext.Idents.getAllocator().getTotalMemory());
7597
7598 // How much memory is used for selectors?
7599 createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_Selectors,
7600 (unsigned long) astContext.Selectors.getTotalMemory());
7601
7602 // How much memory is used by ASTContext's side tables?
7603 createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_AST_SideTables,
7604 (unsigned long) astContext.getSideTableAllocatedMemory());
7605
7606 // How much memory is used for caching global code completion results?
7607 unsigned long completionBytes = 0;
7608 if (GlobalCodeCompletionAllocator *completionAllocator =
Alp Tokerf994cef2014-07-05 03:08:06 +00007609 astUnit->getCachedCompletionAllocator().get()) {
Guy Benyei11169dd2012-12-18 14:30:41 +00007610 completionBytes = completionAllocator->getTotalMemory();
7611 }
7612 createCXTUResourceUsageEntry(*entries,
7613 CXTUResourceUsage_GlobalCompletionResults,
7614 completionBytes);
7615
7616 // How much memory is being used by SourceManager's content cache?
7617 createCXTUResourceUsageEntry(*entries,
7618 CXTUResourceUsage_SourceManagerContentCache,
7619 (unsigned long) astContext.getSourceManager().getContentCacheSize());
7620
7621 // How much memory is being used by the MemoryBuffer's in SourceManager?
7622 const SourceManager::MemoryBufferSizes &srcBufs =
7623 astUnit->getSourceManager().getMemoryBufferSizes();
7624
7625 createCXTUResourceUsageEntry(*entries,
7626 CXTUResourceUsage_SourceManager_Membuffer_Malloc,
7627 (unsigned long) srcBufs.malloc_bytes);
7628 createCXTUResourceUsageEntry(*entries,
7629 CXTUResourceUsage_SourceManager_Membuffer_MMap,
7630 (unsigned long) srcBufs.mmap_bytes);
7631 createCXTUResourceUsageEntry(*entries,
7632 CXTUResourceUsage_SourceManager_DataStructures,
7633 (unsigned long) astContext.getSourceManager()
7634 .getDataStructureSizes());
7635
7636 // How much memory is being used by the ExternalASTSource?
7637 if (ExternalASTSource *esrc = astContext.getExternalSource()) {
7638 const ExternalASTSource::MemoryBufferSizes &sizes =
7639 esrc->getMemoryBufferSizes();
7640
7641 createCXTUResourceUsageEntry(*entries,
7642 CXTUResourceUsage_ExternalASTSource_Membuffer_Malloc,
7643 (unsigned long) sizes.malloc_bytes);
7644 createCXTUResourceUsageEntry(*entries,
7645 CXTUResourceUsage_ExternalASTSource_Membuffer_MMap,
7646 (unsigned long) sizes.mmap_bytes);
7647 }
7648
7649 // How much memory is being used by the Preprocessor?
7650 Preprocessor &pp = astUnit->getPreprocessor();
7651 createCXTUResourceUsageEntry(*entries,
7652 CXTUResourceUsage_Preprocessor,
7653 pp.getTotalMemory());
7654
7655 if (PreprocessingRecord *pRec = pp.getPreprocessingRecord()) {
7656 createCXTUResourceUsageEntry(*entries,
7657 CXTUResourceUsage_PreprocessingRecord,
7658 pRec->getTotalMemory());
7659 }
7660
7661 createCXTUResourceUsageEntry(*entries,
7662 CXTUResourceUsage_Preprocessor_HeaderSearch,
7663 pp.getHeaderSearchInfo().getTotalMemory());
Craig Topper69186e72014-06-08 08:38:04 +00007664
Guy Benyei11169dd2012-12-18 14:30:41 +00007665 CXTUResourceUsage usage = { (void*) entries.get(),
7666 (unsigned) entries->size(),
Alexander Kornienko6ee521c2015-01-23 15:36:10 +00007667 !entries->empty() ? &(*entries)[0] : nullptr };
Ahmed Charles9a16beb2014-03-07 19:33:25 +00007668 entries.release();
Guy Benyei11169dd2012-12-18 14:30:41 +00007669 return usage;
7670}
7671
7672void clang_disposeCXTUResourceUsage(CXTUResourceUsage usage) {
7673 if (usage.data)
7674 delete (MemUsageEntries*) usage.data;
7675}
7676
Argyrios Kyrtzidis0e282ef2013-12-06 18:55:45 +00007677CXSourceRangeList *clang_getSkippedRanges(CXTranslationUnit TU, CXFile file) {
7678 CXSourceRangeList *skipped = new CXSourceRangeList;
Argyrios Kyrtzidis9ef57752013-12-05 08:19:32 +00007679 skipped->count = 0;
Craig Topper69186e72014-06-08 08:38:04 +00007680 skipped->ranges = nullptr;
Argyrios Kyrtzidis9ef57752013-12-05 08:19:32 +00007681
Dmitri Gribenko852d6222014-02-11 15:02:48 +00007682 if (isNotUsableTU(TU)) {
Dmitri Gribenko256454f2014-02-11 14:34:14 +00007683 LOG_BAD_TU(TU);
7684 return skipped;
7685 }
7686
Argyrios Kyrtzidis9ef57752013-12-05 08:19:32 +00007687 if (!file)
7688 return skipped;
7689
7690 ASTUnit *astUnit = cxtu::getASTUnit(TU);
7691 PreprocessingRecord *ppRec = astUnit->getPreprocessor().getPreprocessingRecord();
7692 if (!ppRec)
7693 return skipped;
7694
7695 ASTContext &Ctx = astUnit->getASTContext();
7696 SourceManager &sm = Ctx.getSourceManager();
7697 FileEntry *fileEntry = static_cast<FileEntry *>(file);
7698 FileID wantedFileID = sm.translateFile(fileEntry);
7699
7700 const std::vector<SourceRange> &SkippedRanges = ppRec->getSkippedRanges();
7701 std::vector<SourceRange> wantedRanges;
7702 for (std::vector<SourceRange>::const_iterator i = SkippedRanges.begin(), ei = SkippedRanges.end();
7703 i != ei; ++i) {
7704 if (sm.getFileID(i->getBegin()) == wantedFileID || sm.getFileID(i->getEnd()) == wantedFileID)
7705 wantedRanges.push_back(*i);
7706 }
7707
7708 skipped->count = wantedRanges.size();
7709 skipped->ranges = new CXSourceRange[skipped->count];
7710 for (unsigned i = 0, ei = skipped->count; i != ei; ++i)
7711 skipped->ranges[i] = cxloc::translateSourceRange(Ctx, wantedRanges[i]);
7712
7713 return skipped;
7714}
7715
Argyrios Kyrtzidis0e282ef2013-12-06 18:55:45 +00007716void clang_disposeSourceRangeList(CXSourceRangeList *ranges) {
7717 if (ranges) {
7718 delete[] ranges->ranges;
7719 delete ranges;
Argyrios Kyrtzidis9ef57752013-12-05 08:19:32 +00007720 }
7721}
7722
Guy Benyei11169dd2012-12-18 14:30:41 +00007723} // end extern "C"
7724
7725void clang::PrintLibclangResourceUsage(CXTranslationUnit TU) {
7726 CXTUResourceUsage Usage = clang_getCXTUResourceUsage(TU);
7727 for (unsigned I = 0; I != Usage.numEntries; ++I)
7728 fprintf(stderr, " %s: %lu\n",
7729 clang_getTUResourceUsageName(Usage.entries[I].kind),
7730 Usage.entries[I].amount);
7731
7732 clang_disposeCXTUResourceUsage(Usage);
7733}
7734
7735//===----------------------------------------------------------------------===//
7736// Misc. utility functions.
7737//===----------------------------------------------------------------------===//
7738
7739/// Default to using an 8 MB stack size on "safety" threads.
7740static unsigned SafetyStackThreadSize = 8 << 20;
7741
7742namespace clang {
7743
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00007744bool RunSafely(llvm::CrashRecoveryContext &CRC, llvm::function_ref<void()> Fn,
Guy Benyei11169dd2012-12-18 14:30:41 +00007745 unsigned Size) {
7746 if (!Size)
7747 Size = GetSafetyThreadStackSize();
7748 if (Size)
Benjamin Kramer11a9cd92015-07-25 20:55:44 +00007749 return CRC.RunSafelyOnThread(Fn, Size);
7750 return CRC.RunSafely(Fn);
Guy Benyei11169dd2012-12-18 14:30:41 +00007751}
7752
7753unsigned GetSafetyThreadStackSize() {
7754 return SafetyStackThreadSize;
7755}
7756
7757void SetSafetyThreadStackSize(unsigned Value) {
7758 SafetyStackThreadSize = Value;
7759}
7760
Alexander Kornienkoab9db512015-06-22 23:07:51 +00007761}
Guy Benyei11169dd2012-12-18 14:30:41 +00007762
7763void clang::setThreadBackgroundPriority() {
7764 if (getenv("LIBCLANG_BGPRIO_DISABLE"))
7765 return;
7766
Alp Toker1a86ad22014-07-06 06:24:00 +00007767#ifdef USE_DARWIN_THREADS
Guy Benyei11169dd2012-12-18 14:30:41 +00007768 setpriority(PRIO_DARWIN_THREAD, 0, PRIO_DARWIN_BG);
7769#endif
7770}
7771
7772void cxindex::printDiagsToStderr(ASTUnit *Unit) {
7773 if (!Unit)
7774 return;
7775
7776 for (ASTUnit::stored_diag_iterator D = Unit->stored_diag_begin(),
7777 DEnd = Unit->stored_diag_end();
7778 D != DEnd; ++D) {
Ben Langmuir749323f2014-04-22 17:40:12 +00007779 CXStoredDiagnostic Diag(*D, Unit->getLangOpts());
Guy Benyei11169dd2012-12-18 14:30:41 +00007780 CXString Msg = clang_formatDiagnostic(&Diag,
7781 clang_defaultDiagnosticDisplayOptions());
7782 fprintf(stderr, "%s\n", clang_getCString(Msg));
7783 clang_disposeString(Msg);
7784 }
7785#ifdef LLVM_ON_WIN32
7786 // On Windows, force a flush, since there may be multiple copies of
7787 // stderr and stdout in the file system, all with different buffers
7788 // but writing to the same device.
7789 fflush(stderr);
7790#endif
7791}
7792
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007793MacroInfo *cxindex::getMacroInfo(const IdentifierInfo &II,
7794 SourceLocation MacroDefLoc,
7795 CXTranslationUnit TU){
7796 if (MacroDefLoc.isInvalid() || !TU)
Craig Topper69186e72014-06-08 08:38:04 +00007797 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007798 if (!II.hadMacroDefinition())
Craig Topper69186e72014-06-08 08:38:04 +00007799 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007800
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00007801 ASTUnit *Unit = cxtu::getASTUnit(TU);
Argyrios Kyrtzidis2d77aeb2013-01-07 19:16:30 +00007802 Preprocessor &PP = Unit->getPreprocessor();
Richard Smith20e883e2015-04-29 23:20:19 +00007803 MacroDirective *MD = PP.getLocalMacroDirectiveHistory(&II);
Argyrios Kyrtzidisb6210df2013-03-26 17:17:01 +00007804 if (MD) {
7805 for (MacroDirective::DefInfo
7806 Def = MD->getDefinition(); Def; Def = Def.getPreviousDefinition()) {
7807 if (MacroDefLoc == Def.getMacroInfo()->getDefinitionLoc())
7808 return Def.getMacroInfo();
7809 }
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007810 }
7811
Craig Topper69186e72014-06-08 08:38:04 +00007812 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007813}
7814
Richard Smith66a81862015-05-04 02:25:31 +00007815const MacroInfo *cxindex::getMacroInfo(const MacroDefinitionRecord *MacroDef,
Dmitri Gribenkoba2f7462013-01-11 21:01:49 +00007816 CXTranslationUnit TU) {
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007817 if (!MacroDef || !TU)
Craig Topper69186e72014-06-08 08:38:04 +00007818 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007819 const IdentifierInfo *II = MacroDef->getName();
7820 if (!II)
Craig Topper69186e72014-06-08 08:38:04 +00007821 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007822
7823 return getMacroInfo(*II, MacroDef->getLocation(), TU);
7824}
7825
Richard Smith66a81862015-05-04 02:25:31 +00007826MacroDefinitionRecord *
7827cxindex::checkForMacroInMacroDefinition(const MacroInfo *MI, const Token &Tok,
7828 CXTranslationUnit TU) {
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007829 if (!MI || !TU)
Craig Topper69186e72014-06-08 08:38:04 +00007830 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007831 if (Tok.isNot(tok::raw_identifier))
Craig Topper69186e72014-06-08 08:38:04 +00007832 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007833
7834 if (MI->getNumTokens() == 0)
Craig Topper69186e72014-06-08 08:38:04 +00007835 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007836 SourceRange DefRange(MI->getReplacementToken(0).getLocation(),
7837 MI->getDefinitionEndLoc());
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00007838 ASTUnit *Unit = cxtu::getASTUnit(TU);
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007839
7840 // Check that the token is inside the definition and not its argument list.
7841 SourceManager &SM = Unit->getSourceManager();
7842 if (SM.isBeforeInTranslationUnit(Tok.getLocation(), DefRange.getBegin()))
Craig Topper69186e72014-06-08 08:38:04 +00007843 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007844 if (SM.isBeforeInTranslationUnit(DefRange.getEnd(), Tok.getLocation()))
Craig Topper69186e72014-06-08 08:38:04 +00007845 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007846
7847 Preprocessor &PP = Unit->getPreprocessor();
7848 PreprocessingRecord *PPRec = PP.getPreprocessingRecord();
7849 if (!PPRec)
Craig Topper69186e72014-06-08 08:38:04 +00007850 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007851
Alp Toker2d57cea2014-05-17 04:53:25 +00007852 IdentifierInfo &II = PP.getIdentifierTable().get(Tok.getRawIdentifier());
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007853 if (!II.hadMacroDefinition())
Craig Topper69186e72014-06-08 08:38:04 +00007854 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007855
7856 // Check that the identifier is not one of the macro arguments.
7857 if (std::find(MI->arg_begin(), MI->arg_end(), &II) != MI->arg_end())
Craig Topper69186e72014-06-08 08:38:04 +00007858 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007859
Richard Smith20e883e2015-04-29 23:20:19 +00007860 MacroDirective *InnerMD = PP.getLocalMacroDirectiveHistory(&II);
Argyrios Kyrtzidis09c9e812013-02-20 00:54:57 +00007861 if (!InnerMD)
Craig Topper69186e72014-06-08 08:38:04 +00007862 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007863
Argyrios Kyrtzidisb6210df2013-03-26 17:17:01 +00007864 return PPRec->findMacroDefinition(InnerMD->getMacroInfo());
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007865}
7866
Richard Smith66a81862015-05-04 02:25:31 +00007867MacroDefinitionRecord *
7868cxindex::checkForMacroInMacroDefinition(const MacroInfo *MI, SourceLocation Loc,
7869 CXTranslationUnit TU) {
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007870 if (Loc.isInvalid() || !MI || !TU)
Craig Topper69186e72014-06-08 08:38:04 +00007871 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007872
7873 if (MI->getNumTokens() == 0)
Craig Topper69186e72014-06-08 08:38:04 +00007874 return nullptr;
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00007875 ASTUnit *Unit = cxtu::getASTUnit(TU);
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007876 Preprocessor &PP = Unit->getPreprocessor();
7877 if (!PP.getPreprocessingRecord())
Craig Topper69186e72014-06-08 08:38:04 +00007878 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007879 Loc = Unit->getSourceManager().getSpellingLoc(Loc);
7880 Token Tok;
7881 if (PP.getRawToken(Loc, Tok))
Craig Topper69186e72014-06-08 08:38:04 +00007882 return nullptr;
Argyrios Kyrtzidis579825a2013-01-07 19:16:25 +00007883
7884 return checkForMacroInMacroDefinition(MI, Tok, TU);
7885}
7886
Guy Benyei11169dd2012-12-18 14:30:41 +00007887extern "C" {
7888
7889CXString clang_getClangVersion() {
Dmitri Gribenko2f23e9c2013-02-02 02:19:29 +00007890 return cxstring::createDup(getClangFullVersion());
Guy Benyei11169dd2012-12-18 14:30:41 +00007891}
7892
7893} // end: extern "C"
7894
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007895Logger &cxindex::Logger::operator<<(CXTranslationUnit TU) {
7896 if (TU) {
Dmitri Gribenkoc22ea1c2013-01-26 18:53:38 +00007897 if (ASTUnit *Unit = cxtu::getASTUnit(TU)) {
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007898 LogOS << '<' << Unit->getMainFileName() << '>';
Argyrios Kyrtzidis37f2ab42013-03-05 20:21:14 +00007899 if (Unit->isMainFileAST())
7900 LogOS << " (" << Unit->getASTFileName() << ')';
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007901 return *this;
7902 }
Dmitri Gribenkoea4d1c32014-02-12 19:12:37 +00007903 } else {
7904 LogOS << "<NULL TU>";
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007905 }
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007906 return *this;
7907}
7908
Argyrios Kyrtzidisba4b5f82013-03-08 02:32:26 +00007909Logger &cxindex::Logger::operator<<(const FileEntry *FE) {
7910 *this << FE->getName();
7911 return *this;
7912}
7913
7914Logger &cxindex::Logger::operator<<(CXCursor cursor) {
7915 CXString cursorName = clang_getCursorDisplayName(cursor);
7916 *this << cursorName << "@" << clang_getCursorLocation(cursor);
7917 clang_disposeString(cursorName);
7918 return *this;
7919}
7920
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007921Logger &cxindex::Logger::operator<<(CXSourceLocation Loc) {
7922 CXFile File;
7923 unsigned Line, Column;
Craig Topper69186e72014-06-08 08:38:04 +00007924 clang_getFileLocation(Loc, &File, &Line, &Column, nullptr);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007925 CXString FileName = clang_getFileName(File);
7926 *this << llvm::format("(%s:%d:%d)", clang_getCString(FileName), Line, Column);
7927 clang_disposeString(FileName);
7928 return *this;
7929}
7930
7931Logger &cxindex::Logger::operator<<(CXSourceRange range) {
7932 CXSourceLocation BLoc = clang_getRangeStart(range);
7933 CXSourceLocation ELoc = clang_getRangeEnd(range);
7934
7935 CXFile BFile;
7936 unsigned BLine, BColumn;
Craig Topper69186e72014-06-08 08:38:04 +00007937 clang_getFileLocation(BLoc, &BFile, &BLine, &BColumn, nullptr);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007938
7939 CXFile EFile;
7940 unsigned ELine, EColumn;
Craig Topper69186e72014-06-08 08:38:04 +00007941 clang_getFileLocation(ELoc, &EFile, &ELine, &EColumn, nullptr);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007942
7943 CXString BFileName = clang_getFileName(BFile);
7944 if (BFile == EFile) {
7945 *this << llvm::format("[%s %d:%d-%d:%d]", clang_getCString(BFileName),
7946 BLine, BColumn, ELine, EColumn);
7947 } else {
7948 CXString EFileName = clang_getFileName(EFile);
7949 *this << llvm::format("[%s:%d:%d - ", clang_getCString(BFileName),
7950 BLine, BColumn)
7951 << llvm::format("%s:%d:%d]", clang_getCString(EFileName),
7952 ELine, EColumn);
7953 clang_disposeString(EFileName);
7954 }
7955 clang_disposeString(BFileName);
7956 return *this;
7957}
7958
7959Logger &cxindex::Logger::operator<<(CXString Str) {
7960 *this << clang_getCString(Str);
7961 return *this;
7962}
7963
7964Logger &cxindex::Logger::operator<<(const llvm::format_object_base &Fmt) {
7965 LogOS << Fmt;
7966 return *this;
7967}
7968
Chandler Carruth37ad2582014-06-27 15:14:39 +00007969static llvm::ManagedStatic<llvm::sys::Mutex> LoggingMutex;
7970
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007971cxindex::Logger::~Logger() {
Chandler Carruth37ad2582014-06-27 15:14:39 +00007972 llvm::sys::ScopedLock L(*LoggingMutex);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007973
7974 static llvm::TimeRecord sBeginTR = llvm::TimeRecord::getCurrentTime();
7975
Dmitri Gribenkof8579502013-01-12 19:30:44 +00007976 raw_ostream &OS = llvm::errs();
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007977 OS << "[libclang:" << Name << ':';
7978
Alp Toker1a86ad22014-07-06 06:24:00 +00007979#ifdef USE_DARWIN_THREADS
7980 // TODO: Portability.
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007981 mach_port_t tid = pthread_mach_thread_np(pthread_self());
7982 OS << tid << ':';
7983#endif
7984
7985 llvm::TimeRecord TR = llvm::TimeRecord::getCurrentTime();
7986 OS << llvm::format("%7.4f] ", TR.getWallTime() - sBeginTR.getWallTime());
Yaron Keren09fb7c62015-03-10 07:33:23 +00007987 OS << Msg << '\n';
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007988
7989 if (Trace) {
Zachary Turner1fe2a8d2015-03-05 19:15:09 +00007990 llvm::sys::PrintStackTrace(OS);
Argyrios Kyrtzidisea474352013-01-10 18:54:52 +00007991 OS << "--------------------------------------------------\n";
7992 }
7993}
Benjamin Kramerc1ffdab2016-03-03 08:58:18 +00007994
7995#ifdef CLANG_TOOL_EXTRA_BUILD
7996// This anchor is used to force the linker to link the clang-tidy plugin.
7997extern volatile int ClangTidyPluginAnchorSource;
7998static int LLVM_ATTRIBUTE_UNUSED ClangTidyPluginAnchorDestination =
7999 ClangTidyPluginAnchorSource;
8000#endif