From 30d3db0d339a9a577fd310bf7e96d8cd0b0dede7 Mon Sep 17 00:00:00 2001
From: Son Ho
Date: Fri, 17 Dec 2021 14:00:37 +0100
Subject: Implement the ASharedLoan case of reborrow_shared

---
 src/Interpreter.ml | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/Interpreter.ml b/src/Interpreter.ml
index b58afb4c..caac0de7 100644
--- a/src/Interpreter.ml
+++ b/src/Interpreter.ml
@@ -1082,7 +1082,13 @@ let reborrow_shared (original_bid : V.BorrowId.id) (new_bid : V.BorrowId.id)
           V.SharedLoan (bids', sv))
         else super#visit_SharedLoan env bids sv
 
-      method! visit_Abs _ _ = raise Unimplemented
+      method! visit_ASharedLoan env bids v av =
+        (* This case is similar to the [SharedLoan] case *)
+        if V.BorrowId.Set.mem original_bid bids then (
+          set_ref ();
+          let bids' = V.BorrowId.Set.add new_bid bids in
+          V.ASharedLoan (bids', v, av))
+        else super#visit_ASharedLoan env bids v av
     end
   in
 
-- 
cgit v1.2.3