1 // ConsoleClose.cpp

     2 

     3 #include "StdAfx.h"

     4 

     5 #include "ConsoleClose.h"

     6 

     7 static int g_BreakCounter = 0;

     8 static const int kBreakAbortThreshold = 2;

     9 

    10 namespace NConsoleClose {

    11 

    12 static BOOL WINAPI HandlerRoutine(DWORD ctrlType)

    13 {

    14   if (ctrlType == CTRL_LOGOFF_EVENT)

    15   {

    16     // printf("\nCTRL_LOGOFF_EVENT\n");

    17     return TRUE;

    18   }

    19 

    20   g_BreakCounter++;

    21   if (g_BreakCounter < kBreakAbortThreshold)

    22     return TRUE;

    23   return FALSE;

    24   /*

    25   switch(ctrlType)

    26   {

    27     case CTRL_C_EVENT:

    28     case CTRL_BREAK_EVENT:

    29       if (g_BreakCounter < kBreakAbortThreshold)

    30       return TRUE;

    31   }

    32   return FALSE;

    33   */

    34 }

    35 

    36 bool TestBreakSignal()

    37 {

    38   /*

    39   if (g_BreakCounter > 0)

    40     return true;

    41   */

    42   return (g_BreakCounter > 0);

    43 }

    44 

    45 void CheckCtrlBreak()

    46 {

    47   if (TestBreakSignal())

    48     throw CCtrlBreakException();

    49 }

    50 

    51 CCtrlHandlerSetter::CCtrlHandlerSetter()

    52 {

    53   if(!SetConsoleCtrlHandler(HandlerRoutine, TRUE))

    54     throw "SetConsoleCtrlHandler fails";

    55 }

    56 

    57 CCtrlHandlerSetter::~CCtrlHandlerSetter()

    58 {

    59   if(!SetConsoleCtrlHandler(HandlerRoutine, FALSE))

    60     throw "SetConsoleCtrlHandler fails";

    61 }

    62 

    63 }