From e8a9178ebe4860a8a00a6ec8f77b661fdad84890 Mon Sep 17 00:00:00 2001
From: Nadrieril
Date: Wed, 21 Aug 2019 17:40:56 +0200
Subject: Don't use take_mut::take lightly since normalize_whnf might panic

---
 dhall/src/core/value.rs | 27 ++++++++++++++++++---------
 1 file changed, 18 insertions(+), 9 deletions(-)

(limited to 'dhall')

diff --git a/dhall/src/core/value.rs b/dhall/src/core/value.rs
index e1623a8..69d372a 100644
--- a/dhall/src/core/value.rs
+++ b/dhall/src/core/value.rs
@@ -61,16 +61,25 @@ impl ValueInternal {
     }
 
     fn normalize_whnf(&mut self) {
-        take_mut::take(self, |vint| match &vint.form {
-            Unevaled => ValueInternal {
-                form: WHNF,
-                // TODO: thunk chaining
-                value: normalize_whnf(vint.value).into_whnf(),
-                ty: vint.ty,
+        take_mut::take_or_recover(
+            self,
+            // Dummy value in case the other closure panics
+            || ValueInternal {
+                form: Unevaled,
+                value: ValueF::Const(Const::Type),
+                ty: None,
             },
-            // Already in WHNF
-            WHNF | NF => vint,
-        })
+            |vint| match &vint.form {
+                Unevaled => ValueInternal {
+                    form: WHNF,
+                    // TODO: thunk chaining
+                    value: normalize_whnf(vint.value).into_whnf(),
+                    ty: vint.ty,
+                },
+                // Already in WHNF
+                WHNF | NF => vint,
+            },
+        )
     }
     fn normalize_nf(&mut self) {
         match self.form {
-- 
cgit v1.2.3