blob: 67c9f56368e1f4756a2a70f6dfe4baac38b360e6 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
//
// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
//
/*=============================================================================
**
** Source: test10.c
**
** Dependencies: PAL_Initialize
** PAL_Terminate
** GenerateConsoleCtrlEvent
**
** Purpose:
**
** Test to ensure proper operation of the SetConsoleCtrlHandler()
** API by checking whether a console control handler function is
** actually removed by the API when it returns success for that
** operation.
**
**
**===========================================================================*/
#include <palsuite.h>
/* global test value */
static BOOL g_bFlag = FALSE;
/* handler function */
static BOOL PALAPI CtrlHandler( DWORD CtrlType )
{
if( CtrlType == CTRL_BREAK_EVENT )
{
g_bFlag = TRUE;
return TRUE;
}
return FALSE;
}
/* main entry point function */
int __cdecl main( int argc, char **argv )
{
/* local variables */
BOOL ret = PASS;
/* PAL initialization */
if( (PAL_Initialize(argc, argv)) != 0 )
{
return( FAIL );
}
/* set the console control handler function */
if( ! SetConsoleCtrlHandler( CtrlHandler, TRUE ) )
{
ret = FAIL;
Trace( "ERROR:%lu:SetConsoleCtrlHandler() failed to add "
"CtrlHandler\n",
GetLastError() );
Fail( "Test failed\n" );
}
/* test that the right control handler functions are set */
if( ! GenerateConsoleCtrlEvent( CTRL_BREAK_EVENT, 0 ) )
{
Trace( "ERROR:%lu:GenerateConsoleCtrlEvent() failed\n",
GetLastError() );
ret = FAIL;
goto done;
}
/* give the handlers a chance to execute */
Sleep( 2000 );
/* check the results */
if( ! g_bFlag )
{
Trace( "ERROR:CtrlHandler() was not called but should have been\n" );
ret = FAIL;
}
done:
/* unset the control handle that was set */
if( ! SetConsoleCtrlHandler( CtrlHandler, FALSE ) )
{
ret = FAIL;
Trace( "ERROR:%lu:SetConsoleCtrlHandler() failed to "
"remove CtrlHandler\n",
GetLastError() );
Fail( "Test failed\n" );
}
/* PAL termination */
PAL_TerminateEx(ret);
/* return our result */
return ret;
}
|