1 files changed, 5 insertions, 5 deletions
diff --git a/lib/Sema/SemaDeclAttr.cpp b/lib/Sema/SemaDeclAttr.cpp
index e65b3aa6fd..23fe4010fb 100644
--- a/lib/Sema/SemaDeclAttr.cpp
+++ b/lib/Sema/SemaDeclAttr.cpp
@@ -437,16 +437,16 @@ static void HandleMallocAttr(Decl *d, const AttributeList &Attr, Sema &S) {
     S.Diag(Attr.getLoc(), diag::err_attribute_wrong_number_arguments) << 0;
     return;
   }
-
-  const FunctionDecl *FD = dyn_cast<FunctionDecl>(d);
-
-  if (!FD) {
+  
+  const FunctionType *FT = getFunctionType(d, false);
+  
+  if (!FT) {
     S.Diag(Attr.getLoc(), diag::warn_attribute_wrong_decl_type)
       << Attr.getName() << 0 /*function*/;
     return;
   }
 
-  QualType RetTy = FD->getResultType();
+  QualType RetTy = FT->getResultType();
   
   if (!(RetTy->isAnyPointerType() || RetTy->isBlockPointerType())) {
     S.Diag(Attr.getLoc(), diag::warn_attribute_malloc_pointer_only);