DOS窗口關閉事件

#include <iostream>
#include <windows.h>
#include <winuser.h>
using namespace std;

#define KeyDown(vk_code) ((GetAsyncKeyState(vk_code) & 0x8000) ? 1 : 0)
#define Msg(info) MessageBox(NULL,info,"message",MB_OK)
 
BOOL WINAPI DosWndProc(DWORD event)
{
	printf("event: %d\n", event);
	switch( event )
	{
		// 當按下關閉按鈕時
		case CTRL_CLOSE_EVENT:
			Msg("CTRL_CLOSE_EVENT");
			break;
		
		// 當系統關閉時
		case CTRL_SHUTDOWN_EVENT:
			Msg("CTRL_SHUTDOWN_EVENT");
			break;
 
		default:
			break;
	}
	return TRUE;
}
 
int main(int argc, char *argv[])
{
	if(SetConsoleCtrlHandler((PHANDLER_ROUTINE)DosWndProc,TRUE)==FALSE)
	{
		cout<<"error: unable to initalize message processor!"<<endl;
		return 1;
	}
 
 	while(1)
 	{
 		NULL;
 	}
 	
	return 0;
}