From 856c4076a4c032b9945b93a86f19d98496db647a Mon Sep 17 00:00:00 2001 From: Abramo Bagnara Date: Thu, 4 Oct 2012 21:39:47 +0000 Subject: [PATCH] Fixed friend decl source range. llvm-svn: 165257 --- clang/include/clang/AST/DeclFriend.h | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/clang/include/clang/AST/DeclFriend.h b/clang/include/clang/AST/DeclFriend.h index a5f736739b17..b40144388add 100644 --- a/clang/include/clang/AST/DeclFriend.h +++ b/clang/include/clang/AST/DeclFriend.h @@ -16,6 +16,7 @@ #define LLVM_CLANG_AST_DECLFRIEND_H #include "clang/AST/DeclCXX.h" +#include "clang/AST/DeclTemplate.h" #include "llvm/Support/Compiler.h" namespace clang { @@ -104,10 +105,15 @@ public: /// Retrieves the source range for the friend declaration. SourceRange getSourceRange() const LLVM_READONLY { - // FIXME: If this is a friend function declaration, the 'friend' keyword - // might not be the first token of the declaration. - if (NamedDecl *ND = getFriendDecl()) + if (NamedDecl *ND = getFriendDecl()) { + if (FunctionTemplateDecl *FTD = dyn_cast(ND)) + return FTD->getSourceRange(); + if (DeclaratorDecl *DD = dyn_cast(ND)) { + if (DD->getOuterLocStart() != DD->getInnerLocStart()) + return DD->getSourceRange(); + } return SourceRange(getFriendLoc(), ND->getLocEnd()); + } else if (TypeSourceInfo *TInfo = getFriendType()) return SourceRange(getFriendLoc(), TInfo->getTypeLoc().getEndLoc()); else