Mercurial > hg > truffle

comparison agent/src/os/win32/Reaper.cpp @ 0:a61af66fc99e jdk7-b24

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Initial load
author duke
date Sat, 01 Dec 2007 00:00:00 +0000
parents
children c18cbe5936b8
comparison
equal deleted inserted replaced
-1:000000000000 0:a61af66fc99e
1 /*
2 * Copyright 2000 Sun Microsystems, Inc. All Rights Reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
20 * CA 95054 USA or visit www.sun.com if you need additional information or
21 * have any questions.
22 *
23 */
24
25 #include <iostream>
26 #include "Reaper.hpp"
27
28 using namespace std;
29
30 Reaper::Reaper(ReaperCB* cb) {
31 InitializeCriticalSection(&crit);
32 event = CreateEvent(NULL, TRUE, FALSE, NULL);
33 this->cb = cb;
34
35 active = false;
36 shouldShutDown = false;
37 }
38
39 bool
40 Reaper::start() {
41 bool result = false;
42
43 EnterCriticalSection(&crit);
44
45 if (!active) {
46 DWORD id;
47 HANDLE reaper = CreateThread(NULL, 0, &Reaper::reaperThreadEntry,
48 this, 0, &id);
49 if (reaper != NULL) {
50 result = true;
51 }
52 }
53
54 LeaveCriticalSection(&crit);
55
56 return result;
57 }
58
59 bool
60 Reaper::stop() {
61 bool result = false;
62
63 EnterCriticalSection(&crit);
64
65 if (active) {
66 shouldShutDown = true;
67 SetEvent(event);
68 while (active) {
69 Sleep(1);
70 }
71 shouldShutDown = false;
72 result = true;
73 }
74
75 LeaveCriticalSection(&crit);
76
77 return result;
78 }
79
80 void
81 Reaper::registerProcess(HANDLE processHandle, void* userData) {
82 ProcessInfo info;
83
84 info.handle = processHandle;
85 info.userData = userData;
86
87 EnterCriticalSection(&crit);
88
89 procInfo.push_back(info);
90 SetEvent(event);
91
92 LeaveCriticalSection(&crit);
93 }
94
95 void
96 Reaper::reaperThread() {
97 while (!shouldShutDown) {
98 // Take atomic snapshot of the current process list and user data
99 EnterCriticalSection(&crit);
100
101 int num = procInfo.size();
102 HANDLE* handleList = new HANDLE[1 + num];
103 void** dataList = new void*[num];
104 for (int i = 0; i < num; i++) {
105 handleList[i] = procInfo[i].handle;
106 dataList[i] = procInfo[i].userData;
107 }
108
109 LeaveCriticalSection(&crit);
110
111 // Topmost handle becomes the event object, so other threads can
112 // signal this one to notice differences in the above list (or
113 // shut down)
114 handleList[num] = event;
115
116 // Wait for these objects
117 DWORD idx = WaitForMultipleObjects(1 + num, handleList,
118 FALSE, INFINITE);
119 if ((idx >= WAIT_OBJECT_0) && (idx <= WAIT_OBJECT_0 + num)) {
120 idx -= WAIT_OBJECT_0;
121 if (idx < num) {
122 // A process exited (i.e., it wasn't that we were woken up
123 // just because the event went off)
124 (*cb)(dataList[idx]);
125 // Remove this process from the list (NOTE: requires that
126 // ordering does not change, i.e., that all additions are to
127 // the back of the process list)
128 EnterCriticalSection(&crit);
129
130 std::vector<ProcessInfo>::iterator iter = procInfo.begin();
131 iter += idx;
132 procInfo.erase(iter);
133
134 LeaveCriticalSection(&crit);
135 } else {
136 // Notification from other thread
137 ResetEvent(event);
138 }
139 } else {
140 // Unexpected return value. For now, warn.
141 cerr << "Reaper::reaperThread(): unexpected return value "
142 << idx << " from WaitForMultipleObjects" << endl;
143 }
144
145 // Clean up these lists
146 delete[] handleList;
147 delete[] dataList;
148 }
149
150 // Time to shut down
151 active = false;
152 }
153
154 DWORD WINAPI
155 Reaper::reaperThreadEntry(LPVOID data) {
156 Reaper* reaper = (Reaper*) data;
157 reaper->reaperThread();
158 return 0;
159 }