Home | History | Annotate | Download | only in PathSensitive
      1 //== ProgramState.h - Path-sensitive "State" for tracking values -*- C++ -*--=//
      2 //
      3 //                     The LLVM Compiler Infrastructure
      4 //
      5 // This file is distributed under the University of Illinois Open Source
      6 // License. See LICENSE.TXT for details.
      7 //
      8 //===----------------------------------------------------------------------===//
      9 //
     10 // This file defines the state of the program along the analysisa path.
     11 //
     12 //===----------------------------------------------------------------------===//
     13 
     14 #ifndef LLVM_CLANG_STATICANALYZER_CORE_PATHSENSITIVE_PROGRAMSTATE_H
     15 #define LLVM_CLANG_STATICANALYZER_CORE_PATHSENSITIVE_PROGRAMSTATE_H
     16 
     17 #include "clang/Basic/LLVM.h"
     18 #include "clang/StaticAnalyzer/Core/PathSensitive/ConstraintManager.h"
     19 #include "clang/StaticAnalyzer/Core/PathSensitive/DynamicTypeInfo.h"
     20 #include "clang/StaticAnalyzer/Core/PathSensitive/Environment.h"
     21 #include "clang/StaticAnalyzer/Core/PathSensitive/ProgramState_Fwd.h"
     22 #include "clang/StaticAnalyzer/Core/PathSensitive/SValBuilder.h"
     23 #include "clang/StaticAnalyzer/Core/PathSensitive/Store.h"
     24 #include "clang/StaticAnalyzer/Core/PathSensitive/TaintTag.h"
     25 #include "llvm/ADT/FoldingSet.h"
     26 #include "llvm/ADT/ImmutableMap.h"
     27 #include "llvm/Support/Allocator.h"
     28 #include <utility>
     29 
     30 namespace llvm {
     31 class APSInt;
     32 }
     33 
     34 namespace clang {
     35 class ASTContext;
     36 
     37 namespace ento {
     38 
     39 class CallEvent;
     40 class CallEventManager;
     41 
     42 typedef std::unique_ptr<ConstraintManager>(*ConstraintManagerCreator)(
     43     ProgramStateManager &, SubEngine *);
     44 typedef std::unique_ptr<StoreManager>(*StoreManagerCreator)(
     45     ProgramStateManager &);
     46 typedef llvm::ImmutableMap<const SubRegion*, TaintTagType> TaintedSubRegions;
     47 
     48 //===----------------------------------------------------------------------===//
     49 // ProgramStateTrait - Traits used by the Generic Data Map of a ProgramState.
     50 //===----------------------------------------------------------------------===//
     51 
     52 template <typename T> struct ProgramStatePartialTrait;
     53 
     54 template <typename T> struct ProgramStateTrait {
     55   typedef typename T::data_type data_type;
     56   static inline void *MakeVoidPtr(data_type D) { return (void*) D; }
     57   static inline data_type MakeData(void *const* P) {
     58     return P ? (data_type) *P : (data_type) 0;
     59   }
     60 };
     61 
     62 /// \class ProgramState
     63 /// ProgramState - This class encapsulates:
     64 ///
     65 ///    1. A mapping from expressions to values (Environment)
     66 ///    2. A mapping from locations to values (Store)
     67 ///    3. Constraints on symbolic values (GenericDataMap)
     68 ///
     69 ///  Together these represent the "abstract state" of a program.
     70 ///
     71 ///  ProgramState is intended to be used as a functional object; that is,
     72 ///  once it is created and made "persistent" in a FoldingSet, its
     73 ///  values will never change.
     74 class ProgramState : public llvm::FoldingSetNode {
     75 public:
     76   typedef llvm::ImmutableSet<llvm::APSInt*>                IntSetTy;
     77   typedef llvm::ImmutableMap<void*, void*>                 GenericDataMap;
     78 
     79 private:
     80   void operator=(const ProgramState& R) = delete;
     81 
     82   friend class ProgramStateManager;
     83   friend class ExplodedGraph;
     84   friend class ExplodedNode;
     85 
     86   ProgramStateManager *stateMgr;
     87   Environment Env;           // Maps a Stmt to its current SVal.
     88   Store store;               // Maps a location to its current value.
     89   GenericDataMap   GDM;      // Custom data stored by a client of this class.
     90   unsigned refCount;
     91 
     92   /// makeWithStore - Return a ProgramState with the same values as the current
     93   ///  state with the exception of using the specified Store.
     94   ProgramStateRef makeWithStore(const StoreRef &store) const;
     95 
     96   void setStore(const StoreRef &storeRef);
     97 
     98 public:
     99   /// This ctor is used when creating the first ProgramState object.
    100   ProgramState(ProgramStateManager *mgr, const Environment& env,
    101           StoreRef st, GenericDataMap gdm);
    102 
    103   /// Copy ctor - We must explicitly define this or else the "Next" ptr
    104   ///  in FoldingSetNode will also get copied.
    105   ProgramState(const ProgramState &RHS);
    106 
    107   ~ProgramState();
    108 
    109   /// Return the ProgramStateManager associated with this state.
    110   ProgramStateManager &getStateManager() const {
    111     return *stateMgr;
    112   }
    113 
    114   /// Return the ConstraintManager.
    115   ConstraintManager &getConstraintManager() const;
    116 
    117   /// getEnvironment - Return the environment associated with this state.
    118   ///  The environment is the mapping from expressions to values.
    119   const Environment& getEnvironment() const { return Env; }
    120 
    121   /// Return the store associated with this state.  The store
    122   ///  is a mapping from locations to values.
    123   Store getStore() const { return store; }
    124 
    125 
    126   /// getGDM - Return the generic data map associated with this state.
    127   GenericDataMap getGDM() const { return GDM; }
    128 
    129   void setGDM(GenericDataMap gdm) { GDM = gdm; }
    130 
    131   /// Profile - Profile the contents of a ProgramState object for use in a
    132   ///  FoldingSet.  Two ProgramState objects are considered equal if they
    133   ///  have the same Environment, Store, and GenericDataMap.
    134   static void Profile(llvm::FoldingSetNodeID& ID, const ProgramState *V) {
    135     V->Env.Profile(ID);
    136     ID.AddPointer(V->store);
    137     V->GDM.Profile(ID);
    138   }
    139 
    140   /// Profile - Used to profile the contents of this object for inclusion
    141   ///  in a FoldingSet.
    142   void Profile(llvm::FoldingSetNodeID& ID) const {
    143     Profile(ID, this);
    144   }
    145 
    146   BasicValueFactory &getBasicVals() const;
    147   SymbolManager &getSymbolManager() const;
    148 
    149   //==---------------------------------------------------------------------==//
    150   // Constraints on values.
    151   //==---------------------------------------------------------------------==//
    152   //
    153   // Each ProgramState records constraints on symbolic values.  These constraints
    154   // are managed using the ConstraintManager associated with a ProgramStateManager.
    155   // As constraints gradually accrue on symbolic values, added constraints
    156   // may conflict and indicate that a state is infeasible (as no real values
    157   // could satisfy all the constraints).  This is the principal mechanism
    158   // for modeling path-sensitivity in ExprEngine/ProgramState.
    159   //
    160   // Various "assume" methods form the interface for adding constraints to
    161   // symbolic values.  A call to 'assume' indicates an assumption being placed
    162   // on one or symbolic values.  'assume' methods take the following inputs:
    163   //
    164   //  (1) A ProgramState object representing the current state.
    165   //
    166   //  (2) The assumed constraint (which is specific to a given "assume" method).
    167   //
    168   //  (3) A binary value "Assumption" that indicates whether the constraint is
    169   //      assumed to be true or false.
    170   //
    171   // The output of "assume*" is a new ProgramState object with the added constraints.
    172   // If no new state is feasible, NULL is returned.
    173   //
    174 
    175   /// Assumes that the value of \p cond is zero (if \p assumption is "false")
    176   /// or non-zero (if \p assumption is "true").
    177   ///
    178   /// This returns a new state with the added constraint on \p cond.
    179   /// If no new state is feasible, NULL is returned.
    180   ProgramStateRef assume(DefinedOrUnknownSVal cond, bool assumption) const;
    181 
    182   /// Assumes both "true" and "false" for \p cond, and returns both
    183   /// corresponding states (respectively).
    184   ///
    185   /// This is more efficient than calling assume() twice. Note that one (but not
    186   /// both) of the returned states may be NULL.
    187   std::pair<ProgramStateRef, ProgramStateRef>
    188   assume(DefinedOrUnknownSVal cond) const;
    189 
    190   ProgramStateRef assumeInBound(DefinedOrUnknownSVal idx,
    191                                DefinedOrUnknownSVal upperBound,
    192                                bool assumption,
    193                                QualType IndexType = QualType()) const;
    194 
    195   /// Assumes that the value of \p Val is bounded with [\p From; \p To]
    196   /// (if \p assumption is "true") or it is fully out of this range
    197   /// (if \p assumption is "false").
    198   ///
    199   /// This returns a new state with the added constraint on \p cond.
    200   /// If no new state is feasible, NULL is returned.
    201   ProgramStateRef assumeInclusiveRange(DefinedOrUnknownSVal Val,
    202                                        const llvm::APSInt &From,
    203                                        const llvm::APSInt &To,
    204                                        bool assumption) const;
    205 
    206   /// Assumes given range both "true" and "false" for \p Val, and returns both
    207   /// corresponding states (respectively).
    208   ///
    209   /// This is more efficient than calling assume() twice. Note that one (but not
    210   /// both) of the returned states may be NULL.
    211   std::pair<ProgramStateRef, ProgramStateRef>
    212   assumeInclusiveRange(DefinedOrUnknownSVal Val, const llvm::APSInt &From,
    213                        const llvm::APSInt &To) const;
    214 
    215   /// \brief Check if the given SVal is constrained to zero or is a zero
    216   ///        constant.
    217   ConditionTruthVal isNull(SVal V) const;
    218 
    219   /// Utility method for getting regions.
    220   const VarRegion* getRegion(const VarDecl *D, const LocationContext *LC) const;
    221 
    222   //==---------------------------------------------------------------------==//
    223   // Binding and retrieving values to/from the environment and symbolic store.
    224   //==---------------------------------------------------------------------==//
    225 
    226   /// Create a new state by binding the value 'V' to the statement 'S' in the
    227   /// state's environment.
    228   ProgramStateRef BindExpr(const Stmt *S, const LocationContext *LCtx,
    229                                SVal V, bool Invalidate = true) const;
    230 
    231   ProgramStateRef bindLoc(Loc location,
    232                           SVal V,
    233                           const LocationContext *LCtx,
    234                           bool notifyChanges = true) const;
    235 
    236   ProgramStateRef bindLoc(SVal location, SVal V, const LocationContext *LCtx) const;
    237 
    238   ProgramStateRef bindDefault(SVal loc, SVal V, const LocationContext *LCtx) const;
    239 
    240   ProgramStateRef killBinding(Loc LV) const;
    241 
    242   /// \brief Returns the state with bindings for the given regions
    243   ///  cleared from the store.
    244   ///
    245   /// Optionally invalidates global regions as well.
    246   ///
    247   /// \param Regions the set of regions to be invalidated.
    248   /// \param E the expression that caused the invalidation.
    249   /// \param BlockCount The number of times the current basic block has been
    250   //         visited.
    251   /// \param CausesPointerEscape the flag is set to true when
    252   ///        the invalidation entails escape of a symbol (representing a
    253   ///        pointer). For example, due to it being passed as an argument in a
    254   ///        call.
    255   /// \param IS the set of invalidated symbols.
    256   /// \param Call if non-null, the invalidated regions represent parameters to
    257   ///        the call and should be considered directly invalidated.
    258   /// \param ITraits information about special handling for a particular
    259   ///        region/symbol.
    260   ProgramStateRef
    261   invalidateRegions(ArrayRef<const MemRegion *> Regions, const Expr *E,
    262                     unsigned BlockCount, const LocationContext *LCtx,
    263                     bool CausesPointerEscape, InvalidatedSymbols *IS = nullptr,
    264                     const CallEvent *Call = nullptr,
    265                     RegionAndSymbolInvalidationTraits *ITraits = nullptr) const;
    266 
    267   ProgramStateRef
    268   invalidateRegions(ArrayRef<SVal> Regions, const Expr *E,
    269                     unsigned BlockCount, const LocationContext *LCtx,
    270                     bool CausesPointerEscape, InvalidatedSymbols *IS = nullptr,
    271                     const CallEvent *Call = nullptr,
    272                     RegionAndSymbolInvalidationTraits *ITraits = nullptr) const;
    273 
    274   /// enterStackFrame - Returns the state for entry to the given stack frame,
    275   ///  preserving the current state.
    276   ProgramStateRef enterStackFrame(const CallEvent &Call,
    277                                   const StackFrameContext *CalleeCtx) const;
    278 
    279   /// Get the lvalue for a variable reference.
    280   Loc getLValue(const VarDecl *D, const LocationContext *LC) const;
    281 
    282   Loc getLValue(const CompoundLiteralExpr *literal,
    283                 const LocationContext *LC) const;
    284 
    285   /// Get the lvalue for an ivar reference.
    286   SVal getLValue(const ObjCIvarDecl *decl, SVal base) const;
    287 
    288   /// Get the lvalue for a field reference.
    289   SVal getLValue(const FieldDecl *decl, SVal Base) const;
    290 
    291   /// Get the lvalue for an indirect field reference.
    292   SVal getLValue(const IndirectFieldDecl *decl, SVal Base) const;
    293 
    294   /// Get the lvalue for an array index.
    295   SVal getLValue(QualType ElementType, SVal Idx, SVal Base) const;
    296 
    297   /// Returns the SVal bound to the statement 'S' in the state's environment.
    298   SVal getSVal(const Stmt *S, const LocationContext *LCtx) const;
    299 
    300   SVal getSValAsScalarOrLoc(const Stmt *Ex, const LocationContext *LCtx) const;
    301 
    302   /// \brief Return the value bound to the specified location.
    303   /// Returns UnknownVal() if none found.
    304   SVal getSVal(Loc LV, QualType T = QualType()) const;
    305 
    306   /// Returns the "raw" SVal bound to LV before any value simplfication.
    307   SVal getRawSVal(Loc LV, QualType T= QualType()) const;
    308 
    309   /// \brief Return the value bound to the specified location.
    310   /// Returns UnknownVal() if none found.
    311   SVal getSVal(const MemRegion* R) const;
    312 
    313   SVal getSValAsScalarOrLoc(const MemRegion *R) const;
    314 
    315   /// \brief Visits the symbols reachable from the given SVal using the provided
    316   /// SymbolVisitor.
    317   ///
    318   /// This is a convenience API. Consider using ScanReachableSymbols class
    319   /// directly when making multiple scans on the same state with the same
    320   /// visitor to avoid repeated initialization cost.
    321   /// \sa ScanReachableSymbols
    322   bool scanReachableSymbols(SVal val, SymbolVisitor& visitor) const;
    323 
    324   /// \brief Visits the symbols reachable from the SVals in the given range
    325   /// using the provided SymbolVisitor.
    326   bool scanReachableSymbols(const SVal *I, const SVal *E,
    327                             SymbolVisitor &visitor) const;
    328 
    329   /// \brief Visits the symbols reachable from the regions in the given
    330   /// MemRegions range using the provided SymbolVisitor.
    331   bool scanReachableSymbols(const MemRegion * const *I,
    332                             const MemRegion * const *E,
    333                             SymbolVisitor &visitor) const;
    334 
    335   template <typename CB> CB scanReachableSymbols(SVal val) const;
    336   template <typename CB> CB scanReachableSymbols(const SVal *beg,
    337                                                  const SVal *end) const;
    338 
    339   template <typename CB> CB
    340   scanReachableSymbols(const MemRegion * const *beg,
    341                        const MemRegion * const *end) const;
    342 
    343   /// Create a new state in which the statement is marked as tainted.
    344   ProgramStateRef addTaint(const Stmt *S, const LocationContext *LCtx,
    345                                TaintTagType Kind = TaintTagGeneric) const;
    346 
    347   /// Create a new state in which the value is marked as tainted.
    348   ProgramStateRef addTaint(SVal V, TaintTagType Kind = TaintTagGeneric) const;
    349 
    350   /// Create a new state in which the symbol is marked as tainted.
    351   ProgramStateRef addTaint(SymbolRef S,
    352                                TaintTagType Kind = TaintTagGeneric) const;
    353 
    354   /// Create a new state in which the region symbol is marked as tainted.
    355   ProgramStateRef addTaint(const MemRegion *R,
    356                                TaintTagType Kind = TaintTagGeneric) const;
    357 
    358   /// Create a new state in a which a sub-region of a given symbol is tainted.
    359   /// This might be necessary when referring to regions that can not have an
    360   /// individual symbol, e.g. if they are represented by the default binding of
    361   /// a LazyCompoundVal.
    362   ProgramStateRef addPartialTaint(SymbolRef ParentSym,
    363                                   const SubRegion *SubRegion,
    364                                   TaintTagType Kind = TaintTagGeneric) const;
    365 
    366   /// Check if the statement is tainted in the current state.
    367   bool isTainted(const Stmt *S, const LocationContext *LCtx,
    368                  TaintTagType Kind = TaintTagGeneric) const;
    369   bool isTainted(SVal V, TaintTagType Kind = TaintTagGeneric) const;
    370   bool isTainted(SymbolRef Sym, TaintTagType Kind = TaintTagGeneric) const;
    371   bool isTainted(const MemRegion *Reg, TaintTagType Kind=TaintTagGeneric) const;
    372 
    373   //==---------------------------------------------------------------------==//
    374   // Accessing the Generic Data Map (GDM).
    375   //==---------------------------------------------------------------------==//
    376 
    377   void *const* FindGDM(void *K) const;
    378 
    379   template<typename T>
    380   ProgramStateRef add(typename ProgramStateTrait<T>::key_type K) const;
    381 
    382   template <typename T>
    383   typename ProgramStateTrait<T>::data_type
    384   get() const {
    385     return ProgramStateTrait<T>::MakeData(FindGDM(ProgramStateTrait<T>::GDMIndex()));
    386   }
    387 
    388   template<typename T>
    389   typename ProgramStateTrait<T>::lookup_type
    390   get(typename ProgramStateTrait<T>::key_type key) const {
    391     void *const* d = FindGDM(ProgramStateTrait<T>::GDMIndex());
    392     return ProgramStateTrait<T>::Lookup(ProgramStateTrait<T>::MakeData(d), key);
    393   }
    394 
    395   template <typename T>
    396   typename ProgramStateTrait<T>::context_type get_context() const;
    397 
    398 
    399   template<typename T>
    400   ProgramStateRef remove(typename ProgramStateTrait<T>::key_type K) const;
    401 
    402   template<typename T>
    403   ProgramStateRef remove(typename ProgramStateTrait<T>::key_type K,
    404                         typename ProgramStateTrait<T>::context_type C) const;
    405   template <typename T>
    406   ProgramStateRef remove() const;
    407 
    408   template<typename T>
    409   ProgramStateRef set(typename ProgramStateTrait<T>::data_type D) const;
    410 
    411   template<typename T>
    412   ProgramStateRef set(typename ProgramStateTrait<T>::key_type K,
    413                      typename ProgramStateTrait<T>::value_type E) const;
    414 
    415   template<typename T>
    416   ProgramStateRef set(typename ProgramStateTrait<T>::key_type K,
    417                      typename ProgramStateTrait<T>::value_type E,
    418                      typename ProgramStateTrait<T>::context_type C) const;
    419 
    420   template<typename T>
    421   bool contains(typename ProgramStateTrait<T>::key_type key) const {
    422     void *const* d = FindGDM(ProgramStateTrait<T>::GDMIndex());
    423     return ProgramStateTrait<T>::Contains(ProgramStateTrait<T>::MakeData(d), key);
    424   }
    425 
    426   // Pretty-printing.
    427   void print(raw_ostream &Out, const char *nl = "\n",
    428              const char *sep = "") const;
    429   void printDOT(raw_ostream &Out) const;
    430   void printTaint(raw_ostream &Out, const char *nl = "\n",
    431                   const char *sep = "") const;
    432 
    433   void dump() const;
    434   void dumpTaint() const;
    435 
    436 private:
    437   friend void ProgramStateRetain(const ProgramState *state);
    438   friend void ProgramStateRelease(const ProgramState *state);
    439 
    440   /// \sa invalidateValues()
    441   /// \sa invalidateRegions()
    442   ProgramStateRef
    443   invalidateRegionsImpl(ArrayRef<SVal> Values,
    444                         const Expr *E, unsigned BlockCount,
    445                         const LocationContext *LCtx,
    446                         bool ResultsInSymbolEscape,
    447                         InvalidatedSymbols *IS,
    448                         RegionAndSymbolInvalidationTraits *HTraits,
    449                         const CallEvent *Call) const;
    450 };
    451 
    452 //===----------------------------------------------------------------------===//
    453 // ProgramStateManager - Factory object for ProgramStates.
    454 //===----------------------------------------------------------------------===//
    455 
    456 class ProgramStateManager {
    457   friend class ProgramState;
    458   friend void ProgramStateRelease(const ProgramState *state);
    459 private:
    460   /// Eng - The SubEngine that owns this state manager.
    461   SubEngine *Eng; /* Can be null. */
    462 
    463   EnvironmentManager                   EnvMgr;
    464   std::unique_ptr<StoreManager>        StoreMgr;
    465   std::unique_ptr<ConstraintManager>   ConstraintMgr;
    466 
    467   ProgramState::GenericDataMap::Factory     GDMFactory;
    468   TaintedSubRegions::Factory TSRFactory;
    469 
    470   typedef llvm::DenseMap<void*,std::pair<void*,void (*)(void*)> > GDMContextsTy;
    471   GDMContextsTy GDMContexts;
    472 
    473   /// StateSet - FoldingSet containing all the states created for analyzing
    474   ///  a particular function.  This is used to unique states.
    475   llvm::FoldingSet<ProgramState> StateSet;
    476 
    477   /// Object that manages the data for all created SVals.
    478   std::unique_ptr<SValBuilder> svalBuilder;
    479 
    480   /// Manages memory for created CallEvents.
    481   std::unique_ptr<CallEventManager> CallEventMgr;
    482 
    483   /// A BumpPtrAllocator to allocate states.
    484   llvm::BumpPtrAllocator &Alloc;
    485 
    486   /// A vector of ProgramStates that we can reuse.
    487   std::vector<ProgramState *> freeStates;
    488 
    489 public:
    490   ProgramStateManager(ASTContext &Ctx,
    491                  StoreManagerCreator CreateStoreManager,
    492                  ConstraintManagerCreator CreateConstraintManager,
    493                  llvm::BumpPtrAllocator& alloc,
    494                  SubEngine *subeng);
    495 
    496   ~ProgramStateManager();
    497 
    498   ProgramStateRef getInitialState(const LocationContext *InitLoc);
    499 
    500   ASTContext &getContext() { return svalBuilder->getContext(); }
    501   const ASTContext &getContext() const { return svalBuilder->getContext(); }
    502 
    503   BasicValueFactory &getBasicVals() {
    504     return svalBuilder->getBasicValueFactory();
    505   }
    506 
    507   SValBuilder &getSValBuilder() {
    508     return *svalBuilder;
    509   }
    510 
    511   SymbolManager &getSymbolManager() {
    512     return svalBuilder->getSymbolManager();
    513   }
    514   const SymbolManager &getSymbolManager() const {
    515     return svalBuilder->getSymbolManager();
    516   }
    517 
    518   llvm::BumpPtrAllocator& getAllocator() { return Alloc; }
    519 
    520   MemRegionManager& getRegionManager() {
    521     return svalBuilder->getRegionManager();
    522   }
    523   const MemRegionManager& getRegionManager() const {
    524     return svalBuilder->getRegionManager();
    525   }
    526 
    527   CallEventManager &getCallEventManager() { return *CallEventMgr; }
    528 
    529   StoreManager& getStoreManager() { return *StoreMgr; }
    530   ConstraintManager& getConstraintManager() { return *ConstraintMgr; }
    531   SubEngine* getOwningEngine() { return Eng; }
    532 
    533   ProgramStateRef removeDeadBindings(ProgramStateRef St,
    534                                     const StackFrameContext *LCtx,
    535                                     SymbolReaper& SymReaper);
    536 
    537 public:
    538 
    539   SVal ArrayToPointer(Loc Array, QualType ElementTy) {
    540     return StoreMgr->ArrayToPointer(Array, ElementTy);
    541   }
    542 
    543   // Methods that manipulate the GDM.
    544   ProgramStateRef addGDM(ProgramStateRef St, void *Key, void *Data);
    545   ProgramStateRef removeGDM(ProgramStateRef state, void *Key);
    546 
    547   // Methods that query & manipulate the Store.
    548 
    549   void iterBindings(ProgramStateRef state, StoreManager::BindingsHandler& F) {
    550     StoreMgr->iterBindings(state->getStore(), F);
    551   }
    552 
    553   ProgramStateRef getPersistentState(ProgramState &Impl);
    554   ProgramStateRef getPersistentStateWithGDM(ProgramStateRef FromState,
    555                                            ProgramStateRef GDMState);
    556 
    557   bool haveEqualEnvironments(ProgramStateRef S1, ProgramStateRef S2) {
    558     return S1->Env == S2->Env;
    559   }
    560 
    561   bool haveEqualStores(ProgramStateRef S1, ProgramStateRef S2) {
    562     return S1->store == S2->store;
    563   }
    564 
    565   //==---------------------------------------------------------------------==//
    566   // Generic Data Map methods.
    567   //==---------------------------------------------------------------------==//
    568   //
    569   // ProgramStateManager and ProgramState support a "generic data map" that allows
    570   // different clients of ProgramState objects to embed arbitrary data within a
    571   // ProgramState object.  The generic data map is essentially an immutable map
    572   // from a "tag" (that acts as the "key" for a client) and opaque values.
    573   // Tags/keys and values are simply void* values.  The typical way that clients
    574   // generate unique tags are by taking the address of a static variable.
    575   // Clients are responsible for ensuring that data values referred to by a
    576   // the data pointer are immutable (and thus are essentially purely functional
    577   // data).
    578   //
    579   // The templated methods below use the ProgramStateTrait<T> class
    580   // to resolve keys into the GDM and to return data values to clients.
    581   //
    582 
    583   // Trait based GDM dispatch.
    584   template <typename T>
    585   ProgramStateRef set(ProgramStateRef st, typename ProgramStateTrait<T>::data_type D) {
    586     return addGDM(st, ProgramStateTrait<T>::GDMIndex(),
    587                   ProgramStateTrait<T>::MakeVoidPtr(D));
    588   }
    589 
    590   template<typename T>
    591   ProgramStateRef set(ProgramStateRef st,
    592                      typename ProgramStateTrait<T>::key_type K,
    593                      typename ProgramStateTrait<T>::value_type V,
    594                      typename ProgramStateTrait<T>::context_type C) {
    595 
    596     return addGDM(st, ProgramStateTrait<T>::GDMIndex(),
    597      ProgramStateTrait<T>::MakeVoidPtr(ProgramStateTrait<T>::Set(st->get<T>(), K, V, C)));
    598   }
    599 
    600   template <typename T>
    601   ProgramStateRef add(ProgramStateRef st,
    602                      typename ProgramStateTrait<T>::key_type K,
    603                      typename ProgramStateTrait<T>::context_type C) {
    604     return addGDM(st, ProgramStateTrait<T>::GDMIndex(),
    605         ProgramStateTrait<T>::MakeVoidPtr(ProgramStateTrait<T>::Add(st->get<T>(), K, C)));
    606   }
    607 
    608   template <typename T>
    609   ProgramStateRef remove(ProgramStateRef st,
    610                         typename ProgramStateTrait<T>::key_type K,
    611                         typename ProgramStateTrait<T>::context_type C) {
    612 
    613     return addGDM(st, ProgramStateTrait<T>::GDMIndex(),
    614      ProgramStateTrait<T>::MakeVoidPtr(ProgramStateTrait<T>::Remove(st->get<T>(), K, C)));
    615   }
    616 
    617   template <typename T>
    618   ProgramStateRef remove(ProgramStateRef st) {
    619     return removeGDM(st, ProgramStateTrait<T>::GDMIndex());
    620   }
    621 
    622   void *FindGDMContext(void *index,
    623                        void *(*CreateContext)(llvm::BumpPtrAllocator&),
    624                        void  (*DeleteContext)(void*));
    625 
    626   template <typename T>
    627   typename ProgramStateTrait<T>::context_type get_context() {
    628     void *p = FindGDMContext(ProgramStateTrait<T>::GDMIndex(),
    629                              ProgramStateTrait<T>::CreateContext,
    630                              ProgramStateTrait<T>::DeleteContext);
    631 
    632     return ProgramStateTrait<T>::MakeContext(p);
    633   }
    634 
    635   void EndPath(ProgramStateRef St) {
    636     ConstraintMgr->EndPath(St);
    637   }
    638 };
    639 
    640 
    641 //===----------------------------------------------------------------------===//
    642 // Out-of-line method definitions for ProgramState.
    643 //===----------------------------------------------------------------------===//
    644 
    645 inline ConstraintManager &ProgramState::getConstraintManager() const {
    646   return stateMgr->getConstraintManager();
    647 }
    648 
    649 inline const VarRegion* ProgramState::getRegion(const VarDecl *D,
    650                                                 const LocationContext *LC) const
    651 {
    652   return getStateManager().getRegionManager().getVarRegion(D, LC);
    653 }
    654 
    655 inline ProgramStateRef ProgramState::assume(DefinedOrUnknownSVal Cond,
    656                                       bool Assumption) const {
    657   if (Cond.isUnknown())
    658     return this;
    659 
    660   return getStateManager().ConstraintMgr
    661       ->assume(this, Cond.castAs<DefinedSVal>(), Assumption);
    662 }
    663 
    664 inline std::pair<ProgramStateRef , ProgramStateRef >
    665 ProgramState::assume(DefinedOrUnknownSVal Cond) const {
    666   if (Cond.isUnknown())
    667     return std::make_pair(this, this);
    668 
    669   return getStateManager().ConstraintMgr
    670       ->assumeDual(this, Cond.castAs<DefinedSVal>());
    671 }
    672 
    673 inline ProgramStateRef ProgramState::assumeInclusiveRange(
    674     DefinedOrUnknownSVal Val, const llvm::APSInt &From, const llvm::APSInt &To,
    675     bool Assumption) const {
    676   if (Val.isUnknown())
    677     return this;
    678 
    679   assert(Val.getAs<NonLoc>() && "Only NonLocs are supported!");
    680 
    681   return getStateManager().ConstraintMgr->assumeInclusiveRange(
    682       this, Val.castAs<NonLoc>(), From, To, Assumption);
    683 }
    684 
    685 inline std::pair<ProgramStateRef, ProgramStateRef>
    686 ProgramState::assumeInclusiveRange(DefinedOrUnknownSVal Val,
    687                                    const llvm::APSInt &From,
    688                                    const llvm::APSInt &To) const {
    689   if (Val.isUnknown())
    690     return std::make_pair(this, this);
    691 
    692   assert(Val.getAs<NonLoc>() && "Only NonLocs are supported!");
    693 
    694   return getStateManager().ConstraintMgr->assumeInclusiveRangeDual(
    695       this, Val.castAs<NonLoc>(), From, To);
    696 }
    697 
    698 inline ProgramStateRef ProgramState::bindLoc(SVal LV, SVal V, const LocationContext *LCtx) const {
    699   if (Optional<Loc> L = LV.getAs<Loc>())
    700     return bindLoc(*L, V, LCtx);
    701   return this;
    702 }
    703 
    704 inline Loc ProgramState::getLValue(const VarDecl *VD,
    705                                const LocationContext *LC) const {
    706   return getStateManager().StoreMgr->getLValueVar(VD, LC);
    707 }
    708 
    709 inline Loc ProgramState::getLValue(const CompoundLiteralExpr *literal,
    710                                const LocationContext *LC) const {
    711   return getStateManager().StoreMgr->getLValueCompoundLiteral(literal, LC);
    712 }
    713 
    714 inline SVal ProgramState::getLValue(const ObjCIvarDecl *D, SVal Base) const {
    715   return getStateManager().StoreMgr->getLValueIvar(D, Base);
    716 }
    717 
    718 inline SVal ProgramState::getLValue(const FieldDecl *D, SVal Base) const {
    719   return getStateManager().StoreMgr->getLValueField(D, Base);
    720 }
    721 
    722 inline SVal ProgramState::getLValue(const IndirectFieldDecl *D,
    723                                     SVal Base) const {
    724   StoreManager &SM = *getStateManager().StoreMgr;
    725   for (const auto *I : D->chain()) {
    726     Base = SM.getLValueField(cast<FieldDecl>(I), Base);
    727   }
    728 
    729   return Base;
    730 }
    731 
    732 inline SVal ProgramState::getLValue(QualType ElementType, SVal Idx, SVal Base) const{
    733   if (Optional<NonLoc> N = Idx.getAs<NonLoc>())
    734     return getStateManager().StoreMgr->getLValueElement(ElementType, *N, Base);
    735   return UnknownVal();
    736 }
    737 
    738 inline SVal ProgramState::getSVal(const Stmt *Ex,
    739                                   const LocationContext *LCtx) const{
    740   return Env.getSVal(EnvironmentEntry(Ex, LCtx),
    741                      *getStateManager().svalBuilder);
    742 }
    743 
    744 inline SVal
    745 ProgramState::getSValAsScalarOrLoc(const Stmt *S,
    746                                    const LocationContext *LCtx) const {
    747   if (const Expr *Ex = dyn_cast<Expr>(S)) {
    748     QualType T = Ex->getType();
    749     if (Ex->isGLValue() || Loc::isLocType(T) ||
    750         T->isIntegralOrEnumerationType())
    751       return getSVal(S, LCtx);
    752   }
    753 
    754   return UnknownVal();
    755 }
    756 
    757 inline SVal ProgramState::getRawSVal(Loc LV, QualType T) const {
    758   return getStateManager().StoreMgr->getBinding(getStore(), LV, T);
    759 }
    760 
    761 inline SVal ProgramState::getSVal(const MemRegion* R) const {
    762   return getStateManager().StoreMgr->getBinding(getStore(),
    763                                                 loc::MemRegionVal(R));
    764 }
    765 
    766 inline BasicValueFactory &ProgramState::getBasicVals() const {
    767   return getStateManager().getBasicVals();
    768 }
    769 
    770 inline SymbolManager &ProgramState::getSymbolManager() const {
    771   return getStateManager().getSymbolManager();
    772 }
    773 
    774 template<typename T>
    775 ProgramStateRef ProgramState::add(typename ProgramStateTrait<T>::key_type K) const {
    776   return getStateManager().add<T>(this, K, get_context<T>());
    777 }
    778 
    779 template <typename T>
    780 typename ProgramStateTrait<T>::context_type ProgramState::get_context() const {
    781   return getStateManager().get_context<T>();
    782 }
    783 
    784 template<typename T>
    785 ProgramStateRef ProgramState::remove(typename ProgramStateTrait<T>::key_type K) const {
    786   return getStateManager().remove<T>(this, K, get_context<T>());
    787 }
    788 
    789 template<typename T>
    790 ProgramStateRef ProgramState::remove(typename ProgramStateTrait<T>::key_type K,
    791                                typename ProgramStateTrait<T>::context_type C) const {
    792   return getStateManager().remove<T>(this, K, C);
    793 }
    794 
    795 template <typename T>
    796 ProgramStateRef ProgramState::remove() const {
    797   return getStateManager().remove<T>(this);
    798 }
    799 
    800 template<typename T>
    801 ProgramStateRef ProgramState::set(typename ProgramStateTrait<T>::data_type D) const {
    802   return getStateManager().set<T>(this, D);
    803 }
    804 
    805 template<typename T>
    806 ProgramStateRef ProgramState::set(typename ProgramStateTrait<T>::key_type K,
    807                             typename ProgramStateTrait<T>::value_type E) const {
    808   return getStateManager().set<T>(this, K, E, get_context<T>());
    809 }
    810 
    811 template<typename T>
    812 ProgramStateRef ProgramState::set(typename ProgramStateTrait<T>::key_type K,
    813                             typename ProgramStateTrait<T>::value_type E,
    814                             typename ProgramStateTrait<T>::context_type C) const {
    815   return getStateManager().set<T>(this, K, E, C);
    816 }
    817 
    818 template <typename CB>
    819 CB ProgramState::scanReachableSymbols(SVal val) const {
    820   CB cb(this);
    821   scanReachableSymbols(val, cb);
    822   return cb;
    823 }
    824 
    825 template <typename CB>
    826 CB ProgramState::scanReachableSymbols(const SVal *beg, const SVal *end) const {
    827   CB cb(this);
    828   scanReachableSymbols(beg, end, cb);
    829   return cb;
    830 }
    831 
    832 template <typename CB>
    833 CB ProgramState::scanReachableSymbols(const MemRegion * const *beg,
    834                                  const MemRegion * const *end) const {
    835   CB cb(this);
    836   scanReachableSymbols(beg, end, cb);
    837   return cb;
    838 }
    839 
    840 /// \class ScanReachableSymbols
    841 /// A utility class that visits the reachable symbols using a custom
    842 /// SymbolVisitor. Terminates recursive traversal when the visitor function
    843 /// returns false.
    844 class ScanReachableSymbols {
    845   typedef llvm::DenseSet<const void*> VisitedItems;
    846 
    847   VisitedItems visited;
    848   ProgramStateRef state;
    849   SymbolVisitor &visitor;
    850 public:
    851   ScanReachableSymbols(ProgramStateRef st, SymbolVisitor &v)
    852       : state(std::move(st)), visitor(v) {}
    853 
    854   bool scan(nonloc::LazyCompoundVal val);
    855   bool scan(nonloc::CompoundVal val);
    856   bool scan(SVal val);
    857   bool scan(const MemRegion *R);
    858   bool scan(const SymExpr *sym);
    859 };
    860 
    861 } // end ento namespace
    862 
    863 } // end clang namespace
    864 
    865 #endif
    866