lib: improve signal handling

1 files changed, 40 insertions, 7 deletions

diff --git a/ttwhy/io.cppm b/ttwhy/io.cppm
index fd84dfd..be87d36 100644
--- a/ttwhy/io.cppm
+++ b/ttwhy/io.cppm
@@ -14,19 +14,48 @@ import :scanner;
 namespace ttwhy::io
 {
 
-  export template<typename Stream>
-  auto handle_signals(Stream & stream) -> asio::awaitable<std::error_code>
+  export template<typename Stream, typename TerminalAttributes>
+  auto handle_signals(Stream & stream, TerminalAttributes & attributes) -> asio::awaitable<std::error_code>
   {
     auto executor = co_await asio::this_coro::executor;
     auto signals = asio::signal_set{executor, SIGINT, SIGTERM};
+    signals.add(SIGTSTP);
+    signals.add(SIGCONT);
 
-    auto [error, signal] = co_await signals.async_wait(asio::as_tuple(asio::use_awaitable));
-    if (!error)
+    while (true)
     {
-      auto message = std::format("Received signal {}, exiting ...\n", signal);
-      co_await asio::async_write(stream, asio::buffer(message), asio::use_awaitable);
+      auto [error, signal] = co_await signals.async_wait(asio::as_tuple(asio::use_awaitable));
+      if (error)
+      {
+        co_return error;
+      }
+
+      switch (signal)
+      {
+        case SIGINT:
+        case SIGTERM:
+        {
+          auto message = std::format("Received signal {}, exiting ...\n", signal);
+          co_await asio::async_write(stream, asio::buffer(message), asio::use_awaitable);
+          co_return asio::error_code{};
+        }
+        case SIGTSTP:
+        {
+          attributes.canonical_mode(true).echo(true);
+          ::raise(SIGSTOP);
+          break;
+        }
+        case SIGCONT:
+        {
+          attributes.canonical_mode(false).echo(false);
+          break;
+        }
+      }
+
+      if (!error)
+      {
+      }
     }
-    co_return error;
   }
 
   export template<typename InputStream, typename AppRouter>
@@ -45,6 +74,10 @@ namespace ttwhy::io
 
       if (error)
       {
+        if (error == asio::error::interrupted)
+        {
+          continue;
+        }
         co_return;
       }