diff options
Diffstat (limited to 'lib/Analysis/CFRefCount.cpp')
| -rw-r--r-- | lib/Analysis/CFRefCount.cpp | 18 |
1 files changed, 13 insertions, 5 deletions
diff --git a/lib/Analysis/CFRefCount.cpp b/lib/Analysis/CFRefCount.cpp index acf4b568ed..db8a4598ba 100644 --- a/lib/Analysis/CFRefCount.cpp +++ b/lib/Analysis/CFRefCount.cpp @@ -704,8 +704,12 @@ RetainSummary* RetainSummaryManager::getSummary(FunctionDecl* FD) { } } - if (FName[0] == 'C' && FName[1] == 'F') - S = getCFSummary(FD, FName); + if (FName[0] == 'C') { + if (FName[1] == 'F') + S = getCFSummary(FD, FName); + else if (FName[1] == 'G') + S = getCGSummary(FD, FName); + } else if (FName[0] == 'N' && FName[1] == 'S') S = getNSSummary(FD, FName); } @@ -834,10 +838,14 @@ RetainSummary* RetainSummaryManager::getCFSummaryCreateRule(FunctionDecl* FD) { FunctionType* FT = dyn_cast<FunctionType>(FD->getType().getTypePtr()); - - if (FT && !isCFRefType(FT->getResultType())) - return getPersistentSummary(RetEffect::MakeNoRet()); + if (FT) { + QualType ResTy = FT->getResultType(); + + if (!isCFRefType(ResTy) && !isCGRefType(ResTy)) + return getPersistentSummary(RetEffect::MakeNoRet()); + } + assert (ScratchArgs.empty()); if (FD->getIdentifier() == CFDictionaryCreateII) { |