1 files changed, 28 insertions, 6 deletions

diff --git a/clang/lib/Analysis/Interval.cpp b/clang/lib/Analysis/Interval.cpp
index 3a82344..7096d75 100644
--- a/clang/lib/Analysis/Interval.cpp
+++ b/clang/lib/Analysis/Interval.cpp
@@ -637,10 +637,16 @@ void runOnBlock(const CFGBlock* block, EqnSys& system, BlockVars& block_vars) {
 	  std::vector<EqnExpr*> plusArgs;
 	  for (int negate = 0; negate < 2; ++negate) {
 	    std::string var_name = negate ? negate_string(variables->first) : variables->first;
-	    std::vector<EqnExpr*> minArgs;
-	    minArgs.push_back(vars[var_name]);
-	    minArgs.push_back(&system.constant(cond[var_name]));
-	    plusArgs.push_back(&system.expression(new Minimum<ZBar>(), minArgs));
+            if (cond[var_name] == infinity<ZBar>()) {
+              plusArgs.push_back(vars[var_name]);
+            } else if (cond[var_name] == -infinity<ZBar>()) {
+              plusArgs.push_back(&system.constant(-infinity<ZBar>()));
+            } else {
+              std::vector<EqnExpr*> minArgs;
+              minArgs.push_back(vars[var_name]);
+              minArgs.push_back(&system.constant(cond[var_name]));
+              plusArgs.push_back(&system.expression(new Minimum<ZBar>(), minArgs));
+            }
 	  }
 
 	  std::vector<EqnExpr*> guard_args;
@@ -667,15 +673,31 @@ IntervalAnalysis :: IntervalAnalysis(AnalysisDeclContext &context)
 IntervalAnalysis :: ~IntervalAnalysis() {
 }
 
-void IntervalAnalysis::runOnAllBlocks() {
+void IntervalAnalysis::runOnAllBlocks(const Decl& decl) {
   const CFG *cfg = this->context->getCFG();
 
   cfg->dump(context->getASTContext().getLangOpts(),
             llvm::sys::Process::StandardErrHasColors());
 
-  EqnSys system;
+  EqnSys system; 
   BlockVars block_vars;
 
+  std::vector<EqnExpr*> infArg; 
+  infArg.push_back(&system.constant(infinity<ZBar>()));
+  std::set<std::string>& function_arguments = block_vars[&cfg->getEntry()];
+  std::string block_id = toString(cfg->getEntry().getBlockID());
+  if (const FunctionDecl* func = dyn_cast<const FunctionDecl>(&decl)) {
+    for (unsigned int i = func->getNumParams(); i > 0; i--) {
+      std::string name = func->getParamDecl(i-1)->getNameAsString();
+
+      function_arguments.insert(name); // add it to the first block's vars
+      function_arguments.insert(negate_string(name));  // add it to the first block's vars
+
+      system[system.variable(name + '-' + block_id + "-pre")] = &system.maxExpression(infArg); // set the vars to infinity in the system here
+      system[system.variable(negate_string(name) + '-' + block_id + "-pre")] = &system.maxExpression(infArg); // set the vars to infinity in the system here
+    }
+  }
+
   std::set<const CFGBlock*> seen;
   std::deque<const CFGBlock*> todo;
   todo.push_back(&cfg->getEntry());