21.02

CPP/7zip/UI/Console/ConsoleClose.cpp

@@ -4,8 +4,13 @@
 
 #include "ConsoleClose.h"
 
-#if !defined(UNDER_CE) && defined(_WIN32)
+#ifndef UNDER_CE
+
+#ifdef _WIN32
 #include "../../../Common/MyWindows.h"
+#else
+#include <stdlib.h>
+#include <signal.h>
 #endif
 
 namespace NConsoleClose {
@@ -13,7 +18,8 @@ namespace NConsoleClose {
 unsigned g_BreakCounter = 0;
 static const unsigned kBreakAbortThreshold = 2;
 
-#if !defined(UNDER_CE) && defined(_WIN32)
+#ifdef _WIN32
+
 static BOOL WINAPI HandlerRoutine(DWORD ctrlType)
 {
   if (ctrlType == CTRL_LOGOFF_EVENT)
@@ -37,7 +43,49 @@ static BOOL WINAPI HandlerRoutine(DWORD ctrlType)
   return FALSE;
   */
 }
-#endif
+
+CCtrlHandlerSetter::CCtrlHandlerSetter()
+{
+  if (!SetConsoleCtrlHandler(HandlerRoutine, TRUE))
+    throw "SetConsoleCtrlHandler fails";
+}
+
+CCtrlHandlerSetter::~CCtrlHandlerSetter()
+{
+  if (!SetConsoleCtrlHandler(HandlerRoutine, FALSE))
+  {
+    // warning for throw in destructor.
+    // throw "SetConsoleCtrlHandler fails";
+  }
+}
+
+#else // _WIN32
+
+static void HandlerRoutine(int)
+{
+  g_BreakCounter++;
+  if (g_BreakCounter < kBreakAbortThreshold)
+    return;
+  exit(EXIT_FAILURE);
+}
+
+CCtrlHandlerSetter::CCtrlHandlerSetter()
+{
+  memo_sig_int = signal(SIGINT, HandlerRoutine); // CTRL-C
+  if (memo_sig_int == SIG_ERR)
+    throw "SetConsoleCtrlHandler fails (SIGINT)";
+  memo_sig_term = signal(SIGTERM, HandlerRoutine); // for kill -15 (before "kill -9")
+  if (memo_sig_term == SIG_ERR)
+    throw "SetConsoleCtrlHandler fails (SIGTERM)";
+}
+
+CCtrlHandlerSetter::~CCtrlHandlerSetter()
+{
+  signal(SIGINT, memo_sig_int); // CTRL-C
+  signal(SIGTERM, memo_sig_term); // kill {pid}
+}
+
+#endif // _WIN32
 
 /*
 void CheckCtrlBreak()
@@ -47,23 +95,6 @@ void CheckCtrlBreak()
 }
 */
 
-CCtrlHandlerSetter::CCtrlHandlerSetter()
-{
-  #if !defined(UNDER_CE) && defined(_WIN32)
-  if (!SetConsoleCtrlHandler(HandlerRoutine, TRUE))
-    throw "SetConsoleCtrlHandler fails";
-  #endif
 }
 
-CCtrlHandlerSetter::~CCtrlHandlerSetter()
-{
-  #if !defined(UNDER_CE) && defined(_WIN32)
-  if (!SetConsoleCtrlHandler(HandlerRoutine, FALSE))
-  {
-    // warning for throw in destructor.
-    // throw "SetConsoleCtrlHandler fails";
-  }
-  #endif
-}
-
-}
+#endif