1 /*
2 * Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
3 * Copyright (c) 1995-1999 by Internet Software Consortium
4 *
5 * Permission to use, copy, modify, and distribute this software for any
6 * purpose with or without fee is hereby granted, provided that the above
7 * copyright notice and this permission notice appear in all copies.
8 *
9 * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
10 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR
12 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
15 * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16 */
17
18 /* ev_connects.c - implement asynch connect/accept for the eventlib
19 * vix 16sep96 [initial]
20 */
21
22 #if !defined(LINT) && !defined(CODECENTER)
23 static const char rcsid[] = "$Id: ev_connects.c,v 1.8 2006/03/09 23:57:56 marka Exp $";
24 #endif
25
26 /* Import. */
27
28 #include "port_before.h"
29 #include "fd_setsize.h"
30
31 #include <sys/types.h>
32 #include <sys/socket.h>
33 #include <sys/ioctl.h>
34
35 #include <unistd.h>
36
37 #include <isc/eventlib.h>
38 #include <isc/assertions.h>
39 #include "eventlib_p.h"
40
41 #include "port_after.h"
42
43 /* Macros. */
44
45 #define GETXXXNAME(f, s, sa, len) ( \
46 (f((s), (&sa), (&len)) >= 0) ? 0 : \
47 (errno != EAFNOSUPPORT && errno != EOPNOTSUPP) ? -1 : ( \
48 memset(&(sa), 0, sizeof (sa)), \
49 (len) = sizeof (sa), \
50 (sa).sa_family = AF_UNIX, \
51 0 \
52 ) \
53 )
54
55 /* Forward. */
56
57 static void listener(evContext ctx, void *uap, int fd, int evmask);
58 static void connector(evContext ctx, void *uap, int fd, int evmask);
59
60 /* Public. */
61
62 int
evListen(evContext opaqueCtx,int fd,int maxconn,evConnFunc func,void * uap,evConnID * id)63 evListen(evContext opaqueCtx, int fd, int maxconn,
64 evConnFunc func, void *uap, evConnID *id)
65 {
66 evContext_p *ctx = opaqueCtx.opaque;
67 evConn *new;
68 int mode;
69
70 OKNEW(new);
71 new->flags = EV_CONN_LISTEN;
72 OKFREE(mode = fcntl(fd, F_GETFL, NULL), new); /*%< side effect: validate fd. */
73 /*
74 * Remember the nonblocking status. We assume that either evSelectFD
75 * has not been done to this fd, or that if it has then the caller
76 * will evCancelConn before they evDeselectFD. If our assumptions
77 * are not met, then we might restore the old nonblocking status
78 * incorrectly.
79 */
80 if ((mode & PORT_NONBLOCK) == 0) {
81 #ifdef USE_FIONBIO_IOCTL
82 int on = 1;
83 OKFREE(ioctl(fd, FIONBIO, (char *)&on), new);
84 #else
85 OKFREE(fcntl(fd, F_SETFL, mode | PORT_NONBLOCK), new);
86 #endif
87 new->flags |= EV_CONN_BLOCK;
88 }
89 OKFREE(listen(fd, maxconn), new);
90 if (evSelectFD(opaqueCtx, fd, EV_READ, listener, new, &new->file) < 0){
91 int save = errno;
92
93 FREE(new);
94 errno = save;
95 return (-1);
96 }
97 new->flags |= EV_CONN_SELECTED;
98 new->func = func;
99 new->uap = uap;
100 new->fd = fd;
101 if (ctx->conns != NULL)
102 ctx->conns->prev = new;
103 new->prev = NULL;
104 new->next = ctx->conns;
105 ctx->conns = new;
106 if (id)
107 id->opaque = new;
108 return (0);
109 }
110
111 int
evConnect(evContext opaqueCtx,int fd,const void * ra,int ralen,evConnFunc func,void * uap,evConnID * id)112 evConnect(evContext opaqueCtx, int fd, const void *ra, int ralen,
113 evConnFunc func, void *uap, evConnID *id)
114 {
115 evContext_p *ctx = opaqueCtx.opaque;
116 evConn *new;
117
118 OKNEW(new);
119 new->flags = 0;
120 /* Do the select() first to get the socket into nonblocking mode. */
121 if (evSelectFD(opaqueCtx, fd, EV_MASK_ALL,
122 connector, new, &new->file) < 0) {
123 int save = errno;
124
125 FREE(new);
126 errno = save;
127 return (-1);
128 }
129 new->flags |= EV_CONN_SELECTED;
130 if (connect(fd, ra, ralen) < 0 &&
131 errno != EWOULDBLOCK &&
132 errno != EAGAIN &&
133 errno != EINPROGRESS) {
134 int save = errno;
135
136 (void) evDeselectFD(opaqueCtx, new->file);
137 FREE(new);
138 errno = save;
139 return (-1);
140 }
141 /* No error, or EWOULDBLOCK. select() tells when it's ready. */
142 new->func = func;
143 new->uap = uap;
144 new->fd = fd;
145 if (ctx->conns != NULL)
146 ctx->conns->prev = new;
147 new->prev = NULL;
148 new->next = ctx->conns;
149 ctx->conns = new;
150 if (id)
151 id->opaque = new;
152 return (0);
153 }
154
155 int
evCancelConn(evContext opaqueCtx,evConnID id)156 evCancelConn(evContext opaqueCtx, evConnID id) {
157 evContext_p *ctx = opaqueCtx.opaque;
158 evConn *this = id.opaque;
159 evAccept *acc, *nxtacc;
160 int mode;
161
162 if ((this->flags & EV_CONN_SELECTED) != 0)
163 (void) evDeselectFD(opaqueCtx, this->file);
164 if ((this->flags & EV_CONN_BLOCK) != 0) {
165 mode = fcntl(this->fd, F_GETFL, NULL);
166 if (mode == -1) {
167 if (errno != EBADF)
168 return (-1);
169 } else {
170 #ifdef USE_FIONBIO_IOCTL
171 int off = 0;
172 OK(ioctl(this->fd, FIONBIO, (char *)&off));
173 #else
174 OK(fcntl(this->fd, F_SETFL, mode & ~PORT_NONBLOCK));
175 #endif
176 }
177 }
178
179 /* Unlink from ctx->conns. */
180 if (this->prev != NULL)
181 this->prev->next = this->next;
182 else
183 ctx->conns = this->next;
184 if (this->next != NULL)
185 this->next->prev = this->prev;
186
187 /*
188 * Remove `this' from the ctx->accepts list (zero or more times).
189 */
190 for (acc = HEAD(ctx->accepts), nxtacc = NULL;
191 acc != NULL;
192 acc = nxtacc)
193 {
194 nxtacc = NEXT(acc, link);
195 if (acc->conn == this) {
196 UNLINK(ctx->accepts, acc, link);
197 close(acc->fd);
198 FREE(acc);
199 }
200 }
201
202 /* Wrap up and get out. */
203 FREE(this);
204 return (0);
205 }
206
evHold(evContext opaqueCtx,evConnID id)207 int evHold(evContext opaqueCtx, evConnID id) {
208 evConn *this = id.opaque;
209
210 if ((this->flags & EV_CONN_LISTEN) == 0) {
211 errno = EINVAL;
212 return (-1);
213 }
214 if ((this->flags & EV_CONN_SELECTED) == 0)
215 return (0);
216 this->flags &= ~EV_CONN_SELECTED;
217 return (evDeselectFD(opaqueCtx, this->file));
218 }
219
evUnhold(evContext opaqueCtx,evConnID id)220 int evUnhold(evContext opaqueCtx, evConnID id) {
221 evConn *this = id.opaque;
222 int ret;
223
224 if ((this->flags & EV_CONN_LISTEN) == 0) {
225 errno = EINVAL;
226 return (-1);
227 }
228 if ((this->flags & EV_CONN_SELECTED) != 0)
229 return (0);
230 ret = evSelectFD(opaqueCtx, this->fd, EV_READ, listener, this,
231 &this->file);
232 if (ret == 0)
233 this->flags |= EV_CONN_SELECTED;
234 return (ret);
235 }
236
237 int
evTryAccept(evContext opaqueCtx,evConnID id,int * sys_errno)238 evTryAccept(evContext opaqueCtx, evConnID id, int *sys_errno) {
239 evContext_p *ctx = opaqueCtx.opaque;
240 evConn *conn = id.opaque;
241 evAccept *new;
242
243 if ((conn->flags & EV_CONN_LISTEN) == 0) {
244 errno = EINVAL;
245 return (-1);
246 }
247 OKNEW(new);
248 new->conn = conn;
249 new->ralen = sizeof new->ra;
250 new->fd = accept(conn->fd, &new->ra.sa, &new->ralen);
251 if (new->fd > ctx->highestFD) {
252 close(new->fd);
253 new->fd = -1;
254 new->ioErrno = ENOTSOCK;
255 }
256 if (new->fd >= 0) {
257 new->lalen = sizeof new->la;
258 if (GETXXXNAME(getsockname, new->fd, new->la.sa, new->lalen) < 0) {
259 new->ioErrno = errno;
260 (void) close(new->fd);
261 new->fd = -1;
262 } else
263 new->ioErrno = 0;
264 } else {
265 new->ioErrno = errno;
266 if (errno == EAGAIN || errno == EWOULDBLOCK) {
267 FREE(new);
268 return (-1);
269 }
270 }
271 INIT_LINK(new, link);
272 APPEND(ctx->accepts, new, link);
273 *sys_errno = new->ioErrno;
274 return (0);
275 }
276
277 /* Private. */
278
279 static void
listener(evContext opaqueCtx,void * uap,int fd,int evmask)280 listener(evContext opaqueCtx, void *uap, int fd, int evmask) {
281 evContext_p *ctx = opaqueCtx.opaque;
282 evConn *conn = uap;
283 union {
284 struct sockaddr sa;
285 struct sockaddr_in in;
286 #ifndef NO_SOCKADDR_UN
287 struct sockaddr_un un;
288 #endif
289 } la, ra;
290 int new;
291 ISC_SOCKLEN_T lalen = 0, ralen;
292
293 REQUIRE((evmask & EV_READ) != 0);
294 ralen = sizeof ra;
295 new = accept(fd, &ra.sa, &ralen);
296 if (new > ctx->highestFD) {
297 close(new);
298 new = -1;
299 errno = ENOTSOCK;
300 }
301 if (new >= 0) {
302 lalen = sizeof la;
303 if (GETXXXNAME(getsockname, new, la.sa, lalen) < 0) {
304 int save = errno;
305
306 (void) close(new);
307 errno = save;
308 new = -1;
309 }
310 } else if (errno == EAGAIN || errno == EWOULDBLOCK)
311 return;
312 (*conn->func)(opaqueCtx, conn->uap, new, &la.sa, lalen, &ra.sa, ralen);
313 }
314
315 static void
connector(evContext opaqueCtx,void * uap,int fd,int evmask)316 connector(evContext opaqueCtx, void *uap, int fd, int evmask) {
317 evConn *conn = uap;
318 union {
319 struct sockaddr sa;
320 struct sockaddr_in in;
321 #ifndef NO_SOCKADDR_UN
322 struct sockaddr_un un;
323 #endif
324 } la, ra;
325 ISC_SOCKLEN_T lalen, ralen;
326 #ifndef NETREAD_BROKEN
327 char buf[1];
328 #endif
329 void *conn_uap;
330 evConnFunc conn_func;
331 evConnID id;
332 int socket_errno = 0;
333 ISC_SOCKLEN_T optlen;
334
335 UNUSED(evmask);
336
337 lalen = sizeof la;
338 ralen = sizeof ra;
339 conn_uap = conn->uap;
340 conn_func = conn->func;
341 id.opaque = conn;
342 #ifdef SO_ERROR
343 optlen = sizeof socket_errno;
344 if (fd < 0 &&
345 getsockopt(conn->fd, SOL_SOCKET, SO_ERROR, (char *)&socket_errno,
346 &optlen) < 0)
347 socket_errno = errno;
348 else
349 errno = socket_errno;
350 #endif
351 if (evCancelConn(opaqueCtx, id) < 0 ||
352 socket_errno ||
353 #ifdef NETREAD_BROKEN
354 0 ||
355 #else
356 read(fd, buf, 0) < 0 ||
357 #endif
358 GETXXXNAME(getsockname, fd, la.sa, lalen) < 0 ||
359 GETXXXNAME(getpeername, fd, ra.sa, ralen) < 0) {
360 int save = errno;
361
362 (void) close(fd); /*%< XXX closing caller's fd */
363 errno = save;
364 fd = -1;
365 }
366 (*conn_func)(opaqueCtx, conn_uap, fd, &la.sa, lalen, &ra.sa, ralen);
367 }
368
369 /*! \file */
370