ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/svn/ircd-hybrid/trunk/src/s_bsd.c
Revision: 2228
Committed: Thu Jun 13 19:46:30 2013 UTC (13 years, 2 months ago) by michael
Content type: text/x-csrc
File size: 23949 byte(s)
Log Message:
- Implement certificate fingerprint validation for oper{} and connect{} blocks.
  Some code taken from oftc-hybrid. Hello, stu!

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     #include <netinet/in_systm.h>
27     #include <netinet/ip.h>
28     #include <netinet/tcp.h>
29 michael 1011 #include "list.h"
30 adx 30 #include "fdlist.h"
31     #include "s_bsd.h"
32     #include "client.h"
33     #include "dbuf.h"
34     #include "event.h"
35     #include "irc_string.h"
36     #include "ircd.h"
37     #include "listener.h"
38     #include "numeric.h"
39     #include "packet.h"
40     #include "irc_res.h"
41     #include "restart.h"
42     #include "s_auth.h"
43 michael 1309 #include "conf.h"
44     #include "log.h"
45 adx 30 #include "s_serv.h"
46     #include "send.h"
47     #include "memory.h"
48     #include "s_user.h"
49     #include "hook.h"
50    
51     static const char *comm_err_str[] = { "Comm OK", "Error during bind()",
52     "Error during DNS lookup", "connect timeout", "Error during connect()",
53     "Comm Error" };
54    
55     struct Callback *setup_socket_cb = NULL;
56    
57 michael 1013 static void comm_connect_callback(fde_t *, int);
58 adx 30 static PF comm_connect_timeout;
59 michael 992 static void comm_connect_dns_callback(void *, const struct irc_ssaddr *, const char *);
60 adx 30 static PF comm_connect_tryconnect;
61    
62    
63     /* check_can_use_v6()
64     * Check if the system can open AF_INET6 sockets
65     */
66     void
67     check_can_use_v6(void)
68     {
69     #ifdef IPV6
70     int v6;
71    
72     if ((v6 = socket(AF_INET6, SOCK_STREAM, 0)) < 0)
73     ServerInfo.can_use_v6 = 0;
74     else
75     {
76     ServerInfo.can_use_v6 = 1;
77     close(v6);
78     }
79     #else
80     ServerInfo.can_use_v6 = 0;
81     #endif
82     }
83    
84     /* get_sockerr - get the error value from the socket or the current errno
85     *
86     * Get the *real* error from the socket (well try to anyway..).
87     * This may only work when SO_DEBUG is enabled but its worth the
88     * gamble anyway.
89     */
90     int
91     get_sockerr(int fd)
92     {
93     int errtmp = errno;
94     #ifdef SO_ERROR
95     int err = 0;
96     socklen_t len = sizeof(err);
97    
98 michael 967 if (-1 < fd && !getsockopt(fd, SOL_SOCKET, SO_ERROR, &err, &len))
99 adx 30 {
100     if (err)
101     errtmp = err;
102     }
103     errno = errtmp;
104     #endif
105     return errtmp;
106     }
107    
108     /*
109     * report_error - report an error from an errno.
110     * Record error to log and also send a copy to all *LOCAL* opers online.
111     *
112     * text is a *format* string for outputing error. It must
113     * contain only two '%s', the first will be replaced
114     * by the sockhost from the client_p, and the latter will
115     * be taken from sys_errlist[errno].
116     *
117     * client_p if not NULL, is the *LOCAL* client associated with
118     * the error.
119     *
120     * Cannot use perror() within daemon. stderr is closed in
121     * ircd and cannot be used. And, worse yet, it might have
122     * been reassigned to a normal connection...
123     *
124     * Actually stderr is still there IFF ircd was run with -s --Rodder
125     */
126    
127     void
128     report_error(int level, const char* text, const char* who, int error)
129     {
130     who = (who) ? who : "";
131    
132 michael 1618 sendto_realops_flags(UMODE_DEBUG, level, SEND_NOTICE,
133     text, who, strerror(error));
134 michael 1247 ilog(LOG_TYPE_IRCD, text, who, strerror(error));
135 adx 30 }
136    
137     /*
138     * setup_socket()
139     *
140     * Set the socket non-blocking, and other wonderful bits.
141     */
142     static void *
143     setup_socket(va_list args)
144     {
145     int fd = va_arg(args, int);
146     int opt = 1;
147    
148 michael 967 setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, sizeof(opt));
149 adx 30
150     #ifdef IPTOS_LOWDELAY
151     opt = IPTOS_LOWDELAY;
152 michael 967 setsockopt(fd, IPPROTO_IP, IP_TOS, &opt, sizeof(opt));
153 adx 30 #endif
154    
155     fcntl(fd, F_SETFL, fcntl(fd, F_GETFL, 0) | O_NONBLOCK);
156    
157     return NULL;
158     }
159    
160     /*
161     * init_comm()
162     *
163     * Initializes comm subsystem.
164     */
165     void
166     init_comm(void)
167     {
168     setup_socket_cb = register_callback("setup_socket", setup_socket);
169     init_netio();
170     }
171    
172     /*
173     * close_connection
174     * Close the physical connection. This function must make
175     * MyConnect(client_p) == FALSE, and set client_p->from == NULL.
176     */
177     void
178     close_connection(struct Client *client_p)
179     {
180 michael 1391 dlink_node *ptr = NULL;
181 adx 30
182 michael 1391 assert(client_p);
183 adx 30
184 michael 683 if (!IsDead(client_p))
185     {
186     /* attempt to flush any pending dbufs. Evil, but .. -- adrian */
187     /* there is still a chance that we might send data to this socket
188     * even if it is marked as blocked (COMM_SELECT_READ handler is called
189     * before COMM_SELECT_WRITE). Let's try, nothing to lose.. -adx
190     */
191     ClearSendqBlocked(client_p);
192     send_queued_write(client_p);
193     }
194    
195 michael 1143 if (IsClient(client_p))
196 adx 30 {
197 michael 1143 ++ServerStats.is_cl;
198     ServerStats.is_cbs += client_p->localClient->send.bytes;
199     ServerStats.is_cbr += client_p->localClient->recv.bytes;
200 michael 1241 ServerStats.is_cti += CurrentTime - client_p->localClient->firsttime;
201 michael 1143 }
202     else if (IsServer(client_p))
203     {
204 michael 896 ++ServerStats.is_sv;
205     ServerStats.is_sbs += client_p->localClient->send.bytes;
206     ServerStats.is_sbr += client_p->localClient->recv.bytes;
207 michael 1241 ServerStats.is_sti += CurrentTime - client_p->localClient->firsttime;
208 adx 30
209 michael 1391 DLINK_FOREACH(ptr, server_items.head)
210 adx 30 {
211 michael 1632 struct MaskItem *conf = ptr->data;
212 michael 1391
213     if (irccmp(conf->name, client_p->name))
214     continue;
215    
216 adx 30 /*
217 michael 1391 * Reset next-connect cycle of all connect{} blocks that match
218     * this servername.
219 adx 30 */
220 michael 1649 conf->until = CurrentTime + conf->class->con_freq;
221 adx 30 }
222     }
223     else
224 michael 896 ++ServerStats.is_ni;
225 adx 30
226     #ifdef HAVE_LIBCRYPTO
227     if (client_p->localClient->fd.ssl)
228 michael 451 {
229     SSL_set_shutdown(client_p->localClient->fd.ssl, SSL_RECEIVED_SHUTDOWN);
230    
231     if (!SSL_shutdown(client_p->localClient->fd.ssl))
232     SSL_shutdown(client_p->localClient->fd.ssl);
233     }
234 adx 30 #endif
235     if (client_p->localClient->fd.flags.open)
236     fd_close(&client_p->localClient->fd);
237    
238     dbuf_clear(&client_p->localClient->buf_sendq);
239     dbuf_clear(&client_p->localClient->buf_recvq);
240    
241     MyFree(client_p->localClient->passwd);
242 michael 1632 detach_conf(client_p, CONF_CLIENT|CONF_OPER|CONF_SERVER);
243 adx 30 client_p->from = NULL; /* ...this should catch them! >:) --msa */
244     }
245    
246 michael 2228 /* Base16 encoding is:
247     * Copyright (c) 2001-2004, Roger Dingledine
248     * Copyright (c) 2004-2007, Roger Dingledine, Nick Mathewson
249     *
250     * Redistribution and use in source and binary forms, with or without
251     * modification, are permitted provided that the following conditions are
252     * met:
253     *
254     * Redistributions of source code must retain the above copyright
255     * notice, this list of conditions and the following disclaimer.
256    
257     * Redistributions in binary form must reproduce the above copyright notice,
258     * this list of conditions and the following disclaimer
259     * in the documentation and/or other materials provided with the distribution.
260     *
261     * Neither the names of the copyright owners nor the names of its
262     * contributors may be used to endorse or promote products derived from
263     * this software without specific prior written permission.
264    
265     * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
266     * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
267     * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
268     * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
269     * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
270     * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
271     * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
272     * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
273     * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
274     * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
275     * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
276     */
277     static void
278     base16_encode(char *dest, size_t destlen, const char *src, size_t srclen)
279     {
280     const char *end;
281     char *cp;
282    
283     assert(destlen >= srclen * 2 + 1);
284    
285     cp = dest;
286     end = src + srclen;
287    
288     while (src < end)
289     {
290     *cp++ = "0123456789ABCDEF"[(*(const uint8_t *)src) >> 4];
291     *cp++ = "0123456789ABCDEF"[(*(const uint8_t *)src) & 0xf];
292     ++src;
293     }
294    
295     *cp = '\0';
296     }
297    
298 adx 30 #ifdef HAVE_LIBCRYPTO
299     /*
300     * ssl_handshake - let OpenSSL initialize the protocol. Register for
301     * read/write events if necessary.
302     */
303     static void
304     ssl_handshake(int fd, struct Client *client_p)
305     {
306 michael 2228 X509 *cert = NULL;
307 adx 30 int ret = SSL_accept(client_p->localClient->fd.ssl);
308 michael 2228 int err = SSL_get_error(client_p->localClient->fd.ssl, ret);
309 adx 30
310 michael 2228 ilog(LOG_TYPE_IRCD, "SSL Error %d %s", err, ERR_error_string(err, NULL));
311    
312     if ((cert = SSL_get_peer_certificate(client_p->localClient->fd.ssl)))
313     {
314     int res = SSL_get_verify_result(client_p->localClient->fd.ssl);
315    
316     if (res == X509_V_OK || res == X509_V_ERR_SELF_SIGNED_CERT_IN_CHAIN ||
317     res == X509_V_ERR_UNABLE_TO_VERIFY_LEAF_SIGNATURE ||
318     res == X509_V_ERR_DEPTH_ZERO_SELF_SIGNED_CERT)
319     base16_encode(client_p->certfp, sizeof(client_p->certfp),
320     (const char *)cert->sha1_hash, sizeof(cert->sha1_hash));
321     else
322     ilog(LOG_TYPE_IRCD, "Client %s!%s@%s gave bad SSL client certificate: %d",
323     client_p->name, client_p->username, client_p->host, res);
324     X509_free(cert);
325     }
326    
327 adx 30 if (ret <= 0)
328 michael 2228 {
329 adx 30 switch (SSL_get_error(client_p->localClient->fd.ssl, ret))
330     {
331     case SSL_ERROR_WANT_WRITE:
332     comm_setselect(&client_p->localClient->fd, COMM_SELECT_WRITE,
333     (PF *) ssl_handshake, client_p, 0);
334     return;
335    
336     case SSL_ERROR_WANT_READ:
337     comm_setselect(&client_p->localClient->fd, COMM_SELECT_READ,
338     (PF *) ssl_handshake, client_p, 0);
339     return;
340    
341     default:
342     exit_client(client_p, client_p, "Error during SSL handshake");
343     return;
344     }
345 michael 2228 }
346 adx 30
347 michael 2181 start_auth(client_p);
348 adx 30 }
349     #endif
350    
351     /*
352     * add_connection - creates a client which has just connected to us on
353     * the given fd. The sockhost field is initialized with the ip# of the host.
354     * An unique id is calculated now, in case it is needed for auth.
355     * The client is sent to the auth module for verification, and not put in
356     * any client list yet.
357     */
358     void
359 michael 549 add_connection(struct Listener *listener, struct irc_ssaddr *irn, int fd)
360 adx 30 {
361 michael 1123 struct Client *new_client = make_client(NULL);
362 michael 549
363 adx 30 fd_open(&new_client->localClient->fd, fd, 1,
364     (listener->flags & LISTENER_SSL) ?
365     "Incoming SSL connection" : "Incoming connection");
366    
367 michael 1123 /*
368 adx 30 * copy address to 'sockhost' as a string, copy it to host too
369     * so we have something valid to put into error messages...
370     */
371 michael 549 memcpy(&new_client->localClient->ip, irn, sizeof(struct irc_ssaddr));
372 adx 30
373 michael 1123 getnameinfo((struct sockaddr *)&new_client->localClient->ip,
374     new_client->localClient->ip.ss_len, new_client->sockhost,
375     sizeof(new_client->sockhost), NULL, 0, NI_NUMERICHOST);
376 adx 30 new_client->localClient->aftype = new_client->localClient->ip.ss.ss_family;
377 michael 1072
378 michael 1858 #ifdef HAVE_LIBGEOIP
379     /* XXX IPV6 SUPPORT XXX */
380     if (irn->ss.ss_family == AF_INET && geoip_ctx)
381     {
382     const struct sockaddr_in *v4 = (const struct sockaddr_in *)&new_client->localClient->ip;
383     new_client->localClient->country_id = GeoIP_id_by_ipnum(geoip_ctx, (unsigned long)ntohl(v4->sin_addr.s_addr));
384     }
385     #endif
386    
387 michael 1123 if (new_client->sockhost[0] == ':' && new_client->sockhost[1] == ':')
388 adx 30 {
389 michael 1072 strlcpy(new_client->host, "0", sizeof(new_client->host));
390     strlcpy(new_client->host+1, new_client->sockhost, sizeof(new_client->host)-1);
391 michael 1123 memmove(new_client->sockhost+1, new_client->sockhost, sizeof(new_client->sockhost)-1);
392     new_client->sockhost[0] = '0';
393 michael 549 }
394     else
395 michael 1072 strlcpy(new_client->host, new_client->sockhost, sizeof(new_client->host));
396 adx 30
397     new_client->localClient->listener = listener;
398     ++listener->ref_count;
399    
400     #ifdef HAVE_LIBCRYPTO
401 michael 549 if (listener->flags & LISTENER_SSL)
402 adx 30 {
403 michael 967 if ((new_client->localClient->fd.ssl = SSL_new(ServerInfo.server_ctx)) == NULL)
404 adx 30 {
405 michael 1247 ilog(LOG_TYPE_IRCD, "SSL_new() ERROR! -- %s",
406 adx 30 ERR_error_string(ERR_get_error(), NULL));
407    
408     SetDead(new_client);
409     exit_client(new_client, new_client, "SSL_new failed");
410     return;
411     }
412    
413     SSL_set_fd(new_client->localClient->fd.ssl, fd);
414     ssl_handshake(0, new_client);
415     }
416     else
417     #endif
418 michael 2181 start_auth(new_client);
419 adx 30 }
420    
421     /*
422     * stolen from squid - its a neat (but overused! :) routine which we
423     * can use to see whether we can ignore this errno or not. It is
424     * generally useful for non-blocking network IO related errnos.
425     * -- adrian
426     */
427     int
428     ignoreErrno(int ierrno)
429     {
430     switch (ierrno)
431     {
432     case EINPROGRESS:
433     case EWOULDBLOCK:
434     #if EAGAIN != EWOULDBLOCK
435     case EAGAIN:
436     #endif
437     case EALREADY:
438     case EINTR:
439     #ifdef ERESTART
440     case ERESTART:
441     #endif
442     return 1;
443     default:
444     return 0;
445     }
446     }
447    
448     /*
449     * comm_settimeout() - set the socket timeout
450     *
451     * Set the timeout for the fd
452     */
453     void
454     comm_settimeout(fde_t *fd, time_t timeout, PF *callback, void *cbdata)
455     {
456     assert(fd->flags.open);
457    
458     fd->timeout = CurrentTime + (timeout / 1000);
459     fd->timeout_handler = callback;
460     fd->timeout_data = cbdata;
461     }
462    
463     /*
464     * comm_setflush() - set a flush function
465     *
466     * A flush function is simply a function called if found during
467     * comm_timeouts(). Its basically a second timeout, except in this case
468     * I'm too lazy to implement multiple timeout functions! :-)
469     * its kinda nice to have it separate, since this is designed for
470     * flush functions, and when comm_close() is implemented correctly
471     * with close functions, we _actually_ don't call comm_close() here ..
472     * -- originally Adrian's notes
473     * comm_close() is replaced with fd_close() in fdlist.c
474     */
475     void
476     comm_setflush(fde_t *fd, time_t timeout, PF *callback, void *cbdata)
477     {
478     assert(fd->flags.open);
479    
480     fd->flush_timeout = CurrentTime + (timeout / 1000);
481     fd->flush_handler = callback;
482     fd->flush_data = cbdata;
483     }
484    
485     /*
486     * comm_checktimeouts() - check the socket timeouts
487     *
488     * All this routine does is call the given callback/cbdata, without closing
489     * down the file descriptor. When close handlers have been implemented,
490     * this will happen.
491     */
492     void
493     comm_checktimeouts(void *notused)
494     {
495     int i;
496     fde_t *F;
497     PF *hdl;
498     void *data;
499    
500     for (i = 0; i < FD_HASH_SIZE; i++)
501     for (F = fd_hash[i]; F != NULL; F = fd_next_in_loop)
502     {
503     assert(F->flags.open);
504     fd_next_in_loop = F->hnext;
505    
506     /* check flush functions */
507     if (F->flush_handler && F->flush_timeout > 0 &&
508     F->flush_timeout < CurrentTime)
509     {
510     hdl = F->flush_handler;
511     data = F->flush_data;
512     comm_setflush(F, 0, NULL, NULL);
513     hdl(F, data);
514     }
515    
516     /* check timeouts */
517     if (F->timeout_handler && F->timeout > 0 &&
518     F->timeout < CurrentTime)
519     {
520     /* Call timeout handler */
521     hdl = F->timeout_handler;
522     data = F->timeout_data;
523     comm_settimeout(F, 0, NULL, NULL);
524     hdl(F, data);
525     }
526     }
527     }
528    
529     /*
530     * void comm_connect_tcp(int fd, const char *host, unsigned short port,
531     * struct sockaddr *clocal, int socklen,
532     * CNCB *callback, void *data, int aftype, int timeout)
533     * Input: An fd to connect with, a host and port to connect to,
534     * a local sockaddr to connect from + length(or NULL to use the
535     * default), a callback, the data to pass into the callback, the
536     * address family.
537     * Output: None.
538     * Side-effects: A non-blocking connection to the host is started, and
539     * if necessary, set up for selection. The callback given
540     * may be called now, or it may be called later.
541     */
542     void
543     comm_connect_tcp(fde_t *fd, const char *host, unsigned short port,
544     struct sockaddr *clocal, int socklen, CNCB *callback,
545     void *data, int aftype, int timeout)
546     {
547     struct addrinfo hints, *res;
548 michael 992 char portname[PORTNAMELEN + 1];
549 adx 30
550     assert(callback);
551     fd->connect.callback = callback;
552     fd->connect.data = data;
553    
554     fd->connect.hostaddr.ss.ss_family = aftype;
555     fd->connect.hostaddr.ss_port = htons(port);
556    
557     /* Note that we're using a passed sockaddr here. This is because
558     * generally you'll be bind()ing to a sockaddr grabbed from
559     * getsockname(), so this makes things easier.
560     * XXX If NULL is passed as local, we should later on bind() to the
561     * virtual host IP, for completeness.
562     * -- adrian
563     */
564     if ((clocal != NULL) && (bind(fd->fd, clocal, socklen) < 0))
565     {
566     /* Failure, call the callback with COMM_ERR_BIND */
567     comm_connect_callback(fd, COMM_ERR_BIND);
568     /* ... and quit */
569     return;
570     }
571    
572     /* Next, if we have been given an IP, get the addr and skip the
573     * DNS check (and head direct to comm_connect_tryconnect().
574     */
575     memset(&hints, 0, sizeof(hints));
576     hints.ai_family = AF_UNSPEC;
577     hints.ai_socktype = SOCK_STREAM;
578     hints.ai_flags = AI_PASSIVE | AI_NUMERICHOST;
579    
580 michael 992 snprintf(portname, sizeof(portname), "%d", port);
581 adx 30
582 michael 1123 if (getaddrinfo(host, portname, &hints, &res))
583 adx 30 {
584     /* Send the DNS request, for the next level */
585 db 871 if (aftype == AF_INET6)
586 michael 992 gethost_byname_type(comm_connect_dns_callback, fd, host, T_AAAA);
587 db 871 else
588 michael 992 gethost_byname_type(comm_connect_dns_callback, fd, host, T_A);
589 adx 30 }
590     else
591     {
592     /* We have a valid IP, so we just call tryconnect */
593     /* Make sure we actually set the timeout here .. */
594     assert(res != NULL);
595     memcpy(&fd->connect.hostaddr, res->ai_addr, res->ai_addrlen);
596     fd->connect.hostaddr.ss_len = res->ai_addrlen;
597     fd->connect.hostaddr.ss.ss_family = res->ai_family;
598 michael 1123 freeaddrinfo(res);
599 adx 30 comm_settimeout(fd, timeout*1000, comm_connect_timeout, NULL);
600     comm_connect_tryconnect(fd, NULL);
601     }
602     }
603    
604     /*
605     * comm_connect_callback() - call the callback, and continue with life
606     */
607     static void
608     comm_connect_callback(fde_t *fd, int status)
609     {
610     CNCB *hdl;
611    
612     /* This check is gross..but probably necessary */
613     if (fd->connect.callback == NULL)
614     return;
615    
616     /* Clear the connect flag + handler */
617     hdl = fd->connect.callback;
618     fd->connect.callback = NULL;
619    
620     /* Clear the timeout handler */
621     comm_settimeout(fd, 0, NULL, NULL);
622    
623     /* Call the handler */
624     hdl(fd, status, fd->connect.data);
625     }
626    
627     /*
628     * comm_connect_timeout() - this gets called when the socket connection
629     * times out. This *only* can be called once connect() is initially
630     * called ..
631     */
632     static void
633     comm_connect_timeout(fde_t *fd, void *notused)
634     {
635     /* error! */
636     comm_connect_callback(fd, COMM_ERR_TIMEOUT);
637     }
638    
639     /*
640     * comm_connect_dns_callback() - called at the completion of the DNS request
641     *
642     * The DNS request has completed, so if we've got an error, return it,
643     * otherwise we initiate the connect()
644     */
645     static void
646 michael 992 comm_connect_dns_callback(void *vptr, const struct irc_ssaddr *addr, const char *name)
647 adx 30 {
648     fde_t *F = vptr;
649    
650 michael 992 if (name == NULL)
651 adx 30 {
652     comm_connect_callback(F, COMM_ERR_DNS);
653     return;
654     }
655    
656     /* No error, set a 10 second timeout */
657     comm_settimeout(F, 30*1000, comm_connect_timeout, NULL);
658    
659     /* Copy over the DNS reply info so we can use it in the connect() */
660     /*
661     * Note we don't fudge the refcount here, because we aren't keeping
662     * the DNS record around, and the DNS cache is gone anyway..
663     * -- adrian
664     */
665 michael 992 memcpy(&F->connect.hostaddr, addr, addr->ss_len);
666 adx 30 /* The cast is hacky, but safe - port offset is same on v4 and v6 */
667     ((struct sockaddr_in *) &F->connect.hostaddr)->sin_port =
668     F->connect.hostaddr.ss_port;
669 michael 992 F->connect.hostaddr.ss_len = addr->ss_len;
670 adx 30
671     /* Now, call the tryconnect() routine to try a connect() */
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     /*
700     * If we get EISCONN, then we've already connect()ed the socket,
701     * which is a good thing.
702     * -- adrian
703     */
704     if (errno == EISCONN)
705     comm_connect_callback(fd, COMM_OK);
706     else if (ignoreErrno(errno))
707     /* Ignore error? Reschedule */
708     comm_setselect(fd, COMM_SELECT_WRITE, comm_connect_tryconnect,
709     NULL, 0);
710     else
711     /* Error? Fail with COMM_ERR_CONNECT */
712     comm_connect_callback(fd, COMM_ERR_CONNECT);
713     return;
714     }
715    
716     /* If we get here, we've suceeded, so call with COMM_OK */
717     comm_connect_callback(fd, COMM_OK);
718     }
719    
720     /*
721     * comm_errorstr() - return an error string for the given error condition
722     */
723     const char *
724     comm_errstr(int error)
725     {
726     if (error < 0 || error >= COMM_ERR_MAX)
727     return "Invalid error number!";
728     return comm_err_str[error];
729     }
730    
731     /*
732     * comm_open() - open a socket
733     *
734     * This is a highly highly cut down version of squid's comm_open() which
735     * for the most part emulates socket(), *EXCEPT* it fails if we're about
736     * to run out of file descriptors.
737     */
738     int
739     comm_open(fde_t *F, int family, int sock_type, int proto, const char *note)
740     {
741     int fd;
742    
743     /* First, make sure we aren't going to run out of file descriptors */
744     if (number_fd >= hard_fdlimit)
745     {
746     errno = ENFILE;
747     return -1;
748     }
749    
750     /*
751     * Next, we try to open the socket. We *should* drop the reserved FD
752     * limit if/when we get an error, but we can deal with that later.
753     * XXX !!! -- adrian
754     */
755     fd = socket(family, sock_type, proto);
756     if (fd < 0)
757     return -1; /* errno will be passed through, yay.. */
758    
759     execute_callback(setup_socket_cb, fd);
760    
761     /* update things in our fd tracking */
762     fd_open(F, fd, 1, note);
763     return 0;
764     }
765    
766     /*
767     * comm_accept() - accept an incoming connection
768     *
769     * This is a simple wrapper for accept() which enforces FD limits like
770     * comm_open() does. Returned fd must be either closed or tagged with
771     * fd_open (this function no longer does it).
772     */
773     int
774     comm_accept(struct Listener *lptr, struct irc_ssaddr *pn)
775     {
776     int newfd;
777     socklen_t addrlen = sizeof(struct irc_ssaddr);
778    
779     if (number_fd >= hard_fdlimit)
780     {
781     errno = ENFILE;
782     return -1;
783     }
784    
785     /*
786     * Next, do the accept(). if we get an error, we should drop the
787     * reserved fd limit, but we can deal with that when comm_open()
788     * also does it. XXX -- adrian
789     */
790 michael 1013 newfd = accept(lptr->fd.fd, (struct sockaddr *)pn, &addrlen);
791 adx 30 if (newfd < 0)
792     return -1;
793    
794     #ifdef IPV6
795     remove_ipv6_mapping(pn);
796     #else
797     pn->ss_len = addrlen;
798     #endif
799    
800     execute_callback(setup_socket_cb, newfd);
801    
802     /* .. and return */
803     return newfd;
804     }
805    
806     /*
807     * remove_ipv6_mapping() - Removes IPv4-In-IPv6 mapping from an address
808 michael 1122 * OSes with IPv6 mapping listening on both
809 adx 30 * AF_INET and AF_INET6 map AF_INET connections inside AF_INET6 structures
810     *
811     */
812     #ifdef IPV6
813     void
814     remove_ipv6_mapping(struct irc_ssaddr *addr)
815     {
816     if (addr->ss.ss_family == AF_INET6)
817     {
818 michael 1122 if (IN6_IS_ADDR_V4MAPPED(&((struct sockaddr_in6 *)addr)->sin6_addr))
819     {
820     struct sockaddr_in6 v6;
821     struct sockaddr_in *v4 = (struct sockaddr_in *)addr;
822 adx 30
823 michael 1122 memcpy(&v6, addr, sizeof(v6));
824     memset(v4, 0, sizeof(struct sockaddr_in));
825     memcpy(&v4->sin_addr, &v6.sin6_addr.s6_addr[12], sizeof(v4->sin_addr));
826    
827 adx 30 addr->ss.ss_family = AF_INET;
828     addr->ss_len = sizeof(struct sockaddr_in);
829     }
830     else
831     addr->ss_len = sizeof(struct sockaddr_in6);
832     }
833     else
834     addr->ss_len = sizeof(struct sockaddr_in);
835     }
836     #endif

Properties

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