From a18a185a18e1564e6b2ca113a8e5f8d60176cf51 Mon Sep 17 00:00:00 2001
From: Dmitry Stogov <dmitry@zend.com>
Date: Wed, 17 Apr 2024 08:47:25 +0300
Subject: [PATCH] Dump execution time even if JIT-ed code calls exit()

---
 ir_main.c | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/ir_main.c b/ir_main.c
index bd0d248..63d7c1d 100644
--- a/ir_main.c
+++ b/ir_main.c
@@ -34,9 +34,19 @@ static double ir_time(void)
 	return (double)((((uint64_t)filetime.dwHighDateTime << 32) | (uint64_t)filetime.dwLowDateTime)/10) /
 		1000000.0;
 }
-
 #endif
 
+static double ir_atexit_start = 0.0;
+
+static void ir_atexit(void)
+{
+	if (ir_atexit_start) {
+		double t = ir_time();
+		fprintf(stderr, "\nexecution time = %0.6f\n", t - ir_atexit_start);
+		ir_atexit_start = 0.0;
+	}
+}
+
 static void help(const char *cmd)
 {
 	printf(
@@ -1428,6 +1438,11 @@ int main(int argc, char **argv)
 		char **jit_argv;
 		int (*func)(int, char**) = loader.main;
 
+		if (dump_time) {
+			ir_atexit_start = start;
+			atexit(ir_atexit);
+		}
+
 		if (run_args && argc > run_args) {
 			jit_argc = argc - run_args + 1;
 		}
@@ -1441,6 +1456,7 @@ int main(int argc, char **argv)
 		if (dump_time) {
 			double t = ir_time();
 			fprintf(stderr, "\nexecution time = %0.6f\n", t - start);
+			ir_atexit_start = 0.0;
 		}
 
 #ifndef _WIN32