1 //===- CXCursor.cpp - Routines for manipulating CXCursors -----------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 // 9 // This file defines routines for manipulating CXCursors. It should be the 10 // only file that has internal knowledge of the encoding of the data in 11 // CXCursor. 12 // 13 //===----------------------------------------------------------------------===// 14 15 #include "CXCursor.h" 16 #include "CXString.h" 17 #include "CXTranslationUnit.h" 18 #include "CXType.h" 19 #include "clang-c/Index.h" 20 #include "clang/AST/Attr.h" 21 #include "clang/AST/Decl.h" 22 #include "clang/AST/DeclCXX.h" 23 #include "clang/AST/DeclObjC.h" 24 #include "clang/AST/DeclTemplate.h" 25 #include "clang/AST/Expr.h" 26 #include "clang/AST/ExprCXX.h" 27 #include "clang/AST/ExprObjC.h" 28 #include "clang/Frontend/ASTUnit.h" 29 #include "llvm/Support/ErrorHandling.h" 30 31 using namespace clang; 32 using namespace cxcursor; 33 34 CXCursor cxcursor::MakeCXCursorInvalid(CXCursorKind K, CXTranslationUnit TU) { 35 assert(K >= CXCursor_FirstInvalid && K <= CXCursor_LastInvalid); 36 CXCursor C = {K, 0, {nullptr, nullptr, TU}}; 37 return C; 38 } 39 40 static CXCursorKind GetCursorKind(const Attr *A) { 41 assert(A && "Invalid arguments!"); 42 switch (A->getKind()) { 43 default: 44 break; 45 case attr::IBAction: 46 return CXCursor_IBActionAttr; 47 case attr::IBOutlet: 48 return CXCursor_IBOutletAttr; 49 case attr::IBOutletCollection: 50 return CXCursor_IBOutletCollectionAttr; 51 case attr::Final: 52 return CXCursor_CXXFinalAttr; 53 case attr::Override: 54 return CXCursor_CXXOverrideAttr; 55 case attr::Annotate: 56 return CXCursor_AnnotateAttr; 57 case attr::AsmLabel: 58 return CXCursor_AsmLabelAttr; 59 case attr::Packed: 60 return CXCursor_PackedAttr; 61 case attr::Pure: 62 return CXCursor_PureAttr; 63 case attr::Const: 64 return CXCursor_ConstAttr; 65 case attr::NoDuplicate: 66 return CXCursor_NoDuplicateAttr; 67 case attr::CUDAConstant: 68 return CXCursor_CUDAConstantAttr; 69 case attr::CUDADevice: 70 return CXCursor_CUDADeviceAttr; 71 case attr::CUDAGlobal: 72 return CXCursor_CUDAGlobalAttr; 73 case attr::CUDAHost: 74 return CXCursor_CUDAHostAttr; 75 case attr::CUDAShared: 76 return CXCursor_CUDASharedAttr; 77 case attr::Visibility: 78 return CXCursor_VisibilityAttr; 79 case attr::DLLExport: 80 return CXCursor_DLLExport; 81 case attr::DLLImport: 82 return CXCursor_DLLImport; 83 case attr::NSReturnsRetained: 84 return CXCursor_NSReturnsRetained; 85 case attr::NSReturnsNotRetained: 86 return CXCursor_NSReturnsNotRetained; 87 case attr::NSReturnsAutoreleased: 88 return CXCursor_NSReturnsAutoreleased; 89 case attr::NSConsumesSelf: 90 return CXCursor_NSConsumesSelf; 91 case attr::NSConsumed: 92 return CXCursor_NSConsumed; 93 case attr::ObjCException: 94 return CXCursor_ObjCException; 95 case attr::ObjCNSObject: 96 return CXCursor_ObjCNSObject; 97 case attr::ObjCIndependentClass: 98 return CXCursor_ObjCIndependentClass; 99 case attr::ObjCPreciseLifetime: 100 return CXCursor_ObjCPreciseLifetime; 101 case attr::ObjCReturnsInnerPointer: 102 return CXCursor_ObjCReturnsInnerPointer; 103 case attr::ObjCRequiresSuper: 104 return CXCursor_ObjCRequiresSuper; 105 case attr::ObjCRootClass: 106 return CXCursor_ObjCRootClass; 107 case attr::ObjCSubclassingRestricted: 108 return CXCursor_ObjCSubclassingRestricted; 109 case attr::ObjCExplicitProtocolImpl: 110 return CXCursor_ObjCExplicitProtocolImpl; 111 case attr::ObjCDesignatedInitializer: 112 return CXCursor_ObjCDesignatedInitializer; 113 case attr::ObjCRuntimeVisible: 114 return CXCursor_ObjCRuntimeVisible; 115 case attr::ObjCBoxable: 116 return CXCursor_ObjCBoxable; 117 case attr::FlagEnum: 118 return CXCursor_FlagEnum; 119 case attr::Convergent: 120 return CXCursor_ConvergentAttr; 121 case attr::WarnUnused: 122 return CXCursor_WarnUnusedAttr; 123 case attr::WarnUnusedResult: 124 return CXCursor_WarnUnusedResultAttr; 125 case attr::Aligned: 126 return CXCursor_AlignedAttr; 127 } 128 129 return CXCursor_UnexposedAttr; 130 } 131 132 CXCursor cxcursor::MakeCXCursor(const Attr *A, const Decl *Parent, 133 CXTranslationUnit TU) { 134 assert(A && Parent && TU && "Invalid arguments!"); 135 CXCursor C = {GetCursorKind(A), 0, {Parent, A, TU}}; 136 return C; 137 } 138 139 CXCursor cxcursor::MakeCXCursor(const Decl *D, CXTranslationUnit TU, 140 SourceRange RegionOfInterest, 141 bool FirstInDeclGroup) { 142 assert(D && TU && "Invalid arguments!"); 143 144 CXCursorKind K = getCursorKindForDecl(D); 145 146 if (K == CXCursor_ObjCClassMethodDecl || 147 K == CXCursor_ObjCInstanceMethodDecl) { 148 int SelectorIdIndex = -1; 149 // Check if cursor points to a selector id. 150 if (RegionOfInterest.isValid() && 151 RegionOfInterest.getBegin() == RegionOfInterest.getEnd()) { 152 SmallVector<SourceLocation, 16> SelLocs; 153 cast<ObjCMethodDecl>(D)->getSelectorLocs(SelLocs); 154 SmallVectorImpl<SourceLocation>::iterator I = 155 llvm::find(SelLocs, RegionOfInterest.getBegin()); 156 if (I != SelLocs.end()) 157 SelectorIdIndex = I - SelLocs.begin(); 158 } 159 CXCursor C = {K, 160 SelectorIdIndex, 161 {D, (void *)(intptr_t)(FirstInDeclGroup ? 1 : 0), TU}}; 162 return C; 163 } 164 165 CXCursor C = {K, 0, {D, (void *)(intptr_t)(FirstInDeclGroup ? 1 : 0), TU}}; 166 return C; 167 } 168 169 CXCursor cxcursor::MakeCXCursor(const Stmt *S, const Decl *Parent, 170 CXTranslationUnit TU, 171 SourceRange RegionOfInterest) { 172 assert(S && TU && "Invalid arguments!"); 173 CXCursorKind K = CXCursor_NotImplemented; 174 175 switch (S->getStmtClass()) { 176 case Stmt::NoStmtClass: 177 break; 178 179 case Stmt::CaseStmtClass: 180 K = CXCursor_CaseStmt; 181 break; 182 183 case Stmt::DefaultStmtClass: 184 K = CXCursor_DefaultStmt; 185 break; 186 187 case Stmt::IfStmtClass: 188 K = CXCursor_IfStmt; 189 break; 190 191 case Stmt::SwitchStmtClass: 192 K = CXCursor_SwitchStmt; 193 break; 194 195 case Stmt::WhileStmtClass: 196 K = CXCursor_WhileStmt; 197 break; 198 199 case Stmt::DoStmtClass: 200 K = CXCursor_DoStmt; 201 break; 202 203 case Stmt::ForStmtClass: 204 K = CXCursor_ForStmt; 205 break; 206 207 case Stmt::GotoStmtClass: 208 K = CXCursor_GotoStmt; 209 break; 210 211 case Stmt::IndirectGotoStmtClass: 212 K = CXCursor_IndirectGotoStmt; 213 break; 214 215 case Stmt::ContinueStmtClass: 216 K = CXCursor_ContinueStmt; 217 break; 218 219 case Stmt::BreakStmtClass: 220 K = CXCursor_BreakStmt; 221 break; 222 223 case Stmt::ReturnStmtClass: 224 K = CXCursor_ReturnStmt; 225 break; 226 227 case Stmt::GCCAsmStmtClass: 228 K = CXCursor_GCCAsmStmt; 229 break; 230 231 case Stmt::MSAsmStmtClass: 232 K = CXCursor_MSAsmStmt; 233 break; 234 235 case Stmt::ObjCAtTryStmtClass: 236 K = CXCursor_ObjCAtTryStmt; 237 break; 238 239 case Stmt::ObjCAtCatchStmtClass: 240 K = CXCursor_ObjCAtCatchStmt; 241 break; 242 243 case Stmt::ObjCAtFinallyStmtClass: 244 K = CXCursor_ObjCAtFinallyStmt; 245 break; 246 247 case Stmt::ObjCAtThrowStmtClass: 248 K = CXCursor_ObjCAtThrowStmt; 249 break; 250 251 case Stmt::ObjCAtSynchronizedStmtClass: 252 K = CXCursor_ObjCAtSynchronizedStmt; 253 break; 254 255 case Stmt::ObjCAutoreleasePoolStmtClass: 256 K = CXCursor_ObjCAutoreleasePoolStmt; 257 break; 258 259 case Stmt::ObjCForCollectionStmtClass: 260 K = CXCursor_ObjCForCollectionStmt; 261 break; 262 263 case Stmt::CXXCatchStmtClass: 264 K = CXCursor_CXXCatchStmt; 265 break; 266 267 case Stmt::CXXTryStmtClass: 268 K = CXCursor_CXXTryStmt; 269 break; 270 271 case Stmt::CXXForRangeStmtClass: 272 K = CXCursor_CXXForRangeStmt; 273 break; 274 275 case Stmt::SEHTryStmtClass: 276 K = CXCursor_SEHTryStmt; 277 break; 278 279 case Stmt::SEHExceptStmtClass: 280 K = CXCursor_SEHExceptStmt; 281 break; 282 283 case Stmt::SEHFinallyStmtClass: 284 K = CXCursor_SEHFinallyStmt; 285 break; 286 287 case Stmt::SEHLeaveStmtClass: 288 K = CXCursor_SEHLeaveStmt; 289 break; 290 291 case Stmt::CoroutineBodyStmtClass: 292 case Stmt::CoreturnStmtClass: 293 K = CXCursor_UnexposedStmt; 294 break; 295 296 case Stmt::ArrayTypeTraitExprClass: 297 case Stmt::AsTypeExprClass: 298 case Stmt::AtomicExprClass: 299 case Stmt::BinaryConditionalOperatorClass: 300 case Stmt::TypeTraitExprClass: 301 case Stmt::CoawaitExprClass: 302 case Stmt::ConceptSpecializationExprClass: 303 case Stmt::RequiresExprClass: 304 case Stmt::DependentCoawaitExprClass: 305 case Stmt::CoyieldExprClass: 306 case Stmt::CXXBindTemporaryExprClass: 307 case Stmt::CXXDefaultArgExprClass: 308 case Stmt::CXXDefaultInitExprClass: 309 case Stmt::CXXFoldExprClass: 310 case Stmt::CXXRewrittenBinaryOperatorClass: 311 case Stmt::CXXStdInitializerListExprClass: 312 case Stmt::CXXScalarValueInitExprClass: 313 case Stmt::CXXUuidofExprClass: 314 case Stmt::ChooseExprClass: 315 case Stmt::DesignatedInitExprClass: 316 case Stmt::DesignatedInitUpdateExprClass: 317 case Stmt::ArrayInitLoopExprClass: 318 case Stmt::ArrayInitIndexExprClass: 319 case Stmt::ExprWithCleanupsClass: 320 case Stmt::ExpressionTraitExprClass: 321 case Stmt::ExtVectorElementExprClass: 322 case Stmt::ImplicitCastExprClass: 323 case Stmt::ImplicitValueInitExprClass: 324 case Stmt::NoInitExprClass: 325 case Stmt::MaterializeTemporaryExprClass: 326 case Stmt::ObjCIndirectCopyRestoreExprClass: 327 case Stmt::OffsetOfExprClass: 328 case Stmt::ParenListExprClass: 329 case Stmt::PredefinedExprClass: 330 case Stmt::ShuffleVectorExprClass: 331 case Stmt::SourceLocExprClass: 332 case Stmt::ConvertVectorExprClass: 333 case Stmt::VAArgExprClass: 334 case Stmt::ObjCArrayLiteralClass: 335 case Stmt::ObjCDictionaryLiteralClass: 336 case Stmt::ObjCBoxedExprClass: 337 case Stmt::ObjCSubscriptRefExprClass: 338 case Stmt::RecoveryExprClass: 339 K = CXCursor_UnexposedExpr; 340 break; 341 342 case Stmt::OpaqueValueExprClass: 343 if (Expr *Src = cast<OpaqueValueExpr>(S)->getSourceExpr()) 344 return MakeCXCursor(Src, Parent, TU, RegionOfInterest); 345 K = CXCursor_UnexposedExpr; 346 break; 347 348 case Stmt::PseudoObjectExprClass: 349 return MakeCXCursor(cast<PseudoObjectExpr>(S)->getSyntacticForm(), Parent, 350 TU, RegionOfInterest); 351 352 case Stmt::CompoundStmtClass: 353 K = CXCursor_CompoundStmt; 354 break; 355 356 case Stmt::NullStmtClass: 357 K = CXCursor_NullStmt; 358 break; 359 360 case Stmt::LabelStmtClass: 361 K = CXCursor_LabelStmt; 362 break; 363 364 case Stmt::AttributedStmtClass: 365 K = CXCursor_UnexposedStmt; 366 break; 367 368 case Stmt::DeclStmtClass: 369 K = CXCursor_DeclStmt; 370 break; 371 372 case Stmt::CapturedStmtClass: 373 K = CXCursor_UnexposedStmt; 374 break; 375 376 case Stmt::IntegerLiteralClass: 377 K = CXCursor_IntegerLiteral; 378 break; 379 380 case Stmt::FixedPointLiteralClass: 381 K = CXCursor_FixedPointLiteral; 382 break; 383 384 case Stmt::FloatingLiteralClass: 385 K = CXCursor_FloatingLiteral; 386 break; 387 388 case Stmt::ImaginaryLiteralClass: 389 K = CXCursor_ImaginaryLiteral; 390 break; 391 392 case Stmt::StringLiteralClass: 393 K = CXCursor_StringLiteral; 394 break; 395 396 case Stmt::CharacterLiteralClass: 397 K = CXCursor_CharacterLiteral; 398 break; 399 400 case Stmt::ConstantExprClass: 401 return MakeCXCursor(cast<ConstantExpr>(S)->getSubExpr(), Parent, TU, 402 RegionOfInterest); 403 404 case Stmt::ParenExprClass: 405 K = CXCursor_ParenExpr; 406 break; 407 408 case Stmt::UnaryOperatorClass: 409 K = CXCursor_UnaryOperator; 410 break; 411 412 case Stmt::UnaryExprOrTypeTraitExprClass: 413 case Stmt::CXXNoexceptExprClass: 414 K = CXCursor_UnaryExpr; 415 break; 416 417 case Stmt::MSPropertySubscriptExprClass: 418 case Stmt::ArraySubscriptExprClass: 419 K = CXCursor_ArraySubscriptExpr; 420 break; 421 422 case Stmt::OMPArraySectionExprClass: 423 K = CXCursor_OMPArraySectionExpr; 424 break; 425 426 case Stmt::BinaryOperatorClass: 427 K = CXCursor_BinaryOperator; 428 break; 429 430 case Stmt::CompoundAssignOperatorClass: 431 K = CXCursor_CompoundAssignOperator; 432 break; 433 434 case Stmt::ConditionalOperatorClass: 435 K = CXCursor_ConditionalOperator; 436 break; 437 438 case Stmt::CStyleCastExprClass: 439 K = CXCursor_CStyleCastExpr; 440 break; 441 442 case Stmt::CompoundLiteralExprClass: 443 K = CXCursor_CompoundLiteralExpr; 444 break; 445 446 case Stmt::InitListExprClass: 447 K = CXCursor_InitListExpr; 448 break; 449 450 case Stmt::AddrLabelExprClass: 451 K = CXCursor_AddrLabelExpr; 452 break; 453 454 case Stmt::StmtExprClass: 455 K = CXCursor_StmtExpr; 456 break; 457 458 case Stmt::GenericSelectionExprClass: 459 K = CXCursor_GenericSelectionExpr; 460 break; 461 462 case Stmt::GNUNullExprClass: 463 K = CXCursor_GNUNullExpr; 464 break; 465 466 case Stmt::CXXStaticCastExprClass: 467 K = CXCursor_CXXStaticCastExpr; 468 break; 469 470 case Stmt::CXXDynamicCastExprClass: 471 K = CXCursor_CXXDynamicCastExpr; 472 break; 473 474 case Stmt::CXXReinterpretCastExprClass: 475 K = CXCursor_CXXReinterpretCastExpr; 476 break; 477 478 case Stmt::CXXConstCastExprClass: 479 K = CXCursor_CXXConstCastExpr; 480 break; 481 482 case Stmt::CXXFunctionalCastExprClass: 483 K = CXCursor_CXXFunctionalCastExpr; 484 break; 485 486 case Stmt::CXXTypeidExprClass: 487 K = CXCursor_CXXTypeidExpr; 488 break; 489 490 case Stmt::CXXBoolLiteralExprClass: 491 K = CXCursor_CXXBoolLiteralExpr; 492 break; 493 494 case Stmt::CXXNullPtrLiteralExprClass: 495 K = CXCursor_CXXNullPtrLiteralExpr; 496 break; 497 498 case Stmt::CXXThisExprClass: 499 K = CXCursor_CXXThisExpr; 500 break; 501 502 case Stmt::CXXThrowExprClass: 503 K = CXCursor_CXXThrowExpr; 504 break; 505 506 case Stmt::CXXNewExprClass: 507 K = CXCursor_CXXNewExpr; 508 break; 509 510 case Stmt::CXXDeleteExprClass: 511 K = CXCursor_CXXDeleteExpr; 512 break; 513 514 case Stmt::ObjCStringLiteralClass: 515 K = CXCursor_ObjCStringLiteral; 516 break; 517 518 case Stmt::ObjCEncodeExprClass: 519 K = CXCursor_ObjCEncodeExpr; 520 break; 521 522 case Stmt::ObjCSelectorExprClass: 523 K = CXCursor_ObjCSelectorExpr; 524 break; 525 526 case Stmt::ObjCProtocolExprClass: 527 K = CXCursor_ObjCProtocolExpr; 528 break; 529 530 case Stmt::ObjCBoolLiteralExprClass: 531 K = CXCursor_ObjCBoolLiteralExpr; 532 break; 533 534 case Stmt::ObjCAvailabilityCheckExprClass: 535 K = CXCursor_ObjCAvailabilityCheckExpr; 536 break; 537 538 case Stmt::ObjCBridgedCastExprClass: 539 K = CXCursor_ObjCBridgedCastExpr; 540 break; 541 542 case Stmt::BlockExprClass: 543 K = CXCursor_BlockExpr; 544 break; 545 546 case Stmt::PackExpansionExprClass: 547 K = CXCursor_PackExpansionExpr; 548 break; 549 550 case Stmt::SizeOfPackExprClass: 551 K = CXCursor_SizeOfPackExpr; 552 break; 553 554 case Stmt::DeclRefExprClass: 555 if (const ImplicitParamDecl *IPD = dyn_cast_or_null<ImplicitParamDecl>( 556 cast<DeclRefExpr>(S)->getDecl())) { 557 if (const ObjCMethodDecl *MD = 558 dyn_cast<ObjCMethodDecl>(IPD->getDeclContext())) { 559 if (MD->getSelfDecl() == IPD) { 560 K = CXCursor_ObjCSelfExpr; 561 break; 562 } 563 } 564 } 565 566 K = CXCursor_DeclRefExpr; 567 break; 568 569 case Stmt::DependentScopeDeclRefExprClass: 570 case Stmt::SubstNonTypeTemplateParmExprClass: 571 case Stmt::SubstNonTypeTemplateParmPackExprClass: 572 case Stmt::FunctionParmPackExprClass: 573 case Stmt::UnresolvedLookupExprClass: 574 case Stmt::TypoExprClass: // A typo could actually be a DeclRef or a MemberRef 575 K = CXCursor_DeclRefExpr; 576 break; 577 578 case Stmt::CXXDependentScopeMemberExprClass: 579 case Stmt::CXXPseudoDestructorExprClass: 580 case Stmt::MemberExprClass: 581 case Stmt::MSPropertyRefExprClass: 582 case Stmt::ObjCIsaExprClass: 583 case Stmt::ObjCIvarRefExprClass: 584 case Stmt::ObjCPropertyRefExprClass: 585 case Stmt::UnresolvedMemberExprClass: 586 K = CXCursor_MemberRefExpr; 587 break; 588 589 case Stmt::CallExprClass: 590 case Stmt::CXXOperatorCallExprClass: 591 case Stmt::CXXMemberCallExprClass: 592 case Stmt::CUDAKernelCallExprClass: 593 case Stmt::CXXConstructExprClass: 594 case Stmt::CXXInheritedCtorInitExprClass: 595 case Stmt::CXXTemporaryObjectExprClass: 596 case Stmt::CXXUnresolvedConstructExprClass: 597 case Stmt::UserDefinedLiteralClass: 598 K = CXCursor_CallExpr; 599 break; 600 601 case Stmt::LambdaExprClass: 602 K = CXCursor_LambdaExpr; 603 break; 604 605 case Stmt::ObjCMessageExprClass: { 606 K = CXCursor_ObjCMessageExpr; 607 int SelectorIdIndex = -1; 608 // Check if cursor points to a selector id. 609 if (RegionOfInterest.isValid() && 610 RegionOfInterest.getBegin() == RegionOfInterest.getEnd()) { 611 SmallVector<SourceLocation, 16> SelLocs; 612 cast<ObjCMessageExpr>(S)->getSelectorLocs(SelLocs); 613 SmallVectorImpl<SourceLocation>::iterator I = 614 llvm::find(SelLocs, RegionOfInterest.getBegin()); 615 if (I != SelLocs.end()) 616 SelectorIdIndex = I - SelLocs.begin(); 617 } 618 CXCursor C = {K, 0, {Parent, S, TU}}; 619 return getSelectorIdentifierCursor(SelectorIdIndex, C); 620 } 621 622 case Stmt::MSDependentExistsStmtClass: 623 K = CXCursor_UnexposedStmt; 624 break; 625 case Stmt::OMPParallelDirectiveClass: 626 K = CXCursor_OMPParallelDirective; 627 break; 628 case Stmt::OMPSimdDirectiveClass: 629 K = CXCursor_OMPSimdDirective; 630 break; 631 case Stmt::OMPForDirectiveClass: 632 K = CXCursor_OMPForDirective; 633 break; 634 case Stmt::OMPForSimdDirectiveClass: 635 K = CXCursor_OMPForSimdDirective; 636 break; 637 case Stmt::OMPSectionsDirectiveClass: 638 K = CXCursor_OMPSectionsDirective; 639 break; 640 case Stmt::OMPSectionDirectiveClass: 641 K = CXCursor_OMPSectionDirective; 642 break; 643 case Stmt::OMPSingleDirectiveClass: 644 K = CXCursor_OMPSingleDirective; 645 break; 646 case Stmt::OMPMasterDirectiveClass: 647 K = CXCursor_OMPMasterDirective; 648 break; 649 case Stmt::OMPCriticalDirectiveClass: 650 K = CXCursor_OMPCriticalDirective; 651 break; 652 case Stmt::OMPParallelForDirectiveClass: 653 K = CXCursor_OMPParallelForDirective; 654 break; 655 case Stmt::OMPParallelForSimdDirectiveClass: 656 K = CXCursor_OMPParallelForSimdDirective; 657 break; 658 case Stmt::OMPParallelMasterDirectiveClass: 659 K = CXCursor_OMPParallelMasterDirective; 660 break; 661 case Stmt::OMPParallelSectionsDirectiveClass: 662 K = CXCursor_OMPParallelSectionsDirective; 663 break; 664 case Stmt::OMPTaskDirectiveClass: 665 K = CXCursor_OMPTaskDirective; 666 break; 667 case Stmt::OMPTaskyieldDirectiveClass: 668 K = CXCursor_OMPTaskyieldDirective; 669 break; 670 case Stmt::OMPBarrierDirectiveClass: 671 K = CXCursor_OMPBarrierDirective; 672 break; 673 case Stmt::OMPTaskwaitDirectiveClass: 674 K = CXCursor_OMPTaskwaitDirective; 675 break; 676 case Stmt::OMPTaskgroupDirectiveClass: 677 K = CXCursor_OMPTaskgroupDirective; 678 break; 679 case Stmt::OMPFlushDirectiveClass: 680 K = CXCursor_OMPFlushDirective; 681 break; 682 case Stmt::OMPDepobjDirectiveClass: 683 K = CXCursor_OMPDepobjDirective; 684 break; 685 case Stmt::OMPScanDirectiveClass: 686 K = CXCursor_OMPScanDirective; 687 break; 688 case Stmt::OMPOrderedDirectiveClass: 689 K = CXCursor_OMPOrderedDirective; 690 break; 691 case Stmt::OMPAtomicDirectiveClass: 692 K = CXCursor_OMPAtomicDirective; 693 break; 694 case Stmt::OMPTargetDirectiveClass: 695 K = CXCursor_OMPTargetDirective; 696 break; 697 case Stmt::OMPTargetDataDirectiveClass: 698 K = CXCursor_OMPTargetDataDirective; 699 break; 700 case Stmt::OMPTargetEnterDataDirectiveClass: 701 K = CXCursor_OMPTargetEnterDataDirective; 702 break; 703 case Stmt::OMPTargetExitDataDirectiveClass: 704 K = CXCursor_OMPTargetExitDataDirective; 705 break; 706 case Stmt::OMPTargetParallelDirectiveClass: 707 K = CXCursor_OMPTargetParallelDirective; 708 break; 709 case Stmt::OMPTargetParallelForDirectiveClass: 710 K = CXCursor_OMPTargetParallelForDirective; 711 break; 712 case Stmt::OMPTargetUpdateDirectiveClass: 713 K = CXCursor_OMPTargetUpdateDirective; 714 break; 715 case Stmt::OMPTeamsDirectiveClass: 716 K = CXCursor_OMPTeamsDirective; 717 break; 718 case Stmt::OMPCancellationPointDirectiveClass: 719 K = CXCursor_OMPCancellationPointDirective; 720 break; 721 case Stmt::OMPCancelDirectiveClass: 722 K = CXCursor_OMPCancelDirective; 723 break; 724 case Stmt::OMPTaskLoopDirectiveClass: 725 K = CXCursor_OMPTaskLoopDirective; 726 break; 727 case Stmt::OMPTaskLoopSimdDirectiveClass: 728 K = CXCursor_OMPTaskLoopSimdDirective; 729 break; 730 case Stmt::OMPMasterTaskLoopDirectiveClass: 731 K = CXCursor_OMPMasterTaskLoopDirective; 732 break; 733 case Stmt::OMPMasterTaskLoopSimdDirectiveClass: 734 K = CXCursor_OMPMasterTaskLoopSimdDirective; 735 break; 736 case Stmt::OMPParallelMasterTaskLoopDirectiveClass: 737 K = CXCursor_OMPParallelMasterTaskLoopDirective; 738 break; 739 case Stmt::OMPParallelMasterTaskLoopSimdDirectiveClass: 740 K = CXCursor_OMPParallelMasterTaskLoopSimdDirective; 741 break; 742 case Stmt::OMPDistributeDirectiveClass: 743 K = CXCursor_OMPDistributeDirective; 744 break; 745 case Stmt::OMPDistributeParallelForDirectiveClass: 746 K = CXCursor_OMPDistributeParallelForDirective; 747 break; 748 case Stmt::OMPDistributeParallelForSimdDirectiveClass: 749 K = CXCursor_OMPDistributeParallelForSimdDirective; 750 break; 751 case Stmt::OMPDistributeSimdDirectiveClass: 752 K = CXCursor_OMPDistributeSimdDirective; 753 break; 754 case Stmt::OMPTargetParallelForSimdDirectiveClass: 755 K = CXCursor_OMPTargetParallelForSimdDirective; 756 break; 757 case Stmt::OMPTargetSimdDirectiveClass: 758 K = CXCursor_OMPTargetSimdDirective; 759 break; 760 case Stmt::OMPTeamsDistributeDirectiveClass: 761 K = CXCursor_OMPTeamsDistributeDirective; 762 break; 763 case Stmt::OMPTeamsDistributeSimdDirectiveClass: 764 K = CXCursor_OMPTeamsDistributeSimdDirective; 765 break; 766 case Stmt::OMPTeamsDistributeParallelForSimdDirectiveClass: 767 K = CXCursor_OMPTeamsDistributeParallelForSimdDirective; 768 break; 769 case Stmt::OMPTeamsDistributeParallelForDirectiveClass: 770 K = CXCursor_OMPTeamsDistributeParallelForDirective; 771 break; 772 case Stmt::OMPTargetTeamsDirectiveClass: 773 K = CXCursor_OMPTargetTeamsDirective; 774 break; 775 case Stmt::OMPTargetTeamsDistributeDirectiveClass: 776 K = CXCursor_OMPTargetTeamsDistributeDirective; 777 break; 778 case Stmt::OMPTargetTeamsDistributeParallelForDirectiveClass: 779 K = CXCursor_OMPTargetTeamsDistributeParallelForDirective; 780 break; 781 case Stmt::OMPTargetTeamsDistributeParallelForSimdDirectiveClass: 782 K = CXCursor_OMPTargetTeamsDistributeParallelForSimdDirective; 783 break; 784 case Stmt::OMPTargetTeamsDistributeSimdDirectiveClass: 785 K = CXCursor_OMPTargetTeamsDistributeSimdDirective; 786 break; 787 case Stmt::BuiltinBitCastExprClass: 788 K = CXCursor_BuiltinBitCastExpr; 789 } 790 791 CXCursor C = {K, 0, {Parent, S, TU}}; 792 return C; 793 } 794 795 CXCursor cxcursor::MakeCursorObjCSuperClassRef(ObjCInterfaceDecl *Super, 796 SourceLocation Loc, 797 CXTranslationUnit TU) { 798 assert(Super && TU && "Invalid arguments!"); 799 void *RawLoc = Loc.getPtrEncoding(); 800 CXCursor C = {CXCursor_ObjCSuperClassRef, 0, {Super, RawLoc, TU}}; 801 return C; 802 } 803 804 std::pair<const ObjCInterfaceDecl *, SourceLocation> 805 cxcursor::getCursorObjCSuperClassRef(CXCursor C) { 806 assert(C.kind == CXCursor_ObjCSuperClassRef); 807 return std::make_pair(static_cast<const ObjCInterfaceDecl *>(C.data[0]), 808 SourceLocation::getFromPtrEncoding(C.data[1])); 809 } 810 811 CXCursor cxcursor::MakeCursorObjCProtocolRef(const ObjCProtocolDecl *Proto, 812 SourceLocation Loc, 813 CXTranslationUnit TU) { 814 assert(Proto && TU && "Invalid arguments!"); 815 void *RawLoc = Loc.getPtrEncoding(); 816 CXCursor C = {CXCursor_ObjCProtocolRef, 0, {Proto, RawLoc, TU}}; 817 return C; 818 } 819 820 std::pair<const ObjCProtocolDecl *, SourceLocation> 821 cxcursor::getCursorObjCProtocolRef(CXCursor C) { 822 assert(C.kind == CXCursor_ObjCProtocolRef); 823 return std::make_pair(static_cast<const ObjCProtocolDecl *>(C.data[0]), 824 SourceLocation::getFromPtrEncoding(C.data[1])); 825 } 826 827 CXCursor cxcursor::MakeCursorObjCClassRef(const ObjCInterfaceDecl *Class, 828 SourceLocation Loc, 829 CXTranslationUnit TU) { 830 // 'Class' can be null for invalid code. 831 if (!Class) 832 return MakeCXCursorInvalid(CXCursor_InvalidCode); 833 assert(TU && "Invalid arguments!"); 834 void *RawLoc = Loc.getPtrEncoding(); 835 CXCursor C = {CXCursor_ObjCClassRef, 0, {Class, RawLoc, TU}}; 836 return C; 837 } 838 839 std::pair<const ObjCInterfaceDecl *, SourceLocation> 840 cxcursor::getCursorObjCClassRef(CXCursor C) { 841 assert(C.kind == CXCursor_ObjCClassRef); 842 return std::make_pair(static_cast<const ObjCInterfaceDecl *>(C.data[0]), 843 SourceLocation::getFromPtrEncoding(C.data[1])); 844 } 845 846 CXCursor cxcursor::MakeCursorTypeRef(const TypeDecl *Type, SourceLocation Loc, 847 CXTranslationUnit TU) { 848 assert(Type && TU && "Invalid arguments!"); 849 void *RawLoc = Loc.getPtrEncoding(); 850 CXCursor C = {CXCursor_TypeRef, 0, {Type, RawLoc, TU}}; 851 return C; 852 } 853 854 std::pair<const TypeDecl *, SourceLocation> 855 cxcursor::getCursorTypeRef(CXCursor C) { 856 assert(C.kind == CXCursor_TypeRef); 857 return std::make_pair(static_cast<const TypeDecl *>(C.data[0]), 858 SourceLocation::getFromPtrEncoding(C.data[1])); 859 } 860 861 CXCursor cxcursor::MakeCursorTemplateRef(const TemplateDecl *Template, 862 SourceLocation Loc, 863 CXTranslationUnit TU) { 864 assert(Template && TU && "Invalid arguments!"); 865 void *RawLoc = Loc.getPtrEncoding(); 866 CXCursor C = {CXCursor_TemplateRef, 0, {Template, RawLoc, TU}}; 867 return C; 868 } 869 870 std::pair<const TemplateDecl *, SourceLocation> 871 cxcursor::getCursorTemplateRef(CXCursor C) { 872 assert(C.kind == CXCursor_TemplateRef); 873 return std::make_pair(static_cast<const TemplateDecl *>(C.data[0]), 874 SourceLocation::getFromPtrEncoding(C.data[1])); 875 } 876 877 CXCursor cxcursor::MakeCursorNamespaceRef(const NamedDecl *NS, 878 SourceLocation Loc, 879 CXTranslationUnit TU) { 880 881 assert(NS && (isa<NamespaceDecl>(NS) || isa<NamespaceAliasDecl>(NS)) && TU && 882 "Invalid arguments!"); 883 void *RawLoc = Loc.getPtrEncoding(); 884 CXCursor C = {CXCursor_NamespaceRef, 0, {NS, RawLoc, TU}}; 885 return C; 886 } 887 888 std::pair<const NamedDecl *, SourceLocation> 889 cxcursor::getCursorNamespaceRef(CXCursor C) { 890 assert(C.kind == CXCursor_NamespaceRef); 891 return std::make_pair(static_cast<const NamedDecl *>(C.data[0]), 892 SourceLocation::getFromPtrEncoding(C.data[1])); 893 } 894 895 CXCursor cxcursor::MakeCursorVariableRef(const VarDecl *Var, SourceLocation Loc, 896 CXTranslationUnit TU) { 897 898 assert(Var && TU && "Invalid arguments!"); 899 void *RawLoc = Loc.getPtrEncoding(); 900 CXCursor C = {CXCursor_VariableRef, 0, {Var, RawLoc, TU}}; 901 return C; 902 } 903 904 std::pair<const VarDecl *, SourceLocation> 905 cxcursor::getCursorVariableRef(CXCursor C) { 906 assert(C.kind == CXCursor_VariableRef); 907 return std::make_pair(static_cast<const VarDecl *>(C.data[0]), 908 SourceLocation::getFromPtrEncoding(C.data[1])); 909 } 910 911 CXCursor cxcursor::MakeCursorMemberRef(const FieldDecl *Field, 912 SourceLocation Loc, 913 CXTranslationUnit TU) { 914 915 assert(Field && TU && "Invalid arguments!"); 916 void *RawLoc = Loc.getPtrEncoding(); 917 CXCursor C = {CXCursor_MemberRef, 0, {Field, RawLoc, TU}}; 918 return C; 919 } 920 921 std::pair<const FieldDecl *, SourceLocation> 922 cxcursor::getCursorMemberRef(CXCursor C) { 923 assert(C.kind == CXCursor_MemberRef); 924 return std::make_pair(static_cast<const FieldDecl *>(C.data[0]), 925 SourceLocation::getFromPtrEncoding(C.data[1])); 926 } 927 928 CXCursor cxcursor::MakeCursorCXXBaseSpecifier(const CXXBaseSpecifier *B, 929 CXTranslationUnit TU) { 930 CXCursor C = {CXCursor_CXXBaseSpecifier, 0, {B, nullptr, TU}}; 931 return C; 932 } 933 934 const CXXBaseSpecifier *cxcursor::getCursorCXXBaseSpecifier(CXCursor C) { 935 assert(C.kind == CXCursor_CXXBaseSpecifier); 936 return static_cast<const CXXBaseSpecifier *>(C.data[0]); 937 } 938 939 CXCursor cxcursor::MakePreprocessingDirectiveCursor(SourceRange Range, 940 CXTranslationUnit TU) { 941 CXCursor C = { 942 CXCursor_PreprocessingDirective, 943 0, 944 {Range.getBegin().getPtrEncoding(), Range.getEnd().getPtrEncoding(), TU}}; 945 return C; 946 } 947 948 SourceRange cxcursor::getCursorPreprocessingDirective(CXCursor C) { 949 assert(C.kind == CXCursor_PreprocessingDirective); 950 SourceRange Range(SourceLocation::getFromPtrEncoding(C.data[0]), 951 SourceLocation::getFromPtrEncoding(C.data[1])); 952 ASTUnit *TU = getCursorASTUnit(C); 953 return TU->mapRangeFromPreamble(Range); 954 } 955 956 CXCursor cxcursor::MakeMacroDefinitionCursor(const MacroDefinitionRecord *MI, 957 CXTranslationUnit TU) { 958 CXCursor C = {CXCursor_MacroDefinition, 0, {MI, nullptr, TU}}; 959 return C; 960 } 961 962 const MacroDefinitionRecord *cxcursor::getCursorMacroDefinition(CXCursor C) { 963 assert(C.kind == CXCursor_MacroDefinition); 964 return static_cast<const MacroDefinitionRecord *>(C.data[0]); 965 } 966 967 CXCursor cxcursor::MakeMacroExpansionCursor(MacroExpansion *MI, 968 CXTranslationUnit TU) { 969 CXCursor C = {CXCursor_MacroExpansion, 0, {MI, nullptr, TU}}; 970 return C; 971 } 972 973 CXCursor cxcursor::MakeMacroExpansionCursor(MacroDefinitionRecord *MI, 974 SourceLocation Loc, 975 CXTranslationUnit TU) { 976 assert(Loc.isValid()); 977 CXCursor C = {CXCursor_MacroExpansion, 0, {MI, Loc.getPtrEncoding(), TU}}; 978 return C; 979 } 980 981 const IdentifierInfo *cxcursor::MacroExpansionCursor::getName() const { 982 if (isPseudo()) 983 return getAsMacroDefinition()->getName(); 984 return getAsMacroExpansion()->getName(); 985 } 986 const MacroDefinitionRecord * 987 cxcursor::MacroExpansionCursor::getDefinition() const { 988 if (isPseudo()) 989 return getAsMacroDefinition(); 990 return getAsMacroExpansion()->getDefinition(); 991 } 992 SourceRange cxcursor::MacroExpansionCursor::getSourceRange() const { 993 if (isPseudo()) 994 return getPseudoLoc(); 995 return getAsMacroExpansion()->getSourceRange(); 996 } 997 998 CXCursor cxcursor::MakeInclusionDirectiveCursor(InclusionDirective *ID, 999 CXTranslationUnit TU) { 1000 CXCursor C = {CXCursor_InclusionDirective, 0, {ID, nullptr, TU}}; 1001 return C; 1002 } 1003 1004 const InclusionDirective *cxcursor::getCursorInclusionDirective(CXCursor C) { 1005 assert(C.kind == CXCursor_InclusionDirective); 1006 return static_cast<const InclusionDirective *>(C.data[0]); 1007 } 1008 1009 CXCursor cxcursor::MakeCursorLabelRef(LabelStmt *Label, SourceLocation Loc, 1010 CXTranslationUnit TU) { 1011 1012 assert(Label && TU && "Invalid arguments!"); 1013 void *RawLoc = Loc.getPtrEncoding(); 1014 CXCursor C = {CXCursor_LabelRef, 0, {Label, RawLoc, TU}}; 1015 return C; 1016 } 1017 1018 std::pair<const LabelStmt *, SourceLocation> 1019 cxcursor::getCursorLabelRef(CXCursor C) { 1020 assert(C.kind == CXCursor_LabelRef); 1021 return std::make_pair(static_cast<const LabelStmt *>(C.data[0]), 1022 SourceLocation::getFromPtrEncoding(C.data[1])); 1023 } 1024 1025 CXCursor cxcursor::MakeCursorOverloadedDeclRef(const OverloadExpr *E, 1026 CXTranslationUnit TU) { 1027 assert(E && TU && "Invalid arguments!"); 1028 OverloadedDeclRefStorage Storage(E); 1029 void *RawLoc = E->getNameLoc().getPtrEncoding(); 1030 CXCursor C = { 1031 CXCursor_OverloadedDeclRef, 0, {Storage.getOpaqueValue(), RawLoc, TU}}; 1032 return C; 1033 } 1034 1035 CXCursor cxcursor::MakeCursorOverloadedDeclRef(const Decl *D, 1036 SourceLocation Loc, 1037 CXTranslationUnit TU) { 1038 assert(D && TU && "Invalid arguments!"); 1039 void *RawLoc = Loc.getPtrEncoding(); 1040 OverloadedDeclRefStorage Storage(D); 1041 CXCursor C = { 1042 CXCursor_OverloadedDeclRef, 0, {Storage.getOpaqueValue(), RawLoc, TU}}; 1043 return C; 1044 } 1045 1046 CXCursor cxcursor::MakeCursorOverloadedDeclRef(TemplateName Name, 1047 SourceLocation Loc, 1048 CXTranslationUnit TU) { 1049 assert(Name.getAsOverloadedTemplate() && TU && "Invalid arguments!"); 1050 void *RawLoc = Loc.getPtrEncoding(); 1051 OverloadedDeclRefStorage Storage(Name.getAsOverloadedTemplate()); 1052 CXCursor C = { 1053 CXCursor_OverloadedDeclRef, 0, {Storage.getOpaqueValue(), RawLoc, TU}}; 1054 return C; 1055 } 1056 1057 std::pair<cxcursor::OverloadedDeclRefStorage, SourceLocation> 1058 cxcursor::getCursorOverloadedDeclRef(CXCursor C) { 1059 assert(C.kind == CXCursor_OverloadedDeclRef); 1060 return std::make_pair(OverloadedDeclRefStorage::getFromOpaqueValue( 1061 const_cast<void *>(C.data[0])), 1062 SourceLocation::getFromPtrEncoding(C.data[1])); 1063 } 1064 1065 const Decl *cxcursor::getCursorDecl(CXCursor Cursor) { 1066 return static_cast<const Decl *>(Cursor.data[0]); 1067 } 1068 1069 const Expr *cxcursor::getCursorExpr(CXCursor Cursor) { 1070 return dyn_cast_or_null<Expr>(getCursorStmt(Cursor)); 1071 } 1072 1073 const Stmt *cxcursor::getCursorStmt(CXCursor Cursor) { 1074 if (Cursor.kind == CXCursor_ObjCSuperClassRef || 1075 Cursor.kind == CXCursor_ObjCProtocolRef || 1076 Cursor.kind == CXCursor_ObjCClassRef) 1077 return nullptr; 1078 1079 return static_cast<const Stmt *>(Cursor.data[1]); 1080 } 1081 1082 const Attr *cxcursor::getCursorAttr(CXCursor Cursor) { 1083 return static_cast<const Attr *>(Cursor.data[1]); 1084 } 1085 1086 ASTContext &cxcursor::getCursorContext(CXCursor Cursor) { 1087 return getCursorASTUnit(Cursor)->getASTContext(); 1088 } 1089 1090 ASTUnit *cxcursor::getCursorASTUnit(CXCursor Cursor) { 1091 CXTranslationUnit TU = getCursorTU(Cursor); 1092 if (!TU) 1093 return nullptr; 1094 return cxtu::getASTUnit(TU); 1095 } 1096 1097 CXTranslationUnit cxcursor::getCursorTU(CXCursor Cursor) { 1098 return static_cast<CXTranslationUnit>(const_cast<void *>(Cursor.data[2])); 1099 } 1100 1101 void cxcursor::getOverriddenCursors(CXCursor cursor, 1102 SmallVectorImpl<CXCursor> &overridden) { 1103 assert(clang_isDeclaration(cursor.kind)); 1104 const NamedDecl *D = dyn_cast_or_null<NamedDecl>(getCursorDecl(cursor)); 1105 if (!D) 1106 return; 1107 1108 CXTranslationUnit TU = getCursorTU(cursor); 1109 SmallVector<const NamedDecl *, 8> OverDecls; 1110 D->getASTContext().getOverriddenMethods(D, OverDecls); 1111 1112 for (SmallVectorImpl<const NamedDecl *>::iterator I = OverDecls.begin(), 1113 E = OverDecls.end(); 1114 I != E; ++I) { 1115 overridden.push_back(MakeCXCursor(*I, TU)); 1116 } 1117 } 1118 1119 std::pair<int, SourceLocation> 1120 cxcursor::getSelectorIdentifierIndexAndLoc(CXCursor cursor) { 1121 if (cursor.kind == CXCursor_ObjCMessageExpr) { 1122 if (cursor.xdata != -1) 1123 return std::make_pair(cursor.xdata, 1124 cast<ObjCMessageExpr>(getCursorExpr(cursor)) 1125 ->getSelectorLoc(cursor.xdata)); 1126 } else if (cursor.kind == CXCursor_ObjCClassMethodDecl || 1127 cursor.kind == CXCursor_ObjCInstanceMethodDecl) { 1128 if (cursor.xdata != -1) 1129 return std::make_pair(cursor.xdata, 1130 cast<ObjCMethodDecl>(getCursorDecl(cursor)) 1131 ->getSelectorLoc(cursor.xdata)); 1132 } 1133 1134 return std::make_pair(-1, SourceLocation()); 1135 } 1136 1137 CXCursor cxcursor::getSelectorIdentifierCursor(int SelIdx, CXCursor cursor) { 1138 CXCursor newCursor = cursor; 1139 1140 if (cursor.kind == CXCursor_ObjCMessageExpr) { 1141 if (SelIdx == -1 || 1142 unsigned(SelIdx) >= 1143 cast<ObjCMessageExpr>(getCursorExpr(cursor))->getNumSelectorLocs()) 1144 newCursor.xdata = -1; 1145 else 1146 newCursor.xdata = SelIdx; 1147 } else if (cursor.kind == CXCursor_ObjCClassMethodDecl || 1148 cursor.kind == CXCursor_ObjCInstanceMethodDecl) { 1149 if (SelIdx == -1 || 1150 unsigned(SelIdx) >= 1151 cast<ObjCMethodDecl>(getCursorDecl(cursor))->getNumSelectorLocs()) 1152 newCursor.xdata = -1; 1153 else 1154 newCursor.xdata = SelIdx; 1155 } 1156 1157 return newCursor; 1158 } 1159 1160 CXCursor cxcursor::getTypeRefCursor(CXCursor cursor) { 1161 if (cursor.kind != CXCursor_CallExpr) 1162 return cursor; 1163 1164 if (cursor.xdata == 0) 1165 return cursor; 1166 1167 const Expr *E = getCursorExpr(cursor); 1168 TypeSourceInfo *Type = nullptr; 1169 if (const CXXUnresolvedConstructExpr *UnCtor = 1170 dyn_cast<CXXUnresolvedConstructExpr>(E)) { 1171 Type = UnCtor->getTypeSourceInfo(); 1172 } else if (const CXXTemporaryObjectExpr *Tmp = 1173 dyn_cast<CXXTemporaryObjectExpr>(E)) { 1174 Type = Tmp->getTypeSourceInfo(); 1175 } 1176 1177 if (!Type) 1178 return cursor; 1179 1180 CXTranslationUnit TU = getCursorTU(cursor); 1181 QualType Ty = Type->getType(); 1182 TypeLoc TL = Type->getTypeLoc(); 1183 SourceLocation Loc = TL.getBeginLoc(); 1184 1185 if (const ElaboratedType *ElabT = Ty->getAs<ElaboratedType>()) { 1186 Ty = ElabT->getNamedType(); 1187 ElaboratedTypeLoc ElabTL = TL.castAs<ElaboratedTypeLoc>(); 1188 Loc = ElabTL.getNamedTypeLoc().getBeginLoc(); 1189 } 1190 1191 if (const TypedefType *Typedef = Ty->getAs<TypedefType>()) 1192 return MakeCursorTypeRef(Typedef->getDecl(), Loc, TU); 1193 if (const TagType *Tag = Ty->getAs<TagType>()) 1194 return MakeCursorTypeRef(Tag->getDecl(), Loc, TU); 1195 if (const TemplateTypeParmType *TemplP = Ty->getAs<TemplateTypeParmType>()) 1196 return MakeCursorTypeRef(TemplP->getDecl(), Loc, TU); 1197 1198 return cursor; 1199 } 1200 1201 bool cxcursor::operator==(CXCursor X, CXCursor Y) { 1202 return X.kind == Y.kind && X.data[0] == Y.data[0] && X.data[1] == Y.data[1] && 1203 X.data[2] == Y.data[2]; 1204 } 1205 1206 // FIXME: Remove once we can model DeclGroups and their appropriate ranges 1207 // properly in the ASTs. 1208 bool cxcursor::isFirstInDeclGroup(CXCursor C) { 1209 assert(clang_isDeclaration(C.kind)); 1210 return ((uintptr_t)(C.data[1])) != 0; 1211 } 1212 1213 //===----------------------------------------------------------------------===// 1214 // libclang CXCursor APIs 1215 //===----------------------------------------------------------------------===// 1216 1217 int clang_Cursor_isNull(CXCursor cursor) { 1218 return clang_equalCursors(cursor, clang_getNullCursor()); 1219 } 1220 1221 CXTranslationUnit clang_Cursor_getTranslationUnit(CXCursor cursor) { 1222 return getCursorTU(cursor); 1223 } 1224 1225 int clang_Cursor_getNumArguments(CXCursor C) { 1226 if (clang_isDeclaration(C.kind)) { 1227 const Decl *D = cxcursor::getCursorDecl(C); 1228 if (const ObjCMethodDecl *MD = dyn_cast_or_null<ObjCMethodDecl>(D)) 1229 return MD->param_size(); 1230 if (const FunctionDecl *FD = dyn_cast_or_null<FunctionDecl>(D)) 1231 return FD->param_size(); 1232 } 1233 1234 if (clang_isExpression(C.kind)) { 1235 const Expr *E = cxcursor::getCursorExpr(C); 1236 if (const CallExpr *CE = dyn_cast<CallExpr>(E)) { 1237 return CE->getNumArgs(); 1238 } 1239 if (const CXXConstructExpr *CE = dyn_cast<CXXConstructExpr>(E)) { 1240 return CE->getNumArgs(); 1241 } 1242 } 1243 1244 return -1; 1245 } 1246 1247 CXCursor clang_Cursor_getArgument(CXCursor C, unsigned i) { 1248 if (clang_isDeclaration(C.kind)) { 1249 const Decl *D = cxcursor::getCursorDecl(C); 1250 if (const ObjCMethodDecl *MD = dyn_cast_or_null<ObjCMethodDecl>(D)) { 1251 if (i < MD->param_size()) 1252 return cxcursor::MakeCXCursor(MD->parameters()[i], 1253 cxcursor::getCursorTU(C)); 1254 } else if (const FunctionDecl *FD = dyn_cast_or_null<FunctionDecl>(D)) { 1255 if (i < FD->param_size()) 1256 return cxcursor::MakeCXCursor(FD->parameters()[i], 1257 cxcursor::getCursorTU(C)); 1258 } 1259 } 1260 1261 if (clang_isExpression(C.kind)) { 1262 const Expr *E = cxcursor::getCursorExpr(C); 1263 if (const CallExpr *CE = dyn_cast<CallExpr>(E)) { 1264 if (i < CE->getNumArgs()) { 1265 return cxcursor::MakeCXCursor(CE->getArg(i), getCursorDecl(C), 1266 cxcursor::getCursorTU(C)); 1267 } 1268 } 1269 if (const CXXConstructExpr *CE = dyn_cast<CXXConstructExpr>(E)) { 1270 if (i < CE->getNumArgs()) { 1271 return cxcursor::MakeCXCursor(CE->getArg(i), getCursorDecl(C), 1272 cxcursor::getCursorTU(C)); 1273 } 1274 } 1275 } 1276 1277 return clang_getNullCursor(); 1278 } 1279 1280 int clang_Cursor_getNumTemplateArguments(CXCursor C) { 1281 if (clang_getCursorKind(C) != CXCursor_FunctionDecl) { 1282 return -1; 1283 } 1284 1285 const FunctionDecl *FD = 1286 llvm::dyn_cast_or_null<clang::FunctionDecl>(getCursorDecl(C)); 1287 if (!FD) { 1288 return -1; 1289 } 1290 1291 const FunctionTemplateSpecializationInfo *SpecInfo = 1292 FD->getTemplateSpecializationInfo(); 1293 if (!SpecInfo) { 1294 return -1; 1295 } 1296 1297 return SpecInfo->TemplateArguments->size(); 1298 } 1299 1300 enum CXGetTemplateArgumentStatus { 1301 /** The operation completed successfully */ 1302 CXGetTemplateArgumentStatus_Success = 0, 1303 1304 /** The specified cursor did not represent a FunctionDecl. */ 1305 CXGetTemplateArgumentStatus_CursorNotFunctionDecl = -1, 1306 1307 /** The specified cursor was not castable to a FunctionDecl. */ 1308 CXGetTemplateArgumentStatus_BadFunctionDeclCast = -2, 1309 1310 /** A NULL FunctionTemplateSpecializationInfo was retrieved. */ 1311 CXGetTemplateArgumentStatus_NullTemplSpecInfo = -3, 1312 1313 /** An invalid (OOB) argument index was specified */ 1314 CXGetTemplateArgumentStatus_InvalidIndex = -4 1315 }; 1316 1317 static int clang_Cursor_getTemplateArgument(CXCursor C, unsigned I, 1318 TemplateArgument *TA) { 1319 if (clang_getCursorKind(C) != CXCursor_FunctionDecl) { 1320 return CXGetTemplateArgumentStatus_CursorNotFunctionDecl; 1321 } 1322 1323 const FunctionDecl *FD = 1324 llvm::dyn_cast_or_null<clang::FunctionDecl>(getCursorDecl(C)); 1325 if (!FD) { 1326 return CXGetTemplateArgumentStatus_BadFunctionDeclCast; 1327 } 1328 1329 const FunctionTemplateSpecializationInfo *SpecInfo = 1330 FD->getTemplateSpecializationInfo(); 1331 if (!SpecInfo) { 1332 return CXGetTemplateArgumentStatus_NullTemplSpecInfo; 1333 } 1334 1335 if (I >= SpecInfo->TemplateArguments->size()) { 1336 return CXGetTemplateArgumentStatus_InvalidIndex; 1337 } 1338 1339 *TA = SpecInfo->TemplateArguments->get(I); 1340 return 0; 1341 } 1342 1343 enum CXTemplateArgumentKind clang_Cursor_getTemplateArgumentKind(CXCursor C, 1344 unsigned I) { 1345 TemplateArgument TA; 1346 if (clang_Cursor_getTemplateArgument(C, I, &TA)) { 1347 return CXTemplateArgumentKind_Invalid; 1348 } 1349 1350 switch (TA.getKind()) { 1351 case TemplateArgument::Null: 1352 return CXTemplateArgumentKind_Null; 1353 case TemplateArgument::Type: 1354 return CXTemplateArgumentKind_Type; 1355 case TemplateArgument::Declaration: 1356 return CXTemplateArgumentKind_Declaration; 1357 case TemplateArgument::NullPtr: 1358 return CXTemplateArgumentKind_NullPtr; 1359 case TemplateArgument::Integral: 1360 return CXTemplateArgumentKind_Integral; 1361 case TemplateArgument::Template: 1362 return CXTemplateArgumentKind_Template; 1363 case TemplateArgument::TemplateExpansion: 1364 return CXTemplateArgumentKind_TemplateExpansion; 1365 case TemplateArgument::Expression: 1366 return CXTemplateArgumentKind_Expression; 1367 case TemplateArgument::Pack: 1368 return CXTemplateArgumentKind_Pack; 1369 } 1370 1371 return CXTemplateArgumentKind_Invalid; 1372 } 1373 1374 CXType clang_Cursor_getTemplateArgumentType(CXCursor C, unsigned I) { 1375 TemplateArgument TA; 1376 if (clang_Cursor_getTemplateArgument(C, I, &TA) != 1377 CXGetTemplateArgumentStatus_Success) { 1378 return cxtype::MakeCXType(QualType(), getCursorTU(C)); 1379 } 1380 1381 if (TA.getKind() != TemplateArgument::Type) { 1382 return cxtype::MakeCXType(QualType(), getCursorTU(C)); 1383 } 1384 1385 return cxtype::MakeCXType(TA.getAsType(), getCursorTU(C)); 1386 } 1387 1388 long long clang_Cursor_getTemplateArgumentValue(CXCursor C, unsigned I) { 1389 TemplateArgument TA; 1390 if (clang_Cursor_getTemplateArgument(C, I, &TA) != 1391 CXGetTemplateArgumentStatus_Success) { 1392 assert(0 && "Unable to retrieve TemplateArgument"); 1393 return 0; 1394 } 1395 1396 if (TA.getKind() != TemplateArgument::Integral) { 1397 assert(0 && "Passed template argument is not Integral"); 1398 return 0; 1399 } 1400 1401 return TA.getAsIntegral().getSExtValue(); 1402 } 1403 1404 unsigned long long clang_Cursor_getTemplateArgumentUnsignedValue(CXCursor C, 1405 unsigned I) { 1406 TemplateArgument TA; 1407 if (clang_Cursor_getTemplateArgument(C, I, &TA) != 1408 CXGetTemplateArgumentStatus_Success) { 1409 assert(0 && "Unable to retrieve TemplateArgument"); 1410 return 0; 1411 } 1412 1413 if (TA.getKind() != TemplateArgument::Integral) { 1414 assert(0 && "Passed template argument is not Integral"); 1415 return 0; 1416 } 1417 1418 return TA.getAsIntegral().getZExtValue(); 1419 } 1420 1421 //===----------------------------------------------------------------------===// 1422 // CXCursorSet. 1423 //===----------------------------------------------------------------------===// 1424 1425 typedef llvm::DenseMap<CXCursor, unsigned> CXCursorSet_Impl; 1426 1427 static inline CXCursorSet packCXCursorSet(CXCursorSet_Impl *setImpl) { 1428 return (CXCursorSet)setImpl; 1429 } 1430 static inline CXCursorSet_Impl *unpackCXCursorSet(CXCursorSet set) { 1431 return (CXCursorSet_Impl *)set; 1432 } 1433 namespace llvm { 1434 template <> struct DenseMapInfo<CXCursor> { 1435 public: 1436 static inline CXCursor getEmptyKey() { 1437 return MakeCXCursorInvalid(CXCursor_InvalidFile); 1438 } 1439 static inline CXCursor getTombstoneKey() { 1440 return MakeCXCursorInvalid(CXCursor_NoDeclFound); 1441 } 1442 static inline unsigned getHashValue(const CXCursor &cursor) { 1443 return llvm::DenseMapInfo<std::pair<const void *, const void *>>:: 1444 getHashValue(std::make_pair(cursor.data[0], cursor.data[1])); 1445 } 1446 static inline bool isEqual(const CXCursor &x, const CXCursor &y) { 1447 return x.kind == y.kind && x.data[0] == y.data[0] && x.data[1] == y.data[1]; 1448 } 1449 }; 1450 } // namespace llvm 1451 1452 CXCursorSet clang_createCXCursorSet() { 1453 return packCXCursorSet(new CXCursorSet_Impl()); 1454 } 1455 1456 void clang_disposeCXCursorSet(CXCursorSet set) { 1457 delete unpackCXCursorSet(set); 1458 } 1459 1460 unsigned clang_CXCursorSet_contains(CXCursorSet set, CXCursor cursor) { 1461 CXCursorSet_Impl *setImpl = unpackCXCursorSet(set); 1462 if (!setImpl) 1463 return 0; 1464 return setImpl->find(cursor) != setImpl->end(); 1465 } 1466 1467 unsigned clang_CXCursorSet_insert(CXCursorSet set, CXCursor cursor) { 1468 // Do not insert invalid cursors into the set. 1469 if (cursor.kind >= CXCursor_FirstInvalid && 1470 cursor.kind <= CXCursor_LastInvalid) 1471 return 1; 1472 1473 CXCursorSet_Impl *setImpl = unpackCXCursorSet(set); 1474 if (!setImpl) 1475 return 1; 1476 unsigned &entry = (*setImpl)[cursor]; 1477 unsigned flag = entry == 0 ? 1 : 0; 1478 entry = 1; 1479 return flag; 1480 } 1481 1482 CXCompletionString clang_getCursorCompletionString(CXCursor cursor) { 1483 enum CXCursorKind kind = clang_getCursorKind(cursor); 1484 if (clang_isDeclaration(kind)) { 1485 const Decl *decl = getCursorDecl(cursor); 1486 if (const NamedDecl *namedDecl = dyn_cast_or_null<NamedDecl>(decl)) { 1487 ASTUnit *unit = getCursorASTUnit(cursor); 1488 CodeCompletionResult Result(namedDecl, CCP_Declaration); 1489 CodeCompletionString *String = Result.CreateCodeCompletionString( 1490 unit->getASTContext(), unit->getPreprocessor(), 1491 CodeCompletionContext::CCC_Other, 1492 unit->getCodeCompletionTUInfo().getAllocator(), 1493 unit->getCodeCompletionTUInfo(), true); 1494 return String; 1495 } 1496 } else if (kind == CXCursor_MacroDefinition) { 1497 const MacroDefinitionRecord *definition = getCursorMacroDefinition(cursor); 1498 const IdentifierInfo *Macro = definition->getName(); 1499 ASTUnit *unit = getCursorASTUnit(cursor); 1500 CodeCompletionResult Result( 1501 Macro, 1502 unit->getPreprocessor().getMacroDefinition(Macro).getMacroInfo()); 1503 CodeCompletionString *String = Result.CreateCodeCompletionString( 1504 unit->getASTContext(), unit->getPreprocessor(), 1505 CodeCompletionContext::CCC_Other, 1506 unit->getCodeCompletionTUInfo().getAllocator(), 1507 unit->getCodeCompletionTUInfo(), false); 1508 return String; 1509 } 1510 return nullptr; 1511 } 1512 1513 namespace { 1514 struct OverridenCursorsPool { 1515 typedef SmallVector<CXCursor, 2> CursorVec; 1516 std::vector<CursorVec *> AllCursors; 1517 std::vector<CursorVec *> AvailableCursors; 1518 1519 ~OverridenCursorsPool() { 1520 for (std::vector<CursorVec *>::iterator I = AllCursors.begin(), 1521 E = AllCursors.end(); 1522 I != E; ++I) { 1523 delete *I; 1524 } 1525 } 1526 }; 1527 } // namespace 1528 1529 void *cxcursor::createOverridenCXCursorsPool() { 1530 return new OverridenCursorsPool(); 1531 } 1532 1533 void cxcursor::disposeOverridenCXCursorsPool(void *pool) { 1534 delete static_cast<OverridenCursorsPool *>(pool); 1535 } 1536 1537 void clang_getOverriddenCursors(CXCursor cursor, CXCursor **overridden, 1538 unsigned *num_overridden) { 1539 if (overridden) 1540 *overridden = nullptr; 1541 if (num_overridden) 1542 *num_overridden = 0; 1543 1544 CXTranslationUnit TU = cxcursor::getCursorTU(cursor); 1545 1546 if (!overridden || !num_overridden || !TU) 1547 return; 1548 1549 if (!clang_isDeclaration(cursor.kind)) 1550 return; 1551 1552 OverridenCursorsPool &pool = 1553 *static_cast<OverridenCursorsPool *>(TU->OverridenCursorsPool); 1554 1555 OverridenCursorsPool::CursorVec *Vec = nullptr; 1556 1557 if (!pool.AvailableCursors.empty()) { 1558 Vec = pool.AvailableCursors.back(); 1559 pool.AvailableCursors.pop_back(); 1560 } else { 1561 Vec = new OverridenCursorsPool::CursorVec(); 1562 pool.AllCursors.push_back(Vec); 1563 } 1564 1565 // Clear out the vector, but don't free the memory contents. This 1566 // reduces malloc() traffic. 1567 Vec->clear(); 1568 1569 // Use the first entry to contain a back reference to the vector. 1570 // This is a complete hack. 1571 CXCursor backRefCursor = MakeCXCursorInvalid(CXCursor_InvalidFile, TU); 1572 backRefCursor.data[0] = Vec; 1573 assert(cxcursor::getCursorTU(backRefCursor) == TU); 1574 Vec->push_back(backRefCursor); 1575 1576 // Get the overridden cursors. 1577 cxcursor::getOverriddenCursors(cursor, *Vec); 1578 1579 // Did we get any overridden cursors? If not, return Vec to the pool 1580 // of available cursor vectors. 1581 if (Vec->size() == 1) { 1582 pool.AvailableCursors.push_back(Vec); 1583 return; 1584 } 1585 1586 // Now tell the caller about the overridden cursors. 1587 assert(Vec->size() > 1); 1588 *overridden = &((*Vec)[1]); 1589 *num_overridden = Vec->size() - 1; 1590 } 1591 1592 void clang_disposeOverriddenCursors(CXCursor *overridden) { 1593 if (!overridden) 1594 return; 1595 1596 // Use pointer arithmetic to get back the first faux entry 1597 // which has a back-reference to the TU and the vector. 1598 --overridden; 1599 OverridenCursorsPool::CursorVec *Vec = 1600 static_cast<OverridenCursorsPool::CursorVec *>( 1601 const_cast<void *>(overridden->data[0])); 1602 CXTranslationUnit TU = getCursorTU(*overridden); 1603 1604 assert(Vec && TU); 1605 1606 OverridenCursorsPool &pool = 1607 *static_cast<OverridenCursorsPool *>(TU->OverridenCursorsPool); 1608 1609 pool.AvailableCursors.push_back(Vec); 1610 } 1611 1612 int clang_Cursor_isDynamicCall(CXCursor C) { 1613 const Expr *E = nullptr; 1614 if (clang_isExpression(C.kind)) 1615 E = getCursorExpr(C); 1616 if (!E) 1617 return 0; 1618 1619 if (const ObjCMessageExpr *MsgE = dyn_cast<ObjCMessageExpr>(E)) { 1620 if (MsgE->getReceiverKind() != ObjCMessageExpr::Instance) 1621 return false; 1622 if (auto *RecE = dyn_cast<ObjCMessageExpr>( 1623 MsgE->getInstanceReceiver()->IgnoreParenCasts())) { 1624 if (RecE->getMethodFamily() == OMF_alloc) 1625 return false; 1626 } 1627 return true; 1628 } 1629 1630 if (auto *PropRefE = dyn_cast<ObjCPropertyRefExpr>(E)) { 1631 return !PropRefE->isSuperReceiver(); 1632 } 1633 1634 const MemberExpr *ME = nullptr; 1635 if (isa<MemberExpr>(E)) 1636 ME = cast<MemberExpr>(E); 1637 else if (const CallExpr *CE = dyn_cast<CallExpr>(E)) 1638 ME = dyn_cast_or_null<MemberExpr>(CE->getCallee()); 1639 1640 if (ME) { 1641 if (const CXXMethodDecl *MD = 1642 dyn_cast_or_null<CXXMethodDecl>(ME->getMemberDecl())) 1643 return MD->isVirtual() && 1644 ME->performsVirtualDispatch( 1645 cxcursor::getCursorContext(C).getLangOpts()); 1646 } 1647 1648 return 0; 1649 } 1650 1651 CXType clang_Cursor_getReceiverType(CXCursor C) { 1652 CXTranslationUnit TU = cxcursor::getCursorTU(C); 1653 const Expr *E = nullptr; 1654 if (clang_isExpression(C.kind)) 1655 E = getCursorExpr(C); 1656 1657 if (const ObjCMessageExpr *MsgE = dyn_cast_or_null<ObjCMessageExpr>(E)) 1658 return cxtype::MakeCXType(MsgE->getReceiverType(), TU); 1659 1660 if (auto *PropRefE = dyn_cast<ObjCPropertyRefExpr>(E)) { 1661 return cxtype::MakeCXType( 1662 PropRefE->getReceiverType(cxcursor::getCursorContext(C)), TU); 1663 } 1664 1665 const MemberExpr *ME = nullptr; 1666 if (isa<MemberExpr>(E)) 1667 ME = cast<MemberExpr>(E); 1668 else if (const CallExpr *CE = dyn_cast<CallExpr>(E)) 1669 ME = dyn_cast_or_null<MemberExpr>(CE->getCallee()); 1670 1671 if (ME) { 1672 if (dyn_cast_or_null<CXXMethodDecl>(ME->getMemberDecl())) { 1673 auto receiverTy = ME->getBase()->IgnoreImpCasts()->getType(); 1674 return cxtype::MakeCXType(receiverTy, TU); 1675 } 1676 } 1677 1678 return cxtype::MakeCXType(QualType(), TU); 1679 } 1680