Add two utility methods.

llvm-svn: 83935

2 files changed, 11 insertions, 0 deletions

diff --git a/clang/include/clang/Analysis/PathSensitive/AnalysisManager.h b/clang/include/clang/Analysis/PathSensitive/AnalysisManager.h
index 868e3c06e4b..e97f80576a8 100644
--- a/clang/include/clang/Analysis/PathSensitive/AnalysisManager.h
+++ b/clang/include/clang/Analysis/PathSensitive/AnalysisManager.h
@@ -122,9 +122,17 @@ public:
     return AnaCtxMgr.getContext(D)->getParentMap();
   }
 
+  // Get the top level stack frame.
   StackFrameContext *getStackFrame(Decl const *D) {
     return LocCtxMgr.getStackFrame(AnaCtxMgr.getContext(D), 0, 0);
   }
+
+  // Get a stack frame with parent.
+  StackFrameContext const *getStackFrame(Decl const *D, 
+                                         LocationContext const *Parent,
+                                         Stmt const *S) {
+    return LocCtxMgr.getStackFrame(AnaCtxMgr.getContext(D), Parent, S);
+  }
 };
 
 }
diff --git a/clang/include/clang/Analysis/PathSensitive/GRCoreEngine.h b/clang/include/clang/Analysis/PathSensitive/GRCoreEngine.h
index 48b86b9eaf0..02e0b0275e4 100644
--- a/clang/include/clang/Analysis/PathSensitive/GRCoreEngine.h
+++ b/clang/include/clang/Analysis/PathSensitive/GRCoreEngine.h
@@ -162,6 +162,9 @@ public:
     return LastNode ? (LastNode->isSink() ? NULL : LastNode) : NULL;
   }
 
+  // FIXME: This should not be exposed.
+  GRWorkList *getWorkList() { return Eng.WList; }
+
   void SetCleanedState(const GRState* St) {
     CleanedState = St;
   }