[cfe-commits] r50700 - /cfe/trunk/lib/Analysis/CFRefCount.cpp

[Ted Kremenek]

Author: kremenek
Date: Mon May  5 21:41:27 2008
New Revision: 50700

URL: http://llvm.org/viewvc/llvm-project?rev=50700&view=rev
Log:
Added receiver effects to EvalSummary.

Modified:
    cfe/trunk/lib/Analysis/CFRefCount.cpp

Modified: cfe/trunk/lib/Analysis/CFRefCount.cpp
URL: http://llvm.org/viewvc/llvm-project/cfe/trunk/lib/Analysis/CFRefCount.cpp?rev=50700&r1=50699&r2=50700&view=diff

==============================================================================
--- cfe/trunk/lib/Analysis/CFRefCount.cpp (original)
+++ cfe/trunk/lib/Analysis/CFRefCount.cpp Mon May  5 21:41:27 2008
@@ -48,7 +48,7 @@
 //===----------------------------------------------------------------------===//
 
 namespace {  
-  enum ArgEffect { IncRef, DecRef, DoNothing, AutoRef };
+  enum ArgEffect { IncRef, DecRef, DoNothing, StopTracking };
   typedef std::vector<std::pair<unsigned,ArgEffect> > ArgEffects;
 }
 
@@ -585,7 +585,7 @@
   ObjCMethSummaries[ GetNullarySelector("release", Ctx) ] = Summ;
 
   // Create the "autorelease" selector.
-  Summ = getPersistentSummary(E, isGCEnabled() ? DoNothing : AutoRef);
+  Summ = getPersistentSummary(E, isGCEnabled() ? DoNothing : StopTracking);
   ObjCMethSummaries[ GetNullarySelector("autorelease", Ctx) ] = Summ;
 }
 
@@ -967,6 +967,10 @@
   return Summ ? Summ->getRetEffect() : RetEffect::MakeNoRet();
 }
 
+static inline ArgEffect GetReceiverE(RetainSummary* Summ) {
+  return Summ ? Summ->getReceiverEffect() : DoNothing;
+}
+
 void CFRefCount::ProcessNonLeakError(ExplodedNodeSet<ValueState>& Dst,
                                      GRStmtNodeBuilder<ValueState>& Builder,
                                      Expr* NodeExpr, Expr* ErrorExpr,                        
@@ -1004,15 +1008,12 @@
   ValueStateManager& StateMgr = Eng.getStateManager();  
   ValueState* St = Builder.GetState(Pred);
   
-  // Evaluate the effects of the call.
+
+  // Evaluate the effect of the arguments.
   
   ValueState StVals = *St;
   RefVal::Kind hasErr = (RefVal::Kind) 0;
-  
-  // This function has a summary.  Evaluate the effect of the arguments.
-  
   unsigned idx = 0;
-  
   Expr* ErrorExpr = NULL;
   SymbolID ErrorSym = 0;                                        
   
@@ -1043,8 +1044,33 @@
       StateMgr.Unbind(StVals, cast<nonlval::LValAsInteger>(V).getLVal());
   } 
   
+  // Evaluate the effect on the message receiver.
+  
+  if (!ErrorExpr && Receiver) {
+    RVal V = StateMgr.GetRVal(St, Receiver);
+
+    if (isa<lval::SymbolVal>(V)) {
+      SymbolID Sym = cast<lval::SymbolVal>(V).getSymbol();
+      RefBindings B = GetRefBindings(StVals);      
+      
+      if (RefBindings::TreeTy* T = B.SlimFind(Sym)) {
+        B = Update(B, Sym, T->getValue().second, GetReceiverE(Summ), hasErr);
+        SetRefBindings(StVals, B);
+        
+        if (hasErr) {
+          ErrorExpr = Receiver;
+          ErrorSym = T->getValue().first;
+        }
+      }
+    }
+  }
+
+  // Get the persistent state.
+  
   St = StateMgr.getPersistentState(StVals);
   
+  // Process any errors.
+  
   if (hasErr) {
     ProcessNonLeakError(Dst, Builder, Ex, ErrorExpr, Pred, St,
                         hasErr, ErrorSym);
@@ -1087,6 +1113,13 @@
       break;
     }
       
+    case RetEffect::ReceiverAlias: {
+      assert (Receiver);
+      RVal V = StateMgr.GetRVal(St, Receiver);
+      St = StateMgr.SetRVal(St, Ex, V, Eng.getCFG().isBlkExpr(Ex), false);
+      break;
+    }
+      
     case RetEffect::OwnedSymbol: {
       unsigned Count = Builder.getCurrentBlockCount();
       SymbolID Sym = Eng.getSymbolManager().getConjuredSymbol(Ex, Count);
@@ -1551,6 +1584,9 @@
       
       return B;
       
+    case StopTracking:
+      return RefBFactory.Remove(B, sym);
+      
     case IncRef:      
       switch (V.getKind()) {
         default: