From a5038864477ba9406c196993e51bfccc0ba4988b Mon Sep 17 00:00:00 2001
From: Matthias Straka <mstraka84@gmail.com>
Date: Sun, 12 Apr 2026 21:23:32 +0200
Subject: [PATCH 1/2] cli : Use acquire/release semantics for stopping logic

Previously, seq_cst ordering was enforced, which is slower and has no benefits when using only one atomic variable.
---
 tools/cli/cli.cpp | 21 +++++++++------------
 1 file changed, 9 insertions(+), 12 deletions(-)

diff --git a/tools/cli/cli.cpp b/tools/cli/cli.cpp
index 8b7b58693fc9..64f1dd2c63c0 100644
--- a/tools/cli/cli.cpp
+++ b/tools/cli/cli.cpp
@@ -37,19 +37,22 @@ const char * LLAMA_ASCII_LOGO = R"(
 
 static std::atomic<bool> g_is_interrupted = false;
 static bool should_stop() {
-    return g_is_interrupted.load();
+    return g_is_interrupted.load(std::memory_order_acquire);
+}
+static void reset_stop() {
+    g_is_interrupted.store(false, std::memory_order_release);
 }
 
 #if defined (__unix__) || (defined (__APPLE__) && defined (__MACH__)) || defined (_WIN32)
 static void signal_handler(int) {
-    if (g_is_interrupted.load()) {
+    const bool already_interrupted = g_is_interrupted.exchange(true, std::memory_order_acq_rel);
+    if (already_interrupted) {
         // second Ctrl+C - exit immediately
         // make sure to clear colors before exiting (not using LOG or console.cpp here to avoid deadlock)
         fprintf(stdout, "\033[0m\n");
         fflush(stdout);
         std::exit(130);
     }
-    g_is_interrupted.store(true);
 }
 #endif
 
@@ -60,9 +63,6 @@ struct cli_context {
     task_params defaults;
     bool verbose_prompt;
 
-    // thread for showing "loading" animation
-    std::atomic<bool> loading_show;
-
     cli_context(const common_params & params) {
         defaults.sampling    = params.sampling;
         defaults.speculative = params.speculative;
@@ -150,10 +150,7 @@ struct cli_context {
         std::string curr_content;
         bool is_thinking = false;
 
-        while (result) {
-            if (should_stop()) {
-                break;
-            }
+        while (result && !should_stop()) {
             if (result->is_error()) {
                 json err_data = result->to_json();
                 if (err_data.contains("message")) {
@@ -195,7 +192,7 @@ struct cli_context {
             }
             result = rd.next(should_stop);
         }
-        g_is_interrupted.store(false);
+        reset_stop();
         // server_response_reader automatically cancels pending tasks upon destruction
         return curr_content;
     }
@@ -527,7 +524,7 @@ int llama_cli(int argc, char ** argv) {
         console::log("\n");
 
         if (should_stop()) {
-            g_is_interrupted.store(false);
+            reset_stop();
             break;
         }
 

From 39af19afa79f0bcecd790189591ae6a57a0d019d Mon Sep 17 00:00:00 2001
From: Matthias Straka <mstraka84@gmail.com>
Date: Wed, 15 Apr 2026 20:34:05 +0200
Subject: [PATCH 2/2] cli : use only async-signal-safe calls in signal handler

---
 tools/cli/cli.cpp | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/tools/cli/cli.cpp b/tools/cli/cli.cpp
index 64f1dd2c63c0..df65f90a56f7 100644
--- a/tools/cli/cli.cpp
+++ b/tools/cli/cli.cpp
@@ -12,17 +12,21 @@
 #include <array>
 #include <atomic>
 #include <algorithm>
+#include <csignal>
 #include <filesystem>
 #include <fstream>
 #include <thread>
-#include <signal.h>
 
 #if defined(_WIN32)
 #define WIN32_LEAN_AND_MEAN
 #ifndef NOMINMAX
 #   define NOMINMAX
 #endif
+#include <io.h>
+#include <stdio.h>
 #include <windows.h>
+#elif defined (__unix__) || (defined (__APPLE__) && defined (__MACH__))
+#include <unistd.h>
 #endif
 
 const char * LLAMA_ASCII_LOGO = R"(
@@ -49,9 +53,13 @@ static void signal_handler(int) {
     if (already_interrupted) {
         // second Ctrl+C - exit immediately
         // make sure to clear colors before exiting (not using LOG or console.cpp here to avoid deadlock)
-        fprintf(stdout, "\033[0m\n");
-        fflush(stdout);
-        std::exit(130);
+        static constexpr char color_reset[] = "\033[0m\n";
+#if defined(_WIN32)
+        _write(_fileno(stdout), color_reset, sizeof(color_reset) - 1);
+#else
+        [[maybe_unused]] ssize_t ret = write(STDOUT_FILENO, color_reset, sizeof(color_reset) - 1);
+#endif
+        _exit(128 + SIGINT);
     }
 }
 #endif