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