0
|
1 /*
|
|
2 * Copyright 2000-2003 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 <assert.h>
|
|
26 #include "serverLists.hpp"
|
|
27
|
|
28 //----------------------------------------------------------------------
|
|
29 // Lists
|
|
30 //
|
|
31
|
|
32 CRITICAL_SECTION Lists::crit;
|
|
33
|
|
34 void
|
|
35 Lists::init() {
|
|
36 InitializeCriticalSection(&crit);
|
|
37 }
|
|
38
|
|
39 void
|
|
40 Lists::lock() {
|
|
41 EnterCriticalSection(&crit);
|
|
42 }
|
|
43
|
|
44 void
|
|
45 Lists::unlock() {
|
|
46 LeaveCriticalSection(&crit);
|
|
47 }
|
|
48
|
|
49 //----------------------------------------------------------------------
|
|
50 // ListsLocker
|
|
51 //
|
|
52
|
|
53 ListsLocker::ListsLocker() {
|
|
54 Lists::lock();
|
|
55 }
|
|
56
|
|
57 ListsLocker::~ListsLocker() {
|
|
58 Lists::unlock();
|
|
59 }
|
|
60
|
|
61 //----------------------------------------------------------------------
|
|
62 // ChildInfo
|
|
63 //
|
|
64
|
|
65 ChildInfo::ChildInfo(DWORD pid, HANDLE childProcessHandle,
|
|
66 HANDLE writeToStdinHandle, HANDLE readFromStdoutHandle,
|
|
67 HANDLE auxHandle1, HANDLE auxHandle2) {
|
|
68 this->pid = pid;
|
|
69 this->childProcessHandle = childProcessHandle;
|
|
70 this->writeToStdinHandle = writeToStdinHandle;
|
|
71 this->readFromStdoutHandle = readFromStdoutHandle;
|
|
72 this->auxHandle1 = auxHandle1;
|
|
73 this->auxHandle2 = auxHandle2;
|
|
74 client = NULL;
|
|
75 }
|
|
76
|
|
77 DWORD
|
|
78 ChildInfo::getPid() {
|
|
79 return pid;
|
|
80 }
|
|
81
|
|
82 HANDLE
|
|
83 ChildInfo::getChildProcessHandle() {
|
|
84 return childProcessHandle;
|
|
85 }
|
|
86
|
|
87 HANDLE
|
|
88 ChildInfo::getWriteToStdinHandle() {
|
|
89 return writeToStdinHandle;
|
|
90 }
|
|
91
|
|
92 HANDLE
|
|
93 ChildInfo::getReadFromStdoutHandle() {
|
|
94 return readFromStdoutHandle;
|
|
95 }
|
|
96
|
|
97 void
|
|
98 ChildInfo::setClient(ClientInfo* clientInfo) {
|
|
99 client = clientInfo;
|
|
100 }
|
|
101
|
|
102 ClientInfo*
|
|
103 ChildInfo::getClient() {
|
|
104 return client;
|
|
105 }
|
|
106
|
|
107 void
|
|
108 ChildInfo::closeAll() {
|
|
109 CloseHandle(childProcessHandle);
|
|
110 CloseHandle(writeToStdinHandle);
|
|
111 CloseHandle(readFromStdoutHandle);
|
|
112 CloseHandle(auxHandle1);
|
|
113 CloseHandle(auxHandle2);
|
|
114 }
|
|
115
|
|
116 //----------------------------------------------------------------------
|
|
117 // ChildList
|
|
118 //
|
|
119
|
|
120 ChildList::ChildList() {
|
|
121 }
|
|
122
|
|
123 ChildList::~ChildList() {
|
|
124 }
|
|
125
|
|
126 void
|
|
127 ChildList::addChild(ChildInfo* info) {
|
|
128 // Could store these in binary sorted order by pid for efficiency
|
|
129 childList.push_back(info);
|
|
130 }
|
|
131
|
|
132 ChildInfo*
|
|
133 ChildList::removeChild(HANDLE childProcessHandle) {
|
|
134 for (ChildInfoList::iterator iter = childList.begin(); iter != childList.end();
|
|
135 iter++) {
|
|
136 ChildInfo* info = *iter;
|
|
137 if (info->getChildProcessHandle() == childProcessHandle) {
|
|
138 childList.erase(iter);
|
|
139 return info;
|
|
140 }
|
|
141 }
|
|
142 assert(false);
|
|
143 return NULL;
|
|
144 }
|
|
145
|
|
146 void
|
|
147 ChildList::removeChild(ChildInfo* info) {
|
|
148 for (ChildInfoList::iterator iter = childList.begin(); iter != childList.end();
|
|
149 iter++) {
|
|
150 if (*iter == info) {
|
|
151 childList.erase(iter);
|
|
152 return;
|
|
153 }
|
|
154 }
|
|
155 assert(false);
|
|
156 }
|
|
157
|
|
158 ChildInfo*
|
|
159 ChildList::getChildByPid(DWORD pid) {
|
|
160 for (ChildInfoList::iterator iter = childList.begin(); iter != childList.end();
|
|
161 iter++) {
|
|
162 ChildInfo* info = *iter;
|
|
163 if (info->getPid() == pid) {
|
|
164 return info;
|
|
165 }
|
|
166 }
|
|
167 return NULL;
|
|
168 }
|
|
169
|
|
170 int
|
|
171 ChildList::size() {
|
|
172 return childList.size();
|
|
173 }
|
|
174
|
|
175 ChildInfo*
|
|
176 ChildList::getChildByIndex(int index) {
|
|
177 return childList[index];
|
|
178 }
|
|
179
|
|
180 //----------------------------------------------------------------------
|
|
181 // ClientInfo
|
|
182 //
|
|
183
|
|
184 ClientInfo::ClientInfo(SOCKET dataSocket) {
|
|
185 this->dataSocket = dataSocket;
|
|
186 buf = new IOBuf(32768, 131072);
|
|
187 buf->setSocket(dataSocket);
|
|
188 target = NULL;
|
|
189 }
|
|
190
|
|
191 ClientInfo::~ClientInfo() {
|
|
192 delete buf;
|
|
193 }
|
|
194
|
|
195 SOCKET
|
|
196 ClientInfo::getDataSocket() {
|
|
197 return dataSocket;
|
|
198 }
|
|
199
|
|
200 IOBuf*
|
|
201 ClientInfo::getIOBuf() {
|
|
202 return buf;
|
|
203 }
|
|
204
|
|
205 void
|
|
206 ClientInfo::setTarget(ChildInfo* childInfo) {
|
|
207 target = childInfo;
|
|
208 }
|
|
209
|
|
210 ChildInfo*
|
|
211 ClientInfo::getTarget() {
|
|
212 return target;
|
|
213 }
|
|
214
|
|
215 void
|
|
216 ClientInfo::closeAll() {
|
|
217 shutdown(dataSocket, SD_BOTH);
|
|
218 closesocket(dataSocket);
|
|
219 dataSocket = INVALID_SOCKET;
|
|
220 }
|
|
221
|
|
222 //----------------------------------------------------------------------
|
|
223 // ClientList
|
|
224 //
|
|
225
|
|
226 ClientList::ClientList() {
|
|
227 }
|
|
228
|
|
229 ClientList::~ClientList() {
|
|
230 }
|
|
231
|
|
232 void
|
|
233 ClientList::addClient(ClientInfo* info) {
|
|
234 clientList.push_back(info);
|
|
235 }
|
|
236
|
|
237 bool
|
|
238 ClientList::isAnyDataSocketSet(fd_set* fds, ClientInfo** out) {
|
|
239 for (ClientInfoList::iterator iter = clientList.begin(); iter != clientList.end();
|
|
240 iter++) {
|
|
241 ClientInfo* info = *iter;
|
|
242 if (FD_ISSET(info->getDataSocket(), fds)) {
|
|
243 *out = info;
|
|
244 return true;
|
|
245 }
|
|
246 }
|
|
247 return false;
|
|
248 }
|
|
249
|
|
250 void
|
|
251 ClientList::removeClient(ClientInfo* client) {
|
|
252 for (ClientInfoList::iterator iter = clientList.begin(); iter != clientList.end();
|
|
253 iter++) {
|
|
254 if (*iter == client) {
|
|
255 clientList.erase(iter);
|
|
256 return;
|
|
257 }
|
|
258 }
|
|
259 assert(false);
|
|
260 }
|
|
261
|
|
262 int
|
|
263 ClientList::size() {
|
|
264 return clientList.size();
|
|
265 }
|
|
266
|
|
267 ClientInfo*
|
|
268 ClientList::get(int num) {
|
|
269 return clientList[num];
|
|
270 }
|