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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
|
//
// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
//
/*=============================================================================
**
** Source: test7.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>
static BOOL g_bFlag1 = FALSE;
static BOOL g_bFlag2 = FALSE;
/* first handler function */
static BOOL PALAPI CtrlHandler1( DWORD CtrlType )
{
if( CtrlType == CTRL_C_EVENT )
{
g_bFlag1 = TRUE;
return TRUE;
}
return FALSE;
}
/* second handler function */
static BOOL PALAPI CtrlHandler2( DWORD CtrlType )
{
if( CtrlType == CTRL_C_EVENT )
{
g_bFlag2 = TRUE;
return FALSE;
}
return FALSE;
}
/* main entry point function */
int __cdecl main( int argc, char **argv )
{
/* local variables */
BOOL ret = PASS;
BOOL bSetHandler1 = FALSE;
/* PAL initialization */
if( (PAL_Initialize(argc, argv)) != 0 )
{
return( FAIL );
}
/* set the console control handler functions */
if( SetConsoleCtrlHandler( CtrlHandler1, TRUE ) )
{
bSetHandler1 = TRUE;
}
else
{
ret = FAIL;
Trace( "ERROR:%lu:SetConsoleCtrlHandler() failed to add "
"CtrlHandler1\n",
GetLastError() );
Fail( "Test failed\n" );
}
if( ! SetConsoleCtrlHandler( CtrlHandler2, TRUE ) )
{
ret = FAIL;
Trace( "ERROR:%lu:SetConsoleCtrlHandler() failed to add "
"CtrlHandler2\n",
GetLastError() );
Fail( "Test failed\n" );
}
/* test that the right control handler functions are set */
if( ! GenerateConsoleCtrlEvent( CTRL_C_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_bFlag1 )
{
Trace( "ERROR:CtrlHandler1() was not called but should have been\n" );
ret = FAIL;
goto done;
}
if( ! g_bFlag2 )
{
Trace( "ERROR:CtrlHandler2() was not called but should have been\n" );
ret = FAIL;
goto done;
}
/* reset our flags */
g_bFlag1 = FALSE;
g_bFlag2 = FALSE;
/* try to unset CtrlHandler2 */
if( ! SetConsoleCtrlHandler( CtrlHandler2, FALSE ) )
{
ret = FAIL;
Trace( "ERROR:%lu:SetConsoleCtrlHandler() failed to "
"remove CtrlHandler2\n",
GetLastError() );
goto done;
}
/* make sure that CtrlHandler1 is set and CtrlHandler2 isn't */
if( ! GenerateConsoleCtrlEvent( CTRL_C_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_bFlag2 )
{
Trace( "ERROR:CtrlHandler2() was called after it was unset\n" );
ret = FAIL;
goto done;
}
if( ! g_bFlag1 )
{
Trace( "ERROR:CtrlHandler1() was not called but should have been\n" );
ret = FAIL;
goto done;
}
done:
/* unset CtrlHandler1 if it was set */
if( bSetHandler1 )
{
if( ! SetConsoleCtrlHandler( CtrlHandler1, FALSE ) )
{
ret = FAIL;
Trace( "ERROR:%lu:SetConsoleCtrlHandler() failed to "
"remove CtrlHandler1\n",
GetLastError() );
Fail( "Test failed\n" );
}
}
/* PAL termination */
PAL_TerminateEx(ret);
/* return our result */
return ret;
}
|
