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
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
|
// **********************************************************************
//
// Copyright (c) 2003-2006 ZeroC, Inc. All rights reserved.
//
// This copy of Ice is licensed to you under the terms described in the
// ICE_LICENSE file included in this distribution.
//
// **********************************************************************
#ifdef __sun
# define _POSIX_PTHREAD_SEMANTICS
#endif
#include <IceUtil/CtrlCHandler.h>
#include <IceUtil/StaticMutex.h>
#ifndef _WIN32
# include <signal.h>
#endif
using namespace std;
using namespace IceUtil;
static CtrlCHandlerCallback _callback = 0;
static const CtrlCHandler* _handler = 0;
CtrlCHandlerException::CtrlCHandlerException(const char* file, int line) :
Exception(file, line)
{
}
static const char* ctrlCHandlerName = "IceUtil::CtrlCHandlerException";
string
CtrlCHandlerException::ice_name() const
{
return ctrlCHandlerName;
}
Exception*
CtrlCHandlerException::ice_clone() const
{
return new CtrlCHandlerException(*this);
}
void
CtrlCHandlerException::ice_throw() const
{
throw *this;
}
void
CtrlCHandler::setCallback(CtrlCHandlerCallback callback)
{
StaticMutex::Lock lock(globalMutex);
_callback = callback;
}
CtrlCHandlerCallback
CtrlCHandler::getCallback() const
{
StaticMutex::Lock lock(globalMutex);
return _callback;
}
#ifdef _WIN32
static BOOL WINAPI handlerRoutine(DWORD dwCtrlType)
{
CtrlCHandlerCallback callback = _handler->getCallback();
if(callback != 0)
{
callback(dwCtrlType);
}
return TRUE;
}
CtrlCHandler::CtrlCHandler(CtrlCHandlerCallback callback)
{
StaticMutex::Lock lock(globalMutex);
if(_handler != 0)
{
throw CtrlCHandlerException(__FILE__, __LINE__);
}
else
{
_callback = callback;
_handler = this;
lock.release();
SetConsoleCtrlHandler(handlerRoutine, TRUE);
}
}
CtrlCHandler::~CtrlCHandler()
{
SetConsoleCtrlHandler(handlerRoutine, FALSE);
{
StaticMutex::Lock lock(globalMutex);
_handler = 0;
}
}
#else
extern "C"
{
static void*
sigwaitThread(void*)
{
sigset_t ctrlCLikeSignals;
sigemptyset(&ctrlCLikeSignals);
sigaddset(&ctrlCLikeSignals, SIGHUP);
sigaddset(&ctrlCLikeSignals, SIGINT);
sigaddset(&ctrlCLikeSignals, SIGTERM);
//
// Run until I'm cancelled (in sigwait())
//
for(;;)
{
int signal = 0;
int rc = sigwait(&ctrlCLikeSignals, &signal);
#if defined(__APPLE__)
//
// WORKAROUND: sigwait is not a cancelation point on MacOS X. To cancel this thread, the
// destructor cancels the thread and send a signal to the thread to unblock sigwait, then
// we explicitly test for cancellation.
//
pthread_testcancel();
#endif
//
// Some sigwait() implementations incorrectly return EINTR
// when interrupted by an unblocked caught signal
//
if(rc == EINTR)
{
continue;
}
assert(rc == 0);
rc = pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, 0);
assert(rc == 0);
CtrlCHandlerCallback callback = _handler->getCallback();
if(callback != 0)
{
callback(signal);
}
rc = pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, 0);
assert(rc == 0);
}
return 0;
}
}
static pthread_t _tid;
CtrlCHandler::CtrlCHandler(CtrlCHandlerCallback callback)
{
StaticMutex::Lock lock(globalMutex);
if(_handler != 0)
{
throw CtrlCHandlerException(__FILE__, __LINE__);
}
else
{
_callback = callback;
_handler = this;
lock.release();
// We block these CTRL+C like signals in the main thread,
// and by default all other threads will inherit this signal
// disposition.
sigset_t ctrlCLikeSignals;
sigemptyset(&ctrlCLikeSignals);
sigaddset(&ctrlCLikeSignals, SIGHUP);
sigaddset(&ctrlCLikeSignals, SIGINT);
sigaddset(&ctrlCLikeSignals, SIGTERM);
int rc = pthread_sigmask(SIG_BLOCK, &ctrlCLikeSignals, 0);
assert(rc == 0);
// Joinable thread
rc = pthread_create(&_tid, 0, sigwaitThread, 0);
assert(rc == 0);
}
}
CtrlCHandler::~CtrlCHandler()
{
int rc = pthread_cancel(_tid);
assert(rc == 0);
#if defined(__APPLE__)
//
// WORKAROUND: sigwait isn't a cancellation point on MacOS X, see
// comment in sigwaitThread
//
rc = pthread_kill(_tid, SIGTERM);
//assert(rc == 0); For some reaosns, this assert is sometime triggered
#endif
void* status = 0;
rc = pthread_join(_tid, &status);
assert(rc == 0);
#if !defined(__APPLE__)
assert(status == PTHREAD_CANCELED);
#endif
{
StaticMutex::Lock lock(globalMutex);
_handler = 0;
}
}
#endif
|
