1 //===- DeclObjC.cpp - ObjC Declaration AST Node Implementation ------------===//
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
7 //===----------------------------------------------------------------------===//
9 // This file implements the Objective-C related Decl classes.
11 //===----------------------------------------------------------------------===//
13 #include "clang/AST/DeclObjC.h"
14 #include "clang/AST/ASTContext.h"
15 #include "clang/AST/ASTMutationListener.h"
16 #include "clang/AST/Attr.h"
17 #include "clang/AST/Decl.h"
18 #include "clang/AST/DeclBase.h"
19 #include "clang/AST/Stmt.h"
20 #include "clang/AST/Type.h"
21 #include "clang/AST/TypeLoc.h"
22 #include "clang/Basic/IdentifierTable.h"
23 #include "clang/Basic/LLVM.h"
24 #include "clang/Basic/LangOptions.h"
25 #include "clang/Basic/SourceLocation.h"
26 #include "llvm/ADT/None.h"
27 #include "llvm/ADT/SmallString.h"
28 #include "llvm/ADT/SmallVector.h"
29 #include "llvm/Support/Casting.h"
30 #include "llvm/Support/ErrorHandling.h"
31 #include "llvm/Support/raw_ostream.h"
38 using namespace clang;
40 //===----------------------------------------------------------------------===//
42 //===----------------------------------------------------------------------===//
44 void ObjCListBase::set(void *const* InList, unsigned Elts, ASTContext &Ctx) {
46 if (Elts == 0) return; // Setting to an empty list is a noop.
48 List = new (Ctx) void*[Elts];
50 memcpy(List, InList, sizeof(void*)*Elts);
53 void ObjCProtocolList::set(ObjCProtocolDecl* const* InList, unsigned Elts,
54 const SourceLocation *Locs, ASTContext &Ctx) {
58 Locations = new (Ctx) SourceLocation[Elts];
59 memcpy(Locations, Locs, sizeof(SourceLocation) * Elts);
60 set(InList, Elts, Ctx);
63 //===----------------------------------------------------------------------===//
65 //===----------------------------------------------------------------------===//
67 ObjCContainerDecl::ObjCContainerDecl(Kind DK, DeclContext *DC,
68 IdentifierInfo *Id, SourceLocation nameLoc,
69 SourceLocation atStartLoc)
70 : NamedDecl(DK, DC, nameLoc, Id), DeclContext(DK) {
71 setAtStartLoc(atStartLoc);
74 void ObjCContainerDecl::anchor() {}
76 /// getIvarDecl - This method looks up an ivar in this ContextDecl.
79 ObjCContainerDecl::getIvarDecl(IdentifierInfo *Id) const {
80 lookup_result R = lookup(Id);
81 for (lookup_iterator Ivar = R.begin(), IvarEnd = R.end();
82 Ivar != IvarEnd; ++Ivar) {
83 if (auto *ivar = dyn_cast<ObjCIvarDecl>(*Ivar))
89 // Get the local instance/class method declared in this interface.
91 ObjCContainerDecl::getMethod(Selector Sel, bool isInstance,
92 bool AllowHidden) const {
93 // If this context is a hidden protocol definition, don't find any
95 if (const auto *Proto = dyn_cast<ObjCProtocolDecl>(this)) {
96 if (const ObjCProtocolDecl *Def = Proto->getDefinition())
97 if (Def->isHidden() && !AllowHidden)
101 // Since instance & class methods can have the same name, the loop below
102 // ensures we get the correct method.
104 // @interface Whatever
105 // - (int) class_method;
106 // + (float) class_method;
108 lookup_result R = lookup(Sel);
109 for (lookup_iterator Meth = R.begin(), MethEnd = R.end();
110 Meth != MethEnd; ++Meth) {
111 auto *MD = dyn_cast<ObjCMethodDecl>(*Meth);
112 if (MD && MD->isInstanceMethod() == isInstance)
118 /// This routine returns 'true' if a user declared setter method was
119 /// found in the class, its protocols, its super classes or categories.
120 /// It also returns 'true' if one of its categories has declared a 'readwrite'
121 /// property. This is because, user must provide a setter method for the
122 /// category's 'readwrite' property.
123 bool ObjCContainerDecl::HasUserDeclaredSetterMethod(
124 const ObjCPropertyDecl *Property) const {
125 Selector Sel = Property->getSetterName();
126 lookup_result R = lookup(Sel);
127 for (lookup_iterator Meth = R.begin(), MethEnd = R.end();
128 Meth != MethEnd; ++Meth) {
129 auto *MD = dyn_cast<ObjCMethodDecl>(*Meth);
130 if (MD && MD->isInstanceMethod() && !MD->isImplicit())
134 if (const auto *ID = dyn_cast<ObjCInterfaceDecl>(this)) {
135 // Also look into categories, including class extensions, looking
136 // for a user declared instance method.
137 for (const auto *Cat : ID->visible_categories()) {
138 if (ObjCMethodDecl *MD = Cat->getInstanceMethod(Sel))
139 if (!MD->isImplicit())
141 if (Cat->IsClassExtension())
143 // Also search through the categories looking for a 'readwrite'
144 // declaration of this property. If one found, presumably a setter will
145 // be provided (properties declared in categories will not get
146 // auto-synthesized).
147 for (const auto *P : Cat->properties())
148 if (P->getIdentifier() == Property->getIdentifier()) {
149 if (P->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_readwrite)
155 // Also look into protocols, for a user declared instance method.
156 for (const auto *Proto : ID->all_referenced_protocols())
157 if (Proto->HasUserDeclaredSetterMethod(Property))
160 // And in its super class.
161 ObjCInterfaceDecl *OSC = ID->getSuperClass();
163 if (OSC->HasUserDeclaredSetterMethod(Property))
165 OSC = OSC->getSuperClass();
168 if (const auto *PD = dyn_cast<ObjCProtocolDecl>(this))
169 for (const auto *PI : PD->protocols())
170 if (PI->HasUserDeclaredSetterMethod(Property))
176 ObjCPropertyDecl::findPropertyDecl(const DeclContext *DC,
177 const IdentifierInfo *propertyID,
178 ObjCPropertyQueryKind queryKind) {
179 // If this context is a hidden protocol definition, don't find any
181 if (const auto *Proto = dyn_cast<ObjCProtocolDecl>(DC)) {
182 if (const ObjCProtocolDecl *Def = Proto->getDefinition())
187 // If context is class, then lookup property in its visible extensions.
188 // This comes before property is looked up in primary class.
189 if (auto *IDecl = dyn_cast<ObjCInterfaceDecl>(DC)) {
190 for (const auto *Ext : IDecl->visible_extensions())
191 if (ObjCPropertyDecl *PD = ObjCPropertyDecl::findPropertyDecl(Ext,
197 DeclContext::lookup_result R = DC->lookup(propertyID);
198 ObjCPropertyDecl *classProp = nullptr;
199 for (DeclContext::lookup_iterator I = R.begin(), E = R.end(); I != E;
201 if (auto *PD = dyn_cast<ObjCPropertyDecl>(*I)) {
202 // If queryKind is unknown, we return the instance property if one
203 // exists; otherwise we return the class property.
204 if ((queryKind == ObjCPropertyQueryKind::OBJC_PR_query_unknown &&
205 !PD->isClassProperty()) ||
206 (queryKind == ObjCPropertyQueryKind::OBJC_PR_query_class &&
207 PD->isClassProperty()) ||
208 (queryKind == ObjCPropertyQueryKind::OBJC_PR_query_instance &&
209 !PD->isClassProperty()))
212 if (PD->isClassProperty())
216 if (queryKind == ObjCPropertyQueryKind::OBJC_PR_query_unknown)
217 // We can't find the instance property, return the class property.
224 ObjCPropertyDecl::getDefaultSynthIvarName(ASTContext &Ctx) const {
225 SmallString<128> ivarName;
227 llvm::raw_svector_ostream os(ivarName);
228 os << '_' << getIdentifier()->getName();
230 return &Ctx.Idents.get(ivarName.str());
233 /// FindPropertyDeclaration - Finds declaration of the property given its name
234 /// in 'PropertyId' and returns it. It returns 0, if not found.
235 ObjCPropertyDecl *ObjCContainerDecl::FindPropertyDeclaration(
236 const IdentifierInfo *PropertyId,
237 ObjCPropertyQueryKind QueryKind) const {
238 // Don't find properties within hidden protocol definitions.
239 if (const auto *Proto = dyn_cast<ObjCProtocolDecl>(this)) {
240 if (const ObjCProtocolDecl *Def = Proto->getDefinition())
245 // Search the extensions of a class first; they override what's in
247 if (const auto *ClassDecl = dyn_cast<ObjCInterfaceDecl>(this)) {
248 for (const auto *Ext : ClassDecl->visible_extensions()) {
249 if (auto *P = Ext->FindPropertyDeclaration(PropertyId, QueryKind))
254 if (ObjCPropertyDecl *PD =
255 ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(this), PropertyId,
262 case Decl::ObjCProtocol: {
263 const auto *PID = cast<ObjCProtocolDecl>(this);
264 for (const auto *I : PID->protocols())
265 if (ObjCPropertyDecl *P = I->FindPropertyDeclaration(PropertyId,
270 case Decl::ObjCInterface: {
271 const auto *OID = cast<ObjCInterfaceDecl>(this);
272 // Look through categories (but not extensions; they were handled above).
273 for (const auto *Cat : OID->visible_categories()) {
274 if (!Cat->IsClassExtension())
275 if (ObjCPropertyDecl *P = Cat->FindPropertyDeclaration(
276 PropertyId, QueryKind))
280 // Look through protocols.
281 for (const auto *I : OID->all_referenced_protocols())
282 if (ObjCPropertyDecl *P = I->FindPropertyDeclaration(PropertyId,
286 // Finally, check the super class.
287 if (const ObjCInterfaceDecl *superClass = OID->getSuperClass())
288 return superClass->FindPropertyDeclaration(PropertyId, QueryKind);
291 case Decl::ObjCCategory: {
292 const auto *OCD = cast<ObjCCategoryDecl>(this);
293 // Look through protocols.
294 if (!OCD->IsClassExtension())
295 for (const auto *I : OCD->protocols())
296 if (ObjCPropertyDecl *P = I->FindPropertyDeclaration(PropertyId,
305 void ObjCInterfaceDecl::anchor() {}
307 ObjCTypeParamList *ObjCInterfaceDecl::getTypeParamList() const {
308 // If this particular declaration has a type parameter list, return it.
309 if (ObjCTypeParamList *written = getTypeParamListAsWritten())
312 // If there is a definition, return its type parameter list.
313 if (const ObjCInterfaceDecl *def = getDefinition())
314 return def->getTypeParamListAsWritten();
316 // Otherwise, look at previous declarations to determine whether any
317 // of them has a type parameter list, skipping over those
318 // declarations that do not.
319 for (const ObjCInterfaceDecl *decl = getMostRecentDecl(); decl;
320 decl = decl->getPreviousDecl()) {
321 if (ObjCTypeParamList *written = decl->getTypeParamListAsWritten())
328 void ObjCInterfaceDecl::setTypeParamList(ObjCTypeParamList *TPL) {
332 // Set the declaration context of each of the type parameters.
333 for (auto *typeParam : *TypeParamList)
334 typeParam->setDeclContext(this);
337 ObjCInterfaceDecl *ObjCInterfaceDecl::getSuperClass() const {
338 // FIXME: Should make sure no callers ever do this.
339 if (!hasDefinition())
342 if (data().ExternallyCompleted)
343 LoadExternalDefinition();
345 if (const ObjCObjectType *superType = getSuperClassType()) {
346 if (ObjCInterfaceDecl *superDecl = superType->getInterface()) {
347 if (ObjCInterfaceDecl *superDef = superDecl->getDefinition())
357 SourceLocation ObjCInterfaceDecl::getSuperClassLoc() const {
358 if (TypeSourceInfo *superTInfo = getSuperClassTInfo())
359 return superTInfo->getTypeLoc().getBeginLoc();
361 return SourceLocation();
364 /// FindPropertyVisibleInPrimaryClass - Finds declaration of the property
365 /// with name 'PropertyId' in the primary class; including those in protocols
366 /// (direct or indirect) used by the primary class.
368 ObjCInterfaceDecl::FindPropertyVisibleInPrimaryClass(
369 IdentifierInfo *PropertyId,
370 ObjCPropertyQueryKind QueryKind) const {
371 // FIXME: Should make sure no callers ever do this.
372 if (!hasDefinition())
375 if (data().ExternallyCompleted)
376 LoadExternalDefinition();
378 if (ObjCPropertyDecl *PD =
379 ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(this), PropertyId,
383 // Look through protocols.
384 for (const auto *I : all_referenced_protocols())
385 if (ObjCPropertyDecl *P = I->FindPropertyDeclaration(PropertyId,
392 void ObjCInterfaceDecl::collectPropertiesToImplement(PropertyMap &PM,
393 PropertyDeclOrder &PO) const {
394 for (auto *Prop : properties()) {
395 PM[std::make_pair(Prop->getIdentifier(), Prop->isClassProperty())] = Prop;
398 for (const auto *Ext : known_extensions()) {
399 const ObjCCategoryDecl *ClassExt = Ext;
400 for (auto *Prop : ClassExt->properties()) {
401 PM[std::make_pair(Prop->getIdentifier(), Prop->isClassProperty())] = Prop;
405 for (const auto *PI : all_referenced_protocols())
406 PI->collectPropertiesToImplement(PM, PO);
407 // Note, the properties declared only in class extensions are still copied
408 // into the main @interface's property list, and therefore we don't
409 // explicitly, have to search class extension properties.
412 bool ObjCInterfaceDecl::isArcWeakrefUnavailable() const {
413 const ObjCInterfaceDecl *Class = this;
415 if (Class->hasAttr<ArcWeakrefUnavailableAttr>())
417 Class = Class->getSuperClass();
422 const ObjCInterfaceDecl *ObjCInterfaceDecl::isObjCRequiresPropertyDefs() const {
423 const ObjCInterfaceDecl *Class = this;
425 if (Class->hasAttr<ObjCRequiresPropertyDefsAttr>())
427 Class = Class->getSuperClass();
432 void ObjCInterfaceDecl::mergeClassExtensionProtocolList(
433 ObjCProtocolDecl *const* ExtList, unsigned ExtNum,
435 if (data().ExternallyCompleted)
436 LoadExternalDefinition();
438 if (data().AllReferencedProtocols.empty() &&
439 data().ReferencedProtocols.empty()) {
440 data().AllReferencedProtocols.set(ExtList, ExtNum, C);
444 // Check for duplicate protocol in class's protocol list.
445 // This is O(n*m). But it is extremely rare and number of protocols in
446 // class or its extension are very few.
447 SmallVector<ObjCProtocolDecl *, 8> ProtocolRefs;
448 for (unsigned i = 0; i < ExtNum; i++) {
449 bool protocolExists = false;
450 ObjCProtocolDecl *ProtoInExtension = ExtList[i];
451 for (auto *Proto : all_referenced_protocols()) {
452 if (C.ProtocolCompatibleWithProtocol(ProtoInExtension, Proto)) {
453 protocolExists = true;
457 // Do we want to warn on a protocol in extension class which
458 // already exist in the class? Probably not.
460 ProtocolRefs.push_back(ProtoInExtension);
463 if (ProtocolRefs.empty())
466 // Merge ProtocolRefs into class's protocol list;
467 ProtocolRefs.append(all_referenced_protocol_begin(),
468 all_referenced_protocol_end());
470 data().AllReferencedProtocols.set(ProtocolRefs.data(), ProtocolRefs.size(),C);
473 const ObjCInterfaceDecl *
474 ObjCInterfaceDecl::findInterfaceWithDesignatedInitializers() const {
475 const ObjCInterfaceDecl *IFace = this;
477 if (IFace->hasDesignatedInitializers())
479 if (!IFace->inheritsDesignatedInitializers())
481 IFace = IFace->getSuperClass();
486 static bool isIntroducingInitializers(const ObjCInterfaceDecl *D) {
487 for (const auto *MD : D->instance_methods()) {
488 if (MD->getMethodFamily() == OMF_init && !MD->isOverriding())
491 for (const auto *Ext : D->visible_extensions()) {
492 for (const auto *MD : Ext->instance_methods()) {
493 if (MD->getMethodFamily() == OMF_init && !MD->isOverriding())
497 if (const auto *ImplD = D->getImplementation()) {
498 for (const auto *MD : ImplD->instance_methods()) {
499 if (MD->getMethodFamily() == OMF_init && !MD->isOverriding())
506 bool ObjCInterfaceDecl::inheritsDesignatedInitializers() const {
507 switch (data().InheritedDesignatedInitializers) {
508 case DefinitionData::IDI_Inherited:
510 case DefinitionData::IDI_NotInherited:
512 case DefinitionData::IDI_Unknown:
513 // If the class introduced initializers we conservatively assume that we
514 // don't know if any of them is a designated initializer to avoid possible
515 // misleading warnings.
516 if (isIntroducingInitializers(this)) {
517 data().InheritedDesignatedInitializers = DefinitionData::IDI_NotInherited;
519 if (auto SuperD = getSuperClass()) {
520 data().InheritedDesignatedInitializers =
521 SuperD->declaresOrInheritsDesignatedInitializers() ?
522 DefinitionData::IDI_Inherited :
523 DefinitionData::IDI_NotInherited;
525 data().InheritedDesignatedInitializers =
526 DefinitionData::IDI_NotInherited;
529 assert(data().InheritedDesignatedInitializers
530 != DefinitionData::IDI_Unknown);
531 return data().InheritedDesignatedInitializers ==
532 DefinitionData::IDI_Inherited;
535 llvm_unreachable("unexpected InheritedDesignatedInitializers value");
538 void ObjCInterfaceDecl::getDesignatedInitializers(
539 llvm::SmallVectorImpl<const ObjCMethodDecl *> &Methods) const {
540 // Check for a complete definition and recover if not so.
541 if (!isThisDeclarationADefinition())
543 if (data().ExternallyCompleted)
544 LoadExternalDefinition();
546 const ObjCInterfaceDecl *IFace= findInterfaceWithDesignatedInitializers();
550 for (const auto *MD : IFace->instance_methods())
551 if (MD->isThisDeclarationADesignatedInitializer())
552 Methods.push_back(MD);
553 for (const auto *Ext : IFace->visible_extensions()) {
554 for (const auto *MD : Ext->instance_methods())
555 if (MD->isThisDeclarationADesignatedInitializer())
556 Methods.push_back(MD);
560 bool ObjCInterfaceDecl::isDesignatedInitializer(Selector Sel,
561 const ObjCMethodDecl **InitMethod) const {
562 bool HasCompleteDef = isThisDeclarationADefinition();
563 // During deserialization the data record for the ObjCInterfaceDecl could
564 // be made invariant by reusing the canonical decl. Take this into account
565 // when checking for the complete definition.
566 if (!HasCompleteDef && getCanonicalDecl()->hasDefinition() &&
567 getCanonicalDecl()->getDefinition() == getDefinition())
568 HasCompleteDef = true;
570 // Check for a complete definition and recover if not so.
574 if (data().ExternallyCompleted)
575 LoadExternalDefinition();
577 const ObjCInterfaceDecl *IFace= findInterfaceWithDesignatedInitializers();
581 if (const ObjCMethodDecl *MD = IFace->getInstanceMethod(Sel)) {
582 if (MD->isThisDeclarationADesignatedInitializer()) {
588 for (const auto *Ext : IFace->visible_extensions()) {
589 if (const ObjCMethodDecl *MD = Ext->getInstanceMethod(Sel)) {
590 if (MD->isThisDeclarationADesignatedInitializer()) {
600 void ObjCInterfaceDecl::allocateDefinitionData() {
601 assert(!hasDefinition() && "ObjC class already has a definition");
602 Data.setPointer(new (getASTContext()) DefinitionData());
603 Data.getPointer()->Definition = this;
605 // Make the type point at the definition, now that we have one.
607 cast<ObjCInterfaceType>(TypeForDecl)->Decl = this;
610 void ObjCInterfaceDecl::startDefinition() {
611 allocateDefinitionData();
613 // Update all of the declarations with a pointer to the definition.
614 for (auto *RD : redecls()) {
620 ObjCIvarDecl *ObjCInterfaceDecl::lookupInstanceVariable(IdentifierInfo *ID,
621 ObjCInterfaceDecl *&clsDeclared) {
622 // FIXME: Should make sure no callers ever do this.
623 if (!hasDefinition())
626 if (data().ExternallyCompleted)
627 LoadExternalDefinition();
629 ObjCInterfaceDecl* ClassDecl = this;
630 while (ClassDecl != nullptr) {
631 if (ObjCIvarDecl *I = ClassDecl->getIvarDecl(ID)) {
632 clsDeclared = ClassDecl;
636 for (const auto *Ext : ClassDecl->visible_extensions()) {
637 if (ObjCIvarDecl *I = Ext->getIvarDecl(ID)) {
638 clsDeclared = ClassDecl;
643 ClassDecl = ClassDecl->getSuperClass();
648 /// lookupInheritedClass - This method returns ObjCInterfaceDecl * of the super
649 /// class whose name is passed as argument. If it is not one of the super classes
650 /// the it returns NULL.
651 ObjCInterfaceDecl *ObjCInterfaceDecl::lookupInheritedClass(
652 const IdentifierInfo*ICName) {
653 // FIXME: Should make sure no callers ever do this.
654 if (!hasDefinition())
657 if (data().ExternallyCompleted)
658 LoadExternalDefinition();
660 ObjCInterfaceDecl* ClassDecl = this;
661 while (ClassDecl != nullptr) {
662 if (ClassDecl->getIdentifier() == ICName)
664 ClassDecl = ClassDecl->getSuperClass();
670 ObjCInterfaceDecl::lookupNestedProtocol(IdentifierInfo *Name) {
671 for (auto *P : all_referenced_protocols())
672 if (P->lookupProtocolNamed(Name))
674 ObjCInterfaceDecl *SuperClass = getSuperClass();
675 return SuperClass ? SuperClass->lookupNestedProtocol(Name) : nullptr;
678 /// lookupMethod - This method returns an instance/class method by looking in
679 /// the class, its categories, and its super classes (using a linear search).
680 /// When argument category "C" is specified, any implicit method found
681 /// in this category is ignored.
682 ObjCMethodDecl *ObjCInterfaceDecl::lookupMethod(Selector Sel,
684 bool shallowCategoryLookup,
686 const ObjCCategoryDecl *C) const
688 // FIXME: Should make sure no callers ever do this.
689 if (!hasDefinition())
692 const ObjCInterfaceDecl* ClassDecl = this;
693 ObjCMethodDecl *MethodDecl = nullptr;
695 if (data().ExternallyCompleted)
696 LoadExternalDefinition();
699 // 1. Look through primary class.
700 if ((MethodDecl = ClassDecl->getMethod(Sel, isInstance)))
703 // 2. Didn't find one yet - now look through categories.
704 for (const auto *Cat : ClassDecl->visible_categories())
705 if ((MethodDecl = Cat->getMethod(Sel, isInstance)))
706 if (C != Cat || !MethodDecl->isImplicit())
709 // 3. Didn't find one yet - look through primary class's protocols.
710 for (const auto *I : ClassDecl->protocols())
711 if ((MethodDecl = I->lookupMethod(Sel, isInstance)))
714 // 4. Didn't find one yet - now look through categories' protocols
715 if (!shallowCategoryLookup)
716 for (const auto *Cat : ClassDecl->visible_categories()) {
717 // Didn't find one yet - look through protocols.
718 const ObjCList<ObjCProtocolDecl> &Protocols =
719 Cat->getReferencedProtocols();
720 for (auto *Protocol : Protocols)
721 if ((MethodDecl = Protocol->lookupMethod(Sel, isInstance)))
722 if (C != Cat || !MethodDecl->isImplicit())
730 // 5. Get to the super class (if any).
731 ClassDecl = ClassDecl->getSuperClass();
736 // Will search "local" class/category implementations for a method decl.
737 // If failed, then we search in class's root for an instance method.
738 // Returns 0 if no method is found.
739 ObjCMethodDecl *ObjCInterfaceDecl::lookupPrivateMethod(
741 bool Instance) const {
742 // FIXME: Should make sure no callers ever do this.
743 if (!hasDefinition())
746 if (data().ExternallyCompleted)
747 LoadExternalDefinition();
749 ObjCMethodDecl *Method = nullptr;
750 if (ObjCImplementationDecl *ImpDecl = getImplementation())
751 Method = Instance ? ImpDecl->getInstanceMethod(Sel)
752 : ImpDecl->getClassMethod(Sel);
754 // Look through local category implementations associated with the class.
756 Method = getCategoryMethod(Sel, Instance);
758 // Before we give up, check if the selector is an instance method.
759 // But only in the root. This matches gcc's behavior and what the
761 if (!Instance && !Method && !getSuperClass()) {
762 Method = lookupInstanceMethod(Sel);
763 // Look through local category implementations associated
764 // with the root class.
766 Method = lookupPrivateMethod(Sel, true);
769 if (!Method && getSuperClass())
770 return getSuperClass()->lookupPrivateMethod(Sel, Instance);
774 //===----------------------------------------------------------------------===//
776 //===----------------------------------------------------------------------===//
778 ObjCMethodDecl::ObjCMethodDecl(
779 SourceLocation beginLoc, SourceLocation endLoc, Selector SelInfo,
780 QualType T, TypeSourceInfo *ReturnTInfo, DeclContext *contextDecl,
781 bool isInstance, bool isVariadic, bool isPropertyAccessor,
782 bool isSynthesizedAccessorStub, bool isImplicitlyDeclared, bool isDefined,
783 ImplementationControl impControl, bool HasRelatedResultType)
784 : NamedDecl(ObjCMethod, contextDecl, beginLoc, SelInfo),
785 DeclContext(ObjCMethod), MethodDeclType(T), ReturnTInfo(ReturnTInfo),
788 // Initialized the bits stored in DeclContext.
789 ObjCMethodDeclBits.Family =
790 static_cast<ObjCMethodFamily>(InvalidObjCMethodFamily);
791 setInstanceMethod(isInstance);
792 setVariadic(isVariadic);
793 setPropertyAccessor(isPropertyAccessor);
794 setSynthesizedAccessorStub(isSynthesizedAccessorStub);
795 setDefined(isDefined);
796 setIsRedeclaration(false);
797 setHasRedeclaration(false);
798 setDeclImplementation(impControl);
799 setObjCDeclQualifier(OBJC_TQ_None);
800 setRelatedResultType(HasRelatedResultType);
801 setSelLocsKind(SelLoc_StandardNoSpace);
802 setOverriding(false);
803 setHasSkippedBody(false);
805 setImplicit(isImplicitlyDeclared);
808 ObjCMethodDecl *ObjCMethodDecl::Create(
809 ASTContext &C, SourceLocation beginLoc, SourceLocation endLoc,
810 Selector SelInfo, QualType T, TypeSourceInfo *ReturnTInfo,
811 DeclContext *contextDecl, bool isInstance, bool isVariadic,
812 bool isPropertyAccessor, bool isSynthesizedAccessorStub,
813 bool isImplicitlyDeclared, bool isDefined, ImplementationControl impControl,
814 bool HasRelatedResultType) {
815 return new (C, contextDecl) ObjCMethodDecl(
816 beginLoc, endLoc, SelInfo, T, ReturnTInfo, contextDecl, isInstance,
817 isVariadic, isPropertyAccessor, isSynthesizedAccessorStub,
818 isImplicitlyDeclared, isDefined, impControl, HasRelatedResultType);
821 ObjCMethodDecl *ObjCMethodDecl::CreateDeserialized(ASTContext &C, unsigned ID) {
822 return new (C, ID) ObjCMethodDecl(SourceLocation(), SourceLocation(),
823 Selector(), QualType(), nullptr, nullptr);
826 bool ObjCMethodDecl::isDirectMethod() const {
827 return hasAttr<ObjCDirectAttr>();
830 bool ObjCMethodDecl::isThisDeclarationADesignatedInitializer() const {
831 return getMethodFamily() == OMF_init &&
832 hasAttr<ObjCDesignatedInitializerAttr>();
835 bool ObjCMethodDecl::definedInNSObject(const ASTContext &Ctx) const {
836 if (const auto *PD = dyn_cast<const ObjCProtocolDecl>(getDeclContext()))
837 return PD->getIdentifier() == Ctx.getNSObjectName();
838 if (const auto *ID = dyn_cast<const ObjCInterfaceDecl>(getDeclContext()))
839 return ID->getIdentifier() == Ctx.getNSObjectName();
843 bool ObjCMethodDecl::isDesignatedInitializerForTheInterface(
844 const ObjCMethodDecl **InitMethod) const {
845 if (getMethodFamily() != OMF_init)
847 const DeclContext *DC = getDeclContext();
848 if (isa<ObjCProtocolDecl>(DC))
850 if (const ObjCInterfaceDecl *ID = getClassInterface())
851 return ID->isDesignatedInitializer(getSelector(), InitMethod);
855 Stmt *ObjCMethodDecl::getBody() const {
856 return Body.get(getASTContext().getExternalSource());
859 void ObjCMethodDecl::setAsRedeclaration(const ObjCMethodDecl *PrevMethod) {
861 getASTContext().setObjCMethodRedeclaration(PrevMethod, this);
862 setIsRedeclaration(true);
863 PrevMethod->setHasRedeclaration(true);
866 void ObjCMethodDecl::setParamsAndSelLocs(ASTContext &C,
867 ArrayRef<ParmVarDecl*> Params,
868 ArrayRef<SourceLocation> SelLocs) {
869 ParamsAndSelLocs = nullptr;
870 NumParams = Params.size();
871 if (Params.empty() && SelLocs.empty())
874 static_assert(alignof(ParmVarDecl *) >= alignof(SourceLocation),
875 "Alignment not sufficient for SourceLocation");
877 unsigned Size = sizeof(ParmVarDecl *) * NumParams +
878 sizeof(SourceLocation) * SelLocs.size();
879 ParamsAndSelLocs = C.Allocate(Size);
880 std::copy(Params.begin(), Params.end(), getParams());
881 std::copy(SelLocs.begin(), SelLocs.end(), getStoredSelLocs());
884 void ObjCMethodDecl::getSelectorLocs(
885 SmallVectorImpl<SourceLocation> &SelLocs) const {
886 for (unsigned i = 0, e = getNumSelectorLocs(); i != e; ++i)
887 SelLocs.push_back(getSelectorLoc(i));
890 void ObjCMethodDecl::setMethodParams(ASTContext &C,
891 ArrayRef<ParmVarDecl*> Params,
892 ArrayRef<SourceLocation> SelLocs) {
893 assert((!SelLocs.empty() || isImplicit()) &&
894 "No selector locs for non-implicit method");
896 return setParamsAndSelLocs(C, Params, llvm::None);
898 setSelLocsKind(hasStandardSelectorLocs(getSelector(), SelLocs, Params,
900 if (getSelLocsKind() != SelLoc_NonStandard)
901 return setParamsAndSelLocs(C, Params, llvm::None);
903 setParamsAndSelLocs(C, Params, SelLocs);
906 /// A definition will return its interface declaration.
907 /// An interface declaration will return its definition.
908 /// Otherwise it will return itself.
909 ObjCMethodDecl *ObjCMethodDecl::getNextRedeclarationImpl() {
910 ASTContext &Ctx = getASTContext();
911 ObjCMethodDecl *Redecl = nullptr;
912 if (hasRedeclaration())
913 Redecl = const_cast<ObjCMethodDecl*>(Ctx.getObjCMethodRedeclaration(this));
917 auto *CtxD = cast<Decl>(getDeclContext());
919 if (!CtxD->isInvalidDecl()) {
920 if (auto *IFD = dyn_cast<ObjCInterfaceDecl>(CtxD)) {
921 if (ObjCImplementationDecl *ImplD = Ctx.getObjCImplementation(IFD))
922 if (!ImplD->isInvalidDecl())
923 Redecl = ImplD->getMethod(getSelector(), isInstanceMethod());
925 } else if (auto *CD = dyn_cast<ObjCCategoryDecl>(CtxD)) {
926 if (ObjCCategoryImplDecl *ImplD = Ctx.getObjCImplementation(CD))
927 if (!ImplD->isInvalidDecl())
928 Redecl = ImplD->getMethod(getSelector(), isInstanceMethod());
930 } else if (auto *ImplD = dyn_cast<ObjCImplementationDecl>(CtxD)) {
931 if (ObjCInterfaceDecl *IFD = ImplD->getClassInterface())
932 if (!IFD->isInvalidDecl())
933 Redecl = IFD->getMethod(getSelector(), isInstanceMethod());
935 } else if (auto *CImplD = dyn_cast<ObjCCategoryImplDecl>(CtxD)) {
936 if (ObjCCategoryDecl *CatD = CImplD->getCategoryDecl())
937 if (!CatD->isInvalidDecl())
938 Redecl = CatD->getMethod(getSelector(), isInstanceMethod());
942 // Ensure that the discovered method redeclaration has a valid declaration
943 // context. Used to prevent infinite loops when iterating redeclarations in
944 // a partially invalid AST.
945 if (Redecl && cast<Decl>(Redecl->getDeclContext())->isInvalidDecl())
948 if (!Redecl && isRedeclaration()) {
949 // This is the last redeclaration, go back to the first method.
950 return cast<ObjCContainerDecl>(CtxD)->getMethod(getSelector(),
954 return Redecl ? Redecl : this;
957 ObjCMethodDecl *ObjCMethodDecl::getCanonicalDecl() {
958 auto *CtxD = cast<Decl>(getDeclContext());
959 const auto &Sel = getSelector();
961 if (auto *ImplD = dyn_cast<ObjCImplementationDecl>(CtxD)) {
962 if (ObjCInterfaceDecl *IFD = ImplD->getClassInterface()) {
963 // When the container is the ObjCImplementationDecl (the primary
964 // @implementation), then the canonical Decl is either in
965 // the class Interface, or in any of its extension.
967 // So when we don't find it in the ObjCInterfaceDecl,
968 // sift through extensions too.
969 if (ObjCMethodDecl *MD = IFD->getMethod(Sel, isInstanceMethod()))
971 for (auto *Ext : IFD->known_extensions())
972 if (ObjCMethodDecl *MD = Ext->getMethod(Sel, isInstanceMethod()))
975 } else if (auto *CImplD = dyn_cast<ObjCCategoryImplDecl>(CtxD)) {
976 if (ObjCCategoryDecl *CatD = CImplD->getCategoryDecl())
977 if (ObjCMethodDecl *MD = CatD->getMethod(Sel, isInstanceMethod()))
981 if (isRedeclaration()) {
982 // It is possible that we have not done deserializing the ObjCMethod yet.
984 cast<ObjCContainerDecl>(CtxD)->getMethod(Sel, isInstanceMethod());
985 return MD ? MD : this;
991 SourceLocation ObjCMethodDecl::getEndLoc() const {
992 if (Stmt *Body = getBody())
993 return Body->getEndLoc();
997 ObjCMethodFamily ObjCMethodDecl::getMethodFamily() const {
998 auto family = static_cast<ObjCMethodFamily>(ObjCMethodDeclBits.Family);
999 if (family != static_cast<unsigned>(InvalidObjCMethodFamily))
1002 // Check for an explicit attribute.
1003 if (const ObjCMethodFamilyAttr *attr = getAttr<ObjCMethodFamilyAttr>()) {
1004 // The unfortunate necessity of mapping between enums here is due
1005 // to the attributes framework.
1006 switch (attr->getFamily()) {
1007 case ObjCMethodFamilyAttr::OMF_None: family = OMF_None; break;
1008 case ObjCMethodFamilyAttr::OMF_alloc: family = OMF_alloc; break;
1009 case ObjCMethodFamilyAttr::OMF_copy: family = OMF_copy; break;
1010 case ObjCMethodFamilyAttr::OMF_init: family = OMF_init; break;
1011 case ObjCMethodFamilyAttr::OMF_mutableCopy: family = OMF_mutableCopy; break;
1012 case ObjCMethodFamilyAttr::OMF_new: family = OMF_new; break;
1014 ObjCMethodDeclBits.Family = family;
1018 family = getSelector().getMethodFamily();
1020 case OMF_None: break;
1022 // init only has a conventional meaning for an instance method, and
1023 // it has to return an object.
1025 if (!isInstanceMethod() || !getReturnType()->isObjCObjectPointerType())
1029 // alloc/copy/new have a conventional meaning for both class and
1030 // instance methods, but they require an object return.
1033 case OMF_mutableCopy:
1035 if (!getReturnType()->isObjCObjectPointerType())
1039 // These selectors have a conventional meaning only for instance methods.
1044 case OMF_autorelease:
1045 case OMF_retainCount:
1047 if (!isInstanceMethod())
1051 case OMF_initialize:
1052 if (isInstanceMethod() || !getReturnType()->isVoidType())
1056 case OMF_performSelector:
1057 if (!isInstanceMethod() || !getReturnType()->isObjCIdType())
1060 unsigned noParams = param_size();
1061 if (noParams < 1 || noParams > 3)
1064 ObjCMethodDecl::param_type_iterator it = param_type_begin();
1065 QualType ArgT = (*it);
1066 if (!ArgT->isObjCSelType()) {
1070 while (--noParams) {
1073 if (!ArgT->isObjCIdType()) {
1084 // Cache the result.
1085 ObjCMethodDeclBits.Family = family;
1089 QualType ObjCMethodDecl::getSelfType(ASTContext &Context,
1090 const ObjCInterfaceDecl *OID,
1091 bool &selfIsPseudoStrong,
1092 bool &selfIsConsumed) const {
1094 selfIsPseudoStrong = false;
1095 selfIsConsumed = false;
1096 if (isInstanceMethod()) {
1097 // There may be no interface context due to error in declaration
1098 // of the interface (which has been reported). Recover gracefully.
1100 selfTy = Context.getObjCInterfaceType(OID);
1101 selfTy = Context.getObjCObjectPointerType(selfTy);
1103 selfTy = Context.getObjCIdType();
1105 } else // we have a factory method.
1106 selfTy = Context.getObjCClassType();
1108 if (Context.getLangOpts().ObjCAutoRefCount) {
1109 if (isInstanceMethod()) {
1110 selfIsConsumed = hasAttr<NSConsumesSelfAttr>();
1112 // 'self' is always __strong. It's actually pseudo-strong except
1113 // in init methods (or methods labeled ns_consumes_self), though.
1115 qs.setObjCLifetime(Qualifiers::OCL_Strong);
1116 selfTy = Context.getQualifiedType(selfTy, qs);
1118 // In addition, 'self' is const unless this is an init method.
1119 if (getMethodFamily() != OMF_init && !selfIsConsumed) {
1120 selfTy = selfTy.withConst();
1121 selfIsPseudoStrong = true;
1125 assert(isClassMethod());
1126 // 'self' is always const in class methods.
1127 selfTy = selfTy.withConst();
1128 selfIsPseudoStrong = true;
1134 void ObjCMethodDecl::createImplicitParams(ASTContext &Context,
1135 const ObjCInterfaceDecl *OID) {
1136 bool selfIsPseudoStrong, selfIsConsumed;
1138 getSelfType(Context, OID, selfIsPseudoStrong, selfIsConsumed);
1139 auto *Self = ImplicitParamDecl::Create(Context, this, SourceLocation(),
1140 &Context.Idents.get("self"), selfTy,
1141 ImplicitParamDecl::ObjCSelf);
1145 Self->addAttr(NSConsumedAttr::CreateImplicit(Context));
1147 if (selfIsPseudoStrong)
1148 Self->setARCPseudoStrong(true);
1150 setCmdDecl(ImplicitParamDecl::Create(
1151 Context, this, SourceLocation(), &Context.Idents.get("_cmd"),
1152 Context.getObjCSelType(), ImplicitParamDecl::ObjCCmd));
1155 ObjCInterfaceDecl *ObjCMethodDecl::getClassInterface() {
1156 if (auto *ID = dyn_cast<ObjCInterfaceDecl>(getDeclContext()))
1158 if (auto *CD = dyn_cast<ObjCCategoryDecl>(getDeclContext()))
1159 return CD->getClassInterface();
1160 if (auto *IMD = dyn_cast<ObjCImplDecl>(getDeclContext()))
1161 return IMD->getClassInterface();
1162 if (isa<ObjCProtocolDecl>(getDeclContext()))
1164 llvm_unreachable("unknown method context");
1167 SourceRange ObjCMethodDecl::getReturnTypeSourceRange() const {
1168 const auto *TSI = getReturnTypeSourceInfo();
1170 return TSI->getTypeLoc().getSourceRange();
1171 return SourceRange();
1174 QualType ObjCMethodDecl::getSendResultType() const {
1175 ASTContext &Ctx = getASTContext();
1176 return getReturnType().getNonLValueExprType(Ctx)
1177 .substObjCTypeArgs(Ctx, {}, ObjCSubstitutionContext::Result);
1180 QualType ObjCMethodDecl::getSendResultType(QualType receiverType) const {
1181 // FIXME: Handle related result types here.
1183 return getReturnType().getNonLValueExprType(getASTContext())
1184 .substObjCMemberType(receiverType, getDeclContext(),
1185 ObjCSubstitutionContext::Result);
1188 static void CollectOverriddenMethodsRecurse(const ObjCContainerDecl *Container,
1189 const ObjCMethodDecl *Method,
1190 SmallVectorImpl<const ObjCMethodDecl *> &Methods,
1191 bool MovedToSuper) {
1195 // In categories look for overridden methods from protocols. A method from
1196 // category is not "overridden" since it is considered as the "same" method
1197 // (same USR) as the one from the interface.
1198 if (const auto *Category = dyn_cast<ObjCCategoryDecl>(Container)) {
1199 // Check whether we have a matching method at this category but only if we
1200 // are at the super class level.
1202 if (ObjCMethodDecl *
1203 Overridden = Container->getMethod(Method->getSelector(),
1204 Method->isInstanceMethod(),
1205 /*AllowHidden=*/true))
1206 if (Method != Overridden) {
1207 // We found an override at this category; there is no need to look
1208 // into its protocols.
1209 Methods.push_back(Overridden);
1213 for (const auto *P : Category->protocols())
1214 CollectOverriddenMethodsRecurse(P, Method, Methods, MovedToSuper);
1218 // Check whether we have a matching method at this level.
1219 if (const ObjCMethodDecl *
1220 Overridden = Container->getMethod(Method->getSelector(),
1221 Method->isInstanceMethod(),
1222 /*AllowHidden=*/true))
1223 if (Method != Overridden) {
1224 // We found an override at this level; there is no need to look
1225 // into other protocols or categories.
1226 Methods.push_back(Overridden);
1230 if (const auto *Protocol = dyn_cast<ObjCProtocolDecl>(Container)){
1231 for (const auto *P : Protocol->protocols())
1232 CollectOverriddenMethodsRecurse(P, Method, Methods, MovedToSuper);
1235 if (const auto *Interface = dyn_cast<ObjCInterfaceDecl>(Container)) {
1236 for (const auto *P : Interface->protocols())
1237 CollectOverriddenMethodsRecurse(P, Method, Methods, MovedToSuper);
1239 for (const auto *Cat : Interface->known_categories())
1240 CollectOverriddenMethodsRecurse(Cat, Method, Methods, MovedToSuper);
1242 if (const ObjCInterfaceDecl *Super = Interface->getSuperClass())
1243 return CollectOverriddenMethodsRecurse(Super, Method, Methods,
1244 /*MovedToSuper=*/true);
1248 static inline void CollectOverriddenMethods(const ObjCContainerDecl *Container,
1249 const ObjCMethodDecl *Method,
1250 SmallVectorImpl<const ObjCMethodDecl *> &Methods) {
1251 CollectOverriddenMethodsRecurse(Container, Method, Methods,
1252 /*MovedToSuper=*/false);
1255 static void collectOverriddenMethodsSlow(const ObjCMethodDecl *Method,
1256 SmallVectorImpl<const ObjCMethodDecl *> &overridden) {
1257 assert(Method->isOverriding());
1259 if (const auto *ProtD =
1260 dyn_cast<ObjCProtocolDecl>(Method->getDeclContext())) {
1261 CollectOverriddenMethods(ProtD, Method, overridden);
1263 } else if (const auto *IMD =
1264 dyn_cast<ObjCImplDecl>(Method->getDeclContext())) {
1265 const ObjCInterfaceDecl *ID = IMD->getClassInterface();
1268 // Start searching for overridden methods using the method from the
1269 // interface as starting point.
1270 if (const ObjCMethodDecl *IFaceMeth = ID->getMethod(Method->getSelector(),
1271 Method->isInstanceMethod(),
1272 /*AllowHidden=*/true))
1274 CollectOverriddenMethods(ID, Method, overridden);
1276 } else if (const auto *CatD =
1277 dyn_cast<ObjCCategoryDecl>(Method->getDeclContext())) {
1278 const ObjCInterfaceDecl *ID = CatD->getClassInterface();
1281 // Start searching for overridden methods using the method from the
1282 // interface as starting point.
1283 if (const ObjCMethodDecl *IFaceMeth = ID->getMethod(Method->getSelector(),
1284 Method->isInstanceMethod(),
1285 /*AllowHidden=*/true))
1287 CollectOverriddenMethods(ID, Method, overridden);
1290 CollectOverriddenMethods(
1291 dyn_cast_or_null<ObjCContainerDecl>(Method->getDeclContext()),
1292 Method, overridden);
1296 void ObjCMethodDecl::getOverriddenMethods(
1297 SmallVectorImpl<const ObjCMethodDecl *> &Overridden) const {
1298 const ObjCMethodDecl *Method = this;
1300 if (Method->isRedeclaration()) {
1301 Method = cast<ObjCContainerDecl>(Method->getDeclContext())->
1302 getMethod(Method->getSelector(), Method->isInstanceMethod());
1305 if (Method->isOverriding()) {
1306 collectOverriddenMethodsSlow(Method, Overridden);
1307 assert(!Overridden.empty() &&
1308 "ObjCMethodDecl's overriding bit is not as expected");
1312 const ObjCPropertyDecl *
1313 ObjCMethodDecl::findPropertyDecl(bool CheckOverrides) const {
1314 Selector Sel = getSelector();
1315 unsigned NumArgs = Sel.getNumArgs();
1319 if (isPropertyAccessor()) {
1320 const auto *Container = cast<ObjCContainerDecl>(getParent());
1321 // For accessor stubs, go back to the interface.
1322 if (auto *ImplDecl = dyn_cast<ObjCImplDecl>(Container))
1323 if (isSynthesizedAccessorStub())
1324 Container = ImplDecl->getClassInterface();
1326 bool IsGetter = (NumArgs == 0);
1327 bool IsInstance = isInstanceMethod();
1329 /// Local function that attempts to find a matching property within the
1330 /// given Objective-C container.
1331 auto findMatchingProperty =
1332 [&](const ObjCContainerDecl *Container) -> const ObjCPropertyDecl * {
1334 for (const auto *I : Container->instance_properties()) {
1335 Selector NextSel = IsGetter ? I->getGetterName()
1336 : I->getSetterName();
1341 for (const auto *I : Container->class_properties()) {
1342 Selector NextSel = IsGetter ? I->getGetterName()
1343 : I->getSetterName();
1352 // Look in the container we were given.
1353 if (const auto *Found = findMatchingProperty(Container))
1356 // If we're in a category or extension, look in the main class.
1357 const ObjCInterfaceDecl *ClassDecl = nullptr;
1358 if (const auto *Category = dyn_cast<ObjCCategoryDecl>(Container)) {
1359 ClassDecl = Category->getClassInterface();
1360 if (const auto *Found = findMatchingProperty(ClassDecl))
1363 // Determine whether the container is a class.
1364 ClassDecl = dyn_cast<ObjCInterfaceDecl>(Container);
1367 // If we have a class, check its visible extensions.
1369 for (const auto *Ext : ClassDecl->visible_extensions()) {
1370 if (Ext == Container)
1373 if (const auto *Found = findMatchingProperty(Ext))
1378 assert(isSynthesizedAccessorStub() && "expected an accessor stub");
1379 for (const auto *Cat : ClassDecl->known_categories()) {
1380 if (Cat == Container)
1383 if (const auto *Found = findMatchingProperty(Cat))
1387 llvm_unreachable("Marked as a property accessor but no property found!");
1390 if (!CheckOverrides)
1393 using OverridesTy = SmallVector<const ObjCMethodDecl *, 8>;
1395 OverridesTy Overrides;
1396 getOverriddenMethods(Overrides);
1397 for (const auto *Override : Overrides)
1398 if (const ObjCPropertyDecl *Prop = Override->findPropertyDecl(false))
1404 //===----------------------------------------------------------------------===//
1405 // ObjCTypeParamDecl
1406 //===----------------------------------------------------------------------===//
1408 void ObjCTypeParamDecl::anchor() {}
1410 ObjCTypeParamDecl *ObjCTypeParamDecl::Create(ASTContext &ctx, DeclContext *dc,
1411 ObjCTypeParamVariance variance,
1412 SourceLocation varianceLoc,
1414 SourceLocation nameLoc,
1415 IdentifierInfo *name,
1416 SourceLocation colonLoc,
1417 TypeSourceInfo *boundInfo) {
1419 new (ctx, dc) ObjCTypeParamDecl(ctx, dc, variance, varianceLoc, index,
1420 nameLoc, name, colonLoc, boundInfo);
1421 QualType TPType = ctx.getObjCTypeParamType(TPDecl, {});
1422 TPDecl->setTypeForDecl(TPType.getTypePtr());
1426 ObjCTypeParamDecl *ObjCTypeParamDecl::CreateDeserialized(ASTContext &ctx,
1428 return new (ctx, ID) ObjCTypeParamDecl(ctx, nullptr,
1429 ObjCTypeParamVariance::Invariant,
1430 SourceLocation(), 0, SourceLocation(),
1431 nullptr, SourceLocation(), nullptr);
1434 SourceRange ObjCTypeParamDecl::getSourceRange() const {
1435 SourceLocation startLoc = VarianceLoc;
1436 if (startLoc.isInvalid())
1437 startLoc = getLocation();
1439 if (hasExplicitBound()) {
1440 return SourceRange(startLoc,
1441 getTypeSourceInfo()->getTypeLoc().getEndLoc());
1444 return SourceRange(startLoc);
1447 //===----------------------------------------------------------------------===//
1448 // ObjCTypeParamList
1449 //===----------------------------------------------------------------------===//
1450 ObjCTypeParamList::ObjCTypeParamList(SourceLocation lAngleLoc,
1451 ArrayRef<ObjCTypeParamDecl *> typeParams,
1452 SourceLocation rAngleLoc)
1453 : NumParams(typeParams.size()) {
1454 Brackets.Begin = lAngleLoc.getRawEncoding();
1455 Brackets.End = rAngleLoc.getRawEncoding();
1456 std::copy(typeParams.begin(), typeParams.end(), begin());
1459 ObjCTypeParamList *ObjCTypeParamList::create(
1461 SourceLocation lAngleLoc,
1462 ArrayRef<ObjCTypeParamDecl *> typeParams,
1463 SourceLocation rAngleLoc) {
1465 ctx.Allocate(totalSizeToAlloc<ObjCTypeParamDecl *>(typeParams.size()),
1466 alignof(ObjCTypeParamList));
1467 return new (mem) ObjCTypeParamList(lAngleLoc, typeParams, rAngleLoc);
1470 void ObjCTypeParamList::gatherDefaultTypeArgs(
1471 SmallVectorImpl<QualType> &typeArgs) const {
1472 typeArgs.reserve(size());
1473 for (auto typeParam : *this)
1474 typeArgs.push_back(typeParam->getUnderlyingType());
1477 //===----------------------------------------------------------------------===//
1478 // ObjCInterfaceDecl
1479 //===----------------------------------------------------------------------===//
1481 ObjCInterfaceDecl *ObjCInterfaceDecl::Create(const ASTContext &C,
1483 SourceLocation atLoc,
1485 ObjCTypeParamList *typeParamList,
1486 ObjCInterfaceDecl *PrevDecl,
1487 SourceLocation ClassLoc,
1489 auto *Result = new (C, DC)
1490 ObjCInterfaceDecl(C, DC, atLoc, Id, typeParamList, ClassLoc, PrevDecl,
1492 Result->Data.setInt(!C.getLangOpts().Modules);
1493 C.getObjCInterfaceType(Result, PrevDecl);
1497 ObjCInterfaceDecl *ObjCInterfaceDecl::CreateDeserialized(const ASTContext &C,
1499 auto *Result = new (C, ID)
1500 ObjCInterfaceDecl(C, nullptr, SourceLocation(), nullptr, nullptr,
1501 SourceLocation(), nullptr, false);
1502 Result->Data.setInt(!C.getLangOpts().Modules);
1506 ObjCInterfaceDecl::ObjCInterfaceDecl(const ASTContext &C, DeclContext *DC,
1507 SourceLocation AtLoc, IdentifierInfo *Id,
1508 ObjCTypeParamList *typeParamList,
1509 SourceLocation CLoc,
1510 ObjCInterfaceDecl *PrevDecl,
1512 : ObjCContainerDecl(ObjCInterface, DC, Id, CLoc, AtLoc),
1513 redeclarable_base(C) {
1514 setPreviousDecl(PrevDecl);
1516 // Copy the 'data' pointer over.
1518 Data = PrevDecl->Data;
1520 setImplicit(IsInternal);
1522 setTypeParamList(typeParamList);
1525 void ObjCInterfaceDecl::LoadExternalDefinition() const {
1526 assert(data().ExternallyCompleted && "Class is not externally completed");
1527 data().ExternallyCompleted = false;
1528 getASTContext().getExternalSource()->CompleteType(
1529 const_cast<ObjCInterfaceDecl *>(this));
1532 void ObjCInterfaceDecl::setExternallyCompleted() {
1533 assert(getASTContext().getExternalSource() &&
1534 "Class can't be externally completed without an external source");
1535 assert(hasDefinition() &&
1536 "Forward declarations can't be externally completed");
1537 data().ExternallyCompleted = true;
1540 void ObjCInterfaceDecl::setHasDesignatedInitializers() {
1541 // Check for a complete definition and recover if not so.
1542 if (!isThisDeclarationADefinition())
1544 data().HasDesignatedInitializers = true;
1547 bool ObjCInterfaceDecl::hasDesignatedInitializers() const {
1548 // Check for a complete definition and recover if not so.
1549 if (!isThisDeclarationADefinition())
1551 if (data().ExternallyCompleted)
1552 LoadExternalDefinition();
1554 return data().HasDesignatedInitializers;
1558 ObjCInterfaceDecl::getObjCRuntimeNameAsString() const {
1559 if (const auto *ObjCRTName = getAttr<ObjCRuntimeNameAttr>())
1560 return ObjCRTName->getMetadataName();
1566 ObjCImplementationDecl::getObjCRuntimeNameAsString() const {
1567 if (ObjCInterfaceDecl *ID =
1568 const_cast<ObjCImplementationDecl*>(this)->getClassInterface())
1569 return ID->getObjCRuntimeNameAsString();
1574 ObjCImplementationDecl *ObjCInterfaceDecl::getImplementation() const {
1575 if (const ObjCInterfaceDecl *Def = getDefinition()) {
1576 if (data().ExternallyCompleted)
1577 LoadExternalDefinition();
1579 return getASTContext().getObjCImplementation(
1580 const_cast<ObjCInterfaceDecl*>(Def));
1583 // FIXME: Should make sure no callers ever do this.
1587 void ObjCInterfaceDecl::setImplementation(ObjCImplementationDecl *ImplD) {
1588 getASTContext().setObjCImplementation(getDefinition(), ImplD);
1593 struct SynthesizeIvarChunk {
1597 SynthesizeIvarChunk(uint64_t size, ObjCIvarDecl *ivar)
1598 : Size(size), Ivar(ivar) {}
1601 bool operator<(const SynthesizeIvarChunk & LHS,
1602 const SynthesizeIvarChunk &RHS) {
1603 return LHS.Size < RHS.Size;
1608 /// all_declared_ivar_begin - return first ivar declared in this class,
1609 /// its extensions and its implementation. Lazily build the list on first
1612 /// Caveat: The list returned by this method reflects the current
1613 /// state of the parser. The cache will be updated for every ivar
1614 /// added by an extension or the implementation when they are
1616 /// See also ObjCIvarDecl::Create().
1617 ObjCIvarDecl *ObjCInterfaceDecl::all_declared_ivar_begin() {
1618 // FIXME: Should make sure no callers ever do this.
1619 if (!hasDefinition())
1622 ObjCIvarDecl *curIvar = nullptr;
1623 if (!data().IvarList) {
1624 if (!ivar_empty()) {
1625 ObjCInterfaceDecl::ivar_iterator I = ivar_begin(), E = ivar_end();
1626 data().IvarList = *I; ++I;
1627 for (curIvar = data().IvarList; I != E; curIvar = *I, ++I)
1628 curIvar->setNextIvar(*I);
1631 for (const auto *Ext : known_extensions()) {
1632 if (!Ext->ivar_empty()) {
1633 ObjCCategoryDecl::ivar_iterator
1634 I = Ext->ivar_begin(),
1635 E = Ext->ivar_end();
1636 if (!data().IvarList) {
1637 data().IvarList = *I; ++I;
1638 curIvar = data().IvarList;
1640 for ( ;I != E; curIvar = *I, ++I)
1641 curIvar->setNextIvar(*I);
1644 data().IvarListMissingImplementation = true;
1647 // cached and complete!
1648 if (!data().IvarListMissingImplementation)
1649 return data().IvarList;
1651 if (ObjCImplementationDecl *ImplDecl = getImplementation()) {
1652 data().IvarListMissingImplementation = false;
1653 if (!ImplDecl->ivar_empty()) {
1654 SmallVector<SynthesizeIvarChunk, 16> layout;
1655 for (auto *IV : ImplDecl->ivars()) {
1656 if (IV->getSynthesize() && !IV->isInvalidDecl()) {
1657 layout.push_back(SynthesizeIvarChunk(
1658 IV->getASTContext().getTypeSize(IV->getType()), IV));
1661 if (!data().IvarList)
1662 data().IvarList = IV;
1664 curIvar->setNextIvar(IV);
1668 if (!layout.empty()) {
1669 // Order synthesized ivars by their size.
1670 llvm::stable_sort(layout);
1671 unsigned Ix = 0, EIx = layout.size();
1672 if (!data().IvarList) {
1673 data().IvarList = layout[0].Ivar; Ix++;
1674 curIvar = data().IvarList;
1676 for ( ; Ix != EIx; curIvar = layout[Ix].Ivar, Ix++)
1677 curIvar->setNextIvar(layout[Ix].Ivar);
1681 return data().IvarList;
1684 /// FindCategoryDeclaration - Finds category declaration in the list of
1685 /// categories for this class and returns it. Name of the category is passed
1686 /// in 'CategoryId'. If category not found, return 0;
1689 ObjCInterfaceDecl::FindCategoryDeclaration(IdentifierInfo *CategoryId) const {
1690 // FIXME: Should make sure no callers ever do this.
1691 if (!hasDefinition())
1694 if (data().ExternallyCompleted)
1695 LoadExternalDefinition();
1697 for (auto *Cat : visible_categories())
1698 if (Cat->getIdentifier() == CategoryId)
1705 ObjCInterfaceDecl::getCategoryInstanceMethod(Selector Sel) const {
1706 for (const auto *Cat : visible_categories()) {
1707 if (ObjCCategoryImplDecl *Impl = Cat->getImplementation())
1708 if (ObjCMethodDecl *MD = Impl->getInstanceMethod(Sel))
1715 ObjCMethodDecl *ObjCInterfaceDecl::getCategoryClassMethod(Selector Sel) const {
1716 for (const auto *Cat : visible_categories()) {
1717 if (ObjCCategoryImplDecl *Impl = Cat->getImplementation())
1718 if (ObjCMethodDecl *MD = Impl->getClassMethod(Sel))
1725 /// ClassImplementsProtocol - Checks that 'lProto' protocol
1726 /// has been implemented in IDecl class, its super class or categories (if
1727 /// lookupCategory is true).
1728 bool ObjCInterfaceDecl::ClassImplementsProtocol(ObjCProtocolDecl *lProto,
1729 bool lookupCategory,
1730 bool RHSIsQualifiedID) {
1731 if (!hasDefinition())
1734 ObjCInterfaceDecl *IDecl = this;
1735 // 1st, look up the class.
1736 for (auto *PI : IDecl->protocols()){
1737 if (getASTContext().ProtocolCompatibleWithProtocol(lProto, PI))
1739 // This is dubious and is added to be compatible with gcc. In gcc, it is
1740 // also allowed assigning a protocol-qualified 'id' type to a LHS object
1741 // when protocol in qualified LHS is in list of protocols in the rhs 'id'
1742 // object. This IMO, should be a bug.
1743 // FIXME: Treat this as an extension, and flag this as an error when GCC
1744 // extensions are not enabled.
1745 if (RHSIsQualifiedID &&
1746 getASTContext().ProtocolCompatibleWithProtocol(PI, lProto))
1750 // 2nd, look up the category.
1752 for (const auto *Cat : visible_categories()) {
1753 for (auto *PI : Cat->protocols())
1754 if (getASTContext().ProtocolCompatibleWithProtocol(lProto, PI))
1758 // 3rd, look up the super class(s)
1759 if (IDecl->getSuperClass())
1761 IDecl->getSuperClass()->ClassImplementsProtocol(lProto, lookupCategory,
1767 //===----------------------------------------------------------------------===//
1769 //===----------------------------------------------------------------------===//
1771 void ObjCIvarDecl::anchor() {}
1773 ObjCIvarDecl *ObjCIvarDecl::Create(ASTContext &C, ObjCContainerDecl *DC,
1774 SourceLocation StartLoc,
1775 SourceLocation IdLoc, IdentifierInfo *Id,
1776 QualType T, TypeSourceInfo *TInfo,
1777 AccessControl ac, Expr *BW,
1780 // Ivar's can only appear in interfaces, implementations (via synthesized
1781 // properties), and class extensions (via direct declaration, or synthesized
1784 // FIXME: This should really be asserting this:
1785 // (isa<ObjCCategoryDecl>(DC) &&
1786 // cast<ObjCCategoryDecl>(DC)->IsClassExtension()))
1787 // but unfortunately we sometimes place ivars into non-class extension
1788 // categories on error. This breaks an AST invariant, and should not be
1790 assert((isa<ObjCInterfaceDecl>(DC) || isa<ObjCImplementationDecl>(DC) ||
1791 isa<ObjCCategoryDecl>(DC)) &&
1792 "Invalid ivar decl context!");
1793 // Once a new ivar is created in any of class/class-extension/implementation
1794 // decl contexts, the previously built IvarList must be rebuilt.
1795 auto *ID = dyn_cast<ObjCInterfaceDecl>(DC);
1797 if (auto *IM = dyn_cast<ObjCImplementationDecl>(DC))
1798 ID = IM->getClassInterface();
1800 ID = cast<ObjCCategoryDecl>(DC)->getClassInterface();
1802 ID->setIvarList(nullptr);
1805 return new (C, DC) ObjCIvarDecl(DC, StartLoc, IdLoc, Id, T, TInfo, ac, BW,
1809 ObjCIvarDecl *ObjCIvarDecl::CreateDeserialized(ASTContext &C, unsigned ID) {
1810 return new (C, ID) ObjCIvarDecl(nullptr, SourceLocation(), SourceLocation(),
1811 nullptr, QualType(), nullptr,
1812 ObjCIvarDecl::None, nullptr, false);
1815 const ObjCInterfaceDecl *ObjCIvarDecl::getContainingInterface() const {
1816 const auto *DC = cast<ObjCContainerDecl>(getDeclContext());
1818 switch (DC->getKind()) {
1820 case ObjCCategoryImpl:
1822 llvm_unreachable("invalid ivar container!");
1824 // Ivars can only appear in class extension categories.
1825 case ObjCCategory: {
1826 const auto *CD = cast<ObjCCategoryDecl>(DC);
1827 assert(CD->IsClassExtension() && "invalid container for ivar!");
1828 return CD->getClassInterface();
1831 case ObjCImplementation:
1832 return cast<ObjCImplementationDecl>(DC)->getClassInterface();
1835 return cast<ObjCInterfaceDecl>(DC);
1839 QualType ObjCIvarDecl::getUsageType(QualType objectType) const {
1840 return getType().substObjCMemberType(objectType, getDeclContext(),
1841 ObjCSubstitutionContext::Property);
1844 //===----------------------------------------------------------------------===//
1845 // ObjCAtDefsFieldDecl
1846 //===----------------------------------------------------------------------===//
1848 void ObjCAtDefsFieldDecl::anchor() {}
1851 *ObjCAtDefsFieldDecl::Create(ASTContext &C, DeclContext *DC,
1852 SourceLocation StartLoc, SourceLocation IdLoc,
1853 IdentifierInfo *Id, QualType T, Expr *BW) {
1854 return new (C, DC) ObjCAtDefsFieldDecl(DC, StartLoc, IdLoc, Id, T, BW);
1857 ObjCAtDefsFieldDecl *ObjCAtDefsFieldDecl::CreateDeserialized(ASTContext &C,
1859 return new (C, ID) ObjCAtDefsFieldDecl(nullptr, SourceLocation(),
1860 SourceLocation(), nullptr, QualType(),
1864 //===----------------------------------------------------------------------===//
1866 //===----------------------------------------------------------------------===//
1868 void ObjCProtocolDecl::anchor() {}
1870 ObjCProtocolDecl::ObjCProtocolDecl(ASTContext &C, DeclContext *DC,
1871 IdentifierInfo *Id, SourceLocation nameLoc,
1872 SourceLocation atStartLoc,
1873 ObjCProtocolDecl *PrevDecl)
1874 : ObjCContainerDecl(ObjCProtocol, DC, Id, nameLoc, atStartLoc),
1875 redeclarable_base(C) {
1876 setPreviousDecl(PrevDecl);
1878 Data = PrevDecl->Data;
1881 ObjCProtocolDecl *ObjCProtocolDecl::Create(ASTContext &C, DeclContext *DC,
1883 SourceLocation nameLoc,
1884 SourceLocation atStartLoc,
1885 ObjCProtocolDecl *PrevDecl) {
1887 new (C, DC) ObjCProtocolDecl(C, DC, Id, nameLoc, atStartLoc, PrevDecl);
1888 Result->Data.setInt(!C.getLangOpts().Modules);
1892 ObjCProtocolDecl *ObjCProtocolDecl::CreateDeserialized(ASTContext &C,
1894 ObjCProtocolDecl *Result =
1895 new (C, ID) ObjCProtocolDecl(C, nullptr, nullptr, SourceLocation(),
1896 SourceLocation(), nullptr);
1897 Result->Data.setInt(!C.getLangOpts().Modules);
1901 ObjCProtocolDecl *ObjCProtocolDecl::lookupProtocolNamed(IdentifierInfo *Name) {
1902 ObjCProtocolDecl *PDecl = this;
1904 if (Name == getIdentifier())
1907 for (auto *I : protocols())
1908 if ((PDecl = I->lookupProtocolNamed(Name)))
1914 // lookupMethod - Lookup a instance/class method in the protocol and protocols
1916 ObjCMethodDecl *ObjCProtocolDecl::lookupMethod(Selector Sel,
1917 bool isInstance) const {
1918 ObjCMethodDecl *MethodDecl = nullptr;
1920 // If there is no definition or the definition is hidden, we don't find
1922 const ObjCProtocolDecl *Def = getDefinition();
1923 if (!Def || Def->isHidden())
1926 if ((MethodDecl = getMethod(Sel, isInstance)))
1929 for (const auto *I : protocols())
1930 if ((MethodDecl = I->lookupMethod(Sel, isInstance)))
1935 void ObjCProtocolDecl::allocateDefinitionData() {
1936 assert(!Data.getPointer() && "Protocol already has a definition!");
1937 Data.setPointer(new (getASTContext()) DefinitionData);
1938 Data.getPointer()->Definition = this;
1941 void ObjCProtocolDecl::startDefinition() {
1942 allocateDefinitionData();
1944 // Update all of the declarations with a pointer to the definition.
1945 for (auto *RD : redecls())
1946 RD->Data = this->Data;
1949 void ObjCProtocolDecl::collectPropertiesToImplement(PropertyMap &PM,
1950 PropertyDeclOrder &PO) const {
1951 if (const ObjCProtocolDecl *PDecl = getDefinition()) {
1952 for (auto *Prop : PDecl->properties()) {
1953 // Insert into PM if not there already.
1954 PM.insert(std::make_pair(
1955 std::make_pair(Prop->getIdentifier(), Prop->isClassProperty()),
1959 // Scan through protocol's protocols.
1960 for (const auto *PI : PDecl->protocols())
1961 PI->collectPropertiesToImplement(PM, PO);
1965 void ObjCProtocolDecl::collectInheritedProtocolProperties(
1966 const ObjCPropertyDecl *Property, ProtocolPropertySet &PS,
1967 PropertyDeclOrder &PO) const {
1968 if (const ObjCProtocolDecl *PDecl = getDefinition()) {
1969 if (!PS.insert(PDecl).second)
1971 for (auto *Prop : PDecl->properties()) {
1972 if (Prop == Property)
1974 if (Prop->getIdentifier() == Property->getIdentifier()) {
1979 // Scan through protocol's protocols which did not have a matching property.
1980 for (const auto *PI : PDecl->protocols())
1981 PI->collectInheritedProtocolProperties(Property, PS, PO);
1986 ObjCProtocolDecl::getObjCRuntimeNameAsString() const {
1987 if (const auto *ObjCRTName = getAttr<ObjCRuntimeNameAttr>())
1988 return ObjCRTName->getMetadataName();
1993 //===----------------------------------------------------------------------===//
1995 //===----------------------------------------------------------------------===//
1997 void ObjCCategoryDecl::anchor() {}
1999 ObjCCategoryDecl::ObjCCategoryDecl(DeclContext *DC, SourceLocation AtLoc,
2000 SourceLocation ClassNameLoc,
2001 SourceLocation CategoryNameLoc,
2002 IdentifierInfo *Id, ObjCInterfaceDecl *IDecl,
2003 ObjCTypeParamList *typeParamList,
2004 SourceLocation IvarLBraceLoc,
2005 SourceLocation IvarRBraceLoc)
2006 : ObjCContainerDecl(ObjCCategory, DC, Id, ClassNameLoc, AtLoc),
2007 ClassInterface(IDecl), CategoryNameLoc(CategoryNameLoc),
2008 IvarLBraceLoc(IvarLBraceLoc), IvarRBraceLoc(IvarRBraceLoc) {
2009 setTypeParamList(typeParamList);
2012 ObjCCategoryDecl *ObjCCategoryDecl::Create(ASTContext &C, DeclContext *DC,
2013 SourceLocation AtLoc,
2014 SourceLocation ClassNameLoc,
2015 SourceLocation CategoryNameLoc,
2017 ObjCInterfaceDecl *IDecl,
2018 ObjCTypeParamList *typeParamList,
2019 SourceLocation IvarLBraceLoc,
2020 SourceLocation IvarRBraceLoc) {
2022 new (C, DC) ObjCCategoryDecl(DC, AtLoc, ClassNameLoc, CategoryNameLoc, Id,
2023 IDecl, typeParamList, IvarLBraceLoc,
2026 // Link this category into its class's category list.
2027 CatDecl->NextClassCategory = IDecl->getCategoryListRaw();
2028 if (IDecl->hasDefinition()) {
2029 IDecl->setCategoryListRaw(CatDecl);
2030 if (ASTMutationListener *L = C.getASTMutationListener())
2031 L->AddedObjCCategoryToInterface(CatDecl, IDecl);
2038 ObjCCategoryDecl *ObjCCategoryDecl::CreateDeserialized(ASTContext &C,
2040 return new (C, ID) ObjCCategoryDecl(nullptr, SourceLocation(),
2041 SourceLocation(), SourceLocation(),
2042 nullptr, nullptr, nullptr);
2045 ObjCCategoryImplDecl *ObjCCategoryDecl::getImplementation() const {
2046 return getASTContext().getObjCImplementation(
2047 const_cast<ObjCCategoryDecl*>(this));
2050 void ObjCCategoryDecl::setImplementation(ObjCCategoryImplDecl *ImplD) {
2051 getASTContext().setObjCImplementation(this, ImplD);
2054 void ObjCCategoryDecl::setTypeParamList(ObjCTypeParamList *TPL) {
2055 TypeParamList = TPL;
2058 // Set the declaration context of each of the type parameters.
2059 for (auto *typeParam : *TypeParamList)
2060 typeParam->setDeclContext(this);
2063 //===----------------------------------------------------------------------===//
2064 // ObjCCategoryImplDecl
2065 //===----------------------------------------------------------------------===//
2067 void ObjCCategoryImplDecl::anchor() {}
2069 ObjCCategoryImplDecl *
2070 ObjCCategoryImplDecl::Create(ASTContext &C, DeclContext *DC,
2072 ObjCInterfaceDecl *ClassInterface,
2073 SourceLocation nameLoc,
2074 SourceLocation atStartLoc,
2075 SourceLocation CategoryNameLoc) {
2076 if (ClassInterface && ClassInterface->hasDefinition())
2077 ClassInterface = ClassInterface->getDefinition();
2078 return new (C, DC) ObjCCategoryImplDecl(DC, Id, ClassInterface, nameLoc,
2079 atStartLoc, CategoryNameLoc);
2082 ObjCCategoryImplDecl *ObjCCategoryImplDecl::CreateDeserialized(ASTContext &C,
2084 return new (C, ID) ObjCCategoryImplDecl(nullptr, nullptr, nullptr,
2085 SourceLocation(), SourceLocation(),
2089 ObjCCategoryDecl *ObjCCategoryImplDecl::getCategoryDecl() const {
2090 // The class interface might be NULL if we are working with invalid code.
2091 if (const ObjCInterfaceDecl *ID = getClassInterface())
2092 return ID->FindCategoryDeclaration(getIdentifier());
2096 void ObjCImplDecl::anchor() {}
2098 void ObjCImplDecl::addPropertyImplementation(ObjCPropertyImplDecl *property) {
2099 // FIXME: The context should be correct before we get here.
2100 property->setLexicalDeclContext(this);
2104 void ObjCImplDecl::setClassInterface(ObjCInterfaceDecl *IFace) {
2105 ASTContext &Ctx = getASTContext();
2107 if (auto *ImplD = dyn_cast_or_null<ObjCImplementationDecl>(this)) {
2109 Ctx.setObjCImplementation(IFace, ImplD);
2111 } else if (auto *ImplD = dyn_cast_or_null<ObjCCategoryImplDecl>(this)) {
2112 if (ObjCCategoryDecl *CD = IFace->FindCategoryDeclaration(getIdentifier()))
2113 Ctx.setObjCImplementation(CD, ImplD);
2116 ClassInterface = IFace;
2119 /// FindPropertyImplIvarDecl - This method lookup the ivar in the list of
2120 /// properties implemented in this \@implementation block and returns
2121 /// the implemented property that uses it.
2122 ObjCPropertyImplDecl *ObjCImplDecl::
2123 FindPropertyImplIvarDecl(IdentifierInfo *ivarId) const {
2124 for (auto *PID : property_impls())
2125 if (PID->getPropertyIvarDecl() &&
2126 PID->getPropertyIvarDecl()->getIdentifier() == ivarId)
2131 /// FindPropertyImplDecl - This method looks up a previous ObjCPropertyImplDecl
2132 /// added to the list of those properties \@synthesized/\@dynamic in this
2133 /// category \@implementation block.
2134 ObjCPropertyImplDecl *ObjCImplDecl::
2135 FindPropertyImplDecl(IdentifierInfo *Id,
2136 ObjCPropertyQueryKind QueryKind) const {
2137 ObjCPropertyImplDecl *ClassPropImpl = nullptr;
2138 for (auto *PID : property_impls())
2139 // If queryKind is unknown, we return the instance property if one
2140 // exists; otherwise we return the class property.
2141 if (PID->getPropertyDecl()->getIdentifier() == Id) {
2142 if ((QueryKind == ObjCPropertyQueryKind::OBJC_PR_query_unknown &&
2143 !PID->getPropertyDecl()->isClassProperty()) ||
2144 (QueryKind == ObjCPropertyQueryKind::OBJC_PR_query_class &&
2145 PID->getPropertyDecl()->isClassProperty()) ||
2146 (QueryKind == ObjCPropertyQueryKind::OBJC_PR_query_instance &&
2147 !PID->getPropertyDecl()->isClassProperty()))
2150 if (PID->getPropertyDecl()->isClassProperty())
2151 ClassPropImpl = PID;
2154 if (QueryKind == ObjCPropertyQueryKind::OBJC_PR_query_unknown)
2155 // We can't find the instance property, return the class property.
2156 return ClassPropImpl;
2161 raw_ostream &clang::operator<<(raw_ostream &OS,
2162 const ObjCCategoryImplDecl &CID) {
2163 OS << CID.getName();
2167 //===----------------------------------------------------------------------===//
2168 // ObjCImplementationDecl
2169 //===----------------------------------------------------------------------===//
2171 void ObjCImplementationDecl::anchor() {}
2173 ObjCImplementationDecl *
2174 ObjCImplementationDecl::Create(ASTContext &C, DeclContext *DC,
2175 ObjCInterfaceDecl *ClassInterface,
2176 ObjCInterfaceDecl *SuperDecl,
2177 SourceLocation nameLoc,
2178 SourceLocation atStartLoc,
2179 SourceLocation superLoc,
2180 SourceLocation IvarLBraceLoc,
2181 SourceLocation IvarRBraceLoc) {
2182 if (ClassInterface && ClassInterface->hasDefinition())
2183 ClassInterface = ClassInterface->getDefinition();
2184 return new (C, DC) ObjCImplementationDecl(DC, ClassInterface, SuperDecl,
2185 nameLoc, atStartLoc, superLoc,
2186 IvarLBraceLoc, IvarRBraceLoc);
2189 ObjCImplementationDecl *
2190 ObjCImplementationDecl::CreateDeserialized(ASTContext &C, unsigned ID) {
2191 return new (C, ID) ObjCImplementationDecl(nullptr, nullptr, nullptr,
2192 SourceLocation(), SourceLocation());
2195 void ObjCImplementationDecl::setIvarInitializers(ASTContext &C,
2196 CXXCtorInitializer ** initializers,
2197 unsigned numInitializers) {
2198 if (numInitializers > 0) {
2199 NumIvarInitializers = numInitializers;
2200 auto **ivarInitializers = new (C) CXXCtorInitializer*[NumIvarInitializers];
2201 memcpy(ivarInitializers, initializers,
2202 numInitializers * sizeof(CXXCtorInitializer*));
2203 IvarInitializers = ivarInitializers;
2207 ObjCImplementationDecl::init_const_iterator
2208 ObjCImplementationDecl::init_begin() const {
2209 return IvarInitializers.get(getASTContext().getExternalSource());
2212 raw_ostream &clang::operator<<(raw_ostream &OS,
2213 const ObjCImplementationDecl &ID) {
2218 //===----------------------------------------------------------------------===//
2219 // ObjCCompatibleAliasDecl
2220 //===----------------------------------------------------------------------===//
2222 void ObjCCompatibleAliasDecl::anchor() {}
2224 ObjCCompatibleAliasDecl *
2225 ObjCCompatibleAliasDecl::Create(ASTContext &C, DeclContext *DC,
2228 ObjCInterfaceDecl* AliasedClass) {
2229 return new (C, DC) ObjCCompatibleAliasDecl(DC, L, Id, AliasedClass);
2232 ObjCCompatibleAliasDecl *
2233 ObjCCompatibleAliasDecl::CreateDeserialized(ASTContext &C, unsigned ID) {
2234 return new (C, ID) ObjCCompatibleAliasDecl(nullptr, SourceLocation(),
2238 //===----------------------------------------------------------------------===//
2240 //===----------------------------------------------------------------------===//
2242 void ObjCPropertyDecl::anchor() {}
2244 ObjCPropertyDecl *ObjCPropertyDecl::Create(ASTContext &C, DeclContext *DC,
2247 SourceLocation AtLoc,
2248 SourceLocation LParenLoc,
2250 TypeSourceInfo *TSI,
2251 PropertyControl propControl) {
2252 return new (C, DC) ObjCPropertyDecl(DC, L, Id, AtLoc, LParenLoc, T, TSI,
2256 ObjCPropertyDecl *ObjCPropertyDecl::CreateDeserialized(ASTContext &C,
2258 return new (C, ID) ObjCPropertyDecl(nullptr, SourceLocation(), nullptr,
2259 SourceLocation(), SourceLocation(),
2260 QualType(), nullptr, None);
2263 QualType ObjCPropertyDecl::getUsageType(QualType objectType) const {
2264 return DeclType.substObjCMemberType(objectType, getDeclContext(),
2265 ObjCSubstitutionContext::Property);
2268 //===----------------------------------------------------------------------===//
2269 // ObjCPropertyImplDecl
2270 //===----------------------------------------------------------------------===//
2272 ObjCPropertyImplDecl *ObjCPropertyImplDecl::Create(ASTContext &C,
2274 SourceLocation atLoc,
2276 ObjCPropertyDecl *property,
2279 SourceLocation ivarLoc) {
2280 return new (C, DC) ObjCPropertyImplDecl(DC, atLoc, L, property, PK, ivar,
2284 ObjCPropertyImplDecl *ObjCPropertyImplDecl::CreateDeserialized(ASTContext &C,
2286 return new (C, ID) ObjCPropertyImplDecl(nullptr, SourceLocation(),
2287 SourceLocation(), nullptr, Dynamic,
2288 nullptr, SourceLocation());
2291 SourceRange ObjCPropertyImplDecl::getSourceRange() const {
2292 SourceLocation EndLoc = getLocation();
2293 if (IvarLoc.isValid())
2296 return SourceRange(AtLoc, EndLoc);