Enregistrement d’une fonction du gestionnaire de contrôle
Exemple de gestionnaire de contrôle de base
Il s'agit d'un exemple de la fonction SetConsoleCtrlHandler qui est utilisée pour installer un gestionnaire de contrôle.
Lorsqu’un signal Ctrl+C est reçu, le gestionnaire de contrôle renvoie VRAI, indiquant qu’il a géré le signal. Cela empêche l'appel d'autres gestionnaires de contrôle.
Lorsqu’un signal CTRL_CLOSE_EVENT est reçu, le gestionnaire de contrôle renvoie VRAI et le processus se termine.
Lorsqu'un signal CTRL_BREAK_EVENT, CTRL_LOGOFF_EVENT ou CTRL_SHUTDOWN_EVENT est reçu, le gestionnaire de contrôle renvoie FAUX. Cela entraîne la transmission du signal à la fonction de gestionnaire de contrôle suivante. Si aucun autre gestionnaire de contrôle n'a été enregistré ou si aucun des gestionnaires enregistrés ne renvoie VRAI, le gestionnaire par défaut sera utilisé, ce qui entraînera l'arrêt du processus.
Remarque
Le fait d’appeler AttachConsole, AllocConsole ou FreeConsole rétablit l’état initial de la table des gestionnaires de contrôles dans le processus client. Vous devez réinscrire les gestionnaires quand la session de console attachée change.
// CtrlHandler.cpp : This file contains the 'main' function. Program execution begins and ends there.
//
#include <windows.h>
#include <stdio.h>
BOOL WINAPI CtrlHandler(DWORD fdwCtrlType)
{
switch (fdwCtrlType)
{
// Handle the CTRL-C signal.
case CTRL_C_EVENT:
printf("Ctrl-C event\n\n");
Beep(750, 300);
return TRUE;
// CTRL-CLOSE: confirm that the user wants to exit.
case CTRL_CLOSE_EVENT:
Beep(600, 200);
printf("Ctrl-Close event\n\n");
return TRUE;
// Pass other signals to the next handler.
case CTRL_BREAK_EVENT:
Beep(900, 200);
printf("Ctrl-Break event\n\n");
return FALSE;
case CTRL_LOGOFF_EVENT:
Beep(1000, 200);
printf("Ctrl-Logoff event\n\n");
return FALSE;
case CTRL_SHUTDOWN_EVENT:
Beep(750, 500);
printf("Ctrl-Shutdown event\n\n");
return FALSE;
default:
return FALSE;
}
}
int main(void)
{
if (SetConsoleCtrlHandler(CtrlHandler, TRUE))
{
printf("\nThe Control Handler is installed.\n");
printf("\n -- Now try pressing Ctrl+C or Ctrl+Break, or");
printf("\n try logging off or closing the console...\n");
printf("\n(...waiting in a loop for events...)\n\n");
while (1) {}
}
else
{
printf("\nERROR: Could not set control handler");
return 1;
}
return 0;
}
Exemple d'écoute avec fenêtre masquée
Selon les remarques, si les bibliothèques gdi32.dll ou user32.dll sont chargées, la fonction SetConsoleCtrlHandler n'est pas appelée pour les événements CTRL_LOGOFF_EVENT et CTRL_SHUTDOWN_EVENT. La solution de contournement spécifiée consiste à créer une fenêtre masquée, si aucune fenêtre n’existe déjà, en appelant la méthode CreateWindowEx avec le paramètre dwExStyle défini sur 0 et à écouter les messages de fenêtre WM_QUERYENDSESSION et WM_ENDSESSION. Si une fenêtre existe déjà, ajoutez les deux messages à la procédure de fenêtre existante.
Pour plus d’informations sur la configuration d’une fenêtre et de sa boucle de messagerie, consultez la section Création d’une fenêtre.
// CtrlHandler.cpp : This file contains the 'main' function. Program execution begins and ends there.
//
#include <Windows.h>
#include <stdio.h>
BOOL WINAPI CtrlHandler(DWORD fdwCtrlType)
{
switch (fdwCtrlType)
{
// Handle the CTRL-C signal.
case CTRL_C_EVENT:
printf("Ctrl-C event\n\n");
Beep(750, 300);
return TRUE;
// CTRL-CLOSE: confirm that the user wants to exit.
case CTRL_CLOSE_EVENT:
Beep(600, 200);
printf("Ctrl-Close event\n\n");
return TRUE;
// Pass other signals to the next handler.
case CTRL_BREAK_EVENT:
Beep(900, 200);
printf("Ctrl-Break event\n\n");
return FALSE;
case CTRL_LOGOFF_EVENT:
Beep(1000, 200);
printf("Ctrl-Logoff event\n\n");
return FALSE;
case CTRL_SHUTDOWN_EVENT:
Beep(750, 500);
printf("Ctrl-Shutdown event\n\n");
return FALSE;
default:
return FALSE;
}
}
LRESULT CALLBACK WindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
switch (uMsg)
{
case WM_QUERYENDSESSION:
{
// Check `lParam` for which system shutdown function and handle events.
// See https://learn.microsoft.com/windows/win32/shutdown/wm-queryendsession
return TRUE; // Respect user's intent and allow shutdown.
}
case WM_ENDSESSION:
{
// Check `lParam` for which system shutdown function and handle events.
// See https://learn.microsoft.com/windows/win32/shutdown/wm-endsession
return 0; // We have handled this message.
}
default:
return DefWindowProc(hwnd, uMsg, wParam, lParam);
}
}
int main(void)
{
WNDCLASS sampleClass{ 0 };
sampleClass.lpszClassName = TEXT("CtrlHandlerSampleClass");
sampleClass.lpfnWndProc = WindowProc;
if (!RegisterClass(&sampleClass))
{
printf("\nERROR: Could not register window class");
return 2;
}
HWND hwnd = CreateWindowEx(
0,
sampleClass.lpszClassName,
TEXT("Console Control Handler Sample"),
0,
CW_USEDEFAULT,
CW_USEDEFAULT,
CW_USEDEFAULT,
CW_USEDEFAULT,
nullptr,
nullptr,
nullptr,
nullptr
);
if (!hwnd)
{
printf("\nERROR: Could not create window");
return 3;
}
ShowWindow(hwnd, SW_HIDE);
if (SetConsoleCtrlHandler(CtrlHandler, TRUE))
{
printf("\nThe Control Handler is installed.\n");
printf("\n -- Now try pressing Ctrl+C or Ctrl+Break, or");
printf("\n try logging off or closing the console...\n");
printf("\n(...waiting in a loop for events...)\n\n");
// Pump message loop for the window we created.
MSG msg{};
while (GetMessage(&msg, nullptr, 0, 0) > 0)
{
TranslateMessage(&msg);
DispatchMessage(&msg);
}
return 0;
}
else
{
printf("\nERROR: Could not set control handler");
return 1;
}
}