ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/svn/ircd-hybrid-7.2/src/s_bsd.c
Revision: 451
Committed: Sun Feb 12 16:19:54 2006 UTC (18 years, 1 month ago) by michael
Content type: text/x-csrc
File size: 23195 byte(s)
Log Message:
- Properly close ssl clients

File Contents

# Content
1 /*
2 * ircd-hybrid: an advanced Internet Relay Chat Daemon(ircd).
3 * s_bsd.c: Network functions.
4 *
5 * Copyright (C) 2002 by the past and present ircd coders, and others.
6 *
7 * This program is free software; you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License as published by
9 * the Free Software Foundation; either version 2 of the License, or
10 * (at your option) any later version.
11 *
12 * This program is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 * GNU General Public License for more details.
16 *
17 * You should have received a copy of the GNU General Public License
18 * along with this program; if not, write to the Free Software
19 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
20 * USA
21 *
22 * $Id$
23 */
24
25 #include "stdinc.h"
26 #ifndef _WIN32
27 #include <netinet/in_systm.h>
28 #include <netinet/ip.h>
29 #include <netinet/tcp.h>
30 #endif
31 #include "fdlist.h"
32 #include "s_bsd.h"
33 #include "client.h"
34 #include "common.h"
35 #include "dbuf.h"
36 #include "event.h"
37 #include "irc_string.h"
38 #include "irc_getnameinfo.h"
39 #include "irc_getaddrinfo.h"
40 #include "ircd.h"
41 #include "list.h"
42 #include "listener.h"
43 #include "numeric.h"
44 #include "packet.h"
45 #include "irc_res.h"
46 #include "inet_misc.h"
47 #include "restart.h"
48 #include "s_auth.h"
49 #include "s_conf.h"
50 #include "s_log.h"
51 #include "s_serv.h"
52 #include "s_stats.h"
53 #include "send.h"
54 #include "memory.h"
55 #include "s_user.h"
56 #include "hook.h"
57
58 static const char *comm_err_str[] = { "Comm OK", "Error during bind()",
59 "Error during DNS lookup", "connect timeout", "Error during connect()",
60 "Comm Error" };
61
62 struct Callback *setup_socket_cb = NULL;
63
64 static void comm_connect_callback(fde_t *fd, int status);
65 static PF comm_connect_timeout;
66 static void comm_connect_dns_callback(void *vptr, struct DNSReply *reply);
67 static PF comm_connect_tryconnect;
68
69 extern void init_netio(void);
70
71 /* check_can_use_v6()
72 * Check if the system can open AF_INET6 sockets
73 */
74 void
75 check_can_use_v6(void)
76 {
77 #ifdef IPV6
78 int v6;
79
80 if ((v6 = socket(AF_INET6, SOCK_STREAM, 0)) < 0)
81 ServerInfo.can_use_v6 = 0;
82 else
83 {
84 ServerInfo.can_use_v6 = 1;
85 #ifdef _WIN32
86 closesocket(v6);
87 #else
88 close(v6);
89 #endif
90 }
91 #else
92 ServerInfo.can_use_v6 = 0;
93 #endif
94 }
95
96 /* get_sockerr - get the error value from the socket or the current errno
97 *
98 * Get the *real* error from the socket (well try to anyway..).
99 * This may only work when SO_DEBUG is enabled but its worth the
100 * gamble anyway.
101 */
102 int
103 get_sockerr(int fd)
104 {
105 #ifndef _WIN32
106 int errtmp = errno;
107 #else
108 int errtmp = WSAGetLastError();
109 #endif
110 #ifdef SO_ERROR
111 int err = 0;
112 socklen_t len = sizeof(err);
113
114 if (-1 < fd && !getsockopt(fd, SOL_SOCKET, SO_ERROR, (char*) &err, (socklen_t *)&len))
115 {
116 if (err)
117 errtmp = err;
118 }
119 errno = errtmp;
120 #endif
121 return errtmp;
122 }
123
124 /*
125 * report_error - report an error from an errno.
126 * Record error to log and also send a copy to all *LOCAL* opers online.
127 *
128 * text is a *format* string for outputing error. It must
129 * contain only two '%s', the first will be replaced
130 * by the sockhost from the client_p, and the latter will
131 * be taken from sys_errlist[errno].
132 *
133 * client_p if not NULL, is the *LOCAL* client associated with
134 * the error.
135 *
136 * Cannot use perror() within daemon. stderr is closed in
137 * ircd and cannot be used. And, worse yet, it might have
138 * been reassigned to a normal connection...
139 *
140 * Actually stderr is still there IFF ircd was run with -s --Rodder
141 */
142
143 void
144 report_error(int level, const char* text, const char* who, int error)
145 {
146 who = (who) ? who : "";
147
148 sendto_realops_flags(UMODE_DEBUG, level, text, who, strerror(error));
149 log_oper_action(LOG_IOERR_TYPE, NULL, "%s %s %s\n", who, text, strerror(error));
150 ilog(L_ERROR, text, who, strerror(error));
151 }
152
153 /*
154 * setup_socket()
155 *
156 * Set the socket non-blocking, and other wonderful bits.
157 */
158 static void *
159 setup_socket(va_list args)
160 {
161 int fd = va_arg(args, int);
162 int opt = 1;
163
164 setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, (char *) &opt, sizeof(opt));
165
166 #ifdef IPTOS_LOWDELAY
167 opt = IPTOS_LOWDELAY;
168 setsockopt(fd, IPPROTO_IP, IP_TOS, (char *) &opt, sizeof(opt));
169 #endif
170
171 #ifndef _WIN32
172 fcntl(fd, F_SETFL, fcntl(fd, F_GETFL, 0) | O_NONBLOCK);
173 #endif
174
175 return NULL;
176 }
177
178 /*
179 * init_comm()
180 *
181 * Initializes comm subsystem.
182 */
183 void
184 init_comm(void)
185 {
186 setup_socket_cb = register_callback("setup_socket", setup_socket);
187 init_netio();
188 }
189
190 /*
191 * close_connection
192 * Close the physical connection. This function must make
193 * MyConnect(client_p) == FALSE, and set client_p->from == NULL.
194 */
195 void
196 close_connection(struct Client *client_p)
197 {
198 struct ConfItem *conf;
199 struct AccessItem *aconf;
200 struct ClassItem *aclass;
201
202 assert(NULL != client_p);
203
204 if (IsServer(client_p))
205 {
206 ServerStats->is_sv++;
207 ServerStats->is_sbs += client_p->localClient->send.bytes;
208 ServerStats->is_sbr += client_p->localClient->recv.bytes;
209 ServerStats->is_sti += CurrentTime - client_p->firsttime;
210
211 /* XXX Does this even make any sense at all anymore?
212 * scheduling a 'quick' reconnect could cause a pile of
213 * nick collides under TSora protocol... -db
214 */
215 /*
216 * If the connection has been up for a long amount of time, schedule
217 * a 'quick' reconnect, else reset the next-connect cycle.
218 */
219 if ((conf = find_conf_exact(SERVER_TYPE,
220 client_p->name, client_p->username,
221 client_p->host)))
222 {
223 /*
224 * Reschedule a faster reconnect, if this was a automatically
225 * connected configuration entry. (Note that if we have had
226 * a rehash in between, the status has been changed to
227 * CONF_ILLEGAL). But only do this if it was a "good" link.
228 */
229 aconf = (struct AccessItem *)map_to_conf(conf);
230 aclass = (struct ClassItem *)map_to_conf(aconf->class_ptr);
231 aconf->hold = time(NULL);
232 aconf->hold += (aconf->hold - client_p->since > HANGONGOODLINK) ?
233 HANGONRETRYDELAY : ConFreq(aclass);
234 if (nextconnect > aconf->hold)
235 nextconnect = aconf->hold;
236 }
237 }
238 else if (IsClient(client_p))
239 {
240 ServerStats->is_cl++;
241 ServerStats->is_cbs += client_p->localClient->send.bytes;
242 ServerStats->is_cbr += client_p->localClient->recv.bytes;
243 ServerStats->is_cti += CurrentTime - client_p->firsttime;
244 }
245 else
246 ServerStats->is_ni++;
247
248 if (!IsDead(client_p))
249 {
250 /* attempt to flush any pending dbufs. Evil, but .. -- adrian */
251 /* there is still a chance that we might send data to this socket
252 * even if it is marked as blocked (COMM_SELECT_READ handler is called
253 * before COMM_SELECT_WRITE). Let's try, nothing to lose.. -adx
254 */
255 ClearSendqBlocked(client_p);
256 send_queued_write(client_p);
257 }
258
259 #ifdef HAVE_LIBCRYPTO
260 if (client_p->localClient->fd.ssl)
261 {
262 SSL_set_shutdown(client_p->localClient->fd.ssl, SSL_RECEIVED_SHUTDOWN);
263
264 if (!SSL_shutdown(client_p->localClient->fd.ssl))
265 SSL_shutdown(client_p->localClient->fd.ssl);
266 }
267 #endif
268 if (client_p->localClient->fd.flags.open)
269 fd_close(&client_p->localClient->fd);
270
271 if (HasServlink(client_p))
272 {
273 if (client_p->localClient->ctrlfd.flags.open)
274 fd_close(&client_p->localClient->ctrlfd);
275 }
276
277 dbuf_clear(&client_p->localClient->buf_sendq);
278 dbuf_clear(&client_p->localClient->buf_recvq);
279
280 MyFree(client_p->localClient->passwd);
281 detach_conf(client_p, CONF_TYPE);
282 client_p->from = NULL; /* ...this should catch them! >:) --msa */
283 }
284
285 #ifdef HAVE_LIBCRYPTO
286 /*
287 * ssl_handshake - let OpenSSL initialize the protocol. Register for
288 * read/write events if necessary.
289 */
290 static void
291 ssl_handshake(int fd, struct Client *client_p)
292 {
293 int ret = SSL_accept(client_p->localClient->fd.ssl);
294
295 if (ret <= 0)
296 switch (SSL_get_error(client_p->localClient->fd.ssl, ret))
297 {
298 case SSL_ERROR_WANT_WRITE:
299 comm_setselect(&client_p->localClient->fd, COMM_SELECT_WRITE,
300 (PF *) ssl_handshake, client_p, 0);
301 return;
302
303 case SSL_ERROR_WANT_READ:
304 comm_setselect(&client_p->localClient->fd, COMM_SELECT_READ,
305 (PF *) ssl_handshake, client_p, 0);
306 return;
307
308 default:
309 exit_client(client_p, client_p, "Error during SSL handshake");
310 return;
311 }
312
313 execute_callback(auth_cb, client_p);
314 }
315 #endif
316
317 /*
318 * add_connection - creates a client which has just connected to us on
319 * the given fd. The sockhost field is initialized with the ip# of the host.
320 * An unique id is calculated now, in case it is needed for auth.
321 * The client is sent to the auth module for verification, and not put in
322 * any client list yet.
323 */
324 void
325 add_connection(struct Listener* listener, int fd)
326 {
327 struct Client *new_client;
328 socklen_t len = sizeof(struct irc_ssaddr);
329 struct irc_ssaddr irn;
330 assert(NULL != listener);
331
332 /*
333 * get the client socket name from the socket
334 * the client has already been checked out in accept_connection
335 */
336
337 memset(&irn, 0, sizeof(irn));
338 if (getpeername(fd, (struct sockaddr *)&irn, (socklen_t *)&len))
339 {
340 #ifdef _WIN32
341 errno = WSAGetLastError();
342 #endif
343 report_error(L_ALL, "Failed in adding new connection %s :%s",
344 get_listener_name(listener), errno);
345 ServerStats->is_ref++;
346 #ifdef _WIN32
347 closesocket(fd);
348 #else
349 close(fd);
350 #endif
351 return;
352 }
353
354 #ifdef IPV6
355 remove_ipv6_mapping(&irn);
356 #else
357 irn.ss_len = len;
358 #endif
359 new_client = make_client(NULL);
360 fd_open(&new_client->localClient->fd, fd, 1,
361 (listener->flags & LISTENER_SSL) ?
362 "Incoming SSL connection" : "Incoming connection");
363 memset(&new_client->localClient->ip, 0, sizeof(struct irc_ssaddr));
364
365 /*
366 * copy address to 'sockhost' as a string, copy it to host too
367 * so we have something valid to put into error messages...
368 */
369 memcpy(&new_client->localClient->ip, &irn, sizeof(struct irc_ssaddr));
370
371 irc_getnameinfo((struct sockaddr*)&new_client->localClient->ip,
372 new_client->localClient->ip.ss_len, new_client->sockhost,
373 HOSTIPLEN, NULL, 0, NI_NUMERICHOST);
374 new_client->localClient->aftype = new_client->localClient->ip.ss.ss_family;
375
376 *new_client->host = '\0';
377 #ifdef IPV6
378 if (*new_client->sockhost == ':')
379 strlcat(new_client->host, "0", HOSTLEN+1);
380
381 if (new_client->localClient->aftype == AF_INET6 &&
382 ConfigFileEntry.dot_in_ip6_addr == 1)
383 {
384 strlcat(new_client->host, new_client->sockhost,HOSTLEN+1);
385 strlcat(new_client->host, ".", HOSTLEN+1);
386 } else
387 #endif
388 strlcat(new_client->host, new_client->sockhost,HOSTLEN+1);
389
390 new_client->localClient->listener = listener;
391 ++listener->ref_count;
392
393 connect_id++;
394 new_client->connect_id = connect_id;
395
396 #ifdef HAVE_LIBCRYPTO
397 if ((listener->flags & LISTENER_SSL))
398 {
399 if ((new_client->localClient->fd.ssl = SSL_new(ServerInfo.ctx)) == NULL)
400 {
401 ilog(L_CRIT, "SSL_new() ERROR! -- %s",
402 ERR_error_string(ERR_get_error(), NULL));
403
404 SetDead(new_client);
405 exit_client(new_client, new_client, "SSL_new failed");
406 return;
407 }
408
409 SSL_set_fd(new_client->localClient->fd.ssl, fd);
410 ssl_handshake(0, new_client);
411 }
412 else
413 #endif
414 execute_callback(auth_cb, new_client);
415 }
416
417 /*
418 * stolen from squid - its a neat (but overused! :) routine which we
419 * can use to see whether we can ignore this errno or not. It is
420 * generally useful for non-blocking network IO related errnos.
421 * -- adrian
422 */
423 int
424 ignoreErrno(int ierrno)
425 {
426 switch (ierrno)
427 {
428 case EINPROGRESS:
429 case EWOULDBLOCK:
430 #if EAGAIN != EWOULDBLOCK
431 case EAGAIN:
432 #endif
433 case EALREADY:
434 case EINTR:
435 #ifdef ERESTART
436 case ERESTART:
437 #endif
438 return 1;
439 default:
440 return 0;
441 }
442 }
443
444 /*
445 * comm_settimeout() - set the socket timeout
446 *
447 * Set the timeout for the fd
448 */
449 void
450 comm_settimeout(fde_t *fd, time_t timeout, PF *callback, void *cbdata)
451 {
452 assert(fd->flags.open);
453
454 fd->timeout = CurrentTime + (timeout / 1000);
455 fd->timeout_handler = callback;
456 fd->timeout_data = cbdata;
457 }
458
459 /*
460 * comm_setflush() - set a flush function
461 *
462 * A flush function is simply a function called if found during
463 * comm_timeouts(). Its basically a second timeout, except in this case
464 * I'm too lazy to implement multiple timeout functions! :-)
465 * its kinda nice to have it separate, since this is designed for
466 * flush functions, and when comm_close() is implemented correctly
467 * with close functions, we _actually_ don't call comm_close() here ..
468 * -- originally Adrian's notes
469 * comm_close() is replaced with fd_close() in fdlist.c
470 */
471 void
472 comm_setflush(fde_t *fd, time_t timeout, PF *callback, void *cbdata)
473 {
474 assert(fd->flags.open);
475
476 fd->flush_timeout = CurrentTime + (timeout / 1000);
477 fd->flush_handler = callback;
478 fd->flush_data = cbdata;
479 }
480
481 /*
482 * comm_checktimeouts() - check the socket timeouts
483 *
484 * All this routine does is call the given callback/cbdata, without closing
485 * down the file descriptor. When close handlers have been implemented,
486 * this will happen.
487 */
488 void
489 comm_checktimeouts(void *notused)
490 {
491 int i;
492 fde_t *F;
493 PF *hdl;
494 void *data;
495
496 for (i = 0; i < FD_HASH_SIZE; i++)
497 for (F = fd_hash[i]; F != NULL; F = fd_next_in_loop)
498 {
499 assert(F->flags.open);
500 fd_next_in_loop = F->hnext;
501
502 /* check flush functions */
503 if (F->flush_handler && F->flush_timeout > 0 &&
504 F->flush_timeout < CurrentTime)
505 {
506 hdl = F->flush_handler;
507 data = F->flush_data;
508 comm_setflush(F, 0, NULL, NULL);
509 hdl(F, data);
510 }
511
512 /* check timeouts */
513 if (F->timeout_handler && F->timeout > 0 &&
514 F->timeout < CurrentTime)
515 {
516 /* Call timeout handler */
517 hdl = F->timeout_handler;
518 data = F->timeout_data;
519 comm_settimeout(F, 0, NULL, NULL);
520 hdl(F, data);
521 }
522 }
523 }
524
525 /*
526 * void comm_connect_tcp(int fd, const char *host, unsigned short port,
527 * struct sockaddr *clocal, int socklen,
528 * CNCB *callback, void *data, int aftype, int timeout)
529 * Input: An fd to connect with, a host and port to connect to,
530 * a local sockaddr to connect from + length(or NULL to use the
531 * default), a callback, the data to pass into the callback, the
532 * address family.
533 * Output: None.
534 * Side-effects: A non-blocking connection to the host is started, and
535 * if necessary, set up for selection. The callback given
536 * may be called now, or it may be called later.
537 */
538 void
539 comm_connect_tcp(fde_t *fd, const char *host, unsigned short port,
540 struct sockaddr *clocal, int socklen, CNCB *callback,
541 void *data, int aftype, int timeout)
542 {
543 struct addrinfo hints, *res;
544 char portname[PORTNAMELEN+1];
545
546 assert(callback);
547 fd->connect.callback = callback;
548 fd->connect.data = data;
549
550 fd->connect.hostaddr.ss.ss_family = aftype;
551 fd->connect.hostaddr.ss_port = htons(port);
552
553 /* Note that we're using a passed sockaddr here. This is because
554 * generally you'll be bind()ing to a sockaddr grabbed from
555 * getsockname(), so this makes things easier.
556 * XXX If NULL is passed as local, we should later on bind() to the
557 * virtual host IP, for completeness.
558 * -- adrian
559 */
560 if ((clocal != NULL) && (bind(fd->fd, clocal, socklen) < 0))
561 {
562 /* Failure, call the callback with COMM_ERR_BIND */
563 comm_connect_callback(fd, COMM_ERR_BIND);
564 /* ... and quit */
565 return;
566 }
567
568 /* Next, if we have been given an IP, get the addr and skip the
569 * DNS check (and head direct to comm_connect_tryconnect().
570 */
571 memset(&hints, 0, sizeof(hints));
572 hints.ai_family = AF_UNSPEC;
573 hints.ai_socktype = SOCK_STREAM;
574 hints.ai_flags = AI_PASSIVE | AI_NUMERICHOST;
575
576 snprintf(portname, PORTNAMELEN, "%d", port);
577
578 if (irc_getaddrinfo(host, portname, &hints, &res))
579 {
580 /* Send the DNS request, for the next level */
581 fd->dns_query = MyMalloc(sizeof(struct DNSQuery));
582 fd->dns_query->ptr = fd;
583 fd->dns_query->callback = comm_connect_dns_callback;
584 gethost_byname(host, fd->dns_query);
585 }
586 else
587 {
588 /* We have a valid IP, so we just call tryconnect */
589 /* Make sure we actually set the timeout here .. */
590 assert(res != NULL);
591 memcpy(&fd->connect.hostaddr, res->ai_addr, res->ai_addrlen);
592 fd->connect.hostaddr.ss_len = res->ai_addrlen;
593 fd->connect.hostaddr.ss.ss_family = res->ai_family;
594 irc_freeaddrinfo(res);
595 comm_settimeout(fd, timeout*1000, comm_connect_timeout, NULL);
596 comm_connect_tryconnect(fd, NULL);
597 }
598 }
599
600 /*
601 * comm_connect_callback() - call the callback, and continue with life
602 */
603 static void
604 comm_connect_callback(fde_t *fd, int status)
605 {
606 CNCB *hdl;
607
608 /* This check is gross..but probably necessary */
609 if (fd->connect.callback == NULL)
610 return;
611
612 /* Clear the connect flag + handler */
613 hdl = fd->connect.callback;
614 fd->connect.callback = NULL;
615
616 /* Clear the timeout handler */
617 comm_settimeout(fd, 0, NULL, NULL);
618
619 /* Call the handler */
620 hdl(fd, status, fd->connect.data);
621 }
622
623 /*
624 * comm_connect_timeout() - this gets called when the socket connection
625 * times out. This *only* can be called once connect() is initially
626 * called ..
627 */
628 static void
629 comm_connect_timeout(fde_t *fd, void *notused)
630 {
631 /* error! */
632 comm_connect_callback(fd, COMM_ERR_TIMEOUT);
633 }
634
635 /*
636 * comm_connect_dns_callback() - called at the completion of the DNS request
637 *
638 * The DNS request has completed, so if we've got an error, return it,
639 * otherwise we initiate the connect()
640 */
641 static void
642 comm_connect_dns_callback(void *vptr, struct DNSReply *reply)
643 {
644 fde_t *F = vptr;
645
646 if (reply == NULL)
647 {
648 MyFree(F->dns_query);
649 F->dns_query = NULL;
650 comm_connect_callback(F, COMM_ERR_DNS);
651 return;
652 }
653
654 /* No error, set a 10 second timeout */
655 comm_settimeout(F, 30*1000, comm_connect_timeout, NULL);
656
657 /* Copy over the DNS reply info so we can use it in the connect() */
658 /*
659 * Note we don't fudge the refcount here, because we aren't keeping
660 * the DNS record around, and the DNS cache is gone anyway..
661 * -- adrian
662 */
663 memcpy(&F->connect.hostaddr, &reply->addr, reply->addr.ss_len);
664 /* The cast is hacky, but safe - port offset is same on v4 and v6 */
665 ((struct sockaddr_in *) &F->connect.hostaddr)->sin_port =
666 F->connect.hostaddr.ss_port;
667 F->connect.hostaddr.ss_len = reply->addr.ss_len;
668
669 /* Now, call the tryconnect() routine to try a connect() */
670 MyFree(F->dns_query);
671 F->dns_query = NULL;
672 comm_connect_tryconnect(F, NULL);
673 }
674
675 /* static void comm_connect_tryconnect(int fd, void *notused)
676 * Input: The fd, the handler data(unused).
677 * Output: None.
678 * Side-effects: Try and connect with pending connect data for the FD. If
679 * we succeed or get a fatal error, call the callback.
680 * Otherwise, it is still blocking or something, so register
681 * to select for a write event on this FD.
682 */
683 static void
684 comm_connect_tryconnect(fde_t *fd, void *notused)
685 {
686 int retval;
687
688 /* This check is needed or re-entrant s_bsd_* like sigio break it. */
689 if (fd->connect.callback == NULL)
690 return;
691
692 /* Try the connect() */
693 retval = connect(fd->fd, (struct sockaddr *) &fd->connect.hostaddr,
694 fd->connect.hostaddr.ss_len);
695
696 /* Error? */
697 if (retval < 0)
698 {
699 #ifdef _WIN32
700 errno = WSAGetLastError();
701 #endif
702 /*
703 * If we get EISCONN, then we've already connect()ed the socket,
704 * which is a good thing.
705 * -- adrian
706 */
707 if (errno == EISCONN)
708 comm_connect_callback(fd, COMM_OK);
709 else if (ignoreErrno(errno))
710 /* Ignore error? Reschedule */
711 comm_setselect(fd, COMM_SELECT_WRITE, comm_connect_tryconnect,
712 NULL, 0);
713 else
714 /* Error? Fail with COMM_ERR_CONNECT */
715 comm_connect_callback(fd, COMM_ERR_CONNECT);
716 return;
717 }
718
719 /* If we get here, we've suceeded, so call with COMM_OK */
720 comm_connect_callback(fd, COMM_OK);
721 }
722
723 /*
724 * comm_errorstr() - return an error string for the given error condition
725 */
726 const char *
727 comm_errstr(int error)
728 {
729 if (error < 0 || error >= COMM_ERR_MAX)
730 return "Invalid error number!";
731 return comm_err_str[error];
732 }
733
734 /*
735 * comm_open() - open a socket
736 *
737 * This is a highly highly cut down version of squid's comm_open() which
738 * for the most part emulates socket(), *EXCEPT* it fails if we're about
739 * to run out of file descriptors.
740 */
741 int
742 comm_open(fde_t *F, int family, int sock_type, int proto, const char *note)
743 {
744 int fd;
745
746 /* First, make sure we aren't going to run out of file descriptors */
747 if (number_fd >= hard_fdlimit)
748 {
749 errno = ENFILE;
750 return -1;
751 }
752
753 /*
754 * Next, we try to open the socket. We *should* drop the reserved FD
755 * limit if/when we get an error, but we can deal with that later.
756 * XXX !!! -- adrian
757 */
758 fd = socket(family, sock_type, proto);
759 if (fd < 0)
760 {
761 #ifdef _WIN32
762 errno = WSAGetLastError();
763 #endif
764 return -1; /* errno will be passed through, yay.. */
765 }
766
767 execute_callback(setup_socket_cb, fd);
768
769 /* update things in our fd tracking */
770 fd_open(F, fd, 1, note);
771 return 0;
772 }
773
774 /*
775 * comm_accept() - accept an incoming connection
776 *
777 * This is a simple wrapper for accept() which enforces FD limits like
778 * comm_open() does. Returned fd must be either closed or tagged with
779 * fd_open (this function no longer does it).
780 */
781 int
782 comm_accept(struct Listener *lptr, struct irc_ssaddr *pn)
783 {
784 int newfd;
785 socklen_t addrlen = sizeof(struct irc_ssaddr);
786
787 if (number_fd >= hard_fdlimit)
788 {
789 errno = ENFILE;
790 return -1;
791 }
792
793 /*
794 * Next, do the accept(). if we get an error, we should drop the
795 * reserved fd limit, but we can deal with that when comm_open()
796 * also does it. XXX -- adrian
797 */
798 newfd = accept(lptr->fd.fd, (struct sockaddr *)pn, (socklen_t *)&addrlen);
799 if (newfd < 0)
800 {
801 #ifdef _WIN32
802 errno = WSAGetLastError();
803 #endif
804 return -1;
805 }
806
807 #ifdef IPV6
808 remove_ipv6_mapping(pn);
809 #else
810 pn->ss_len = addrlen;
811 #endif
812
813 execute_callback(setup_socket_cb, newfd);
814
815 /* .. and return */
816 return newfd;
817 }
818
819 /*
820 * remove_ipv6_mapping() - Removes IPv4-In-IPv6 mapping from an address
821 * This function should really inspect the struct itself rather than relying
822 * on inet_pton and inet_ntop. OSes with IPv6 mapping listening on both
823 * AF_INET and AF_INET6 map AF_INET connections inside AF_INET6 structures
824 *
825 */
826 #ifdef IPV6
827 void
828 remove_ipv6_mapping(struct irc_ssaddr *addr)
829 {
830 if (addr->ss.ss_family == AF_INET6)
831 {
832 struct sockaddr_in6 *v6;
833
834 v6 = (struct sockaddr_in6*)addr;
835 if (IN6_IS_ADDR_V4MAPPED(&v6->sin6_addr))
836 {
837 char v4ip[HOSTIPLEN];
838 struct sockaddr_in *v4 = (struct sockaddr_in*)addr;
839 inetntop(AF_INET6, &v6->sin6_addr, v4ip, HOSTIPLEN);
840 inet_pton(AF_INET, v4ip, &v4->sin_addr);
841 addr->ss.ss_family = AF_INET;
842 addr->ss_len = sizeof(struct sockaddr_in);
843 }
844 else
845 addr->ss_len = sizeof(struct sockaddr_in6);
846 }
847 else
848 addr->ss_len = sizeof(struct sockaddr_in);
849 }
850 #endif

Properties

Name Value
svn:eol-style native
svn:keywords Id Revision