14f7df9beSJordan Rose //===- Calls.cpp - Wrapper for all function and method calls ------*- C++ -*--//
24f7df9beSJordan Rose //
34f7df9beSJordan Rose //                     The LLVM Compiler Infrastructure
44f7df9beSJordan Rose //
54f7df9beSJordan Rose // This file is distributed under the University of Illinois Open Source
64f7df9beSJordan Rose // License. See LICENSE.TXT for details.
74f7df9beSJordan Rose //
84f7df9beSJordan Rose //===----------------------------------------------------------------------===//
94f7df9beSJordan Rose //
104f7df9beSJordan Rose /// \file This file defines CallEvent and its subclasses, which represent path-
114f7df9beSJordan Rose /// sensitive instances of different kinds of function and method calls
124f7df9beSJordan Rose /// (C, C++, and Objective-C).
134f7df9beSJordan Rose //
144f7df9beSJordan Rose //===----------------------------------------------------------------------===//
154f7df9beSJordan Rose 
164f7df9beSJordan Rose #include "clang/StaticAnalyzer/Core/PathSensitive/CallEvent.h"
174f7df9beSJordan Rose #include "clang/Analysis/ProgramPoint.h"
184f7df9beSJordan Rose #include "clang/AST/ParentMap.h"
194f7df9beSJordan Rose #include "llvm/ADT/SmallSet.h"
204f7df9beSJordan Rose #include "llvm/ADT/StringExtras.h"
214f7df9beSJordan Rose 
224f7df9beSJordan Rose using namespace clang;
234f7df9beSJordan Rose using namespace ento;
244f7df9beSJordan Rose 
254f7df9beSJordan Rose QualType CallEvent::getResultType() const {
262da56438SJordan Rose   const Expr *E = getOriginExpr();
272da56438SJordan Rose   assert(E && "Calls without origin expressions do not have results");
282da56438SJordan Rose   QualType ResultTy = E->getType();
294f7df9beSJordan Rose 
302da56438SJordan Rose   ASTContext &Ctx = getState()->getStateManager().getContext();
312da56438SJordan Rose 
322da56438SJordan Rose   // A function that returns a reference to 'int' will have a result type
332da56438SJordan Rose   // of simply 'int'. Check the origin expr's value kind to recover the
342da56438SJordan Rose   // proper type.
352da56438SJordan Rose   switch (E->getValueKind()) {
362da56438SJordan Rose   case VK_LValue:
372da56438SJordan Rose     ResultTy = Ctx.getLValueReferenceType(ResultTy);
382da56438SJordan Rose     break;
392da56438SJordan Rose   case VK_XValue:
402da56438SJordan Rose     ResultTy = Ctx.getRValueReferenceType(ResultTy);
412da56438SJordan Rose     break;
422da56438SJordan Rose   case VK_RValue:
432da56438SJordan Rose     // No adjustment is necessary.
442da56438SJordan Rose     break;
452da56438SJordan Rose   }
464f7df9beSJordan Rose 
474f7df9beSJordan Rose   return ResultTy;
484f7df9beSJordan Rose }
494f7df9beSJordan Rose 
504f7df9beSJordan Rose static bool isCallbackArg(SVal V, QualType T) {
514f7df9beSJordan Rose   // If the parameter is 0, it's harmless.
524f7df9beSJordan Rose   if (V.isZeroConstant())
534f7df9beSJordan Rose     return false;
544f7df9beSJordan Rose 
554f7df9beSJordan Rose   // If a parameter is a block or a callback, assume it can modify pointer.
564f7df9beSJordan Rose   if (T->isBlockPointerType() ||
574f7df9beSJordan Rose       T->isFunctionPointerType() ||
584f7df9beSJordan Rose       T->isObjCSelType())
594f7df9beSJordan Rose     return true;
604f7df9beSJordan Rose 
614f7df9beSJordan Rose   // Check if a callback is passed inside a struct (for both, struct passed by
624f7df9beSJordan Rose   // reference and by value). Dig just one level into the struct for now.
634f7df9beSJordan Rose 
64d1a08b6eSJordan Rose   if (T->isAnyPointerType() || T->isReferenceType())
654f7df9beSJordan Rose     T = T->getPointeeType();
664f7df9beSJordan Rose 
674f7df9beSJordan Rose   if (const RecordType *RT = T->getAsStructureType()) {
684f7df9beSJordan Rose     const RecordDecl *RD = RT->getDecl();
694f7df9beSJordan Rose     for (RecordDecl::field_iterator I = RD->field_begin(), E = RD->field_end();
704f7df9beSJordan Rose          I != E; ++I) {
714f7df9beSJordan Rose       QualType FieldT = I->getType();
724f7df9beSJordan Rose       if (FieldT->isBlockPointerType() || FieldT->isFunctionPointerType())
734f7df9beSJordan Rose         return true;
744f7df9beSJordan Rose     }
754f7df9beSJordan Rose   }
764f7df9beSJordan Rose 
774f7df9beSJordan Rose   return false;
784f7df9beSJordan Rose }
794f7df9beSJordan Rose 
804f7df9beSJordan Rose bool CallEvent::hasNonZeroCallbackArg() const {
814f7df9beSJordan Rose   unsigned NumOfArgs = getNumArgs();
824f7df9beSJordan Rose 
834f7df9beSJordan Rose   // If calling using a function pointer, assume the function does not
844f7df9beSJordan Rose   // have a callback. TODO: We could check the types of the arguments here.
854f7df9beSJordan Rose   if (!getDecl())
864f7df9beSJordan Rose     return false;
874f7df9beSJordan Rose 
884f7df9beSJordan Rose   unsigned Idx = 0;
894f7df9beSJordan Rose   for (CallEvent::param_type_iterator I = param_type_begin(),
904f7df9beSJordan Rose                                        E = param_type_end();
914f7df9beSJordan Rose        I != E && Idx < NumOfArgs; ++I, ++Idx) {
924f7df9beSJordan Rose     if (NumOfArgs <= Idx)
934f7df9beSJordan Rose       break;
944f7df9beSJordan Rose 
954f7df9beSJordan Rose     if (isCallbackArg(getArgSVal(Idx), *I))
964f7df9beSJordan Rose       return true;
974f7df9beSJordan Rose   }
984f7df9beSJordan Rose 
994f7df9beSJordan Rose   return false;
1004f7df9beSJordan Rose }
1014f7df9beSJordan Rose 
1024f7df9beSJordan Rose /// \brief Returns true if a type is a pointer-to-const or reference-to-const
1034f7df9beSJordan Rose /// with no further indirection.
1044f7df9beSJordan Rose static bool isPointerToConst(QualType Ty) {
1054f7df9beSJordan Rose   QualType PointeeTy = Ty->getPointeeType();
1064f7df9beSJordan Rose   if (PointeeTy == QualType())
1074f7df9beSJordan Rose     return false;
1084f7df9beSJordan Rose   if (!PointeeTy.isConstQualified())
1094f7df9beSJordan Rose     return false;
1104f7df9beSJordan Rose   if (PointeeTy->isAnyPointerType())
1114f7df9beSJordan Rose     return false;
1124f7df9beSJordan Rose   return true;
1134f7df9beSJordan Rose }
1144f7df9beSJordan Rose 
1154f7df9beSJordan Rose // Try to retrieve the function declaration and find the function parameter
1164f7df9beSJordan Rose // types which are pointers/references to a non-pointer const.
1174f7df9beSJordan Rose // We will not invalidate the corresponding argument regions.
1184f7df9beSJordan Rose static void findPtrToConstParams(llvm::SmallSet<unsigned, 1> &PreserveArgs,
1194f7df9beSJordan Rose                                  const CallEvent &Call) {
1204f7df9beSJordan Rose   unsigned Idx = 0;
1214f7df9beSJordan Rose   for (CallEvent::param_type_iterator I = Call.param_type_begin(),
1224f7df9beSJordan Rose                                       E = Call.param_type_end();
1234f7df9beSJordan Rose        I != E; ++I, ++Idx) {
1244f7df9beSJordan Rose     if (isPointerToConst(*I))
1254f7df9beSJordan Rose       PreserveArgs.insert(Idx);
1264f7df9beSJordan Rose   }
1274f7df9beSJordan Rose }
1284f7df9beSJordan Rose 
1294f7df9beSJordan Rose ProgramStateRef CallEvent::invalidateRegions(unsigned BlockCount,
1304f7df9beSJordan Rose                                               ProgramStateRef Orig) const {
1314f7df9beSJordan Rose   ProgramStateRef Result = (Orig ? Orig : getState());
1324f7df9beSJordan Rose 
1334f7df9beSJordan Rose   SmallVector<const MemRegion *, 8> RegionsToInvalidate;
1344f7df9beSJordan Rose   getExtraInvalidatedRegions(RegionsToInvalidate);
1354f7df9beSJordan Rose 
1364f7df9beSJordan Rose   // Indexes of arguments whose values will be preserved by the call.
1374f7df9beSJordan Rose   llvm::SmallSet<unsigned, 1> PreserveArgs;
1384f7df9beSJordan Rose   if (!argumentsMayEscape())
1394f7df9beSJordan Rose     findPtrToConstParams(PreserveArgs, *this);
1404f7df9beSJordan Rose 
1414f7df9beSJordan Rose   for (unsigned Idx = 0, Count = getNumArgs(); Idx != Count; ++Idx) {
1424f7df9beSJordan Rose     if (PreserveArgs.count(Idx))
1434f7df9beSJordan Rose       continue;
1444f7df9beSJordan Rose 
1454f7df9beSJordan Rose     SVal V = getArgSVal(Idx);
1464f7df9beSJordan Rose 
1474f7df9beSJordan Rose     // If we are passing a location wrapped as an integer, unwrap it and
1484f7df9beSJordan Rose     // invalidate the values referred by the location.
1494f7df9beSJordan Rose     if (nonloc::LocAsInteger *Wrapped = dyn_cast<nonloc::LocAsInteger>(&V))
1504f7df9beSJordan Rose       V = Wrapped->getLoc();
1514f7df9beSJordan Rose     else if (!isa<Loc>(V))
1524f7df9beSJordan Rose       continue;
1534f7df9beSJordan Rose 
1544f7df9beSJordan Rose     if (const MemRegion *R = V.getAsRegion()) {
1554f7df9beSJordan Rose       // Invalidate the value of the variable passed by reference.
1564f7df9beSJordan Rose 
1574f7df9beSJordan Rose       // Are we dealing with an ElementRegion?  If the element type is
1584f7df9beSJordan Rose       // a basic integer type (e.g., char, int) and the underlying region
1594f7df9beSJordan Rose       // is a variable region then strip off the ElementRegion.
1604f7df9beSJordan Rose       // FIXME: We really need to think about this for the general case
1614f7df9beSJordan Rose       //   as sometimes we are reasoning about arrays and other times
1624f7df9beSJordan Rose       //   about (char*), etc., is just a form of passing raw bytes.
1634f7df9beSJordan Rose       //   e.g., void *p = alloca(); foo((char*)p);
1644f7df9beSJordan Rose       if (const ElementRegion *ER = dyn_cast<ElementRegion>(R)) {
1654f7df9beSJordan Rose         // Checking for 'integral type' is probably too promiscuous, but
1664f7df9beSJordan Rose         // we'll leave it in for now until we have a systematic way of
1674f7df9beSJordan Rose         // handling all of these cases.  Eventually we need to come up
1684f7df9beSJordan Rose         // with an interface to StoreManager so that this logic can be
1694f7df9beSJordan Rose         // appropriately delegated to the respective StoreManagers while
1704f7df9beSJordan Rose         // still allowing us to do checker-specific logic (e.g.,
1714f7df9beSJordan Rose         // invalidating reference counts), probably via callbacks.
1724f7df9beSJordan Rose         if (ER->getElementType()->isIntegralOrEnumerationType()) {
1734f7df9beSJordan Rose           const MemRegion *superReg = ER->getSuperRegion();
1744f7df9beSJordan Rose           if (isa<VarRegion>(superReg) || isa<FieldRegion>(superReg) ||
1754f7df9beSJordan Rose               isa<ObjCIvarRegion>(superReg))
1764f7df9beSJordan Rose             R = cast<TypedRegion>(superReg);
1774f7df9beSJordan Rose         }
1784f7df9beSJordan Rose         // FIXME: What about layers of ElementRegions?
1794f7df9beSJordan Rose       }
1804f7df9beSJordan Rose 
1814f7df9beSJordan Rose       // Mark this region for invalidation.  We batch invalidate regions
1824f7df9beSJordan Rose       // below for efficiency.
1834f7df9beSJordan Rose       RegionsToInvalidate.push_back(R);
1844f7df9beSJordan Rose     }
1854f7df9beSJordan Rose   }
1864f7df9beSJordan Rose 
1874f7df9beSJordan Rose   // Invalidate designated regions using the batch invalidation API.
1884f7df9beSJordan Rose   // NOTE: Even if RegionsToInvalidate is empty, we may still invalidate
1894f7df9beSJordan Rose   //  global variables.
1904f7df9beSJordan Rose   return Result->invalidateRegions(RegionsToInvalidate, getOriginExpr(),
1914f7df9beSJordan Rose                                    BlockCount, getLocationContext(),
1924f7df9beSJordan Rose                                    /*Symbols=*/0, this);
1934f7df9beSJordan Rose }
1944f7df9beSJordan Rose 
1954f7df9beSJordan Rose ProgramPoint CallEvent::getProgramPoint(bool IsPreVisit,
1964f7df9beSJordan Rose                                         const ProgramPointTag *Tag) const {
1974f7df9beSJordan Rose   if (const Expr *E = getOriginExpr()) {
1984f7df9beSJordan Rose     if (IsPreVisit)
1994f7df9beSJordan Rose       return PreStmt(E, getLocationContext(), Tag);
2004f7df9beSJordan Rose     return PostStmt(E, getLocationContext(), Tag);
2014f7df9beSJordan Rose   }
2024f7df9beSJordan Rose 
2034f7df9beSJordan Rose   const Decl *D = getDecl();
2044f7df9beSJordan Rose   assert(D && "Cannot get a program point without a statement or decl");
2054f7df9beSJordan Rose 
2064f7df9beSJordan Rose   SourceLocation Loc = getSourceRange().getBegin();
2074f7df9beSJordan Rose   if (IsPreVisit)
2084f7df9beSJordan Rose     return PreImplicitCall(D, Loc, getLocationContext(), Tag);
2094f7df9beSJordan Rose   return PostImplicitCall(D, Loc, getLocationContext(), Tag);
2104f7df9beSJordan Rose }
2114f7df9beSJordan Rose 
21272ce8e2dSJordan Rose SVal CallEvent::getArgSVal(unsigned Index) const {
21372ce8e2dSJordan Rose   const Expr *ArgE = getArgExpr(Index);
21472ce8e2dSJordan Rose   if (!ArgE)
21572ce8e2dSJordan Rose     return UnknownVal();
21672ce8e2dSJordan Rose   return getSVal(ArgE);
21772ce8e2dSJordan Rose }
21872ce8e2dSJordan Rose 
21972ce8e2dSJordan Rose SourceRange CallEvent::getArgSourceRange(unsigned Index) const {
22072ce8e2dSJordan Rose   const Expr *ArgE = getArgExpr(Index);
22172ce8e2dSJordan Rose   if (!ArgE)
22272ce8e2dSJordan Rose     return SourceRange();
22372ce8e2dSJordan Rose   return ArgE->getSourceRange();
22472ce8e2dSJordan Rose }
22572ce8e2dSJordan Rose 
2269299d8c2SBenjamin Kramer void CallEvent::dump() const {
2279299d8c2SBenjamin Kramer   dump(llvm::errs());
2289299d8c2SBenjamin Kramer }
2299299d8c2SBenjamin Kramer 
23072ce8e2dSJordan Rose void CallEvent::dump(raw_ostream &Out) const {
23172ce8e2dSJordan Rose   ASTContext &Ctx = getState()->getStateManager().getContext();
23272ce8e2dSJordan Rose   if (const Expr *E = getOriginExpr()) {
233235341bcSRichard Smith     E->printPretty(Out, 0, Ctx.getPrintingPolicy());
23472ce8e2dSJordan Rose     Out << "\n";
23572ce8e2dSJordan Rose     return;
23672ce8e2dSJordan Rose   }
23772ce8e2dSJordan Rose 
23872ce8e2dSJordan Rose   if (const Decl *D = getDecl()) {
23972ce8e2dSJordan Rose     Out << "Call to ";
24072ce8e2dSJordan Rose     D->print(Out, Ctx.getPrintingPolicy());
24172ce8e2dSJordan Rose     return;
24272ce8e2dSJordan Rose   }
24372ce8e2dSJordan Rose 
24472ce8e2dSJordan Rose   // FIXME: a string representation of the kind would be nice.
24572ce8e2dSJordan Rose   Out << "Unknown call (type " << getKind() << ")";
24672ce8e2dSJordan Rose }
24772ce8e2dSJordan Rose 
2484f7df9beSJordan Rose 
249e537cc05SJordan Rose bool CallEvent::isCallStmt(const Stmt *S) {
25072ce8e2dSJordan Rose   return isa<CallExpr>(S) || isa<ObjCMessageExpr>(S)
251e537cc05SJordan Rose                           || isa<CXXConstructExpr>(S)
252e537cc05SJordan Rose                           || isa<CXXNewExpr>(S);
2534f7df9beSJordan Rose }
2544f7df9beSJordan Rose 
2555d2964e7SAnna Zaks /// \brief Returns the result type, adjusted for references.
2565d2964e7SAnna Zaks QualType CallEvent::getDeclaredResultType(const Decl *D) {
2575d2964e7SAnna Zaks   assert(D);
2585d2964e7SAnna Zaks   if (const FunctionDecl* FD = dyn_cast<FunctionDecl>(D))
2595d2964e7SAnna Zaks     return FD->getResultType();
2605d2964e7SAnna Zaks   else if (const ObjCMethodDecl* MD = dyn_cast<ObjCMethodDecl>(D))
2615d2964e7SAnna Zaks     return MD->getResultType();
2625d2964e7SAnna Zaks   return QualType();
2635d2964e7SAnna Zaks }
2645d2964e7SAnna Zaks 
26542e8d649SJordan Rose static void addParameterValuesToBindings(const StackFrameContext *CalleeCtx,
26642e8d649SJordan Rose                                          CallEvent::BindingsTy &Bindings,
26742e8d649SJordan Rose                                          SValBuilder &SVB,
26842e8d649SJordan Rose                                          const CallEvent &Call,
26942e8d649SJordan Rose                                          CallEvent::param_iterator I,
27042e8d649SJordan Rose                                          CallEvent::param_iterator E) {
27142e8d649SJordan Rose   MemRegionManager &MRMgr = SVB.getRegionManager();
2724f7df9beSJordan Rose 
27342e8d649SJordan Rose   unsigned Idx = 0;
27442e8d649SJordan Rose   for (; I != E; ++I, ++Idx) {
27542e8d649SJordan Rose     const ParmVarDecl *ParamDecl = *I;
27642e8d649SJordan Rose     assert(ParamDecl && "Formal parameter has no decl?");
2774f7df9beSJordan Rose 
27842e8d649SJordan Rose     SVal ArgVal = Call.getArgSVal(Idx);
27942e8d649SJordan Rose     if (!ArgVal.isUnknown()) {
28042e8d649SJordan Rose       Loc ParamLoc = SVB.makeLoc(MRMgr.getVarRegion(ParamDecl, CalleeCtx));
28142e8d649SJordan Rose       Bindings.push_back(std::make_pair(ParamLoc, ArgVal));
28242e8d649SJordan Rose     }
2834f7df9beSJordan Rose   }
2844f7df9beSJordan Rose 
28542e8d649SJordan Rose   // FIXME: Variadic arguments are not handled at all right now.
28642e8d649SJordan Rose }
28742e8d649SJordan Rose 
28842e8d649SJordan Rose 
28942e8d649SJordan Rose CallEvent::param_iterator AnyFunctionCall::param_begin() const {
29042e8d649SJordan Rose   const FunctionDecl *D = getDecl();
2914f7df9beSJordan Rose   if (!D)
2924f7df9beSJordan Rose     return 0;
2934f7df9beSJordan Rose 
29442e8d649SJordan Rose   return D->param_begin();
29542e8d649SJordan Rose }
29642e8d649SJordan Rose 
29742e8d649SJordan Rose CallEvent::param_iterator AnyFunctionCall::param_end() const {
29842e8d649SJordan Rose   const FunctionDecl *D = getDecl();
29942e8d649SJordan Rose   if (!D)
30042e8d649SJordan Rose     return 0;
30142e8d649SJordan Rose 
30242e8d649SJordan Rose   return D->param_end();
30342e8d649SJordan Rose }
30442e8d649SJordan Rose 
30542e8d649SJordan Rose void AnyFunctionCall::getInitialStackFrameContents(
30642e8d649SJordan Rose                                         const StackFrameContext *CalleeCtx,
30742e8d649SJordan Rose                                         BindingsTy &Bindings) const {
30842e8d649SJordan Rose   const FunctionDecl *D = cast<FunctionDecl>(CalleeCtx->getDecl());
30942e8d649SJordan Rose   SValBuilder &SVB = getState()->getStateManager().getSValBuilder();
31042e8d649SJordan Rose   addParameterValuesToBindings(CalleeCtx, Bindings, SVB, *this,
31142e8d649SJordan Rose                                D->param_begin(), D->param_end());
3124f7df9beSJordan Rose }
3134f7df9beSJordan Rose 
3144f7df9beSJordan Rose bool AnyFunctionCall::argumentsMayEscape() const {
3154f7df9beSJordan Rose   if (hasNonZeroCallbackArg())
3164f7df9beSJordan Rose     return true;
3174f7df9beSJordan Rose 
3184f7df9beSJordan Rose   const FunctionDecl *D = getDecl();
3194f7df9beSJordan Rose   if (!D)
3204f7df9beSJordan Rose     return true;
3214f7df9beSJordan Rose 
3224f7df9beSJordan Rose   const IdentifierInfo *II = D->getIdentifier();
3234f7df9beSJordan Rose   if (!II)
3244f7df9beSJordan Rose     return true;
3254f7df9beSJordan Rose 
3264f7df9beSJordan Rose   // This set of "escaping" APIs is
3274f7df9beSJordan Rose 
3284f7df9beSJordan Rose   // - 'int pthread_setspecific(ptheread_key k, const void *)' stores a
3294f7df9beSJordan Rose   //   value into thread local storage. The value can later be retrieved with
3304f7df9beSJordan Rose   //   'void *ptheread_getspecific(pthread_key)'. So even thought the
3314f7df9beSJordan Rose   //   parameter is 'const void *', the region escapes through the call.
3324f7df9beSJordan Rose   if (II->isStr("pthread_setspecific"))
3334f7df9beSJordan Rose     return true;
3344f7df9beSJordan Rose 
3354f7df9beSJordan Rose   // - xpc_connection_set_context stores a value which can be retrieved later
3364f7df9beSJordan Rose   //   with xpc_connection_get_context.
3374f7df9beSJordan Rose   if (II->isStr("xpc_connection_set_context"))
3384f7df9beSJordan Rose     return true;
3394f7df9beSJordan Rose 
3404f7df9beSJordan Rose   // - funopen - sets a buffer for future IO calls.
3414f7df9beSJordan Rose   if (II->isStr("funopen"))
3424f7df9beSJordan Rose     return true;
3434f7df9beSJordan Rose 
3444f7df9beSJordan Rose   StringRef FName = II->getName();
3454f7df9beSJordan Rose 
3464f7df9beSJordan Rose   // - CoreFoundation functions that end with "NoCopy" can free a passed-in
3474f7df9beSJordan Rose   //   buffer even if it is const.
3484f7df9beSJordan Rose   if (FName.endswith("NoCopy"))
3494f7df9beSJordan Rose     return true;
3504f7df9beSJordan Rose 
3514f7df9beSJordan Rose   // - NSXXInsertXX, for example NSMapInsertIfAbsent, since they can
3524f7df9beSJordan Rose   //   be deallocated by NSMapRemove.
3534f7df9beSJordan Rose   if (FName.startswith("NS") && (FName.find("Insert") != StringRef::npos))
3544f7df9beSJordan Rose     return true;
3554f7df9beSJordan Rose 
3564f7df9beSJordan Rose   // - Many CF containers allow objects to escape through custom
3574f7df9beSJordan Rose   //   allocators/deallocators upon container construction. (PR12101)
3584f7df9beSJordan Rose   if (FName.startswith("CF") || FName.startswith("CG")) {
3594f7df9beSJordan Rose     return StrInStrNoCase(FName, "InsertValue")  != StringRef::npos ||
3604f7df9beSJordan Rose            StrInStrNoCase(FName, "AddValue")     != StringRef::npos ||
3614f7df9beSJordan Rose            StrInStrNoCase(FName, "SetValue")     != StringRef::npos ||
3624f7df9beSJordan Rose            StrInStrNoCase(FName, "WithData")     != StringRef::npos ||
3634f7df9beSJordan Rose            StrInStrNoCase(FName, "AppendValue")  != StringRef::npos ||
3644f7df9beSJordan Rose            StrInStrNoCase(FName, "SetAttribute") != StringRef::npos;
3654f7df9beSJordan Rose   }
3664f7df9beSJordan Rose 
3674f7df9beSJordan Rose   return false;
3684f7df9beSJordan Rose }
3694f7df9beSJordan Rose 
3704f7df9beSJordan Rose 
3714f7df9beSJordan Rose const FunctionDecl *SimpleCall::getDecl() const {
3724f7df9beSJordan Rose   const FunctionDecl *D = getOriginExpr()->getDirectCallee();
3734f7df9beSJordan Rose   if (D)
3744f7df9beSJordan Rose     return D;
3754f7df9beSJordan Rose 
3764f7df9beSJordan Rose   return getSVal(getOriginExpr()->getCallee()).getAsFunctionDecl();
3774f7df9beSJordan Rose }
3784f7df9beSJordan Rose 
3794f7df9beSJordan Rose 
380ce6c99a5SJordan Rose const FunctionDecl *CXXInstanceCall::getDecl() const {
381ce6c99a5SJordan Rose   const CallExpr *CE = cast_or_null<CallExpr>(getOriginExpr());
382ce6c99a5SJordan Rose   if (!CE)
383ce6c99a5SJordan Rose     return AnyFunctionCall::getDecl();
384ce6c99a5SJordan Rose 
385ce6c99a5SJordan Rose   const FunctionDecl *D = CE->getDirectCallee();
386ce6c99a5SJordan Rose   if (D)
387ce6c99a5SJordan Rose     return D;
388ce6c99a5SJordan Rose 
389ce6c99a5SJordan Rose   return getSVal(CE->getCallee()).getAsFunctionDecl();
390ce6c99a5SJordan Rose }
391ce6c99a5SJordan Rose 
3924f7df9beSJordan Rose void CXXInstanceCall::getExtraInvalidatedRegions(RegionList &Regions) const {
3934f7df9beSJordan Rose   if (const MemRegion *R = getCXXThisVal().getAsRegion())
3944f7df9beSJordan Rose     Regions.push_back(R);
3954f7df9beSJordan Rose }
3964f7df9beSJordan Rose 
397fcdda361SJordan Rose SVal CXXInstanceCall::getCXXThisVal() const {
398fcdda361SJordan Rose   const Expr *Base = getCXXThisExpr();
399fcdda361SJordan Rose   // FIXME: This doesn't handle an overloaded ->* operator.
400fcdda361SJordan Rose   if (!Base)
401fcdda361SJordan Rose     return UnknownVal();
402fcdda361SJordan Rose 
403fcdda361SJordan Rose   SVal ThisVal = getSVal(Base);
404*29fc261cSJordan Rose   assert(ThisVal.isUnknownOrUndef() || isa<Loc>(ThisVal));
405fcdda361SJordan Rose   return ThisVal;
406fcdda361SJordan Rose }
407fcdda361SJordan Rose 
4084f7df9beSJordan Rose 
409089c5510STed Kremenek RuntimeDefinition CXXInstanceCall::getRuntimeDefinition() const {
4100f6d63beSJordan Rose   // Do we have a decl at all?
41151bcb226SJordan Rose   const Decl *D = getDecl();
4124f7df9beSJordan Rose   if (!D)
413123af098SAnna Zaks     return RuntimeDefinition();
4144f7df9beSJordan Rose 
4150f6d63beSJordan Rose   // If the method is non-virtual, we know we can inline it.
4164f7df9beSJordan Rose   const CXXMethodDecl *MD = cast<CXXMethodDecl>(D);
4174f7df9beSJordan Rose   if (!MD->isVirtual())
418089c5510STed Kremenek     return AnyFunctionCall::getRuntimeDefinition();
4194f7df9beSJordan Rose 
4200f6d63beSJordan Rose   // Do we know the implicit 'this' object being called?
4210f6d63beSJordan Rose   const MemRegion *R = getCXXThisVal().getAsRegion();
4220f6d63beSJordan Rose   if (!R)
423123af098SAnna Zaks     return RuntimeDefinition();
4240f6d63beSJordan Rose 
4250f6d63beSJordan Rose   // Do we know anything about the type of 'this'?
4260f6d63beSJordan Rose   DynamicTypeInfo DynType = getState()->getDynamicTypeInfo(R);
4270f6d63beSJordan Rose   if (!DynType.isValid())
4280f6d63beSJordan Rose     return RuntimeDefinition();
4290f6d63beSJordan Rose 
4300f6d63beSJordan Rose   // Is the type a C++ class? (This is mostly a defensive check.)
4310f6d63beSJordan Rose   QualType RegionType = DynType.getType()->getPointeeType();
432d1a08b6eSJordan Rose   assert(!RegionType.isNull() && "DynamicTypeInfo should always be a pointer.");
433d1a08b6eSJordan Rose 
4340f6d63beSJordan Rose   const CXXRecordDecl *RD = RegionType->getAsCXXRecordDecl();
43599107208SJordan Rose   if (!RD || !RD->hasDefinition())
4360f6d63beSJordan Rose     return RuntimeDefinition();
4370f6d63beSJordan Rose 
43899c6c2b4SJordan Rose   // Find the decl for this method in that class.
43999c6c2b4SJordan Rose   const CXXMethodDecl *Result = MD->getCorrespondingMethodInClass(RD, true);
44081456d9fSJordan Rose   if (!Result) {
44181456d9fSJordan Rose     // We might not even get the original statically-resolved method due to
44281456d9fSJordan Rose     // some particularly nasty casting (e.g. casts to sister classes).
44381456d9fSJordan Rose     // However, we should at least be able to search up and down our own class
44481456d9fSJordan Rose     // hierarchy, and some real bugs have been caught by checking this.
44581456d9fSJordan Rose     assert(!RD->isDerivedFrom(MD->getParent()) && "Couldn't find known method");
446d44977efSJordan Rose 
447d44977efSJordan Rose     // FIXME: This is checking that our DynamicTypeInfo is at least as good as
448d44977efSJordan Rose     // the static type. However, because we currently don't update
449d44977efSJordan Rose     // DynamicTypeInfo when an object is cast, we can't actually be sure the
450d44977efSJordan Rose     // DynamicTypeInfo is up to date. This assert should be re-enabled once
451d44977efSJordan Rose     // this is fixed. <rdar://problem/12287087>
452d44977efSJordan Rose     //assert(!MD->getParent()->isDerivedFrom(RD) && "Bad DynamicTypeInfo");
453d44977efSJordan Rose 
45481456d9fSJordan Rose     return RuntimeDefinition();
45581456d9fSJordan Rose   }
4560f6d63beSJordan Rose 
4570f6d63beSJordan Rose   // Does the decl that we found have an implementation?
4580f6d63beSJordan Rose   const FunctionDecl *Definition;
4590f6d63beSJordan Rose   if (!Result->hasBody(Definition))
4600f6d63beSJordan Rose     return RuntimeDefinition();
4610f6d63beSJordan Rose 
4620f6d63beSJordan Rose   // We found a definition. If we're not sure that this devirtualization is
4630f6d63beSJordan Rose   // actually what will happen at runtime, make sure to provide the region so
4640f6d63beSJordan Rose   // that ExprEngine can decide what to do with it.
4650f6d63beSJordan Rose   if (DynType.canBeASubClass())
4660f6d63beSJordan Rose     return RuntimeDefinition(Definition, R->StripCasts());
4670f6d63beSJordan Rose   return RuntimeDefinition(Definition, /*DispatchRegion=*/0);
4684f7df9beSJordan Rose }
4694f7df9beSJordan Rose 
47042e8d649SJordan Rose void CXXInstanceCall::getInitialStackFrameContents(
47142e8d649SJordan Rose                                             const StackFrameContext *CalleeCtx,
47242e8d649SJordan Rose                                             BindingsTy &Bindings) const {
47342e8d649SJordan Rose   AnyFunctionCall::getInitialStackFrameContents(CalleeCtx, Bindings);
47442e8d649SJordan Rose 
47502e5309bSJordan Rose   // Handle the binding of 'this' in the new stack frame.
47642e8d649SJordan Rose   SVal ThisVal = getCXXThisVal();
47742e8d649SJordan Rose   if (!ThisVal.isUnknown()) {
47802e5309bSJordan Rose     ProgramStateManager &StateMgr = getState()->getStateManager();
47902e5309bSJordan Rose     SValBuilder &SVB = StateMgr.getSValBuilder();
48002e5309bSJordan Rose 
48142e8d649SJordan Rose     const CXXMethodDecl *MD = cast<CXXMethodDecl>(CalleeCtx->getDecl());
48242e8d649SJordan Rose     Loc ThisLoc = SVB.getCXXThis(MD, CalleeCtx);
48302e5309bSJordan Rose 
48431c3fa9cSJordan Rose     // If we devirtualized to a different member function, we need to make sure
48531c3fa9cSJordan Rose     // we have the proper layering of CXXBaseObjectRegions.
48631c3fa9cSJordan Rose     if (MD->getCanonicalDecl() != getDecl()->getCanonicalDecl()) {
487710f6b12SJordan Rose       ASTContext &Ctx = SVB.getContext();
48802e5309bSJordan Rose       const CXXRecordDecl *Class = MD->getParent();
489710f6b12SJordan Rose       QualType Ty = Ctx.getPointerType(Ctx.getRecordType(Class));
49002e5309bSJordan Rose 
491710f6b12SJordan Rose       // FIXME: CallEvent maybe shouldn't be directly accessing StoreManager.
492710f6b12SJordan Rose       bool Failed;
493710f6b12SJordan Rose       ThisVal = StateMgr.getStoreManager().evalDynamicCast(ThisVal, Ty, Failed);
494710f6b12SJordan Rose       assert(!Failed && "Calling an incorrectly devirtualized method");
49502e5309bSJordan Rose     }
49602e5309bSJordan Rose 
49731c3fa9cSJordan Rose     if (!ThisVal.isUnknown())
49842e8d649SJordan Rose       Bindings.push_back(std::make_pair(ThisLoc, ThisVal));
49942e8d649SJordan Rose   }
50042e8d649SJordan Rose }
50142e8d649SJordan Rose 
50242e8d649SJordan Rose 
5034f7df9beSJordan Rose 
50492e1449bSJordan Rose const Expr *CXXMemberCall::getCXXThisExpr() const {
50592e1449bSJordan Rose   return getOriginExpr()->getImplicitObjectArgument();
5064f7df9beSJordan Rose }
5074f7df9beSJordan Rose 
508089c5510STed Kremenek RuntimeDefinition CXXMemberCall::getRuntimeDefinition() const {
50912f669e3SJordan Rose   // C++11 [expr.call]p1: ...If the selected function is non-virtual, or if the
51012f669e3SJordan Rose   // id-expression in the class member access expression is a qualified-id,
51112f669e3SJordan Rose   // that function is called. Otherwise, its final overrider in the dynamic type
51212f669e3SJordan Rose   // of the object expression is called.
51312f669e3SJordan Rose   if (const MemberExpr *ME = dyn_cast<MemberExpr>(getOriginExpr()->getCallee()))
51412f669e3SJordan Rose     if (ME->hasQualifier())
515089c5510STed Kremenek       return AnyFunctionCall::getRuntimeDefinition();
51612f669e3SJordan Rose 
517089c5510STed Kremenek   return CXXInstanceCall::getRuntimeDefinition();
51812f669e3SJordan Rose }
51912f669e3SJordan Rose 
5204f7df9beSJordan Rose 
52192e1449bSJordan Rose const Expr *CXXMemberOperatorCall::getCXXThisExpr() const {
52292e1449bSJordan Rose   return getOriginExpr()->getArg(0);
5234f7df9beSJordan Rose }
5244f7df9beSJordan Rose 
5254f7df9beSJordan Rose 
5264f7df9beSJordan Rose const BlockDataRegion *BlockCall::getBlockRegion() const {
5274f7df9beSJordan Rose   const Expr *Callee = getOriginExpr()->getCallee();
5284f7df9beSJordan Rose   const MemRegion *DataReg = getSVal(Callee).getAsRegion();
5294f7df9beSJordan Rose 
5304f7df9beSJordan Rose   return dyn_cast_or_null<BlockDataRegion>(DataReg);
5314f7df9beSJordan Rose }
5324f7df9beSJordan Rose 
53342e8d649SJordan Rose CallEvent::param_iterator BlockCall::param_begin() const {
5344f7df9beSJordan Rose   const BlockDecl *D = getBlockDecl();
5354f7df9beSJordan Rose   if (!D)
5364f7df9beSJordan Rose     return 0;
5374f7df9beSJordan Rose   return D->param_begin();
5384f7df9beSJordan Rose }
5394f7df9beSJordan Rose 
54042e8d649SJordan Rose CallEvent::param_iterator BlockCall::param_end() const {
5414f7df9beSJordan Rose   const BlockDecl *D = getBlockDecl();
5424f7df9beSJordan Rose   if (!D)
5434f7df9beSJordan Rose     return 0;
5444f7df9beSJordan Rose   return D->param_end();
5454f7df9beSJordan Rose }
5464f7df9beSJordan Rose 
5474f7df9beSJordan Rose void BlockCall::getExtraInvalidatedRegions(RegionList &Regions) const {
5484f7df9beSJordan Rose   // FIXME: This also needs to invalidate captured globals.
5494f7df9beSJordan Rose   if (const MemRegion *R = getBlockRegion())
5504f7df9beSJordan Rose     Regions.push_back(R);
5514f7df9beSJordan Rose }
5524f7df9beSJordan Rose 
55342e8d649SJordan Rose void BlockCall::getInitialStackFrameContents(const StackFrameContext *CalleeCtx,
55442e8d649SJordan Rose                                              BindingsTy &Bindings) const {
55542e8d649SJordan Rose   const BlockDecl *D = cast<BlockDecl>(CalleeCtx->getDecl());
55642e8d649SJordan Rose   SValBuilder &SVB = getState()->getStateManager().getSValBuilder();
55742e8d649SJordan Rose   addParameterValuesToBindings(CalleeCtx, Bindings, SVB, *this,
55842e8d649SJordan Rose                                D->param_begin(), D->param_end());
55942e8d649SJordan Rose }
56042e8d649SJordan Rose 
56142e8d649SJordan Rose 
5624f7df9beSJordan Rose SVal CXXConstructorCall::getCXXThisVal() const {
5634f7df9beSJordan Rose   if (Data)
5644f7df9beSJordan Rose     return loc::MemRegionVal(static_cast<const MemRegion *>(Data));
5654f7df9beSJordan Rose   return UnknownVal();
5664f7df9beSJordan Rose }
5674f7df9beSJordan Rose 
5684f7df9beSJordan Rose void CXXConstructorCall::getExtraInvalidatedRegions(RegionList &Regions) const {
5694f7df9beSJordan Rose   if (Data)
5704f7df9beSJordan Rose     Regions.push_back(static_cast<const MemRegion *>(Data));
5714f7df9beSJordan Rose }
5724f7df9beSJordan Rose 
57342e8d649SJordan Rose void CXXConstructorCall::getInitialStackFrameContents(
57442e8d649SJordan Rose                                              const StackFrameContext *CalleeCtx,
57542e8d649SJordan Rose                                              BindingsTy &Bindings) const {
57642e8d649SJordan Rose   AnyFunctionCall::getInitialStackFrameContents(CalleeCtx, Bindings);
57742e8d649SJordan Rose 
57842e8d649SJordan Rose   SVal ThisVal = getCXXThisVal();
57942e8d649SJordan Rose   if (!ThisVal.isUnknown()) {
58042e8d649SJordan Rose     SValBuilder &SVB = getState()->getStateManager().getSValBuilder();
58142e8d649SJordan Rose     const CXXMethodDecl *MD = cast<CXXMethodDecl>(CalleeCtx->getDecl());
58242e8d649SJordan Rose     Loc ThisLoc = SVB.getCXXThis(MD, CalleeCtx);
58342e8d649SJordan Rose     Bindings.push_back(std::make_pair(ThisLoc, ThisVal));
58442e8d649SJordan Rose   }
58542e8d649SJordan Rose }
58642e8d649SJordan Rose 
58742e8d649SJordan Rose 
5884f7df9beSJordan Rose 
5894f7df9beSJordan Rose SVal CXXDestructorCall::getCXXThisVal() const {
5904f7df9beSJordan Rose   if (Data)
5912bc9674bSJordan Rose     return loc::MemRegionVal(DtorDataTy::getFromOpaqueValue(Data).getPointer());
5924f7df9beSJordan Rose   return UnknownVal();
5934f7df9beSJordan Rose }
5944f7df9beSJordan Rose 
595089c5510STed Kremenek RuntimeDefinition CXXDestructorCall::getRuntimeDefinition() const {
5962bc9674bSJordan Rose   // Base destructors are always called non-virtually.
5972bc9674bSJordan Rose   // Skip CXXInstanceCall's devirtualization logic in this case.
5982bc9674bSJordan Rose   if (isBaseDestructor())
599089c5510STed Kremenek     return AnyFunctionCall::getRuntimeDefinition();
6002bc9674bSJordan Rose 
601089c5510STed Kremenek   return CXXInstanceCall::getRuntimeDefinition();
6022bc9674bSJordan Rose }
6032bc9674bSJordan Rose 
60442e8d649SJordan Rose 
60542e8d649SJordan Rose CallEvent::param_iterator ObjCMethodCall::param_begin() const {
60642e8d649SJordan Rose   const ObjCMethodDecl *D = getDecl();
6074f7df9beSJordan Rose   if (!D)
6084f7df9beSJordan Rose     return 0;
6094f7df9beSJordan Rose 
61042e8d649SJordan Rose   return D->param_begin();
61142e8d649SJordan Rose }
61242e8d649SJordan Rose 
61342e8d649SJordan Rose CallEvent::param_iterator ObjCMethodCall::param_end() const {
61442e8d649SJordan Rose   const ObjCMethodDecl *D = getDecl();
61542e8d649SJordan Rose   if (!D)
61642e8d649SJordan Rose     return 0;
61742e8d649SJordan Rose 
61842e8d649SJordan Rose   return D->param_end();
6194f7df9beSJordan Rose }
6204f7df9beSJordan Rose 
6214f7df9beSJordan Rose void
6224f7df9beSJordan Rose ObjCMethodCall::getExtraInvalidatedRegions(RegionList &Regions) const {
6234f7df9beSJordan Rose   if (const MemRegion *R = getReceiverSVal().getAsRegion())
6244f7df9beSJordan Rose     Regions.push_back(R);
6254f7df9beSJordan Rose }
6264f7df9beSJordan Rose 
6273d5d3d3eSAnna Zaks SVal ObjCMethodCall::getSelfSVal() const {
6283d5d3d3eSAnna Zaks   const LocationContext *LCtx = getLocationContext();
6293d5d3d3eSAnna Zaks   const ImplicitParamDecl *SelfDecl = LCtx->getSelfDecl();
6303d5d3d3eSAnna Zaks   if (!SelfDecl)
6313d5d3d3eSAnna Zaks     return SVal();
6323d5d3d3eSAnna Zaks   return getState()->getSVal(getState()->getRegion(SelfDecl, LCtx));
6333d5d3d3eSAnna Zaks }
6343d5d3d3eSAnna Zaks 
6354f7df9beSJordan Rose SVal ObjCMethodCall::getReceiverSVal() const {
6364f7df9beSJordan Rose   // FIXME: Is this the best way to handle class receivers?
6374f7df9beSJordan Rose   if (!isInstanceMessage())
6384f7df9beSJordan Rose     return UnknownVal();
6394f7df9beSJordan Rose 
640472dbcf1SAnna Zaks   if (const Expr *RecE = getOriginExpr()->getInstanceReceiver())
641472dbcf1SAnna Zaks     return getSVal(RecE);
6424f7df9beSJordan Rose 
6434f7df9beSJordan Rose   // An instance message with no expression means we are sending to super.
6444f7df9beSJordan Rose   // In this case the object reference is the same as 'self'.
6453d5d3d3eSAnna Zaks   assert(getOriginExpr()->getReceiverKind() == ObjCMessageExpr::SuperInstance);
6463d5d3d3eSAnna Zaks   SVal SelfVal = getSelfSVal();
6473d5d3d3eSAnna Zaks   assert(SelfVal.isValid() && "Calling super but not in ObjC method");
6483d5d3d3eSAnna Zaks   return SelfVal;
6493d5d3d3eSAnna Zaks }
6503d5d3d3eSAnna Zaks 
6513d5d3d3eSAnna Zaks bool ObjCMethodCall::isReceiverSelfOrSuper() const {
6523d5d3d3eSAnna Zaks   if (getOriginExpr()->getReceiverKind() == ObjCMessageExpr::SuperInstance ||
6533d5d3d3eSAnna Zaks       getOriginExpr()->getReceiverKind() == ObjCMessageExpr::SuperClass)
6543d5d3d3eSAnna Zaks       return true;
6553d5d3d3eSAnna Zaks 
6563d5d3d3eSAnna Zaks   if (!isInstanceMessage())
6573d5d3d3eSAnna Zaks     return false;
6583d5d3d3eSAnna Zaks 
6593d5d3d3eSAnna Zaks   SVal RecVal = getSVal(getOriginExpr()->getInstanceReceiver());
6603d5d3d3eSAnna Zaks 
6613d5d3d3eSAnna Zaks   return (RecVal == getSelfSVal());
6624f7df9beSJordan Rose }
6634f7df9beSJordan Rose 
6644f7df9beSJordan Rose SourceRange ObjCMethodCall::getSourceRange() const {
6654f7df9beSJordan Rose   switch (getMessageKind()) {
6664f7df9beSJordan Rose   case OCM_Message:
6674f7df9beSJordan Rose     return getOriginExpr()->getSourceRange();
6684f7df9beSJordan Rose   case OCM_PropertyAccess:
6694f7df9beSJordan Rose   case OCM_Subscript:
6704f7df9beSJordan Rose     return getContainingPseudoObjectExpr()->getSourceRange();
6714f7df9beSJordan Rose   }
6724f7df9beSJordan Rose   llvm_unreachable("unknown message kind");
6734f7df9beSJordan Rose }
6744f7df9beSJordan Rose 
6754f7df9beSJordan Rose typedef llvm::PointerIntPair<const PseudoObjectExpr *, 2> ObjCMessageDataTy;
6764f7df9beSJordan Rose 
6774f7df9beSJordan Rose const PseudoObjectExpr *ObjCMethodCall::getContainingPseudoObjectExpr() const {
6784f7df9beSJordan Rose   assert(Data != 0 && "Lazy lookup not yet performed.");
6794f7df9beSJordan Rose   assert(getMessageKind() != OCM_Message && "Explicit message send.");
6804f7df9beSJordan Rose   return ObjCMessageDataTy::getFromOpaqueValue(Data).getPointer();
6814f7df9beSJordan Rose }
6824f7df9beSJordan Rose 
6834f7df9beSJordan Rose ObjCMessageKind ObjCMethodCall::getMessageKind() const {
6844f7df9beSJordan Rose   if (Data == 0) {
6854f7df9beSJordan Rose     ParentMap &PM = getLocationContext()->getParentMap();
6864f7df9beSJordan Rose     const Stmt *S = PM.getParent(getOriginExpr());
6874f7df9beSJordan Rose     if (const PseudoObjectExpr *POE = dyn_cast_or_null<PseudoObjectExpr>(S)) {
6884f7df9beSJordan Rose       const Expr *Syntactic = POE->getSyntacticForm();
6894f7df9beSJordan Rose 
6904f7df9beSJordan Rose       // This handles the funny case of assigning to the result of a getter.
6914f7df9beSJordan Rose       // This can happen if the getter returns a non-const reference.
6924f7df9beSJordan Rose       if (const BinaryOperator *BO = dyn_cast<BinaryOperator>(Syntactic))
6934f7df9beSJordan Rose         Syntactic = BO->getLHS();
6944f7df9beSJordan Rose 
6954f7df9beSJordan Rose       ObjCMessageKind K;
6964f7df9beSJordan Rose       switch (Syntactic->getStmtClass()) {
6974f7df9beSJordan Rose       case Stmt::ObjCPropertyRefExprClass:
6984f7df9beSJordan Rose         K = OCM_PropertyAccess;
6994f7df9beSJordan Rose         break;
7004f7df9beSJordan Rose       case Stmt::ObjCSubscriptRefExprClass:
7014f7df9beSJordan Rose         K = OCM_Subscript;
7024f7df9beSJordan Rose         break;
7034f7df9beSJordan Rose       default:
7044f7df9beSJordan Rose         // FIXME: Can this ever happen?
7054f7df9beSJordan Rose         K = OCM_Message;
7064f7df9beSJordan Rose         break;
7074f7df9beSJordan Rose       }
7084f7df9beSJordan Rose 
7094f7df9beSJordan Rose       if (K != OCM_Message) {
7104f7df9beSJordan Rose         const_cast<ObjCMethodCall *>(this)->Data
7114f7df9beSJordan Rose           = ObjCMessageDataTy(POE, K).getOpaqueValue();
7124f7df9beSJordan Rose         assert(getMessageKind() == K);
7134f7df9beSJordan Rose         return K;
7144f7df9beSJordan Rose       }
7154f7df9beSJordan Rose     }
7164f7df9beSJordan Rose 
7174f7df9beSJordan Rose     const_cast<ObjCMethodCall *>(this)->Data
7184f7df9beSJordan Rose       = ObjCMessageDataTy(0, 1).getOpaqueValue();
7194f7df9beSJordan Rose     assert(getMessageKind() == OCM_Message);
7204f7df9beSJordan Rose     return OCM_Message;
7214f7df9beSJordan Rose   }
7224f7df9beSJordan Rose 
7234f7df9beSJordan Rose   ObjCMessageDataTy Info = ObjCMessageDataTy::getFromOpaqueValue(Data);
7244f7df9beSJordan Rose   if (!Info.getPointer())
7254f7df9beSJordan Rose     return OCM_Message;
7264f7df9beSJordan Rose   return static_cast<ObjCMessageKind>(Info.getInt());
7274f7df9beSJordan Rose }
7284f7df9beSJordan Rose 
729920af014SAnna Zaks 
730920af014SAnna Zaks bool ObjCMethodCall::canBeOverridenInSubclass(ObjCInterfaceDecl *IDecl,
731920af014SAnna Zaks                                              Selector Sel) const {
732920af014SAnna Zaks   assert(IDecl);
733920af014SAnna Zaks   const SourceManager &SM =
734920af014SAnna Zaks     getState()->getStateManager().getContext().getSourceManager();
735920af014SAnna Zaks 
736920af014SAnna Zaks   // If the class interface is declared inside the main file, assume it is not
737920af014SAnna Zaks   // subcassed.
738920af014SAnna Zaks   // TODO: It could actually be subclassed if the subclass is private as well.
739920af014SAnna Zaks   // This is probably very rare.
740920af014SAnna Zaks   SourceLocation InterfLoc = IDecl->getEndOfDefinitionLoc();
741920af014SAnna Zaks   if (InterfLoc.isValid() && SM.isFromMainFile(InterfLoc))
742920af014SAnna Zaks     return false;
743920af014SAnna Zaks 
7446ddb6b1aSAnna Zaks   // Assume that property accessors are not overridden.
7456ddb6b1aSAnna Zaks   if (getMessageKind() == OCM_PropertyAccess)
7466ddb6b1aSAnna Zaks     return false;
747920af014SAnna Zaks 
748920af014SAnna Zaks   // We assume that if the method is public (declared outside of main file) or
749920af014SAnna Zaks   // has a parent which publicly declares the method, the method could be
750920af014SAnna Zaks   // overridden in a subclass.
751920af014SAnna Zaks 
752920af014SAnna Zaks   // Find the first declaration in the class hierarchy that declares
753920af014SAnna Zaks   // the selector.
754920af014SAnna Zaks   ObjCMethodDecl *D = 0;
755920af014SAnna Zaks   while (true) {
756920af014SAnna Zaks     D = IDecl->lookupMethod(Sel, true);
757920af014SAnna Zaks 
758920af014SAnna Zaks     // Cannot find a public definition.
759920af014SAnna Zaks     if (!D)
760920af014SAnna Zaks       return false;
761920af014SAnna Zaks 
762920af014SAnna Zaks     // If outside the main file,
763920af014SAnna Zaks     if (D->getLocation().isValid() && !SM.isFromMainFile(D->getLocation()))
764920af014SAnna Zaks       return true;
765920af014SAnna Zaks 
766920af014SAnna Zaks     if (D->isOverriding()) {
767920af014SAnna Zaks       // Search in the superclass on the next iteration.
768920af014SAnna Zaks       IDecl = D->getClassInterface();
769920af014SAnna Zaks       if (!IDecl)
770920af014SAnna Zaks         return false;
771920af014SAnna Zaks 
772920af014SAnna Zaks       IDecl = IDecl->getSuperClass();
773920af014SAnna Zaks       if (!IDecl)
774920af014SAnna Zaks         return false;
775920af014SAnna Zaks 
776920af014SAnna Zaks       continue;
777920af014SAnna Zaks     }
778920af014SAnna Zaks 
779920af014SAnna Zaks     return false;
780920af014SAnna Zaks   };
781920af014SAnna Zaks 
782920af014SAnna Zaks   llvm_unreachable("The while loop should always terminate.");
783920af014SAnna Zaks }
784920af014SAnna Zaks 
785089c5510STed Kremenek RuntimeDefinition ObjCMethodCall::getRuntimeDefinition() const {
78663282aefSAnna Zaks   const ObjCMessageExpr *E = getOriginExpr();
78763282aefSAnna Zaks   assert(E);
7885808eb80SAnna Zaks   Selector Sel = E->getSelector();
7894f7df9beSJordan Rose 
79063282aefSAnna Zaks   if (E->isInstanceMessage()) {
7915808eb80SAnna Zaks 
7925808eb80SAnna Zaks     // Find the the receiver type.
7935808eb80SAnna Zaks     const ObjCObjectPointerType *ReceiverT = 0;
79475f49a9cSAnna Zaks     bool CanBeSubClassed = false;
7955808eb80SAnna Zaks     QualType SupersType = E->getSuperType();
796123af098SAnna Zaks     const MemRegion *Receiver = 0;
797123af098SAnna Zaks 
7985808eb80SAnna Zaks     if (!SupersType.isNull()) {
799123af098SAnna Zaks       // Super always means the type of immediate predecessor to the method
800123af098SAnna Zaks       // where the call occurs.
80175930b65SAnna Zaks       ReceiverT = cast<ObjCObjectPointerType>(SupersType);
8025808eb80SAnna Zaks     } else {
803123af098SAnna Zaks       Receiver = getReceiverSVal().getAsRegion();
8046a97d92eSJordan Rose       if (!Receiver)
805123af098SAnna Zaks         return RuntimeDefinition();
8066a97d92eSJordan Rose 
80775f49a9cSAnna Zaks       DynamicTypeInfo DTI = getState()->getDynamicTypeInfo(Receiver);
80875f49a9cSAnna Zaks       QualType DynType = DTI.getType();
80975f49a9cSAnna Zaks       CanBeSubClassed = DTI.canBeASubClass();
81075930b65SAnna Zaks       ReceiverT = dyn_cast<ObjCObjectPointerType>(DynType);
81175f49a9cSAnna Zaks 
81275f49a9cSAnna Zaks       if (ReceiverT && CanBeSubClassed)
81375f49a9cSAnna Zaks         if (ObjCInterfaceDecl *IDecl = ReceiverT->getInterfaceDecl())
81475f49a9cSAnna Zaks           if (!canBeOverridenInSubclass(IDecl, Sel))
81575f49a9cSAnna Zaks             CanBeSubClassed = false;
8164f7df9beSJordan Rose     }
8174f7df9beSJordan Rose 
8185808eb80SAnna Zaks     // Lookup the method implementation.
8195808eb80SAnna Zaks     if (ReceiverT)
820920af014SAnna Zaks       if (ObjCInterfaceDecl *IDecl = ReceiverT->getInterfaceDecl()) {
82175f49a9cSAnna Zaks         const ObjCMethodDecl *MD = IDecl->lookupPrivateMethod(Sel);
82275f49a9cSAnna Zaks         if (CanBeSubClassed)
82375f49a9cSAnna Zaks           return RuntimeDefinition(MD, Receiver);
824920af014SAnna Zaks         else
82575f49a9cSAnna Zaks           return RuntimeDefinition(MD, 0);
826920af014SAnna Zaks       }
8275808eb80SAnna Zaks 
82863282aefSAnna Zaks   } else {
82963282aefSAnna Zaks     // This is a class method.
83063282aefSAnna Zaks     // If we have type info for the receiver class, we are calling via
83163282aefSAnna Zaks     // class name.
83263282aefSAnna Zaks     if (ObjCInterfaceDecl *IDecl = E->getReceiverInterface()) {
83363282aefSAnna Zaks       // Find/Return the method implementation.
83485383182SAnna Zaks       return RuntimeDefinition(IDecl->lookupPrivateClassMethod(Sel));
8354f7df9beSJordan Rose     }
83663282aefSAnna Zaks   }
83763282aefSAnna Zaks 
838123af098SAnna Zaks   return RuntimeDefinition();
8394f7df9beSJordan Rose }
8404f7df9beSJordan Rose 
84142e8d649SJordan Rose void ObjCMethodCall::getInitialStackFrameContents(
84242e8d649SJordan Rose                                              const StackFrameContext *CalleeCtx,
84342e8d649SJordan Rose                                              BindingsTy &Bindings) const {
84442e8d649SJordan Rose   const ObjCMethodDecl *D = cast<ObjCMethodDecl>(CalleeCtx->getDecl());
84542e8d649SJordan Rose   SValBuilder &SVB = getState()->getStateManager().getSValBuilder();
84642e8d649SJordan Rose   addParameterValuesToBindings(CalleeCtx, Bindings, SVB, *this,
84742e8d649SJordan Rose                                D->param_begin(), D->param_end());
84842e8d649SJordan Rose 
84942e8d649SJordan Rose   SVal SelfVal = getReceiverSVal();
85042e8d649SJordan Rose   if (!SelfVal.isUnknown()) {
85142e8d649SJordan Rose     const VarDecl *SelfD = CalleeCtx->getAnalysisDeclContext()->getSelfDecl();
85242e8d649SJordan Rose     MemRegionManager &MRMgr = SVB.getRegionManager();
85342e8d649SJordan Rose     Loc SelfLoc = SVB.makeLoc(MRMgr.getVarRegion(SelfD, CalleeCtx));
85442e8d649SJordan Rose     Bindings.push_back(std::make_pair(SelfLoc, SelfVal));
85542e8d649SJordan Rose   }
85642e8d649SJordan Rose }
85742e8d649SJordan Rose 
858ce6c99a5SJordan Rose CallEventRef<>
859fcd016e5SJordan Rose CallEventManager::getSimpleCall(const CallExpr *CE, ProgramStateRef State,
860fcd016e5SJordan Rose                                 const LocationContext *LCtx) {
861fcd016e5SJordan Rose   if (const CXXMemberCallExpr *MCE = dyn_cast<CXXMemberCallExpr>(CE))
862fcd016e5SJordan Rose     return create<CXXMemberCall>(MCE, State, LCtx);
863fcd016e5SJordan Rose 
864fcd016e5SJordan Rose   if (const CXXOperatorCallExpr *OpCE = dyn_cast<CXXOperatorCallExpr>(CE)) {
865fcd016e5SJordan Rose     const FunctionDecl *DirectCallee = OpCE->getDirectCallee();
866fcd016e5SJordan Rose     if (const CXXMethodDecl *MD = dyn_cast<CXXMethodDecl>(DirectCallee))
867fcd016e5SJordan Rose       if (MD->isInstance())
868fcd016e5SJordan Rose         return create<CXXMemberOperatorCall>(OpCE, State, LCtx);
869fcd016e5SJordan Rose 
870fcd016e5SJordan Rose   } else if (CE->getCallee()->getType()->isBlockPointerType()) {
871fcd016e5SJordan Rose     return create<BlockCall>(CE, State, LCtx);
872fcd016e5SJordan Rose   }
873fcd016e5SJordan Rose 
874fcd016e5SJordan Rose   // Otherwise, it's a normal function call, static member function call, or
875fcd016e5SJordan Rose   // something we can't reason about.
876fcd016e5SJordan Rose   return create<FunctionCall>(CE, State, LCtx);
877fcd016e5SJordan Rose }
878c2d249ceSJordan Rose 
879c2d249ceSJordan Rose 
880c2d249ceSJordan Rose CallEventRef<>
881c2d249ceSJordan Rose CallEventManager::getCaller(const StackFrameContext *CalleeCtx,
882c2d249ceSJordan Rose                             ProgramStateRef State) {
883c2d249ceSJordan Rose   const LocationContext *ParentCtx = CalleeCtx->getParent();
884c2d249ceSJordan Rose   const LocationContext *CallerCtx = ParentCtx->getCurrentStackFrame();
885c2d249ceSJordan Rose   assert(CallerCtx && "This should not be used for top-level stack frames");
886c2d249ceSJordan Rose 
887c2d249ceSJordan Rose   const Stmt *CallSite = CalleeCtx->getCallSite();
888c2d249ceSJordan Rose 
889c2d249ceSJordan Rose   if (CallSite) {
890c2d249ceSJordan Rose     if (const CallExpr *CE = dyn_cast<CallExpr>(CallSite))
891c2d249ceSJordan Rose       return getSimpleCall(CE, State, CallerCtx);
892c2d249ceSJordan Rose 
893c2d249ceSJordan Rose     switch (CallSite->getStmtClass()) {
8948d48938bSJordan Rose     case Stmt::CXXConstructExprClass:
8958d48938bSJordan Rose     case Stmt::CXXTemporaryObjectExprClass: {
896c2d249ceSJordan Rose       SValBuilder &SVB = State->getStateManager().getSValBuilder();
897c2d249ceSJordan Rose       const CXXMethodDecl *Ctor = cast<CXXMethodDecl>(CalleeCtx->getDecl());
898c2d249ceSJordan Rose       Loc ThisPtr = SVB.getCXXThis(Ctor, CalleeCtx);
899c2d249ceSJordan Rose       SVal ThisVal = State->getSVal(ThisPtr);
900c2d249ceSJordan Rose 
901c2d249ceSJordan Rose       return getCXXConstructorCall(cast<CXXConstructExpr>(CallSite),
902c2d249ceSJordan Rose                                    ThisVal.getAsRegion(), State, CallerCtx);
903c2d249ceSJordan Rose     }
904c2d249ceSJordan Rose     case Stmt::CXXNewExprClass:
905c2d249ceSJordan Rose       return getCXXAllocatorCall(cast<CXXNewExpr>(CallSite), State, CallerCtx);
906c2d249ceSJordan Rose     case Stmt::ObjCMessageExprClass:
907c2d249ceSJordan Rose       return getObjCMethodCall(cast<ObjCMessageExpr>(CallSite),
908c2d249ceSJordan Rose                                State, CallerCtx);
909c2d249ceSJordan Rose     default:
910c2d249ceSJordan Rose       llvm_unreachable("This is not an inlineable statement.");
911c2d249ceSJordan Rose     }
912c2d249ceSJordan Rose   }
913c2d249ceSJordan Rose 
914c2d249ceSJordan Rose   // Fall back to the CFG. The only thing we haven't handled yet is
915c2d249ceSJordan Rose   // destructors, though this could change in the future.
916c2d249ceSJordan Rose   const CFGBlock *B = CalleeCtx->getCallSiteBlock();
917c2d249ceSJordan Rose   CFGElement E = (*B)[CalleeCtx->getIndex()];
918c2d249ceSJordan Rose   assert(isa<CFGImplicitDtor>(E) && "All other CFG elements should have exprs");
919c2d249ceSJordan Rose   assert(!isa<CFGTemporaryDtor>(E) && "We don't handle temporaries yet");
920c2d249ceSJordan Rose 
921c2d249ceSJordan Rose   SValBuilder &SVB = State->getStateManager().getSValBuilder();
922c2d249ceSJordan Rose   const CXXDestructorDecl *Dtor = cast<CXXDestructorDecl>(CalleeCtx->getDecl());
923c2d249ceSJordan Rose   Loc ThisPtr = SVB.getCXXThis(Dtor, CalleeCtx);
924c2d249ceSJordan Rose   SVal ThisVal = State->getSVal(ThisPtr);
925c2d249ceSJordan Rose 
926c2d249ceSJordan Rose   const Stmt *Trigger;
927c2d249ceSJordan Rose   if (const CFGAutomaticObjDtor *AutoDtor = dyn_cast<CFGAutomaticObjDtor>(&E))
928c2d249ceSJordan Rose     Trigger = AutoDtor->getTriggerStmt();
929c2d249ceSJordan Rose   else
930c2d249ceSJordan Rose     Trigger = Dtor->getBody();
931c2d249ceSJordan Rose 
932c2d249ceSJordan Rose   return getCXXDestructorCall(Dtor, Trigger, ThisVal.getAsRegion(),
9332bc9674bSJordan Rose                               isa<CFGBaseDtor>(E), State, CallerCtx);
934c2d249ceSJordan Rose }
935