From 957063890262fed3fdb21fa66430f695cce10def Mon Sep 17 00:00:00 2001
From: Steven de Oliveira <de.oliveira.steven@gmail.com>
Date: Fri, 5 Apr 2024 16:56:31 +0200
Subject: [PATCH] Test

---
 src/bin/common/solving_loop.ml | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/src/bin/common/solving_loop.ml b/src/bin/common/solving_loop.ml
index 7956fc4ecf..60a0f8660e 100644
--- a/src/bin/common/solving_loop.ml
+++ b/src/bin/common/solving_loop.ml
@@ -1458,9 +1458,12 @@ let main () =
             cmd_on_modes st [Assert; Sat; Unsat] "goal";
             let st = pop_if_post_query st in
             let st = push_before_query st in
-            let st = handle_query st id loc attrs contents in
+            let st, raised_timeout =
+              try handle_query st id loc attrs contents, false with
+              | Util.Timeout -> st, true
+            in
             let () =
-              if has_timeout st then
+              if raised_timeout || has_timeout st then
                 if Options.get_timelimit_per_goal ()
                 then begin
                   Options.Time.start ();
@@ -1475,9 +1478,12 @@ let main () =
             cmd_on_modes st [Assert; Unsat; Sat] "check-sat";
             let st = pop_if_post_query st in
             let st = push_before_query st in
-            let st = handle_solve st id contents loc attrs in
+            let st, raised_timeout =
+              try handle_solve st id contents loc attrs, false with
+              | Util.Timeout -> st, true
+            in
             let () =
-              if has_timeout st then
+              if raised_timeout || has_timeout st then
                 if Options.get_timelimit_per_goal ()
                 then begin
                   Options.Time.start ();