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

File Contents

# User Rev Content
1 adx 30 /*
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 knight 31 * $Id$
23 adx 30 */
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 michael 451 {
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 adx 30 #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