From 5ec766ceb381f15e6ab4cf568b0f6ab919ca6b68 Mon Sep 17 00:00:00 2001
From: Thomas Bauereiss
Date: Tue, 29 Aug 2017 17:17:49 +0100
Subject: Fix bug in rewriting local variable updates

Don't pull out local variables that are introduced inside a sub-block.
---
 src/rewriter.ml | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/rewriter.ml b/src/rewriter.ml
index 4af90f51..eb2a0bdf 100644
--- a/src/rewriter.ml
+++ b/src/rewriter.ml
@@ -2939,10 +2939,23 @@ let eqidtyp (id1,_) (id2,_) =
   let name2 = match id2 with Id_aux ((Id name | DeIid name),_) -> name in
   name1 = name2
 
-let find_updated_vars exp =
+let find_introduced_vars exp =
   let lEXP_aux ((ids,lexp),annot) =
     let ids = match lexp, annot with
     | LEXP_id id, (_, Some (env, _, _)) ->
+      (match Env.lookup_id id env with
+      | Unbound -> IdSet.add id ids
+      | _ -> ids)
+    | _ -> ids in
+    (ids, LEXP_aux (lexp, annot)) in
+  fst (fold_exp
+    { (compute_exp_alg IdSet.empty IdSet.union) with lEXP_aux = lEXP_aux } exp)
+
+let find_updated_vars exp =
+  let intros = find_introduced_vars exp in
+  let lEXP_aux ((ids,lexp),annot) =
+    let ids = match lexp, annot with
+    | LEXP_id id, (_, Some (env, _, _)) when not (IdSet.mem id intros) ->
       (match Env.lookup_id id env with
       | Local (Mutable, _) -> (id, annot) :: ids
       | _ -> ids)
-- 
cgit v1.2.3