1 //===- DeclFriend.h - Classes for C++ friend declarations -------*- C++ -*-===//
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 defines the section of the AST representing C++ friend
12 //===----------------------------------------------------------------------===//
14 #ifndef LLVM_CLANG_AST_DECLFRIEND_H
15 #define LLVM_CLANG_AST_DECLFRIEND_H
17 #include "clang/AST/Decl.h"
18 #include "clang/AST/DeclBase.h"
19 #include "clang/AST/DeclCXX.h"
20 #include "clang/AST/DeclTemplate.h"
21 #include "clang/AST/ExternalASTSource.h"
22 #include "clang/AST/TypeLoc.h"
23 #include "clang/Basic/LLVM.h"
24 #include "clang/Basic/SourceLocation.h"
25 #include "llvm/ADT/ArrayRef.h"
26 #include "llvm/ADT/None.h"
27 #include "llvm/ADT/PointerUnion.h"
28 #include "llvm/Support/Casting.h"
29 #include "llvm/Support/Compiler.h"
30 #include "llvm/Support/TrailingObjects.h"
38 /// FriendDecl - Represents the declaration of a friend entity,
39 /// which can be a function, a type, or a templated function or type.
43 /// template <typename T> class A {
44 /// friend int foo(T);
46 /// friend T; // only in C++0x
47 /// template <typename U> friend class C;
48 /// template <typename U> friend A& operator+=(A&, const U&) { ... }
52 /// The semantic context of a friend decl is its declaring class.
53 class FriendDecl final
55 private llvm::TrailingObjects<FriendDecl, TemplateParameterList *> {
56 virtual void anchor();
59 using FriendUnion = llvm::PointerUnion<NamedDecl *, TypeSourceInfo *>;
62 friend class CXXRecordDecl;
63 friend class CXXRecordDecl::friend_iterator;
65 // The declaration that's a friend of this class.
68 // A pointer to the next friend in the sequence.
69 LazyDeclPtr NextFriend;
71 // Location of the 'friend' specifier.
72 SourceLocation FriendLoc;
74 /// True if this 'friend' declaration is unsupported. Eventually we
75 /// will support every possible friend declaration, but for now we
76 /// silently ignore some and set this flag to authorize all access.
77 unsigned UnsupportedFriend : 1;
79 // The number of "outer" template parameter lists in non-templatic
80 // (currently unsupported) friend type declarations, such as
81 // template <class T> friend class A<T>::B;
82 unsigned NumTPLists : 31;
84 FriendDecl(DeclContext *DC, SourceLocation L, FriendUnion Friend,
85 SourceLocation FriendL,
86 ArrayRef<TemplateParameterList *> FriendTypeTPLists)
87 : Decl(Decl::Friend, DC, L), Friend(Friend), FriendLoc(FriendL),
88 UnsupportedFriend(false), NumTPLists(FriendTypeTPLists.size()) {
89 for (unsigned i = 0; i < NumTPLists; ++i)
90 getTrailingObjects<TemplateParameterList *>()[i] = FriendTypeTPLists[i];
93 FriendDecl(EmptyShell Empty, unsigned NumFriendTypeTPLists)
94 : Decl(Decl::Friend, Empty), UnsupportedFriend(false),
95 NumTPLists(NumFriendTypeTPLists) {}
97 FriendDecl *getNextFriend() {
98 if (!NextFriend.isOffset())
99 return cast_or_null<FriendDecl>(NextFriend.get(nullptr));
100 return getNextFriendSlowCase();
103 FriendDecl *getNextFriendSlowCase();
106 friend class ASTDeclReader;
107 friend class ASTDeclWriter;
108 friend class ASTNodeImporter;
109 friend TrailingObjects;
111 static FriendDecl *Create(ASTContext &C, DeclContext *DC,
112 SourceLocation L, FriendUnion Friend_,
113 SourceLocation FriendL,
114 ArrayRef<TemplateParameterList*> FriendTypeTPLists
116 static FriendDecl *CreateDeserialized(ASTContext &C, unsigned ID,
117 unsigned FriendTypeNumTPLists);
119 /// If this friend declaration names an (untemplated but possibly
120 /// dependent) type, return the type; otherwise return null. This
121 /// is used for elaborated-type-specifiers and, in C++0x, for
122 /// arbitrary friend type declarations.
123 TypeSourceInfo *getFriendType() const {
124 return Friend.dyn_cast<TypeSourceInfo*>();
127 unsigned getFriendTypeNumTemplateParameterLists() const {
131 TemplateParameterList *getFriendTypeTemplateParameterList(unsigned N) const {
132 assert(N < NumTPLists);
133 return getTrailingObjects<TemplateParameterList *>()[N];
136 /// If this friend declaration doesn't name a type, return the inner
138 NamedDecl *getFriendDecl() const {
139 return Friend.dyn_cast<NamedDecl *>();
142 /// Retrieves the location of the 'friend' keyword.
143 SourceLocation getFriendLoc() const {
147 /// Retrieves the source range for the friend declaration.
148 SourceRange getSourceRange() const override LLVM_READONLY {
149 if (NamedDecl *ND = getFriendDecl()) {
150 if (const auto *FD = dyn_cast<FunctionDecl>(ND))
151 return FD->getSourceRange();
152 if (const auto *FTD = dyn_cast<FunctionTemplateDecl>(ND))
153 return FTD->getSourceRange();
154 if (const auto *CTD = dyn_cast<ClassTemplateDecl>(ND))
155 return CTD->getSourceRange();
156 if (const auto *DD = dyn_cast<DeclaratorDecl>(ND)) {
157 if (DD->getOuterLocStart() != DD->getInnerLocStart())
158 return DD->getSourceRange();
160 return SourceRange(getFriendLoc(), ND->getEndLoc());
162 else if (TypeSourceInfo *TInfo = getFriendType()) {
163 SourceLocation StartL =
164 (NumTPLists == 0) ? getFriendLoc()
165 : getTrailingObjects<TemplateParameterList *>()[0]
167 return SourceRange(StartL, TInfo->getTypeLoc().getEndLoc());
170 return SourceRange(getFriendLoc(), getLocation());
173 /// Determines if this friend kind is unsupported.
174 bool isUnsupportedFriend() const {
175 return UnsupportedFriend;
177 void setUnsupportedFriend(bool Unsupported) {
178 UnsupportedFriend = Unsupported;
181 // Implement isa/cast/dyncast/etc.
182 static bool classof(const Decl *D) { return classofKind(D->getKind()); }
183 static bool classofKind(Kind K) { return K == Decl::Friend; }
186 /// An iterator over the friend declarations of a class.
187 class CXXRecordDecl::friend_iterator {
188 friend class CXXRecordDecl;
192 explicit friend_iterator(FriendDecl *Ptr) : Ptr(Ptr) {}
195 friend_iterator() = default;
197 using value_type = FriendDecl *;
198 using reference = FriendDecl *;
199 using pointer = FriendDecl *;
200 using difference_type = int;
201 using iterator_category = std::forward_iterator_tag;
203 reference operator*() const { return Ptr; }
205 friend_iterator &operator++() {
206 assert(Ptr && "attempt to increment past end of friend list");
207 Ptr = Ptr->getNextFriend();
211 friend_iterator operator++(int) {
212 friend_iterator tmp = *this;
217 bool operator==(const friend_iterator &Other) const {
218 return Ptr == Other.Ptr;
221 bool operator!=(const friend_iterator &Other) const {
222 return Ptr != Other.Ptr;
225 friend_iterator &operator+=(difference_type N) {
226 assert(N >= 0 && "cannot rewind a CXXRecordDecl::friend_iterator");
232 friend_iterator operator+(difference_type N) const {
233 friend_iterator tmp = *this;
239 inline CXXRecordDecl::friend_iterator CXXRecordDecl::friend_begin() const {
240 return friend_iterator(getFirstFriend());
243 inline CXXRecordDecl::friend_iterator CXXRecordDecl::friend_end() const {
244 return friend_iterator(nullptr);
247 inline CXXRecordDecl::friend_range CXXRecordDecl::friends() const {
248 return friend_range(friend_begin(), friend_end());
251 inline void CXXRecordDecl::pushFriendDecl(FriendDecl *FD) {
252 assert(!FD->NextFriend && "friend already has next friend?");
253 FD->NextFriend = data().FirstFriend;
254 data().FirstFriend = FD;
259 #endif // LLVM_CLANG_AST_DECLFRIEND_H