]> arthur.barton.de Git - ngircd-alex.git/blob - src/ngircd/conn.c
Correctly retry outgoing conenctions when forking a resolver failed
[ngircd-alex.git] / src / ngircd / conn.c
1 /*
2  * ngIRCd -- The Next Generation IRC Daemon
3  * Copyright (c)2001-2018 Alexander Barton (alex@barton.de) and Contributors.
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License as published by
7  * the Free Software Foundation; either version 2 of the License, or
8  * (at your option) any later version.
9  * Please read the file COPYING, README and AUTHORS for more information.
10  */
11
12 #define CONN_MODULE
13
14 #include "portab.h"
15
16 /**
17  * @file
18  * Connection management
19  */
20
21 /* Additionan debug messages related to buffer handling: 0=off / 1=on */
22 #define DEBUG_BUFFER 0
23
24 #include <assert.h>
25 #ifdef PROTOTYPES
26 # include <stdarg.h>
27 #else
28 # include <varargs.h>
29 #endif
30 #include <stdio.h>
31 #include <stdlib.h>
32 #include <unistd.h>
33 #include <errno.h>
34 #include <string.h>
35 #include <strings.h>
36 #include <sys/socket.h>
37 #include <sys/stat.h>
38 #include <sys/types.h>
39 #include <time.h>
40 #include <netinet/in.h>
41
42 #ifdef HAVE_NETINET_IP_H
43 # ifdef HAVE_NETINET_IN_SYSTM_H
44 #  include <netinet/in_systm.h>
45 # endif
46 # include <netinet/ip.h>
47 #endif
48
49 #ifdef TCPWRAP
50 # include <tcpd.h>                      /* for TCP Wrappers */
51 #endif
52
53 #include "conn.h"
54
55 #include "ngircd.h"
56 #include "class.h"
57 #ifdef ICONV
58 # include "conn-encoding.h"
59 #endif
60 #include "conn-ssl.h"
61 #include "conn-zip.h"
62 #include "conn-func.h"
63 #include "io.h"
64 #include "log.h"
65 #include "ng_ipaddr.h"
66 #include "parse.h"
67 #include "resolve.h"
68
69 #define SERVER_WAIT (NONE - 1)          /** "Wait for outgoing connection" flag */
70
71 #define MAX_COMMANDS 3                  /** Max. commands per loop for users */
72 #define MAX_COMMANDS_SERVER_MIN 10      /** Min. commands per loop for servers */
73 #define MAX_COMMANDS_SERVICE 10         /** Max. commands per loop for services */
74
75 #define SD_LISTEN_FDS_START 3           /** systemd(8) socket activation offset */
76
77 #define THROTTLE_CMDS 1                 /** Throttling: max commands reached */
78 #define THROTTLE_BPS 2                  /** Throttling: max bps reached */
79
80 static bool Handle_Write PARAMS(( CONN_ID Idx ));
81 static bool Conn_Write PARAMS(( CONN_ID Idx, char *Data, size_t Len ));
82 static int New_Connection PARAMS(( int Sock, bool IsSSL ));
83 static CONN_ID Socket2Index PARAMS(( int Sock ));
84 static void Read_Request PARAMS(( CONN_ID Idx ));
85 static unsigned int Handle_Buffer PARAMS(( CONN_ID Idx ));
86 static void Check_Connections PARAMS(( void ));
87 static void Check_Servers PARAMS(( void ));
88 static void Init_Conn_Struct PARAMS(( CONN_ID Idx ));
89 static bool Init_Socket PARAMS(( int Sock ));
90 static void New_Server PARAMS(( int Server, ng_ipaddr_t *dest ));
91 static void Simple_Message PARAMS(( int Sock, const char *Msg ));
92 static int NewListener PARAMS(( const char *listen_addr, UINT16 Port ));
93 static void Account_Connection PARAMS((void));
94 static void Throttle_Connection PARAMS((const CONN_ID Idx, CLIENT *Client,
95                                         const int Reason, unsigned int Value));
96
97 static array My_Listeners;
98 static array My_ConnArray;
99 static size_t NumConnections, NumConnectionsMax, NumConnectionsAccepted;
100
101 #ifdef TCPWRAP
102 int allow_severity = LOG_INFO;
103 int deny_severity = LOG_ERR;
104 #endif
105
106 static void server_login PARAMS((CONN_ID idx));
107
108 #ifdef SSL_SUPPORT
109 extern struct SSLOptions Conf_SSLOptions;
110 static bool SSL_WantRead PARAMS((const CONNECTION *c));
111 static bool SSL_WantWrite PARAMS((const CONNECTION *c));
112 static void cb_listen_ssl PARAMS((int sock, short irrelevant));
113 static void cb_connserver_login_ssl PARAMS((int sock, short what));
114 static void cb_clientserver_ssl PARAMS((int sock, short what));
115 #endif
116 static void cb_Read_Resolver_Result PARAMS((int sock, UNUSED short what));
117 static void cb_Connect_to_Server PARAMS((int sock, UNUSED short what));
118 static void cb_clientserver PARAMS((int sock, short what));
119
120 time_t idle_t = 0;
121
122 /**
123  * Get number of sockets available from systemd(8).
124  *
125  * ngIRCd needs to implement its own sd_listen_fds(3) function and can't
126  * use the one provided by systemd itself, because the sockets will be
127  * used in a forked child process with a new PID, and this would trigger
128  * an error in the standard implementation.
129  *
130  * @return Number of sockets available, -1 if sockets have already been
131  *         initialized, or 0 when no sockets have been passed.
132  */
133 static int
134 my_sd_listen_fds(void)
135 {
136         const char *e;
137         int count;
138
139         /* Check if LISTEN_PID exists; but we ignore the result, because
140          * normally ngircd forks a child before checking this, and therefore
141          * the PID set in the environment is always wrong ... */
142         e = getenv("LISTEN_PID");
143         if (!e || !*e)
144                 return 0;
145
146         e = getenv("LISTEN_FDS");
147         if (!e || !*e)
148                 return -1;
149         count = atoi(e);
150 #ifdef HAVE_UNSETENV
151         unsetenv("LISTEN_FDS");
152 #endif
153
154         return count;
155 }
156
157 /**
158  * IO callback for listening sockets: handle new connections. This callback
159  * gets called when a new non-SSL connection should be accepted.
160  *
161  * @param sock          Socket descriptor.
162  * @param irrelevant    (ignored IO specification)
163  */
164 static void
165 cb_listen(int sock, short irrelevant)
166 {
167         (void) irrelevant;
168         (void) New_Connection(sock, false);
169 }
170
171 /**
172  * IO callback for new outgoing non-SSL server connections.
173  *
174  * @param sock  Socket descriptor.
175  * @param what  IO specification (IO_WANTREAD/IO_WANTWRITE/...).
176  */
177 static void
178 cb_connserver(int sock, UNUSED short what)
179 {
180         int res, err, server;
181         socklen_t sock_len;
182         CONN_ID idx = Socket2Index( sock );
183
184         if (idx <= NONE) {
185                 io_close(sock);
186                 return;
187         }
188
189         assert(what & IO_WANTWRITE);
190
191         /* Make sure that the server is still configured; it could have been
192          * removed in the meantime! */
193         server = Conf_GetServer(idx);
194         if (server < 0) {
195                 Log(LOG_ERR, "Connection on socket %d to \"%s\" aborted!",
196                     sock, My_Connections[idx].host);
197                 Conn_Close(idx, "Connection aborted", NULL, false);
198                 return;
199         }
200
201         /* connect() finished, get result. */
202         sock_len = (socklen_t)sizeof(err);
203         res = getsockopt(My_Connections[idx].sock, SOL_SOCKET, SO_ERROR,
204                          &err, &sock_len );
205         assert(sock_len == sizeof(err));
206
207         /* Error while connecting? */
208         if ((res != 0) || (err != 0)) {
209                 if (res != 0)
210                         Log(LOG_CRIT, "getsockopt (connection %d): %s!",
211                             idx, strerror(errno));
212                 else
213                         Log(LOG_CRIT,
214                             "Can't connect socket to \"%s:%d\" (connection %d): %s!",
215                             My_Connections[idx].host, Conf_Server[server].port,
216                             idx, strerror(err));
217
218                 Conn_Close(idx, "Can't connect", NULL, false);
219
220                 if (ng_ipaddr_af(&Conf_Server[server].dst_addr[0])) {
221                         /* more addresses to try... */
222                         New_Server(server, &Conf_Server[server].dst_addr[0]);
223                         /* connection to dst_addr[0] is now in progress, so
224                          * remove this address... */
225                         Conf_Server[server].dst_addr[0] =
226                                 Conf_Server[server].dst_addr[1];
227                         memset(&Conf_Server[server].dst_addr[1], 0,
228                                sizeof(Conf_Server[server].dst_addr[1]));
229                 }
230                 return;
231         }
232
233         /* connect() succeeded, remove all additional addresses */
234         memset(&Conf_Server[server].dst_addr, 0,
235                sizeof(Conf_Server[server].dst_addr));
236
237         Conn_OPTION_DEL( &My_Connections[idx], CONN_ISCONNECTING );
238 #ifdef SSL_SUPPORT
239         if ( Conn_OPTION_ISSET( &My_Connections[idx], CONN_SSL_CONNECT )) {
240                 io_event_setcb( sock, cb_connserver_login_ssl );
241                 io_event_add( sock, IO_WANTWRITE|IO_WANTREAD );
242                 return;
243         }
244 #endif
245         server_login(idx);
246 }
247
248 /**
249  * Login to a remote server.
250  *
251  * @param idx   Connection index.
252  */
253 static void
254 server_login(CONN_ID idx)
255 {
256         Log(LOG_INFO,
257             "Connection %d (socket %d) with \"%s:%d\" established. Now logging in ...",
258             idx, My_Connections[idx].sock, My_Connections[idx].host,
259             Conf_Server[Conf_GetServer(idx)].port);
260
261         io_event_setcb( My_Connections[idx].sock, cb_clientserver);
262         io_event_add( My_Connections[idx].sock, IO_WANTREAD|IO_WANTWRITE);
263
264         /* Send PASS and SERVER command to peer */
265         Conn_WriteStr(idx, "PASS %s %s",
266                       Conf_Server[Conf_GetServer( idx )].pwd_out, NGIRCd_ProtoID);
267         Conn_WriteStr(idx, "SERVER %s :%s",
268                       Conf_ServerName, Conf_ServerInfo);
269 }
270
271 /**
272  * IO callback for established non-SSL client and server connections.
273  *
274  * @param sock  Socket descriptor.
275  * @param what  IO specification (IO_WANTREAD/IO_WANTWRITE/...).
276  */
277 static void
278 cb_clientserver(int sock, short what)
279 {
280         CONN_ID idx = Socket2Index(sock);
281
282         if (idx <= NONE) {
283                 io_close(sock);
284                 return;
285         }
286
287 #ifdef SSL_SUPPORT
288         if (what & IO_WANTREAD
289             || (Conn_OPTION_ISSET(&My_Connections[idx], CONN_SSL_WANT_WRITE))) {
290                 /* if TLS layer needs to write additional data, call
291                  * Read_Request() instead so that SSL/TLS can continue */
292                 Read_Request(idx);
293         }
294 #else
295         if (what & IO_WANTREAD)
296                 Read_Request(idx);
297 #endif
298         if (what & IO_WANTWRITE)
299                 Handle_Write(idx);
300 }
301
302 /**
303  * Initialize connection module.
304  */
305 GLOBAL void
306 Conn_Init( void )
307 {
308         int size;
309
310         /* Initialize the "connection pool".
311          * FIXME: My_Connetions/Pool_Size is needed by other parts of the
312          * code; remove them! */
313         Pool_Size = 0;
314         size = Conf_MaxConnections > 0 ? Conf_MaxConnections : CONNECTION_POOL;
315         if (Socket2Index(size) <= NONE) {
316                 Log(LOG_EMERG, "Failed to initialize connection pool!");
317                 exit(1);
318         }
319
320         /* Initialize "listener" array. */
321         array_free( &My_Listeners );
322 } /* Conn_Init */
323
324 /**
325  * Clean up connection module.
326  */
327 GLOBAL void
328 Conn_Exit( void )
329 {
330         CONN_ID idx;
331
332         Conn_ExitListeners();
333
334         LogDebug("Shutting down all connections ..." );
335         for( idx = 0; idx < Pool_Size; idx++ ) {
336                 if( My_Connections[idx].sock > NONE ) {
337                         Conn_Close( idx, NULL, NGIRCd_SignalRestart ?
338                                 "Server going down (restarting)":"Server going down", true );
339                 }
340         }
341
342         array_free(&My_ConnArray);
343         My_Connections = NULL;
344         Pool_Size = 0;
345         io_library_shutdown();
346 } /* Conn_Exit */
347
348 /**
349  * Close all sockets (file descriptors) of open connections.
350  * This is useful in forked child processes, for example, to make sure that
351  * they don't hold connections open that the main process wants to close.
352  */
353 GLOBAL void
354 Conn_CloseAllSockets(int ExceptOf)
355 {
356         CONN_ID idx;
357
358         for(idx = 0; idx < Pool_Size; idx++) {
359                 if(My_Connections[idx].sock > NONE &&
360                    My_Connections[idx].sock != ExceptOf)
361                         close(My_Connections[idx].sock);
362         }
363 }
364
365 /**
366  * Initialize listening ports.
367  *
368  * @param a             Array containing the ports the daemon should listen on.
369  * @param listen_addr   Address the socket should listen on (can be "0.0.0.0").
370  * @param func          IO callback function to register.
371  * @returns             Number of listening sockets created.
372  */
373 static unsigned int
374 Init_Listeners(array *a, const char *listen_addr, void (*func)(int,short))
375 {
376         unsigned int created = 0;
377         size_t len;
378         int fd;
379         UINT16 *port;
380
381         len = array_length(a, sizeof (UINT16));
382         port = array_start(a);
383         while (len--) {
384                 fd = NewListener(listen_addr, *port);
385                 if (fd < 0) {
386                         port++;
387                         continue;
388                 }
389                 if (!io_event_create( fd, IO_WANTREAD, func )) {
390                         Log(LOG_ERR,
391                             "io_event_create(): Can't add fd %d (port %u): %s!",
392                             fd, (unsigned int) *port, strerror(errno));
393                         close(fd);
394                         port++;
395                         continue;
396                 }
397                 created++;
398                 port++;
399         }
400         return created;
401 }
402
403 /**
404  * Initialize all listening sockets.
405  *
406  * @returns     Number of created listening sockets
407  */
408 GLOBAL unsigned int
409 Conn_InitListeners( void )
410 {
411         /* Initialize ports on which the server should accept connections */
412         unsigned int created = 0;
413         char *af_str, *copy, *listen_addr;
414         int count, fd, i, addr_len;
415         ng_ipaddr_t addr;
416
417         assert(Conf_ListenAddress);
418
419         count = my_sd_listen_fds();
420         if (count < 0) {
421                 Log(LOG_INFO,
422                     "Not re-initializing listening sockets of systemd(8) ...");
423                 return 0;
424         }
425         if (count > 0) {
426                 /* systemd(8) passed sockets to us, so don't try to initialize
427                  * listening sockets on our own but use the passed ones */
428                 LogDebug("Initializing %d systemd sockets ...", count);
429                 for (i = 0; i < count; i++) {
430                         fd = SD_LISTEN_FDS_START + i;
431                         addr_len = (int)sizeof(addr);
432                         getsockname(fd, (struct sockaddr *)&addr,
433                                     (socklen_t*)&addr_len);
434 #ifdef WANT_IPV6
435                         if (addr.sin4.sin_family != AF_INET
436                             && addr.sin4.sin_family != AF_INET6)
437 #else
438                         if (addr.sin4.sin_family != AF_INET)
439 #endif
440                         {
441                                 /* Socket is of unsupported type! For example,
442                                  * systemd passed in an IPv6 socket but ngIRCd
443                                  * isn't compiled with IPv6 support. */
444                                 switch (addr.sin4.sin_family)
445                                 {
446                                         case AF_UNSPEC: af_str = "AF_UNSPEC"; break;
447                                         case AF_UNIX: af_str = "AF_UNIX"; break;
448                                         case AF_INET: af_str = "AF_INET"; break;
449 #ifdef AF_INET6
450                                         case AF_INET6: af_str = "AF_INET6"; break;
451 #endif
452 #ifdef AF_NETLINK
453                                         case AF_NETLINK: af_str = "AF_NETLINK"; break;
454 #endif
455                                         default: af_str = "unknown"; break;
456                                 }
457                                 Log(LOG_CRIT,
458                                     "Socket %d is of unsupported type \"%s\" (%d), have to ignore it!",
459                                     fd, af_str, addr.sin4.sin_family);
460                                 close(fd);
461                                 continue;
462                         }
463
464                         Init_Socket(fd);
465                         if (!io_event_create(fd, IO_WANTREAD, cb_listen)) {
466                                 Log(LOG_ERR,
467                                     "io_event_create(): Can't add fd %d: %s!",
468                                     fd, strerror(errno));
469                                 continue;
470                         }
471                         Log(LOG_INFO,
472                             "Initialized socket %d from systemd(8): %s:%d.", fd,
473                             ng_ipaddr_tostr(&addr), ng_ipaddr_getport(&addr));
474                         created++;
475                 }
476                 return created;
477         }
478
479         /* not using systemd socket activation, initialize listening sockets: */
480
481         /* can't use Conf_ListenAddress directly, see below */
482         copy = strdup(Conf_ListenAddress);
483         if (!copy) {
484                 Log(LOG_CRIT, "Cannot copy %s: %s", Conf_ListenAddress,
485                     strerror(errno));
486                 return 0;
487         }
488         listen_addr = strtok(copy, ",");
489
490         while (listen_addr) {
491                 ngt_TrimStr(listen_addr);
492                 if (*listen_addr) {
493                         created += Init_Listeners(&Conf_ListenPorts,
494                                                   listen_addr, cb_listen);
495 #ifdef SSL_SUPPORT
496                         created += Init_Listeners(&Conf_SSLOptions.ListenPorts,
497                                                   listen_addr, cb_listen_ssl);
498 #endif
499                 }
500
501                 listen_addr = strtok(NULL, ",");
502         }
503
504         /* Can't free() Conf_ListenAddress here: on REHASH, if the config file
505          * cannot be re-loaded, we'd end up with a NULL Conf_ListenAddress.
506          * Instead, free() takes place in conf.c, before the config file
507          * is being parsed. */
508         free(copy);
509
510         return created;
511 } /* Conn_InitListeners */
512
513 /**
514  * Shut down all listening sockets.
515  */
516 GLOBAL void
517 Conn_ExitListeners( void )
518 {
519         /* Close down all listening sockets */
520         int *fd;
521         size_t arraylen;
522
523         /* Get number of listening sockets to shut down. There can be none
524          * if ngIRCd has been "socket activated" by systemd. */
525         arraylen = array_length(&My_Listeners, sizeof (int));
526         if (arraylen < 1)
527                 return;
528
529         Log(LOG_INFO,
530             "Shutting down all listening sockets (%d total) ...", arraylen);
531         fd = array_start(&My_Listeners);
532         while(arraylen--) {
533                 assert(fd != NULL);
534                 assert(*fd >= 0);
535                 io_close(*fd);
536                 LogDebug("Listening socket %d closed.", *fd );
537                 fd++;
538         }
539         array_free(&My_Listeners);
540 } /* Conn_ExitListeners */
541
542 /**
543  * Bind a socket to a specific (source) address.
544  *
545  * @param addr                  Address structure.
546  * @param listen_addrstr        Source address as string.
547  * @param Port                  Port number.
548  * @returns                     true on success, false otherwise.
549  */
550 static bool
551 InitSinaddrListenAddr(ng_ipaddr_t *addr, const char *listen_addrstr, UINT16 Port)
552 {
553         bool ret;
554
555         ret = ng_ipaddr_init(addr, listen_addrstr, Port);
556         if (!ret) {
557                 assert(listen_addrstr);
558                 Log(LOG_CRIT,
559                     "Can't bind to [%s]:%u: can't convert ip address \"%s\"!",
560                     listen_addrstr, Port, listen_addrstr);
561         }
562         return ret;
563 }
564
565 /**
566  * Set a socket to "IPv6 only". If the given socket doesn't belong to the
567  * AF_INET6 family, or the operating system doesn't support this functionality,
568  * this function retruns silently.
569  *
570  * @param af    Address family of the socket.
571  * @param sock  Socket handle.
572  */
573 static void
574 set_v6_only(int af, int sock)
575 {
576 #if defined(IPV6_V6ONLY) && defined(WANT_IPV6)
577         int on = 1;
578
579         if (af != AF_INET6)
580                 return;
581
582         if (setsockopt(sock, IPPROTO_IPV6, IPV6_V6ONLY, &on, (socklen_t)sizeof(on)))
583                 Log(LOG_ERR, "Could not set IPV6_V6ONLY: %s", strerror(errno));
584 #else
585         (void)af;
586         (void)sock;
587 #endif
588 }
589
590 /**
591  * Initialize new listening port.
592  *
593  * @param listen_addr   Local address to bind the socet to (can be 0.0.0.0).
594  * @param Port          Port number on which the new socket should be listening.
595  * @returns             file descriptor of the socket or -1 on failure.
596  */
597 static int
598 NewListener(const char *listen_addr, UINT16 Port)
599 {
600         /* Create new listening socket on specified port */
601         ng_ipaddr_t addr;
602         int sock, af;
603
604         if (!InitSinaddrListenAddr(&addr, listen_addr, Port))
605                 return -1;
606
607         af = ng_ipaddr_af(&addr);
608         sock = socket(af, SOCK_STREAM, 0);
609         if (sock < 0) {
610                 Log(LOG_CRIT, "Can't create socket (af %d) : %s!", af,
611                     strerror(errno));
612                 return -1;
613         }
614
615         set_v6_only(af, sock);
616
617         if (!Init_Socket(sock))
618                 return -1;
619
620         if (bind(sock, (struct sockaddr *)&addr, ng_ipaddr_salen(&addr)) != 0) {
621                 Log(LOG_CRIT, "Can't bind socket to address %s:%d - %s!",
622                     ng_ipaddr_tostr(&addr), Port, strerror(errno));
623                 close(sock);
624                 return -1;
625         }
626
627         if (listen(sock, 10) != 0) {
628                 Log(LOG_CRIT, "Can't listen on socket: %s!", strerror(errno));
629                 close(sock);
630                 return -1;
631         }
632
633         /* keep fd in list so we can close it when ngircd restarts/shuts down */
634         if (!array_catb(&My_Listeners, (char *)&sock, sizeof(int))) {
635                 Log(LOG_CRIT, "Can't add socket to My_Listeners array: %s!",
636                     strerror(errno));
637                 close(sock);
638                 return -1;
639         }
640
641         Log(LOG_INFO, "Now listening on [%s]:%d (socket %d).",
642             ng_ipaddr_tostr(&addr), Port, sock);
643         return sock;
644 } /* NewListener */
645
646 /**
647  * "Main Loop": Loop until shutdown or restart is signalled.
648  *
649  * This function loops until a shutdown or restart of ngIRCd is signalled and
650  * calls io_dispatch() to check for readable and writable sockets every second.
651  * It checks for status changes on pending connections (e. g. when a hostname
652  * has been resolved), checks for "penalties" and timeouts, and handles the
653  * input buffers.
654  */
655 GLOBAL void
656 Conn_Handler(void)
657 {
658         int i;
659         size_t wdatalen;
660         struct timeval tv;
661         time_t t;
662
663         while (!NGIRCd_SignalQuit && !NGIRCd_SignalRestart) {
664                 t = time(NULL);
665
666                 /* Check configured servers and established links */
667                 Check_Servers();
668                 Check_Connections();
669
670                 /* Expire outdated class/list items */
671                 Class_Expire();
672
673                 /* Look for non-empty read buffers ... */
674                 for (i = 0; i < Pool_Size; i++) {
675                         if ((My_Connections[i].sock > NONE)
676                             && (array_bytes(&My_Connections[i].rbuf) > 0)) {
677                                 /* ... and try to handle the received data */
678                                 Handle_Buffer(i);
679                         }
680                 }
681
682                 /* Look for non-empty write buffers ... */
683                 for (i = 0; i < Pool_Size; i++) {
684                         if (My_Connections[i].sock <= NONE)
685                                 continue;
686
687                         wdatalen = array_bytes(&My_Connections[i].wbuf);
688 #ifdef ZLIB
689                         if (wdatalen > 0 ||
690                             array_bytes(&My_Connections[i].zip.wbuf) > 0)
691 #else
692                         if (wdatalen > 0)
693 #endif
694                         {
695 #ifdef SSL_SUPPORT
696                                 if (SSL_WantRead(&My_Connections[i]))
697                                         continue;
698 #endif
699                                 io_event_add(My_Connections[i].sock,
700                                              IO_WANTWRITE);
701                         }
702                 }
703
704                 /* Check from which sockets we possibly could read ... */
705                 for (i = 0; i < Pool_Size; i++) {
706                         if (My_Connections[i].sock <= NONE)
707                                 continue;
708 #ifdef SSL_SUPPORT
709                         if (SSL_WantWrite(&My_Connections[i]))
710                                 /* TLS/SSL layer needs to write data; deal
711                                  * with this first! */
712                                 continue;
713 #endif
714                         if (Proc_InProgress(&My_Connections[i].proc_stat)) {
715                                 /* Wait for completion of forked subprocess
716                                  * and ignore the socket in the meantime ... */
717                                 io_event_del(My_Connections[i].sock,
718                                              IO_WANTREAD);
719                                 continue;
720                         }
721
722                         if (Conn_OPTION_ISSET(&My_Connections[i], CONN_ISCONNECTING))
723                                 /* Wait for completion of connect() ... */
724                                 continue;
725
726                         if (My_Connections[i].delaytime > t) {
727                                 /* There is a "penalty time" set: ignore socket! */
728                                 io_event_del(My_Connections[i].sock,
729                                              IO_WANTREAD);
730                                 continue;
731                         }
732
733                         io_event_add(My_Connections[i].sock, IO_WANTREAD);
734                 }
735
736                 /* Set the timeout for reading from the network to 1 second,
737                  * which is the granularity with witch we handle "penalty
738                  * times" for example.
739                  * Note: tv_sec/usec are undefined(!) after io_dispatch()
740                  * returns, so we have to set it before each call to it! */
741                 tv.tv_usec = 0;
742                 tv.tv_sec = 1;
743
744                 /* Wait for activity ... */
745                 i = io_dispatch(&tv);
746                 if (i == -1 && errno != EINTR) {
747                         Log(LOG_EMERG, "Conn_Handler(): io_dispatch(): %s!",
748                             strerror(errno));
749                         Log(LOG_ALERT, "%s exiting due to fatal errors!",
750                             PACKAGE_NAME);
751                         exit(1);
752                 }
753
754                 /* Should ngIRCd timeout when idle? */
755                 if (Conf_IdleTimeout > 0 && NumConnectionsAccepted > 0
756                     && idle_t > 0 && time(NULL) - idle_t >= Conf_IdleTimeout) {
757                         LogDebug("Server idle timeout reached: %d second%s. Initiating shutdown ...",
758                                  Conf_IdleTimeout,
759                                  Conf_IdleTimeout == 1 ? "" : "s");
760                         NGIRCd_SignalQuit = true;
761                 }
762         }
763
764         if (NGIRCd_SignalQuit)
765                 Log(LOG_NOTICE | LOG_snotice, "Server going down NOW!");
766         else if (NGIRCd_SignalRestart)
767                 Log(LOG_NOTICE | LOG_snotice, "Server restarting NOW!");
768 } /* Conn_Handler */
769
770 /**
771  * Write a text string into the socket of a connection.
772  *
773  * This function automatically appends CR+LF to the string and validates that
774  * the result is a valid IRC message (oversized messages are shortened, for
775  * example). Then it calls the Conn_Write() function to do the actual sending.
776  *
777  * @param Idx           Index fo the connection.
778  * @param Format        Format string, see printf().
779  * @returns             true on success, false otherwise.
780  */
781 #ifdef PROTOTYPES
782 GLOBAL bool
783 Conn_WriteStr(CONN_ID Idx, const char *Format, ...)
784 #else
785 GLOBAL bool
786 Conn_WriteStr(Idx, Format, va_alist)
787 CONN_ID Idx;
788 const char *Format;
789 va_dcl
790 #endif
791 {
792         char buffer[COMMAND_LEN];
793 #ifdef ICONV
794         char *ptr, *message;
795 #endif
796         size_t len;
797         bool ok;
798         va_list ap;
799         int r;
800
801         assert( Idx > NONE );
802         assert( Format != NULL );
803
804 #ifdef PROTOTYPES
805         va_start( ap, Format );
806 #else
807         va_start( ap );
808 #endif
809         r = vsnprintf(buffer, COMMAND_LEN - 2, Format, ap);
810         if (r >= COMMAND_LEN - 2 || r == -1) {
811                 /*
812                  * The string that should be written to the socket is longer
813                  * than the allowed size of COMMAND_LEN bytes (including both
814                  * the CR and LF characters). This can be caused by the
815                  * IRC_WriteXXX() functions when the prefix of this server had
816                  * to be added to an already "quite long" command line which
817                  * has been received from a regular IRC client, for example.
818                  *
819                  * We are not allowed to send such "oversized" messages to
820                  * other servers and clients, see RFC 2812 2.3 and 2813 3.3
821                  * ("these messages SHALL NOT exceed 512 characters in length,
822                  * counting all characters including the trailing CR-LF").
823                  *
824                  * So we have a big problem here: we should send more bytes
825                  * to the network than we are allowed to and we don't know
826                  * the originator (any more). The "old" behavior of blaming
827                  * the receiver ("next hop") is a bad idea (it could be just
828                  * an other server only routing the message!), so the only
829                  * option left is to shorten the string and to hope that the
830                  * result is still somewhat useful ...
831                  *
832                  * Note:
833                  * C99 states that vsnprintf() "returns the number of characters
834                  * that would have been printed if the n were unlimited"; but
835                  * according to the Linux manual page "glibc until 2.0.6 would
836                  * return -1 when the output was truncated" -- so we have to
837                  * handle both cases ...
838                  *                                                   -alex-
839                  */
840
841                 strcpy (buffer + sizeof(buffer) - strlen(CUT_TXTSUFFIX) - 2 - 1,
842                         CUT_TXTSUFFIX);
843         }
844
845 #ifdef ICONV
846         ptr = strchr(buffer + 1, ':');
847         if (ptr) {
848                 ptr++;
849                 message = Conn_EncodingTo(Idx, ptr);
850                 if (message != ptr)
851                         strlcpy(ptr, message, sizeof(buffer) - (ptr - buffer));
852         }
853 #endif
854
855 #ifdef SNIFFER
856         if (NGIRCd_Sniffer)
857                 Log(LOG_DEBUG, " -> connection %d: '%s'.", Idx, buffer);
858 #endif
859
860         len = strlcat( buffer, "\r\n", sizeof( buffer ));
861         ok = Conn_Write(Idx, buffer, len);
862         My_Connections[Idx].msg_out++;
863
864         va_end( ap );
865         return ok;
866 } /* Conn_WriteStr */
867
868 GLOBAL char*
869 Conn_Password( CONN_ID Idx )
870 {
871         assert( Idx > NONE );
872         if (My_Connections[Idx].pwd == NULL)
873                 return (char*)"\0";
874         else
875                 return My_Connections[Idx].pwd;
876 } /* Conn_Password */
877
878 GLOBAL void
879 Conn_SetPassword( CONN_ID Idx, const char *Pwd )
880 {
881         assert( Idx > NONE );
882
883         if (My_Connections[Idx].pwd)
884                 free(My_Connections[Idx].pwd);
885
886         My_Connections[Idx].pwd = strdup(Pwd);
887         if (My_Connections[Idx].pwd == NULL) {
888                 Log(LOG_EMERG, "Can't allocate memory! [Conn_SetPassword]");
889                 exit(1);
890         }
891 } /* Conn_SetPassword */
892
893 /**
894  * Append Data to the outbound write buffer of a connection.
895  *
896  * @param Idx   Index of the connection.
897  * @param Data  pointer to the data.
898  * @param Len   length of Data.
899  * @returns     true on success, false otherwise.
900  */
901 static bool
902 Conn_Write( CONN_ID Idx, char *Data, size_t Len )
903 {
904         CLIENT *c;
905         size_t writebuf_limit = WRITEBUFFER_MAX_LEN;
906         assert( Idx > NONE );
907         assert( Data != NULL );
908         assert( Len > 0 );
909
910         /* Is the socket still open? A previous call to Conn_Write()
911          * may have closed the connection due to a fatal error.
912          * In this case it is sufficient to return an error, as well. */
913         if (My_Connections[Idx].sock <= NONE) {
914                 LogDebug("Skipped write on closed socket (connection %d).", Idx);
915                 return false;
916         }
917
918         /* Make sure that there still exists a CLIENT structure associated
919          * with this connection and check if this is a server or not: */
920         c = Conn_GetClient(Idx);
921         if (c) {
922                 /* Servers do get special write buffer limits, so they can
923                  * generate all the messages that are required while peering. */
924                 if (Client_Type(c) == CLIENT_SERVER)
925                         writebuf_limit = WRITEBUFFER_SLINK_LEN;
926         } else
927                 LogDebug("Write on socket without client (connection %d)!?", Idx);
928
929 #ifdef ZLIB
930         if ( Conn_OPTION_ISSET( &My_Connections[Idx], CONN_ZIP )) {
931                 /* Compressed link:
932                  * Zip_Buffer() does all the dirty work for us: it flushes
933                  * the (pre-)compression buffers if required and handles
934                  * all error conditions. */
935                 if (!Zip_Buffer(Idx, Data, Len))
936                         return false;
937         }
938         else
939 #endif
940         {
941                 /* Uncompressed link:
942                  * Check if outbound buffer has enough space for the data. */
943                 if (array_bytes(&My_Connections[Idx].wbuf) + Len >=
944                     WRITEBUFFER_FLUSH_LEN) {
945                         /* Buffer is full, flush it. Handle_Write deals with
946                          * low-level errors, if any. */
947                         if (!Handle_Write(Idx))
948                                 return false;
949                 }
950
951                 /* When the write buffer is still too big after flushing it,
952                  * the connection will be killed. */
953                 if (array_bytes(&My_Connections[Idx].wbuf) + Len >=
954                     writebuf_limit) {
955                         Log(LOG_NOTICE,
956                             "Write buffer space exhausted (connection %d, limit is %lu bytes, %lu bytes new, %lu bytes pending)",
957                             Idx, writebuf_limit, Len,
958                             (unsigned long)array_bytes(&My_Connections[Idx].wbuf));
959                         Conn_Close(Idx, "Write buffer space exhausted", NULL, false);
960                         return false;
961                 }
962
963                 /* Copy data to write buffer */
964                 if (!array_catb(&My_Connections[Idx].wbuf, Data, Len))
965                         return false;
966
967                 My_Connections[Idx].bytes_out += Len;
968         }
969
970         /* Adjust global write counter */
971         WCounter += Len;
972
973         return true;
974 } /* Conn_Write */
975
976 /**
977  * Shut down a connection.
978  *
979  * @param Idx           Connection index.
980  * @param LogMsg        Message to write to the log or NULL. If no LogMsg
981  *                      is given, the FwdMsg is logged.
982  * @param FwdMsg        Message to forward to remote servers.
983  * @param InformClient  If true, inform the client on the connection which is
984  *                      to be shut down of the reason (FwdMsg) and send
985  *                      connection statistics before disconnecting it.
986  */
987 GLOBAL void
988 Conn_Close(CONN_ID Idx, const char *LogMsg, const char *FwdMsg, bool InformClient)
989 {
990         /* Close connection. Open pipes of asynchronous resolver
991          * sub-processes are closed down. */
992
993         CLIENT *c;
994         double in_k, out_k;
995         UINT16 port;
996 #ifdef ZLIB
997         double in_z_k, out_z_k;
998         int in_p, out_p;
999 #endif
1000
1001         assert( Idx > NONE );
1002
1003         /* Is this link already shutting down? */
1004         if( Conn_OPTION_ISSET( &My_Connections[Idx], CONN_ISCLOSING )) {
1005                 /* Conn_Close() has been called recursively for this link;
1006                  * probable reason: Handle_Write() failed -- see below. */
1007                 LogDebug("Recursive request to close connection %d!", Idx );
1008                 return;
1009         }
1010
1011         assert( My_Connections[Idx].sock > NONE );
1012
1013         /* Mark link as "closing" */
1014         Conn_OPTION_ADD( &My_Connections[Idx], CONN_ISCLOSING );
1015
1016         port = ng_ipaddr_getport(&My_Connections[Idx].addr);
1017         Log(LOG_INFO, "Shutting down connection %d (%s) with \"%s:%d\" ...", Idx,
1018             LogMsg ? LogMsg : FwdMsg, My_Connections[Idx].host, port);
1019
1020         /* Search client, if any */
1021         c = Conn_GetClient( Idx );
1022
1023         /* Should the client be informed? */
1024         if (InformClient) {
1025 #ifndef STRICT_RFC
1026                 /* Send statistics to client if registered as user: */
1027                 if ((c != NULL) && (Client_Type(c) == CLIENT_USER)) {
1028                         Conn_WriteStr( Idx,
1029                          ":%s NOTICE %s :%sConnection statistics: client %.1f kb, server %.1f kb.",
1030                          Client_ID(Client_ThisServer()), Client_ID(c),
1031                          NOTICE_TXTPREFIX,
1032                          (double)My_Connections[Idx].bytes_in / 1024,
1033                          (double)My_Connections[Idx].bytes_out / 1024);
1034                 }
1035 #endif
1036                 /* Send ERROR to client (see RFC 2812, section 3.1.7) */
1037                 if (FwdMsg)
1038                         Conn_WriteStr(Idx, "ERROR :%s", FwdMsg);
1039                 else
1040                         Conn_WriteStr(Idx, "ERROR :Closing connection");
1041         }
1042
1043         /* Try to write out the write buffer. Note: Handle_Write() eventually
1044          * removes the CLIENT structure associated with this connection if an
1045          * error occurs! So we have to re-check if there is still an valid
1046          * CLIENT structure after calling Handle_Write() ...*/
1047         (void)Handle_Write( Idx );
1048
1049         /* Search client, if any (re-check!) */
1050         c = Conn_GetClient( Idx );
1051 #ifdef SSL_SUPPORT
1052         if ( Conn_OPTION_ISSET( &My_Connections[Idx], CONN_SSL )) {
1053                 LogDebug("SSL connection %d shutting down ...", Idx);
1054                 ConnSSL_Free(&My_Connections[Idx]);
1055         }
1056 #endif
1057         /* Shut down socket */
1058         if (! io_close(My_Connections[Idx].sock)) {
1059                 /* Oops, we can't close the socket!? This is ... ugly! */
1060                 Log(LOG_CRIT,
1061                     "Error closing connection %d (socket %d) with %s:%d - %s! (ignored)",
1062                     Idx, My_Connections[Idx].sock, My_Connections[Idx].host,
1063                     port, strerror(errno));
1064         }
1065
1066         /* Mark socket as invalid: */
1067         My_Connections[Idx].sock = NONE;
1068
1069         /* If there is still a client, unregister it now */
1070         if (c)
1071                 Client_Destroy(c, LogMsg, FwdMsg, true);
1072
1073         /* Calculate statistics and log information */
1074         in_k = (double)My_Connections[Idx].bytes_in / 1024;
1075         out_k = (double)My_Connections[Idx].bytes_out / 1024;
1076 #ifdef ZLIB
1077         if (Conn_OPTION_ISSET( &My_Connections[Idx], CONN_ZIP)) {
1078                 in_z_k = (double)My_Connections[Idx].zip.bytes_in / 1024;
1079                 out_z_k = (double)My_Connections[Idx].zip.bytes_out / 1024;
1080                 /* Make sure that no division by zero can occur during
1081                  * the calculation of in_p and out_p: in_z_k and out_z_k
1082                  * are non-zero, that's guaranteed by the protocol until
1083                  * compression can be enabled. */
1084                 if (! in_z_k)
1085                         in_z_k = in_k;
1086                 if (! out_z_k)
1087                         out_z_k = out_k;
1088                 in_p = (int)(( in_k * 100 ) / in_z_k );
1089                 out_p = (int)(( out_k * 100 ) / out_z_k );
1090                 Log(LOG_INFO,
1091                     "Connection %d with \"%s:%d\" closed (in: %.1fk/%.1fk/%d%%, out: %.1fk/%.1fk/%d%%).",
1092                     Idx, My_Connections[Idx].host, port,
1093                     in_k, in_z_k, in_p, out_k, out_z_k, out_p);
1094         }
1095         else
1096 #endif
1097         {
1098                 Log(LOG_INFO,
1099                     "Connection %d with \"%s:%d\" closed (in: %.1fk, out: %.1fk).",
1100                     Idx, My_Connections[Idx].host, port,
1101                     in_k, out_k);
1102         }
1103
1104         /* Servers: Modify time of next connect attempt? */
1105         Conf_UnsetServer( Idx );
1106
1107 #ifdef ZLIB
1108         /* Clean up zlib, if link was compressed */
1109         if ( Conn_OPTION_ISSET( &My_Connections[Idx], CONN_ZIP )) {
1110                 inflateEnd( &My_Connections[Idx].zip.in );
1111                 deflateEnd( &My_Connections[Idx].zip.out );
1112                 array_free(&My_Connections[Idx].zip.rbuf);
1113                 array_free(&My_Connections[Idx].zip.wbuf);
1114         }
1115 #endif
1116
1117         array_free(&My_Connections[Idx].rbuf);
1118         array_free(&My_Connections[Idx].wbuf);
1119         if (My_Connections[Idx].pwd != NULL)
1120                 free(My_Connections[Idx].pwd);
1121
1122         /* Clean up connection structure (=free it) */
1123         Init_Conn_Struct( Idx );
1124
1125         assert(NumConnections > 0);
1126         if (NumConnections)
1127                 NumConnections--;
1128         LogDebug("Shutdown of connection %d completed, %ld connection%s left.",
1129                  Idx, NumConnections, NumConnections != 1 ? "s" : "");
1130
1131         idle_t = NumConnections > 0 ? 0 : time(NULL);
1132 } /* Conn_Close */
1133
1134 /**
1135  * Get current number of connections.
1136  *
1137  * @returns     Number of current connections.
1138  */
1139 GLOBAL long
1140 Conn_Count(void)
1141 {
1142         return NumConnections;
1143 } /* Conn_Count */
1144
1145 /**
1146  * Get number of maximum simultaneous connections.
1147  *
1148  * @returns     Number of maximum simultaneous connections.
1149  */
1150 GLOBAL long
1151 Conn_CountMax(void)
1152 {
1153         return NumConnectionsMax;
1154 } /* Conn_CountMax */
1155
1156 /**
1157  * Get number of connections accepted since the daemon startet.
1158  *
1159  * @returns     Number of connections accepted.
1160  */
1161 GLOBAL long
1162 Conn_CountAccepted(void)
1163 {
1164         return NumConnectionsAccepted;
1165 } /* Conn_CountAccepted */
1166
1167 /**
1168  * Synchronize established connections and configured server structures
1169  * after a configuration update and store the correct connection IDs, if any.
1170  */
1171 GLOBAL void
1172 Conn_SyncServerStruct(void)
1173 {
1174         CLIENT *client;
1175         CONN_ID i;
1176         int c;
1177
1178         for (i = 0; i < Pool_Size; i++) {
1179                 if (My_Connections[i].sock == NONE)
1180                         continue;
1181
1182                 /* Server link? */
1183                 client = Conn_GetClient(i);
1184                 if (!client || Client_Type(client) != CLIENT_SERVER)
1185                         continue;
1186
1187                 for (c = 0; c < MAX_SERVERS; c++) {
1188                         /* Configured server? */
1189                         if (!Conf_Server[c].host[0])
1190                                 continue;
1191
1192                         if (strcasecmp(Conf_Server[c].name, Client_ID(client)) == 0)
1193                                 Conf_Server[c].conn_id = i;
1194                 }
1195         }
1196 } /* SyncServerStruct */
1197
1198 /**
1199  * Get IP address string of a connection.
1200  *
1201  * @param Idx Connection index.
1202  * @return Pointer to a global buffer containing the IP address as string.
1203  */
1204 GLOBAL const char *
1205 Conn_GetIPAInfo(CONN_ID Idx)
1206 {
1207         assert(Idx > NONE);
1208         return ng_ipaddr_tostr(&My_Connections[Idx].addr);
1209 }
1210
1211 /**
1212  * Send out data of write buffer; connect new sockets.
1213  *
1214  * @param Idx   Connection index.
1215  * @returns     true on success, false otherwise.
1216  */
1217 static bool
1218 Handle_Write( CONN_ID Idx )
1219 {
1220         ssize_t len;
1221         size_t wdatalen;
1222
1223         assert( Idx > NONE );
1224         if ( My_Connections[Idx].sock < 0 ) {
1225                 LogDebug("Handle_Write() on closed socket, connection %d", Idx);
1226                 return false;
1227         }
1228         assert( My_Connections[Idx].sock > NONE );
1229
1230         wdatalen = array_bytes(&My_Connections[Idx].wbuf );
1231
1232 #ifdef ZLIB
1233         if (wdatalen == 0) {
1234                 /* Write buffer is empty, so we try to flush the compression
1235                  * buffer and get some data to work with from there :-) */
1236                 if (!Zip_Flush(Idx))
1237                         return false;
1238
1239                 /* Now the write buffer most probably has changed: */
1240                 wdatalen = array_bytes(&My_Connections[Idx].wbuf);
1241         }
1242 #endif
1243
1244         if (wdatalen == 0) {
1245                 /* Still no data, fine. */
1246                 io_event_del(My_Connections[Idx].sock, IO_WANTWRITE );
1247                 return true;
1248         }
1249
1250 #if DEBUG_BUFFER
1251         LogDebug
1252             ("Handle_Write() called for connection %d, %ld bytes pending ...",
1253              Idx, wdatalen);
1254 #endif
1255
1256 #ifdef SSL_SUPPORT
1257         if ( Conn_OPTION_ISSET( &My_Connections[Idx], CONN_SSL )) {
1258                 len = ConnSSL_Write(&My_Connections[Idx],
1259                                     array_start(&My_Connections[Idx].wbuf),
1260                                     wdatalen);
1261         } else
1262 #endif
1263         {
1264                 len = write(My_Connections[Idx].sock,
1265                             array_start(&My_Connections[Idx].wbuf), wdatalen );
1266         }
1267         if( len < 0 ) {
1268                 if (errno == EAGAIN || errno == EINTR)
1269                         return true;
1270
1271                 if (!Conn_OPTION_ISSET(&My_Connections[Idx], CONN_ISCLOSING))
1272                         Log(LOG_ERR,
1273                             "Write error on connection %d (socket %d): %s!",
1274                             Idx, My_Connections[Idx].sock, strerror(errno));
1275                 else
1276                         LogDebug("Recursive write error on connection %d (socket %d): %s!",
1277                                  Idx, My_Connections[Idx].sock, strerror(errno));
1278                 Conn_Close(Idx, "Write error", NULL, false);
1279                 return false;
1280         }
1281
1282         /* move any data not yet written to beginning */
1283         array_moveleft(&My_Connections[Idx].wbuf, 1, (size_t)len);
1284
1285         return true;
1286 } /* Handle_Write */
1287
1288 /**
1289  * Count established connections to a specific IP address.
1290  *
1291  * @returns     Number of established connections.
1292  */
1293 static int
1294 Count_Connections(ng_ipaddr_t *a)
1295 {
1296         int i, cnt;
1297
1298         cnt = 0;
1299         for (i = 0; i < Pool_Size; i++) {
1300                 if (My_Connections[i].sock <= NONE)
1301                         continue;
1302                 if (ng_ipaddr_ipequal(&My_Connections[i].addr, a))
1303                         cnt++;
1304         }
1305         return cnt;
1306 } /* Count_Connections */
1307
1308 /**
1309  * Initialize new client connection on a listening socket.
1310  *
1311  * @param Sock  Listening socket descriptor.
1312  * @param IsSSL true if this socket expects SSL-encrypted data.
1313  * @returns     Accepted socket descriptor or -1 on error.
1314  */
1315 static int
1316 New_Connection(int Sock, UNUSED bool IsSSL)
1317 {
1318 #ifdef TCPWRAP
1319         struct request_info req;
1320 #endif
1321         ng_ipaddr_t new_addr;
1322         char ip_str[NG_INET_ADDRSTRLEN];
1323         int new_sock, new_sock_len;
1324         CLIENT *c;
1325         long cnt;
1326
1327         assert(Sock > NONE);
1328
1329         LogDebug("Accepting new connection on socket %d ...", Sock);
1330
1331         new_sock_len = (int)sizeof(new_addr);
1332         new_sock = accept(Sock, (struct sockaddr *)&new_addr,
1333                           (socklen_t *)&new_sock_len);
1334         if (new_sock < 0) {
1335                 Log(LOG_CRIT, "Can't accept connection: %s!", strerror(errno));
1336                 return -1;
1337         }
1338         NumConnectionsAccepted++;
1339
1340         if (!ng_ipaddr_tostr_r(&new_addr, ip_str)) {
1341                 Log(LOG_CRIT, "fd %d: Can't convert IP address!", new_sock);
1342                 Simple_Message(new_sock, "ERROR :Internal Server Error");
1343                 close(new_sock);
1344                 return -1;
1345         }
1346
1347 #ifdef TCPWRAP
1348         /* Validate socket using TCP Wrappers */
1349         request_init(&req, RQ_DAEMON, PACKAGE_NAME, RQ_FILE, new_sock,
1350                      RQ_CLIENT_SIN, &new_addr, NULL);
1351         fromhost(&req);
1352         if (!hosts_access(&req)) {
1353                 Log(deny_severity,
1354                     "Refused connection from %s (by TCP Wrappers)!", ip_str);
1355                 Simple_Message(new_sock, "ERROR :Connection refused");
1356                 close(new_sock);
1357                 return -1;
1358         }
1359 #endif
1360
1361         if (!Init_Socket(new_sock))
1362                 return -1;
1363
1364         /* Check global connection limit */
1365         if ((Conf_MaxConnections > 0) &&
1366             (NumConnections >= (size_t) Conf_MaxConnections)) {
1367                 Log(LOG_ALERT, "Can't accept new connection on socket %d: Limit (%d) reached!",
1368                     Sock, Conf_MaxConnections);
1369                 Simple_Message(new_sock, "ERROR :Connection limit reached");
1370                 close(new_sock);
1371                 return -1;
1372         }
1373
1374         /* Check IP-based connection limit */
1375         cnt = Count_Connections(&new_addr);
1376         if ((Conf_MaxConnectionsIP > 0) && (cnt >= Conf_MaxConnectionsIP)) {
1377                 /* Access denied, too many connections from this IP address! */
1378                 Log(LOG_ERR,
1379                     "Refused connection from %s: too may connections (%ld) from this IP address!",
1380                     ip_str, cnt);
1381                 Simple_Message(new_sock,
1382                                "ERROR :Connection refused, too many connections from your IP address");
1383                 close(new_sock);
1384                 return -1;
1385         }
1386
1387         if (Socket2Index(new_sock) <= NONE) {
1388                 Simple_Message(new_sock, "ERROR: Internal error");
1389                 close(new_sock);
1390                 return -1;
1391         }
1392
1393         /* register callback */
1394         if (!io_event_create(new_sock, IO_WANTREAD, cb_clientserver)) {
1395                 Log(LOG_ALERT,
1396                     "Can't accept connection: io_event_create failed!");
1397                 Simple_Message(new_sock, "ERROR :Internal error");
1398                 close(new_sock);
1399                 return -1;
1400         }
1401
1402         c = Client_NewLocal(new_sock, NULL, CLIENT_UNKNOWN, false);
1403         if (!c) {
1404                 Log(LOG_ALERT,
1405                     "Can't accept connection: can't create client structure!");
1406                 Simple_Message(new_sock, "ERROR :Internal error");
1407                 io_close(new_sock);
1408                 return -1;
1409         }
1410
1411         Init_Conn_Struct(new_sock);
1412         My_Connections[new_sock].sock = new_sock;
1413         My_Connections[new_sock].addr = new_addr;
1414         My_Connections[new_sock].client = c;
1415
1416         /* Set initial hostname to IP address. This becomes overwritten when
1417          * the DNS lookup is enabled and succeeds, but is used otherwise. */
1418         if (ng_ipaddr_af(&new_addr) != AF_INET)
1419                 snprintf(My_Connections[new_sock].host,
1420                          sizeof(My_Connections[new_sock].host), "[%s]", ip_str);
1421         else
1422                 strlcpy(My_Connections[new_sock].host, ip_str,
1423                         sizeof(My_Connections[new_sock].host));
1424
1425         Client_SetHostname(c, My_Connections[new_sock].host);
1426
1427         Log(LOG_INFO, "Accepted connection %d from \"%s:%d\" on socket %d.",
1428             new_sock, My_Connections[new_sock].host,
1429             ng_ipaddr_getport(&new_addr), Sock);
1430         Account_Connection();
1431
1432 #ifdef SSL_SUPPORT
1433         /* Delay connection initalization until SSL handshake is finished */
1434         if (!IsSSL)
1435 #endif
1436                 Conn_StartLogin(new_sock);
1437
1438         return new_sock;
1439 } /* New_Connection */
1440
1441 /**
1442  * Finish connection initialization, start resolver subprocess.
1443  *
1444  * @param Idx Connection index.
1445  */
1446 GLOBAL void
1447 Conn_StartLogin(CONN_ID Idx)
1448 {
1449         int ident_sock = -1;
1450
1451         assert(Idx >= 0);
1452
1453         /* Nothing to do if DNS (and resolver subprocess) is disabled */
1454         if (!Conf_DNS)
1455                 return;
1456
1457 #ifdef IDENTAUTH
1458         /* Should we make an IDENT request? */
1459         if (Conf_Ident)
1460                 ident_sock = My_Connections[Idx].sock;
1461 #endif
1462
1463         if (Conf_NoticeBeforeRegistration) {
1464                 /* Send "NOTICE *" messages to the client */
1465 #ifdef IDENTAUTH
1466                 if (Conf_Ident)
1467                         (void)Conn_WriteStr(Idx,
1468                                 "NOTICE * :*** Looking up your hostname and checking ident");
1469                 else
1470 #endif
1471                         (void)Conn_WriteStr(Idx,
1472                                 "NOTICE * :*** Looking up your hostname");
1473                 /* Send buffered data to the client, but break on errors
1474                  * because Handle_Write() would have closed the connection
1475                  * again in this case! */
1476                 if (!Handle_Write(Idx))
1477                         return;
1478         }
1479
1480         Resolve_Addr(&My_Connections[Idx].proc_stat, &My_Connections[Idx].addr,
1481                      ident_sock, cb_Read_Resolver_Result);
1482 }
1483
1484 /**
1485  * Update global connection counters.
1486  */
1487 static void
1488 Account_Connection(void)
1489 {
1490         NumConnections++;
1491         idle_t = 0;
1492         if (NumConnections > NumConnectionsMax)
1493                 NumConnectionsMax = NumConnections;
1494         LogDebug("Total number of connections now %lu (max %lu).",
1495                  NumConnections, NumConnectionsMax);
1496 } /* Account_Connection */
1497
1498 /**
1499  * Translate socket handle into connection index (for historical reasons, it is
1500  * a 1:1 mapping today) and enlarge the "connection pool" accordingly.
1501  *
1502  * @param Sock  Socket handle.
1503  * @returns     Connecion index or NONE when the pool is too small.
1504  */
1505 static CONN_ID
1506 Socket2Index( int Sock )
1507 {
1508         assert(Sock > 0);
1509         assert(Pool_Size >= 0);
1510
1511         if (Sock < Pool_Size)
1512                 return Sock;
1513
1514         /* Try to allocate more memory ... */
1515         if (!array_alloc(&My_ConnArray, sizeof(CONNECTION), (size_t)Sock)) {
1516                 Log(LOG_EMERG,
1517                     "Can't allocate memory to enlarge connection pool!");
1518                 return NONE;
1519         }
1520         LogDebug("Enlarged connection pool for %ld sockets (%ld items, %ld bytes)",
1521                  Sock, array_length(&My_ConnArray, sizeof(CONNECTION)),
1522                  array_bytes(&My_ConnArray));
1523
1524         /* Adjust pointer to new block, update size and initialize new items. */
1525         My_Connections = array_start(&My_ConnArray);
1526         while (Pool_Size <= Sock)
1527                 Init_Conn_Struct(Pool_Size++);
1528
1529         return Sock;
1530 }
1531
1532 /**
1533  * Read data from the network to the read buffer. If an error occurs,
1534  * the socket of this connection will be shut down.
1535  *
1536  * @param Idx   Connection index.
1537  */
1538 static void
1539 Read_Request( CONN_ID Idx )
1540 {
1541         ssize_t len;
1542         static const unsigned int maxbps = COMMAND_LEN / 2;
1543         char readbuf[READBUFFER_LEN];
1544         time_t t;
1545         CLIENT *c;
1546         assert( Idx > NONE );
1547         assert( My_Connections[Idx].sock > NONE );
1548
1549 #ifdef ZLIB
1550         if ((array_bytes(&My_Connections[Idx].rbuf) >= READBUFFER_LEN) ||
1551                 (array_bytes(&My_Connections[Idx].zip.rbuf) >= READBUFFER_LEN))
1552 #else
1553         if (array_bytes(&My_Connections[Idx].rbuf) >= READBUFFER_LEN)
1554 #endif
1555         {
1556                 /* Read buffer is full */
1557                 Log(LOG_ERR,
1558                     "Receive buffer space exhausted (connection %d): %d/%d bytes",
1559                     Idx, array_bytes(&My_Connections[Idx].rbuf), READBUFFER_LEN);
1560                 Conn_Close(Idx, "Receive buffer space exhausted", NULL, false);
1561                 return;
1562         }
1563
1564 #ifdef SSL_SUPPORT
1565         if (Conn_OPTION_ISSET(&My_Connections[Idx], CONN_SSL))
1566                 len = ConnSSL_Read( &My_Connections[Idx], readbuf, sizeof(readbuf));
1567         else
1568 #endif
1569         len = read(My_Connections[Idx].sock, readbuf, sizeof(readbuf));
1570         if (len == 0) {
1571                 LogDebug("Client \"%s:%u\" is closing connection %d ...",
1572                          My_Connections[Idx].host,
1573                          ng_ipaddr_getport(&My_Connections[Idx].addr), Idx);
1574                 Conn_Close(Idx, NULL, "Client closed connection", false);
1575                 return;
1576         }
1577
1578         if (len < 0) {
1579                 if( errno == EAGAIN ) return;
1580                 Log(LOG_ERR, "Read error on connection %d (socket %d): %s!",
1581                     Idx, My_Connections[Idx].sock, strerror(errno));
1582                 Conn_Close(Idx, "Read error", "Client closed connection",
1583                            false);
1584                 return;
1585         }
1586 #ifdef ZLIB
1587         if (Conn_OPTION_ISSET(&My_Connections[Idx], CONN_ZIP)) {
1588                 if (!array_catb(&My_Connections[Idx].zip.rbuf, readbuf,
1589                                 (size_t) len)) {
1590                         Log(LOG_ERR,
1591                             "Could not append received data to zip input buffer (connection %d): %d bytes!",
1592                             Idx, len);
1593                         Conn_Close(Idx, "Receive buffer space exhausted", NULL,
1594                                    false);
1595                         return;
1596                 }
1597         } else
1598 #endif
1599         {
1600                 if (!array_catb( &My_Connections[Idx].rbuf, readbuf, len)) {
1601                         Log(LOG_ERR,
1602                             "Could not append received data to input buffer (connection %d): %d bytes!",
1603                             Idx, len);
1604                         Conn_Close(Idx, "Receive buffer space exhausted", NULL,
1605                                    false );
1606                 }
1607         }
1608
1609         /* Update connection statistics */
1610         My_Connections[Idx].bytes_in += len;
1611
1612         /* Handle read buffer */
1613         My_Connections[Idx].bps += Handle_Buffer(Idx);
1614
1615         /* Make sure that there is still a valid client registered */
1616         c = Conn_GetClient(Idx);
1617         if (!c)
1618                 return;
1619
1620         /* Update timestamp of last data received if this connection is
1621          * registered as a user, server or service connection. Don't update
1622          * otherwise, so users have at least Conf_PongTimeout seconds time to
1623          * register with the IRC server -- see Check_Connections().
1624          * Update "lastping", too, if time shifted backwards ... */
1625         if (Client_Type(c) == CLIENT_USER
1626             || Client_Type(c) == CLIENT_SERVER
1627             || Client_Type(c) == CLIENT_SERVICE) {
1628                 t = time(NULL);
1629                 if (My_Connections[Idx].lastdata != t)
1630                         My_Connections[Idx].bps = 0;
1631
1632                 My_Connections[Idx].lastdata = t;
1633                 if (My_Connections[Idx].lastping > t)
1634                         My_Connections[Idx].lastping = t;
1635         }
1636
1637         /* Look at the data in the (read-) buffer of this connection */
1638         if (My_Connections[Idx].bps >= maxbps)
1639                 Throttle_Connection(Idx, c, THROTTLE_BPS, maxbps);
1640 } /* Read_Request */
1641
1642 /**
1643  * Handle all data in the connection read-buffer.
1644  *
1645  * Data is processed until no complete command is left in the read buffer,
1646  * or MAX_COMMANDS[_SERVER|_SERVICE] commands were processed.
1647  * When a fatal error occurs, the connection is shut down.
1648  *
1649  * @param Idx   Index of the connection.
1650  * @returns     Number of bytes processed.
1651  */
1652 static unsigned int
1653 Handle_Buffer(CONN_ID Idx)
1654 {
1655 #ifndef STRICT_RFC
1656         char *ptr1, *ptr2, *first_eol;
1657 #endif
1658         char *ptr;
1659         size_t len, delta;
1660         time_t starttime;
1661 #ifdef ZLIB
1662         bool old_z;
1663 #endif
1664         unsigned int i, maxcmd = MAX_COMMANDS, len_processed = 0;
1665         CLIENT *c;
1666
1667         c = Conn_GetClient(Idx);
1668         starttime = time(NULL);
1669
1670         assert(c != NULL);
1671
1672         /* Servers get special command limits that depend on the user count */
1673         switch (Client_Type(c)) {
1674             case CLIENT_SERVER:
1675                 maxcmd = (int)(Client_UserCount() / 5)
1676                        + MAX_COMMANDS_SERVER_MIN;
1677                 /* Allow servers to handle even more commands while peering
1678                  * to speed up server login and network synchronization. */
1679                 if (Conn_LastPing(Idx) == 0)
1680                         maxcmd *= 5;
1681                 break;
1682             case CLIENT_SERVICE:
1683                 maxcmd = MAX_COMMANDS_SERVICE;
1684                 break;
1685             case CLIENT_USER:
1686                 if (Client_HasMode(c, 'F'))
1687                         maxcmd = MAX_COMMANDS_SERVICE;
1688                 break;
1689         }
1690
1691         for (i=0; i < maxcmd; i++) {
1692                 /* Check penalty */
1693                 if (My_Connections[Idx].delaytime > starttime)
1694                         return 0;
1695 #ifdef ZLIB
1696                 /* Unpack compressed data, if compression is in use */
1697                 if (Conn_OPTION_ISSET(&My_Connections[Idx], CONN_ZIP)) {
1698                         /* When unzipping fails, Unzip_Buffer() shuts
1699                          * down the connection itself */
1700                         if (!Unzip_Buffer(Idx))
1701                                 return 0;
1702                 }
1703 #endif
1704
1705                 if (0 == array_bytes(&My_Connections[Idx].rbuf))
1706                         break;
1707
1708                 /* Make sure that the buffer is NULL terminated */
1709                 if (!array_cat0_temporary(&My_Connections[Idx].rbuf)) {
1710                         Conn_Close(Idx, NULL,
1711                                    "Can't allocate memory [Handle_Buffer]",
1712                                    true);
1713                         return 0;
1714                 }
1715
1716                 /* RFC 2812, section "2.3 Messages", 5th paragraph:
1717                  * "IRC messages are always lines of characters terminated
1718                  * with a CR-LF (Carriage Return - Line Feed) pair [...]". */
1719                 delta = 2;
1720                 ptr = strstr(array_start(&My_Connections[Idx].rbuf), "\r\n");
1721
1722 #ifndef STRICT_RFC
1723                 /* Check for non-RFC-compliant request (only CR or LF)?
1724                  * Unfortunately, there are quite a few clients out there
1725                  * that do this -- e. g. mIRC, BitchX, and Trillian :-( */
1726                 ptr1 = strchr(array_start(&My_Connections[Idx].rbuf), '\r');
1727                 ptr2 = strchr(array_start(&My_Connections[Idx].rbuf), '\n');
1728                 if (ptr) {
1729                         /* Check if there is a single CR or LF _before_ the
1730                          * correct CR+LF line terminator:  */
1731                         first_eol = ptr1 < ptr2 ? ptr1 : ptr2;
1732                         if (first_eol < ptr) {
1733                                 /* Single CR or LF before CR+LF found */
1734                                 ptr = first_eol;
1735                                 delta = 1;
1736                         }
1737                 } else if (ptr1 || ptr2) {
1738                         /* No CR+LF terminated command found, but single
1739                          * CR or LF found ... */
1740                         if (ptr1 && ptr2)
1741                                 ptr = ptr1 < ptr2 ? ptr1 : ptr2;
1742                         else
1743                                 ptr = ptr1 ? ptr1 : ptr2;
1744                         delta = 1;
1745                 }
1746 #endif
1747
1748                 if (!ptr)
1749                         break;
1750
1751                 /* Complete (=line terminated) request found, handle it! */
1752                 *ptr = '\0';
1753
1754                 len = ptr - (char *)array_start(&My_Connections[Idx].rbuf) + delta;
1755
1756                 if (len > (COMMAND_LEN - 1)) {
1757                         /* Request must not exceed 512 chars (incl. CR+LF!),
1758                          * see RFC 2812. Disconnect Client if this happens. */
1759                         Log(LOG_ERR,
1760                             "Request too long (connection %d): %d bytes (max. %d expected)!",
1761                             Idx, array_bytes(&My_Connections[Idx].rbuf),
1762                             COMMAND_LEN - 1);
1763                         Conn_Close(Idx, NULL, "Request too long", true);
1764                         return 0;
1765                 }
1766
1767                 len_processed += (unsigned int)len;
1768                 if (len <= delta) {
1769                         /* Request is empty (only '\r\n', '\r' or '\n');
1770                          * delta is 2 ('\r\n') or 1 ('\r' or '\n'), see above */
1771                         array_moveleft(&My_Connections[Idx].rbuf, 1, len);
1772                         continue;
1773                 }
1774 #ifdef ZLIB
1775                 /* remember if stream is already compressed */
1776                 old_z = My_Connections[Idx].options & CONN_ZIP;
1777 #endif
1778
1779                 My_Connections[Idx].msg_in++;
1780                 if (!Parse_Request
1781                     (Idx, (char *)array_start(&My_Connections[Idx].rbuf)))
1782                         return 0; /* error -> connection has been closed */
1783
1784                 array_moveleft(&My_Connections[Idx].rbuf, 1, len);
1785 #ifdef ZLIB
1786                 if ((!old_z) && (My_Connections[Idx].options & CONN_ZIP) &&
1787                     (array_bytes(&My_Connections[Idx].rbuf) > 0)) {
1788                         /* The last command activated socket compression.
1789                          * Data that was read after that needs to be copied
1790                          * to the unzip buffer for decompression: */
1791                         if (!array_copy
1792                             (&My_Connections[Idx].zip.rbuf,
1793                              &My_Connections[Idx].rbuf)) {
1794                                 Conn_Close(Idx, NULL,
1795                                            "Can't allocate memory [Handle_Buffer]",
1796                                            true);
1797                                 return 0;
1798                         }
1799
1800                         array_trunc(&My_Connections[Idx].rbuf);
1801                         LogDebug
1802                             ("Moved already received data (%u bytes) to uncompression buffer.",
1803                              array_bytes(&My_Connections[Idx].zip.rbuf));
1804                 }
1805 #endif
1806         }
1807 #if DEBUG_BUFFER
1808         LogDebug("Connection %d: Processed %ld commands (max=%ld), %ld bytes. %ld bytes left in read buffer.",
1809                  Idx, i, maxcmd, len_processed,
1810                  array_bytes(&My_Connections[Idx].rbuf));
1811 #endif
1812
1813         /* If data has been processed but there is still data in the read
1814          * buffer, the command limit triggered. Enforce the penalty time: */
1815         if (len_processed && array_bytes(&My_Connections[Idx].rbuf) > 2)
1816                 Throttle_Connection(Idx, c, THROTTLE_CMDS, maxcmd);
1817
1818         return len_processed;
1819 } /* Handle_Buffer */
1820
1821 /**
1822  * Check whether established connections are still alive or not.
1823  * If not, play PING-PONG first; and if that doesn't help either,
1824  * disconnect the respective peer.
1825  */
1826 static void
1827 Check_Connections(void)
1828 {
1829         CLIENT *c;
1830         CONN_ID i;
1831         char msg[64];
1832
1833         for (i = 0; i < Pool_Size; i++) {
1834                 if (My_Connections[i].sock < 0)
1835                         continue;
1836
1837                 c = Conn_GetClient(i);
1838                 if (c && ((Client_Type(c) == CLIENT_USER)
1839                           || (Client_Type(c) == CLIENT_SERVER)
1840                           || (Client_Type(c) == CLIENT_SERVICE))) {
1841                         /* connected User, Server or Service */
1842                         if (My_Connections[i].lastping >
1843                             My_Connections[i].lastdata) {
1844                                 /* We already sent a ping */
1845                                 if (My_Connections[i].lastping <
1846                                     time(NULL) - Conf_PongTimeout) {
1847                                         /* Timeout */
1848                                         snprintf(msg, sizeof(msg),
1849                                                  "Ping timeout: %d seconds",
1850                                                  Conf_PongTimeout);
1851                                         LogDebug("Connection %d: %s.", i, msg);
1852                                         Conn_Close(i, NULL, msg, true);
1853                                 }
1854                         } else if (My_Connections[i].lastdata <
1855                                    time(NULL) - Conf_PingTimeout) {
1856                                 /* We need to send a PING ... */
1857                                 LogDebug("Connection %d: sending PING ...", i);
1858                                 Conn_UpdatePing(i);
1859                                 Conn_WriteStr(i, "PING :%s",
1860                                               Client_ID(Client_ThisServer()));
1861                         }
1862                 } else {
1863                         /* The connection is not fully established yet, so
1864                          * we don't do the PING-PONG game here but instead
1865                          * disconnect the client after "a short time" if it's
1866                          * still not registered. */
1867
1868                         if (My_Connections[i].lastdata <
1869                             time(NULL) - Conf_PongTimeout) {
1870                                 LogDebug
1871                                     ("Unregistered connection %d timed out ...",
1872                                      i);
1873                                 Conn_Close(i, NULL, "Timeout", false);
1874                         }
1875                 }
1876         }
1877 } /* Check_Connections */
1878
1879 /**
1880  * Check if further server links should be established.
1881  */
1882 static void
1883 Check_Servers(void)
1884 {
1885         int i, n;
1886         time_t time_now;
1887
1888         time_now = time(NULL);
1889
1890         /* Check all configured servers */
1891         for (i = 0; i < MAX_SERVERS; i++) {
1892                 if (Conf_Server[i].conn_id != NONE)
1893                         continue;       /* Already establishing or connected */
1894                 if (!Conf_Server[i].host[0] || Conf_Server[i].port <= 0)
1895                         continue;       /* No host and/or port configured */
1896                 if (Conf_Server[i].flags & CONF_SFLAG_DISABLED)
1897                         continue;       /* Disabled configuration entry */
1898                 if (Conf_Server[i].lasttry > (time_now - Conf_ConnectRetry))
1899                         continue;       /* We have to wait a little bit ... */
1900
1901                 /* Is there already a connection in this group? */
1902                 if (Conf_Server[i].group > NONE) {
1903                         for (n = 0; n < MAX_SERVERS; n++) {
1904                                 if (n == i)
1905                                         continue;
1906                                 if ((Conf_Server[n].conn_id != NONE) &&
1907                                     (Conf_Server[n].group == Conf_Server[i].group))
1908                                         break;
1909                         }
1910                         if (n < MAX_SERVERS)
1911                                 continue;
1912                 }
1913
1914                 /* Okay, try to connect now */
1915                 Log(LOG_NOTICE,
1916                     "Preparing to establish a new server link for \"%s\" ...",
1917                     Conf_Server[i].name);
1918                 Conf_Server[i].lasttry = time_now;
1919                 Conf_Server[i].conn_id = SERVER_WAIT;
1920                 assert(Proc_GetPipeFd(&Conf_Server[i].res_stat) < 0);
1921
1922                 /* Start resolver subprocess ... */
1923                 if (!Resolve_Name(&Conf_Server[i].res_stat, Conf_Server[i].host,
1924                                   cb_Connect_to_Server))
1925                         Conf_Server[i].conn_id = NONE;
1926         }
1927 } /* Check_Servers */
1928
1929 /**
1930  * Establish a new outgoing server connection.
1931  *
1932  * @param Server        Configuration index of the server.
1933  * @param dest          Destination IP address to connect to.
1934  */
1935 static void
1936 New_Server( int Server , ng_ipaddr_t *dest)
1937 {
1938         /* Establish new server link */
1939         char ip_str[NG_INET_ADDRSTRLEN];
1940         int af_dest, res, new_sock;
1941         CLIENT *c;
1942
1943         assert( Server > NONE );
1944
1945         /* Make sure that the remote server hasn't re-linked to this server
1946          * asynchronously on its own */
1947         if (Conf_Server[Server].conn_id > NONE) {
1948                 Log(LOG_INFO,
1949                         "Connection to \"%s\" meanwhile re-established, aborting preparation.");
1950                 return;
1951         }
1952
1953         if (!ng_ipaddr_tostr_r(dest, ip_str)) {
1954                 Log(LOG_WARNING, "New_Server: Could not convert IP to string");
1955                 Conf_Server[Server].conn_id = NONE;
1956                 return;
1957         }
1958
1959         af_dest = ng_ipaddr_af(dest);
1960         new_sock = socket(af_dest, SOCK_STREAM, 0);
1961
1962         Log(LOG_INFO,
1963             "Establishing connection for \"%s\" to \"%s:%d\" (%s), socket %d ...",
1964             Conf_Server[Server].name, Conf_Server[Server].host,
1965             Conf_Server[Server].port, ip_str, new_sock);
1966
1967         if (new_sock < 0) {
1968                 Log(LOG_CRIT, "Can't create socket (af %d): %s!",
1969                     af_dest, strerror(errno));
1970                 Conf_Server[Server].conn_id = NONE;
1971                 return;
1972         }
1973
1974         if (!Init_Socket(new_sock)) {
1975                 Conf_Server[Server].conn_id = NONE;
1976                 return;
1977         }
1978
1979         /* is a bind address configured? */
1980         res = ng_ipaddr_af(&Conf_Server[Server].bind_addr);
1981
1982         /* if yes, bind now. If it fails, warn and let connect() pick a
1983          * source address */
1984         if (res && bind(new_sock, (struct sockaddr *) &Conf_Server[Server].bind_addr,
1985                                 ng_ipaddr_salen(&Conf_Server[Server].bind_addr)))
1986         {
1987                 ng_ipaddr_tostr_r(&Conf_Server[Server].bind_addr, ip_str);
1988                 Log(LOG_WARNING, "Can't bind socket to %s: %s!", ip_str,
1989                     strerror(errno));
1990         }
1991         ng_ipaddr_setport(dest, Conf_Server[Server].port);
1992         res = connect(new_sock, (struct sockaddr *) dest, ng_ipaddr_salen(dest));
1993         if(( res != 0 ) && ( errno != EINPROGRESS )) {
1994                 Log( LOG_CRIT, "Can't connect socket: %s!", strerror( errno ));
1995                 close( new_sock );
1996                 Conf_Server[Server].conn_id = NONE;
1997                 return;
1998         }
1999
2000         if (Socket2Index(new_sock) <= NONE) {
2001                 close( new_sock );
2002                 Conf_Server[Server].conn_id = NONE;
2003                 return;
2004         }
2005
2006         if (!io_event_create( new_sock, IO_WANTWRITE, cb_connserver)) {
2007                 Log(LOG_ALERT, "io_event_create(): could not add fd %d",
2008                     strerror(errno));
2009                 close(new_sock);
2010                 Conf_Server[Server].conn_id = NONE;
2011                 return;
2012         }
2013
2014         assert(My_Connections[new_sock].sock <= 0);
2015
2016         Init_Conn_Struct(new_sock);
2017
2018         ng_ipaddr_tostr_r(dest, ip_str);
2019         c = Client_NewLocal(new_sock, ip_str, CLIENT_UNKNOWNSERVER, false);
2020         if (!c) {
2021                 Log( LOG_ALERT, "Can't establish connection: can't create client structure!" );
2022                 io_close(new_sock);
2023                 Conf_Server[Server].conn_id = NONE;
2024                 return;
2025         }
2026
2027         /* Conn_Close() decrements this counter again */
2028         Account_Connection();
2029         Client_SetIntroducer( c, c );
2030         Client_SetToken( c, TOKEN_OUTBOUND );
2031
2032         /* Register connection */
2033         if (!Conf_SetServer(Server, new_sock))
2034                 return;
2035         My_Connections[new_sock].sock = new_sock;
2036         My_Connections[new_sock].addr = *dest;
2037         My_Connections[new_sock].client = c;
2038         strlcpy( My_Connections[new_sock].host, Conf_Server[Server].host,
2039                                 sizeof(My_Connections[new_sock].host ));
2040
2041 #ifdef SSL_SUPPORT
2042         if (Conf_Server[Server].SSLConnect &&
2043             !ConnSSL_PrepareConnect(&My_Connections[new_sock], &Conf_Server[Server]))
2044         {
2045                 Log(LOG_ALERT, "Could not initialize SSL for outgoing connection");
2046                 Conn_Close(new_sock, "Could not initialize SSL for outgoing connection",
2047                            NULL, false);
2048                 Init_Conn_Struct(new_sock);
2049                 Conf_Server[Server].conn_id = NONE;
2050                 return;
2051         }
2052 #endif
2053         LogDebug("Registered new connection %d on socket %d (%ld in total).",
2054                  new_sock, My_Connections[new_sock].sock, NumConnections);
2055         Conn_OPTION_ADD( &My_Connections[new_sock], CONN_ISCONNECTING );
2056 } /* New_Server */
2057
2058 /**
2059  * Initialize connection structure.
2060  *
2061  * @param Idx   Connection index.
2062  */
2063 static void
2064 Init_Conn_Struct(CONN_ID Idx)
2065 {
2066         time_t now = time(NULL);
2067
2068         memset(&My_Connections[Idx], 0, sizeof(CONNECTION));
2069         My_Connections[Idx].sock = -1;
2070         My_Connections[Idx].signon = now;
2071         My_Connections[Idx].lastdata = now;
2072         My_Connections[Idx].lastprivmsg = now;
2073         Proc_InitStruct(&My_Connections[Idx].proc_stat);
2074
2075 #ifdef ICONV
2076         My_Connections[Idx].iconv_from = (iconv_t)(-1);
2077         My_Connections[Idx].iconv_to = (iconv_t)(-1);
2078 #endif
2079 } /* Init_Conn_Struct */
2080
2081 /**
2082  * Initialize options of a new socket.
2083  *
2084  * For example, we try to set socket options SO_REUSEADDR and IPTOS_LOWDELAY.
2085  * The socket is automatically closed if a fatal error is encountered.
2086  *
2087  * @param Sock  Socket handle.
2088  * @returns false if socket was closed due to fatal error.
2089  */
2090 static bool
2091 Init_Socket( int Sock )
2092 {
2093         int value;
2094
2095         if (!io_setnonblock(Sock)) {
2096                 Log(LOG_CRIT, "Can't enable non-blocking mode for socket: %s!",
2097                     strerror(errno));
2098                 close(Sock);
2099                 return false;
2100         }
2101
2102         /* Don't block this port after socket shutdown */
2103         value = 1;
2104         if (setsockopt(Sock, SOL_SOCKET, SO_REUSEADDR, &value,
2105                        (socklen_t)sizeof(value)) != 0) {
2106                 Log(LOG_ERR, "Can't set socket option SO_REUSEADDR: %s!",
2107                     strerror(errno));
2108                 /* ignore this error */
2109         }
2110
2111         /* Set type of service (TOS) */
2112 #if defined(IPPROTO_IP) && defined(IPTOS_LOWDELAY)
2113         value = IPTOS_LOWDELAY;
2114         if (setsockopt(Sock, IPPROTO_IP, IP_TOS, &value,
2115                        (socklen_t) sizeof(value))) {
2116                 LogDebug("Can't set socket option IP_TOS: %s!",
2117                          strerror(errno));
2118                 /* ignore this error */
2119         } else
2120                 LogDebug("IP_TOS on socket %d has been set to IPTOS_LOWDELAY.",
2121                          Sock);
2122 #endif
2123
2124         return true;
2125 } /* Init_Socket */
2126
2127 /**
2128  * Read results of a resolver sub-process and try to initiate a new server
2129  * connection.
2130  *
2131  * @param fd            File descriptor of the pipe to the sub-process.
2132  * @param events        (ignored IO specification)
2133  */
2134 static void
2135 cb_Connect_to_Server(int fd, UNUSED short events)
2136 {
2137         int i;
2138         size_t len;
2139
2140         /* we can handle at most 3 addresses; but we read up to 4 so we can
2141          * log the 'more than we can handle' condition. First result is tried
2142          * immediately, rest is saved for later if needed. */
2143         ng_ipaddr_t dest_addrs[4];
2144
2145         LogDebug("Resolver: Got forward lookup callback on fd %d, events %d",
2146                  fd, events);
2147
2148         for (i=0; i < MAX_SERVERS; i++) {
2149                   if (Proc_GetPipeFd(&Conf_Server[i].res_stat) == fd )
2150                           break;
2151         }
2152
2153         if( i >= MAX_SERVERS) {
2154                 /* Ops, no matching server found?! */
2155                 io_close( fd );
2156                 LogDebug("Resolver: Got Forward Lookup callback for unknown server!?");
2157                 return;
2158         }
2159
2160         /* Read result from pipe */
2161         len = Proc_Read(&Conf_Server[i].res_stat, dest_addrs, sizeof(dest_addrs));
2162         Proc_Close(&Conf_Server[i].res_stat);
2163         if (len == 0) {
2164                 /* Error resolving hostname: reset server structure */
2165                 Conf_Server[i].conn_id = NONE;
2166                 return;
2167         }
2168
2169         assert((len % sizeof(ng_ipaddr_t)) == 0);
2170
2171         LogDebug("Got result from resolver: %u structs (%u bytes).",
2172                  len/sizeof(ng_ipaddr_t), len);
2173
2174         memset(&Conf_Server[i].dst_addr, 0, sizeof(Conf_Server[i].dst_addr));
2175         if (len > sizeof(ng_ipaddr_t)) {
2176                 /* more than one address for this hostname, remember them
2177                  * in case first address is unreachable/not available */
2178                 len -= sizeof(ng_ipaddr_t);
2179                 if (len > sizeof(Conf_Server[i].dst_addr)) {
2180                         len = sizeof(Conf_Server[i].dst_addr);
2181                         Log(LOG_NOTICE,
2182                                 "Notice: Resolver returned more IP Addresses for host than we can handle, additional addresses dropped.");
2183                 }
2184                 memcpy(&Conf_Server[i].dst_addr, &dest_addrs[1], len);
2185         }
2186         /* connect() */
2187         New_Server(i, dest_addrs);
2188 } /* cb_Read_Forward_Lookup */
2189
2190 /**
2191  * Read results of a resolver sub-process from the pipe and update the
2192  * appropriate connection/client structure(s): hostname and/or IDENT user name.
2193  *
2194  * @param r_fd          File descriptor of the pipe to the sub-process.
2195  * @param events        (ignored IO specification)
2196  */
2197 static void
2198 cb_Read_Resolver_Result( int r_fd, UNUSED short events )
2199 {
2200         CLIENT *c;
2201         CONN_ID i;
2202         size_t len;
2203         char *identptr;
2204 #ifdef IDENTAUTH
2205         char readbuf[HOST_LEN + 2 + CLIENT_USER_LEN];
2206         char *ptr;
2207 #else
2208         char readbuf[HOST_LEN + 1];
2209 #endif
2210
2211         LogDebug("Resolver: Got callback on fd %d, events %d", r_fd, events );
2212         i = Conn_GetFromProc(r_fd);
2213         if (i == NONE) {
2214                 /* Ops, none found? Probably the connection has already
2215                  * been closed!? We'll ignore that ... */
2216                 io_close( r_fd );
2217                 LogDebug("Resolver: Got callback for unknown connection!?");
2218                 return;
2219         }
2220
2221         /* Read result from pipe */
2222         len = Proc_Read(&My_Connections[i].proc_stat, readbuf, sizeof readbuf -1);
2223         Proc_Close(&My_Connections[i].proc_stat);
2224         if (len == 0)
2225                 return;
2226
2227         readbuf[len] = '\0';
2228         identptr = strchr(readbuf, '\n');
2229         assert(identptr != NULL);
2230         if (!identptr) {
2231                 Log( LOG_CRIT, "Resolver: Got malformed result!");
2232                 return;
2233         }
2234
2235         *identptr = '\0';
2236         LogDebug("Got result from resolver: \"%s\" (%u bytes read).", readbuf, len);
2237         /* Okay, we got a complete result: this is a host name for outgoing
2238          * connections and a host name and IDENT user name (if enabled) for
2239          * incoming connections.*/
2240         assert ( My_Connections[i].sock >= 0 );
2241         /* Incoming connection. Search client ... */
2242         c = Conn_GetClient( i );
2243         assert( c != NULL );
2244
2245         /* Only update client information of unregistered clients.
2246          * Note: user commands (e. g. WEBIRC) are always read _after_ reading
2247          * the resolver results, so we don't have to worry to override settings
2248          * from these commands here. */
2249         if(Client_Type(c) == CLIENT_UNKNOWN) {
2250                 strlcpy(My_Connections[i].host, readbuf,
2251                         sizeof(My_Connections[i].host));
2252                 Client_SetHostname(c, readbuf);
2253                 if (Conf_NoticeBeforeRegistration)
2254                         (void)Conn_WriteStr(i,
2255                                         "NOTICE * :*** Found your hostname: %s",
2256                                         My_Connections[i].host);
2257 #ifdef IDENTAUTH
2258                 ++identptr;
2259                 if (*identptr) {
2260                         ptr = identptr;
2261                         while (*ptr) {
2262                                 if ((*ptr < '0' || *ptr > '9') &&
2263                                     (*ptr < 'A' || *ptr > 'Z') &&
2264                                     (*ptr < 'a' || *ptr > 'z'))
2265                                         break;
2266                                 ptr++;
2267                         }
2268                         if (*ptr) {
2269                                 /* Erroneous IDENT reply */
2270                                 Log(LOG_NOTICE,
2271                                     "Got invalid IDENT reply for connection %d! Ignored.",
2272                                     i);
2273                         } else {
2274                                 Log(LOG_INFO,
2275                                     "IDENT lookup for connection %d: \"%s\".",
2276                                     i, identptr);
2277                                 Client_SetUser(c, identptr, true);
2278                         }
2279                         if (Conf_NoticeBeforeRegistration) {
2280                                 (void)Conn_WriteStr(i,
2281                                         "NOTICE * :*** Got %sident response%s%s",
2282                                         *ptr ? "invalid " : "",
2283                                         *ptr ? "" : ": ",
2284                                         *ptr ? "" : identptr);
2285                         }
2286                 } else if(Conf_Ident) {
2287                         Log(LOG_INFO, "IDENT lookup for connection %d: no result.", i);
2288                         if (Conf_NoticeBeforeRegistration)
2289                                 (void)Conn_WriteStr(i,
2290                                         "NOTICE * :*** No ident response");
2291                 }
2292 #endif
2293
2294                 if (Conf_NoticeBeforeRegistration) {
2295                         /* Send buffered data to the client, but break on
2296                          * errors because Handle_Write() would have closed
2297                          * the connection again in this case! */
2298                         if (!Handle_Write(i))
2299                                 return;
2300                 }
2301
2302                 Class_HandleServerBans(c);
2303         }
2304 #ifdef DEBUG
2305         else
2306                 LogDebug("Resolver: discarding result for already registered connection %d.", i);
2307 #endif
2308 } /* cb_Read_Resolver_Result */
2309
2310 /**
2311  * Write a "simple" (error) message to a socket.
2312  *
2313  * The message is sent without using the connection write buffers, without
2314  * compression/encryption, and even without any error reporting. It is
2315  * designed for error messages of e.g. New_Connection().
2316  *
2317  * @param Sock  Socket handle.
2318  * @param Msg   Message string to send.
2319  */
2320 static void
2321 Simple_Message(int Sock, const char *Msg)
2322 {
2323         char buf[COMMAND_LEN];
2324         size_t len;
2325
2326         assert(Sock > NONE);
2327         assert(Msg != NULL);
2328
2329         strlcpy(buf, Msg, sizeof buf - 2);
2330         len = strlcat(buf, "\r\n", sizeof buf);
2331         if (write(Sock, buf, len) < 0) {
2332                 /* Because this function most probably got called to log
2333                  * an error message, any write error is ignored here to
2334                  * avoid an endless loop. But casting the result of write()
2335                  * to "void" doesn't satisfy the GNU C code attribute
2336                  * "warn_unused_result" which is used by some versions of
2337                  * glibc (e.g. 2.11.1), therefore this silly error
2338                  * "handling" code here :-( */
2339                 return;
2340         }
2341 } /* Simple_Error */
2342
2343 /**
2344  * Get CLIENT structure that belongs to a local connection identified by its
2345  * index number. Each connection belongs to a client by definition, so it is
2346  * not required that the caller checks for NULL return values.
2347  *
2348  * @param Idx   Connection index number.
2349  * @returns     Pointer to CLIENT structure.
2350  */
2351 GLOBAL CLIENT *
2352 Conn_GetClient( CONN_ID Idx )
2353 {
2354         CONNECTION *c;
2355
2356         assert(Idx >= 0);
2357         c = array_get(&My_ConnArray, sizeof (CONNECTION), (size_t)Idx);
2358         assert(c != NULL);
2359         return c ? c->client : NULL;
2360 }
2361
2362 /**
2363  * Get PROC_STAT sub-process structure of a connection.
2364  *
2365  * @param Idx   Connection index number.
2366  * @returns     PROC_STAT structure.
2367  */
2368 GLOBAL PROC_STAT *
2369 Conn_GetProcStat(CONN_ID Idx)
2370 {
2371         CONNECTION *c;
2372
2373         assert(Idx >= 0);
2374         c = array_get(&My_ConnArray, sizeof (CONNECTION), (size_t)Idx);
2375         assert(c != NULL);
2376         return &c->proc_stat;
2377 } /* Conn_GetProcStat */
2378
2379 /**
2380  * Get CONN_ID from file descriptor associated to a subprocess structure.
2381  *
2382  * @param fd    File descriptor.
2383  * @returns     CONN_ID or NONE (-1).
2384  */
2385 GLOBAL CONN_ID
2386 Conn_GetFromProc(int fd)
2387 {
2388         int i;
2389
2390         assert(fd > 0);
2391         for (i = 0; i < Pool_Size; i++) {
2392                 if ((My_Connections[i].sock != NONE)
2393                     && (Proc_GetPipeFd(&My_Connections[i].proc_stat) == fd))
2394                         return i;
2395         }
2396         return NONE;
2397 } /* Conn_GetFromProc */
2398
2399 /**
2400  * Throttle a connection because of excessive usage.
2401  *
2402  * @param Reason The reason, see THROTTLE_xxx constants.
2403  * @param Idx The connection index.
2404  * @param Client The client of this connection.
2405  * @param Seconds The time to delay this connection.
2406  */
2407 static void
2408 Throttle_Connection(const CONN_ID Idx, CLIENT *Client, const int Reason,
2409                     unsigned int Value)
2410 {
2411         assert(Idx > NONE);
2412         assert(Client != NULL);
2413
2414         /* Never throttle servers or services, only interrupt processing */
2415         if (Client_Type(Client) == CLIENT_SERVER
2416             || Client_Type(Client) == CLIENT_UNKNOWNSERVER
2417             || Client_Type(Client) == CLIENT_SERVICE)
2418                 return;
2419
2420         /* Don't throttle clients with user mode 'F' set */
2421         if (Client_HasMode(Client, 'F'))
2422                 return;
2423
2424         LogDebug("Throttling connection %d: code %d, value %d!", Idx,
2425                  Reason, Value);
2426         Conn_SetPenalty(Idx, 1);
2427 }
2428
2429 #ifndef STRICT_RFC
2430
2431 GLOBAL long
2432 Conn_GetAuthPing(CONN_ID Idx)
2433 {
2434         assert (Idx != NONE);
2435         return My_Connections[Idx].auth_ping;
2436 } /* Conn_GetAuthPing */
2437
2438 GLOBAL void
2439 Conn_SetAuthPing(CONN_ID Idx, long ID)
2440 {
2441         assert (Idx != NONE);
2442         My_Connections[Idx].auth_ping = ID;
2443 } /* Conn_SetAuthPing */
2444
2445 #endif /* STRICT_RFC */
2446
2447 #ifdef SSL_SUPPORT
2448
2449 /**
2450  * IO callback for new SSL-enabled client and server connections.
2451  *
2452  * @param sock  Socket descriptor.
2453  * @param what  IO specification (IO_WANTREAD/IO_WANTWRITE/...).
2454  */
2455 static void
2456 cb_clientserver_ssl(int sock, UNUSED short what)
2457 {
2458         CONN_ID idx = Socket2Index(sock);
2459
2460         if (idx <= NONE) {
2461                 io_close(sock);
2462                 return;
2463         }
2464
2465         switch (ConnSSL_Accept(&My_Connections[idx])) {
2466                 case 1:
2467                         break;  /* OK */
2468                 case 0:
2469                         return; /* EAGAIN: callback will be invoked again by IO layer */
2470                 default:
2471                         Conn_Close(idx,
2472                                    "SSL accept error, closing socket", "SSL accept error",
2473                                    false);
2474                         return;
2475         }
2476
2477         io_event_setcb(sock, cb_clientserver);  /* SSL handshake completed */
2478 }
2479
2480 /**
2481  * IO callback for listening SSL sockets: handle new connections. This callback
2482  * gets called when a new SSL-enabled connection should be accepted.
2483  *
2484  * @param sock          Socket descriptor.
2485  * @param irrelevant    (ignored IO specification)
2486  */
2487 static void
2488 cb_listen_ssl(int sock, short irrelevant)
2489 {
2490         int fd;
2491
2492         (void) irrelevant;
2493         fd = New_Connection(sock, true);
2494         if (fd < 0)
2495                 return;
2496         io_event_setcb(My_Connections[fd].sock, cb_clientserver_ssl);
2497 }
2498
2499 /**
2500  * IO callback for new outgoing SSL-enabled server connections.
2501  *
2502  * @param sock          Socket descriptor.
2503  * @param unused        (ignored IO specification)
2504  */
2505 static void
2506 cb_connserver_login_ssl(int sock, short unused)
2507 {
2508         CONN_ID idx = Socket2Index(sock);
2509
2510         (void) unused;
2511
2512         if (idx <= NONE) {
2513                 io_close(sock);
2514                 return;
2515         }
2516
2517         switch (ConnSSL_Connect( &My_Connections[idx])) {
2518                 case 1: break;
2519                 case 0: LogDebug("ConnSSL_Connect: not ready");
2520                         return;
2521                 case -1:
2522                         Log(LOG_ERR, "SSL connection on socket %d failed!", sock);
2523                         Conn_Close(idx, "Can't connect", NULL, false);
2524                         return;
2525         }
2526
2527         Log( LOG_INFO, "SSL connection %d with \"%s:%d\" established.", idx,
2528             My_Connections[idx].host, Conf_Server[Conf_GetServer( idx )].port );
2529
2530         server_login(idx);
2531 }
2532
2533
2534 /**
2535  * Check if SSL library needs to read SSL-protocol related data.
2536  *
2537  * SSL/TLS connections require extra treatment:
2538  * When either CONN_SSL_WANT_WRITE or CONN_SSL_WANT_READ is set, we
2539  * need to take care of that first, before checking read/write buffers.
2540  * For instance, while we might have data in our write buffer, the
2541  * TLS/SSL protocol might need to read internal data first for TLS/SSL
2542  * writes to succeed.
2543  *
2544  * If this function returns true, such a condition is met and we have
2545  * to reverse the condition (check for read even if we've data to write,
2546  * do not check for read but writeability even if write-buffer is empty).
2547  *
2548  * @param c     Connection to check.
2549  * @returns     true if SSL-library has to read protocol data.
2550  */
2551 static bool
2552 SSL_WantRead(const CONNECTION *c)
2553 {
2554         if (Conn_OPTION_ISSET(c, CONN_SSL_WANT_READ)) {
2555                 io_event_add(c->sock, IO_WANTREAD);
2556                 return true;
2557         }
2558         return false;
2559 }
2560
2561 /**
2562  * Check if SSL library needs to write SSL-protocol related data.
2563  *
2564  * Please see description of SSL_WantRead() for full description!
2565  *
2566  * @param c     Connection to check.
2567  * @returns     true if SSL-library has to write protocol data.
2568  */
2569 static bool
2570 SSL_WantWrite(const CONNECTION *c)
2571 {
2572         if (Conn_OPTION_ISSET(c, CONN_SSL_WANT_WRITE)) {
2573                 io_event_add(c->sock, IO_WANTWRITE);
2574                 return true;
2575         }
2576         return false;
2577 }
2578
2579 /**
2580  * Get information about used SSL cipher.
2581  *
2582  * @param Idx   Connection index number.
2583  * @param buf   Buffer for returned information text.
2584  * @param len   Size of return buffer "buf".
2585  * @returns     true on success, false otherwise.
2586  */
2587 GLOBAL bool
2588 Conn_GetCipherInfo(CONN_ID Idx, char *buf, size_t len)
2589 {
2590         if (Idx < 0)
2591                 return false;
2592         assert(Idx < (int) array_length(&My_ConnArray, sizeof(CONNECTION)));
2593         return ConnSSL_GetCipherInfo(&My_Connections[Idx], buf, len);
2594 }
2595
2596 /**
2597  * Check if a connection is SSL-enabled or not.
2598  *
2599  * @param Idx   Connection index number.
2600  * @return      true if connection is SSL-enabled, false otherwise.
2601  */
2602 GLOBAL bool
2603 Conn_UsesSSL(CONN_ID Idx)
2604 {
2605         if (Idx < 0)
2606                 return false;
2607         assert(Idx < (int) array_length(&My_ConnArray, sizeof(CONNECTION)));
2608         return Conn_OPTION_ISSET(&My_Connections[Idx], CONN_SSL);
2609 }
2610
2611 GLOBAL char *
2612 Conn_GetCertFp(CONN_ID Idx)
2613 {
2614         if (Idx < 0)
2615                 return NULL;
2616         assert(Idx < (int) array_length(&My_ConnArray, sizeof(CONNECTION)));
2617         return ConnSSL_GetCertFp(&My_Connections[Idx]);
2618 }
2619
2620 GLOBAL bool
2621 Conn_SetCertFp(CONN_ID Idx, const char *fingerprint)
2622 {
2623         if (Idx < 0)
2624                 return false;
2625         assert(Idx < (int) array_length(&My_ConnArray, sizeof(CONNECTION)));
2626         return ConnSSL_SetCertFp(&My_Connections[Idx], fingerprint);
2627 }
2628
2629 #else /* SSL_SUPPORT */
2630
2631 GLOBAL bool
2632 Conn_UsesSSL(UNUSED CONN_ID Idx)
2633 {
2634         return false;
2635 }
2636
2637 GLOBAL char *
2638 Conn_GetCertFp(UNUSED CONN_ID Idx)
2639 {
2640         return NULL;
2641 }
2642
2643 GLOBAL bool
2644 Conn_SetCertFp(UNUSED CONN_ID Idx, UNUSED const char *fingerprint)
2645 {
2646         return true;
2647 }
2648
2649 #endif /* SSL_SUPPORT */
2650
2651 #ifdef DEBUG
2652
2653 /**
2654  * Dump internal state of the "connection module".
2655  */
2656 GLOBAL void
2657 Conn_DebugDump(void)
2658 {
2659         int i;
2660
2661         Log(LOG_DEBUG, "Connection status:");
2662         for (i = 0; i < Pool_Size; i++) {
2663                 if (My_Connections[i].sock == NONE)
2664                         continue;
2665                 Log(LOG_DEBUG,
2666                     " - %d: host=%s, lastdata=%ld, lastping=%ld, delaytime=%ld, flag=%d, options=%d, bps=%d, client=%s",
2667                     My_Connections[i].sock, My_Connections[i].host,
2668                     My_Connections[i].lastdata, My_Connections[i].lastping,
2669                     My_Connections[i].delaytime, My_Connections[i].flag,
2670                     My_Connections[i].options, My_Connections[i].bps,
2671                     My_Connections[i].client ? Client_ID(My_Connections[i].client) : "-");
2672         }
2673 } /* Conn_DumpClients */
2674
2675 #endif /* DEBUG */
2676
2677 /* -eof- */