]> arthur.barton.de Git - ngircd-alex.git/blobdiff - src/ngircd/conn.c
New configuration option "OperServerMode".
[ngircd-alex.git] / src / ngircd / conn.c
index 7b9dcdb3c0a053a43b6c7be7396e225f326b1473..ba874f8a71bd5f57b9b13595677494f88c6b9d80 100644 (file)
@@ -1,6 +1,6 @@
 /*
  * ngIRCd -- The Next Generation IRC Daemon
- * Copyright (c)2001-2003 by Alexander Barton (alex@barton.de)
+ * Copyright (c)2001-2004 Alexander Barton <alex@barton.de>
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
 
 #include "portab.h"
 
-static char UNUSED id[] = "$Id: conn.c,v 1.129 2003/12/27 13:01:12 alex Exp $";
+static char UNUSED id[] = "$Id: conn.c,v 1.143 2005/01/19 23:33:54 alex Exp $";
 
 #include "imp.h"
 #include <assert.h>
-#include <stdarg.h>
+#ifdef PROTOTYPES
+#      include <stdarg.h>
+#else
+#      include <varargs.h>
+#endif
 #include <stdio.h>
 #include <stdlib.h>
 #include <unistd.h>
@@ -33,18 +37,22 @@ static char UNUSED id[] = "$Id: conn.c,v 1.129 2003/12/27 13:01:12 alex Exp $";
 #include <time.h>
 #include <netinet/in.h>
 
+#ifdef HAVE_NETINET_IP_H
+# include <netinet/ip.h>
+#endif
+
 #ifdef HAVE_ARPA_INET_H
-#include <arpa/inet.h>
+# include <arpa/inet.h>
 #else
-#define PF_INET AF_INET
+# define PF_INET AF_INET
 #endif
 
 #ifdef HAVE_STDINT_H
-#include <stdint.h>                    /* e.g. for Mac OS X */
+# include <stdint.h>                   /* e.g. for Mac OS X */
 #endif
 
 #ifdef TCPWRAP
-#include <tcpd.h>                      /* for TCP Wrappers */
+# include <tcpd.h>                     /* for TCP Wrappers */
 #endif
 
 #include "defines.h"
@@ -64,7 +72,7 @@ static char UNUSED id[] = "$Id: conn.c,v 1.129 2003/12/27 13:01:12 alex Exp $";
 #include "tool.h"
 
 #ifdef RENDEZVOUS
-#include "rendezvous.h"
+# include "rendezvous.h"
 #endif
 
 #include "exp.h"
@@ -113,7 +121,7 @@ Conn_Init( VOID )
                /* konfiguriertes Limit beachten */
                if( Pool_Size > Conf_MaxConnections ) Pool_Size = Conf_MaxConnections;
        }
-       My_Connections = malloc( sizeof( CONNECTION ) * Pool_Size );
+       My_Connections = (CONNECTION *)malloc( sizeof( CONNECTION ) * Pool_Size );
        if( ! My_Connections )
        {
                /* Speicher konnte nicht alloziert werden! */
@@ -521,7 +529,7 @@ va_dcl
 #else
        va_start( ap );
 #endif
-       if( vsnprintf( buffer, COMMAND_LEN - 2, Format, ap ) == COMMAND_LEN - 2 )
+       if( vsnprintf( buffer, COMMAND_LEN - 2, Format, ap ) >= COMMAND_LEN - 2 )
        {
                Log( LOG_CRIT, "Text too long to send (connection %d)!", Idx );
                Conn_Close( Idx, "Text too long to send!", NULL, FALSE );
@@ -610,6 +618,7 @@ Conn_Close( CONN_ID Idx, CHAR *LogMsg, CHAR *FwdMsg, BOOLEAN InformClient )
         * sub-processes are closed down. */
 
        CLIENT *c;
+       CHAR *txt;
        DOUBLE in_k, out_k;
 #ifdef ZLIB
        DOUBLE in_z_k, out_z_k;
@@ -617,18 +626,28 @@ Conn_Close( CONN_ID Idx, CHAR *LogMsg, CHAR *FwdMsg, BOOLEAN InformClient )
 #endif
 
        assert( Idx > NONE );
-       assert( My_Connections[Idx].sock > NONE );
 
        /* Is this link already shutting down? */
        if( My_Connections[Idx].options & CONN_ISCLOSING )
        {
                /* Conn_Close() has been called recursively for this link;
                 * probabe reason: Try_Write() failed  -- see below. */
+#ifdef DEBUG
+               Log( LOG_DEBUG, "Recursive request to close connection: %d", Idx );
+#endif
                return;
        }
 
+       assert( My_Connections[Idx].sock > NONE );
+
        /* Mark link as "closing" */
        My_Connections[Idx].options |= CONN_ISCLOSING;
+               
+       if( LogMsg ) txt = LogMsg;
+       else txt = FwdMsg;
+       if( ! txt ) txt = "Reason unknown";
+
+       Log( LOG_INFO, "Shutting down connection %d (%s) with %s:%d ...", Idx, LogMsg ? LogMsg : FwdMsg, My_Connections[Idx].host, ntohs( My_Connections[Idx].addr.sin_port ));
 
        /* Search client, if any */
        c = Client_GetFromConn( Idx );
@@ -647,7 +666,6 @@ Conn_Close( CONN_ID Idx, CHAR *LogMsg, CHAR *FwdMsg, BOOLEAN InformClient )
                /* Send ERROR to client (see RFC!) */
                if( FwdMsg ) Conn_WriteStr( Idx, "ERROR :%s", FwdMsg );
                else Conn_WriteStr( Idx, "ERROR :Closing connection." );
-               if( My_Connections[Idx].sock == NONE ) return;
        }
 
        /* Try to write out the write buffer */
@@ -712,6 +730,10 @@ Conn_Close( CONN_ID Idx, CHAR *LogMsg, CHAR *FwdMsg, BOOLEAN InformClient )
 
        /* Clean up connection structure (=free it) */
        Init_Conn_Struct( Idx );
+
+#ifdef DEBUG
+       Log( LOG_DEBUG, "Shutdown of connection %d completed.", Idx );
+#endif
 } /* Conn_Close */
 
 
@@ -868,11 +890,9 @@ Handle_Write( CONN_ID Idx )
 
                        return FALSE;
                }
-#ifdef DEBUG
-               Log( LOG_DEBUG, "Connection %d with \"%s:%d\" established, now sendig PASS and SERVER ...", Idx, My_Connections[Idx].host, Conf_Server[Conf_GetServer( Idx )].port );
-#endif
+               Log( LOG_INFO, "Connection %d with \"%s:%d\" established. Now logging in ...", Idx, My_Connections[Idx].host, Conf_Server[Conf_GetServer( Idx )].port );
 
-               /* PASS und SERVER verschicken */
+               /* Send PASS and SERVER command to peer */
                Conn_WriteStr( Idx, "PASS %s %s", Conf_Server[Conf_GetServer( Idx )].pwd_out, NGIRCd_ProtoID );
                return Conn_WriteStr( Idx, "SERVER %s :%s", Conf_ServerName, Conf_ServerInfo );
        }
@@ -937,6 +957,7 @@ New_Connection( INT Sock )
 #ifdef TCPWRAP
        /* Validate socket using TCP Wrappers */
        request_init( &req, RQ_DAEMON, PACKAGE_NAME, RQ_FILE, new_sock, RQ_CLIENT_SIN, &new_addr, NULL );
+       fromhost(&req);
        if( ! hosts_access( &req ))
        {
                /* Access denied! */
@@ -954,9 +975,9 @@ New_Connection( INT Sock )
        cnt = Count_Connections( new_addr );
        if(( Conf_MaxConnectionsIP > 0 ) && ( cnt >= Conf_MaxConnectionsIP ))
        {
-               /* Access denied, too many connections from this IP! */
-               Log( LOG_ERR, "Refused connection from %s: too may connections (%ld) from this IP!", inet_ntoa( new_addr.sin_addr ), cnt);
-               Simple_Message( new_sock, "ERROR :Connection refused, too many connections from your IP!" );
+               /* Access denied, too many connections from this IP address! */
+               Log( LOG_ERR, "Refused connection from %s: too may connections (%ld) from this IP address!", inet_ntoa( new_addr.sin_addr ), cnt);
+               Simple_Message( new_sock, "ERROR :Connection refused, too many connections from your IP address!" );
                close( new_sock );
                return;
        }
@@ -991,36 +1012,21 @@ New_Connection( INT Sock )
                        return;
                }
 
-               /* zunaechst realloc() versuchen; wenn das scheitert, malloc() versuchen
-                * und Daten ggf. "haendisch" umkopieren. (Haesslich! Eine wirklich
-                * dynamische Verwaltung waere wohl _deutlich_ besser ...) */
-               ptr = realloc( My_Connections, sizeof( CONNECTION ) * new_size );
+               ptr = (POINTER *)realloc( My_Connections, sizeof( CONNECTION ) * new_size );
                if( ! ptr )
                {
-                       /* realloc() ist fehlgeschlagen. Nun malloc() probieren: */
-                       ptr = malloc( sizeof( CONNECTION ) * new_size );
-                       if( ! ptr )
-                       {
-                               /* Offenbar steht kein weiterer Sepeicher zur Verfuegung :-( */
-                               Log( LOG_EMERG, "Can't allocate memory! [New_Connection]" );
-                               Simple_Message( new_sock, "ERROR: Internal error" );
-                               close( new_sock );
-                               return;
-                       }
-
-                       /* Struktur umkopieren ... */
-                       memcpy( ptr, My_Connections, sizeof( CONNECTION ) * Pool_Size );
-
-#ifdef DEBUG
-                       Log( LOG_DEBUG, "Allocated new connection pool for %ld items (%ld bytes). [malloc()/memcpy()]", new_size, sizeof( CONNECTION ) * new_size );
-#endif
+                       Log( LOG_EMERG, "Can't allocate memory! [New_Connection]" );
+                       Simple_Message( new_sock, "ERROR: Internal error" );
+                       close( new_sock );
+                       return;
                }
+
 #ifdef DEBUG
-               else Log( LOG_DEBUG, "Allocated new connection pool for %ld items (%ld bytes). [realloc()]", new_size, sizeof( CONNECTION ) * new_size );
+               Log( LOG_DEBUG, "Allocated new connection pool for %ld items (%ld bytes). [realloc()]", new_size, sizeof( CONNECTION ) * new_size );
 #endif
 
                /* Adjust pointer to new block */
-               My_Connections = ptr;
+               My_Connections = (CONNECTION *)ptr;
 
                /* Initialize new items */
                for( idx = Pool_Size; idx < new_size; idx++ ) Init_Conn_Struct( idx );
@@ -1123,8 +1129,8 @@ Read_Request( CONN_ID Idx )
 #endif
        {
                /* Der Lesepuffer ist voll */
-               Log( LOG_ERR, "Read buffer overflow (connection %d): %d bytes!", Idx, My_Connections[Idx].rdatalen );
-               Conn_Close( Idx, "Read buffer overflow!", NULL, FALSE );
+               Log( LOG_ERR, "Receive buffer overflow (connection %d): %d bytes!", Idx, My_Connections[Idx].rdatalen );
+               Conn_Close( Idx, "Receive buffer overflow!", NULL, FALSE );
                return;
        }
 
@@ -1262,22 +1268,20 @@ Handle_Buffer( CONN_ID Idx )
                                /* Mit dem letzten Befehl wurde Socket-Kompression aktiviert.
                                 * Evtl. schon vom Socket gelesene Daten in den Unzip-Puffer
                                 * umkopieren, damit diese nun zunaechst entkomprimiert werden */
+                               if( My_Connections[Idx].rdatalen > ZREADBUFFER_LEN )
                                {
-                                       if( My_Connections[Idx].rdatalen > ZREADBUFFER_LEN )
-                                       {
-                                               /* Hupsa! Soviel Platz haben wir aber gar nicht! */
-                                               Log( LOG_ALERT, "Can't move read buffer: No space left in unzip buffer (need %d bytes)!", My_Connections[Idx].rdatalen );
-                                               return FALSE;
-                                       }
-                                       memcpy( My_Connections[Idx].zip.rbuf, My_Connections[Idx].rbuf, My_Connections[Idx].rdatalen );
-                                       My_Connections[Idx].zip.rdatalen = My_Connections[Idx].rdatalen;
-                                       My_Connections[Idx].rdatalen = 0;
-#ifdef DEBUG
-                                       Log( LOG_DEBUG, "Moved already received data (%d bytes) to uncompression buffer.", My_Connections[Idx].zip.rdatalen );
-#endif
+                                       /* Hupsa! Soviel Platz haben wir aber gar nicht! */
+                                       Log( LOG_ALERT, "Can't move receive buffer: No space left in unzip buffer (need %d bytes)!", My_Connections[Idx].rdatalen );
+                                       return FALSE;
                                }
+                               memcpy( My_Connections[Idx].zip.rbuf, My_Connections[Idx].rbuf, My_Connections[Idx].rdatalen );
+                               My_Connections[Idx].zip.rdatalen = My_Connections[Idx].rdatalen;
+                               My_Connections[Idx].rdatalen = 0;
+#ifdef DEBUG
+                               Log( LOG_DEBUG, "Moved already received data (%d bytes) to uncompression buffer.", My_Connections[Idx].zip.rdatalen );
+#endif /* DEBUG */
                        }
-#endif
+#endif /* ZLIB */
                }
 
                if( action ) result = TRUE;
@@ -1549,23 +1553,39 @@ Init_Conn_Struct( CONN_ID Idx )
 LOCAL BOOLEAN
 Init_Socket( INT Sock )
 {
-       /* Socket-Optionen setzen */
+       /* Initialize socket (set options) */
 
-       INT on = 1;
+       INT value;
 
-#ifdef O_NONBLOCK      /* A/UX kennt das nicht? */
+#ifdef O_NONBLOCK      /* unknown on A/UX */
        if( fcntl( Sock, F_SETFL, O_NONBLOCK ) != 0 )
        {
-               Log( LOG_CRIT, "Can't enable non-blocking mode: %s!", strerror( errno ));
+               Log( LOG_CRIT, "Can't enable non-blocking mode for socket: %s!", strerror( errno ));
                close( Sock );
                return FALSE;
        }
 #endif
-       if( setsockopt( Sock, SOL_SOCKET, SO_REUSEADDR, &on, (socklen_t)sizeof( on )) != 0)
+
+       /* Don't block this port after socket shutdown */
+       value = 1;
+       if( setsockopt( Sock, SOL_SOCKET, SO_REUSEADDR, &value, (socklen_t)sizeof( value )) != 0 )
+       {
+               Log( LOG_ERR, "Can't set socket option SO_REUSEADDR: %s!", strerror( errno ));
+               /* ignore this error */
+       }
+
+       /* Set type of service (TOS) */
+#if defined(IP_TOS) && defined(IPTOS_LOWDELAY)
+       value = IPTOS_LOWDELAY;
+#ifdef DEBUG
+       Log( LOG_DEBUG, "Setting option IP_TOS on socket %d to IPTOS_LOWDELAY (%d).", Sock, value );
+#endif
+       if( setsockopt( Sock, SOL_IP, IP_TOS, &value, (socklen_t)sizeof( value )) != 0 )
        {
-               Log( LOG_ERR, "Can't set socket options: %s!", strerror( errno ));
-               /* dieser Fehler kann ignoriert werden. */
+               Log( LOG_ERR, "Can't set socket option IP_TOS: %s!", strerror( errno ));
+               /* ignore this error */
        }
+#endif
 
        return TRUE;
 } /* Init_Socket */
@@ -1574,35 +1594,28 @@ Init_Socket( INT Sock )
 LOCAL VOID
 Read_Resolver_Result( INT r_fd )
 {
-       /* Ergebnis von Resolver Sub-Prozess aus Pipe lesen
-        * und entsprechende Connection aktualisieren */
+       /* Read result of resolver sub-process from pipe and update the
+        * apropriate connection/client structure(s): hostname and/or
+        * IDENT user name.*/
 
-       CHAR result[HOST_LEN];
        CLIENT *c;
        INT len, i, n;
-
-       FD_CLR( r_fd, &Resolver_FDs );
-
-       /* Read result from pipe */
-       len = read( r_fd, result, HOST_LEN - 1 );
-       if( len < 0 )
-       {
-               /* Error! */
-               close( r_fd );
-               Log( LOG_CRIT, "Resolver: Can't read result: %s!", strerror( errno ));
-               return;
-       }
-       result[len] = '\0';
+       RES_STAT *s;
+       CHAR *ptr;
 
        /* Search associated connection ... */
        for( i = 0; i < Pool_Size; i++ )
        {
-               if(( My_Connections[i].sock != NONE ) && ( My_Connections[i].res_stat ) && ( My_Connections[i].res_stat->pipe[0] == r_fd )) break;
+               if(( My_Connections[i].sock != NONE )
+                 && ( My_Connections[i].res_stat != NULL )
+                 && ( My_Connections[i].res_stat->pipe[0] == r_fd ))
+                       break;
        }
        if( i >= Pool_Size )
        {
                /* Ops, none found? Probably the connection has already
-                * been closed. */
+                * been closed!? We'll ignore that ... */
+               FD_CLR( r_fd, &Resolver_FDs );
                close( r_fd );
 #ifdef DEBUG
                Log( LOG_DEBUG, "Resolver: Got result for unknown connection!?" );
@@ -1610,48 +1623,102 @@ Read_Resolver_Result( INT r_fd )
                return;
        }
 
-#ifdef DEBUG
-       Log( LOG_DEBUG, "Resolver: %s is \"%s\".", My_Connections[i].host, result );
+       /* Get resolver structure */
+       s = My_Connections[i].res_stat;
+       assert( s != NULL );
+
+       /* Read result from pipe */
+       len = read( r_fd, s->buffer + s->bufpos, sizeof( s->buffer ) - s->bufpos - 1 );
+       if( len < 0 )
+       {
+               /* Error! */
+               close( r_fd );
+               Log( LOG_CRIT, "Resolver: Can't read result: %s!", strerror( errno ));
+               return;
+       }
+       s->bufpos += len;
+       s->buffer[s->bufpos] = '\0';
+
+       /* If the result string is incomplete, return to main loop and
+        * wait until we can read in more bytes. */
+#ifdef IDENTAUTH
+try_resolve:
 #endif
+       ptr = strchr( s->buffer, '\n' );
+       if( ! ptr ) return;
+       *ptr = '\0';
 
-       /* Clean up ... */
-       close( My_Connections[i].res_stat->pipe[0] );
-       close( My_Connections[i].res_stat->pipe[1] );
-       free( My_Connections[i].res_stat );
-       My_Connections[i].res_stat = NULL;
+#ifdef DEBUG
+       Log( LOG_DEBUG, "Got result from resolver: \"%s\" (%d bytes), stage %d.", s->buffer, len, s->stage );
+#endif
 
+       /* Okay, we got a complete result: this is a host name for outgoing
+        * connections and a host name or IDENT user name (if enabled) for
+        * incoming conneciions.*/
        if( My_Connections[i].sock > NONE )
        {
-#ifdef IDENTAUTH
-               CHAR *ident;
-#endif
-               /* Incoming connection: set hostname */
+               /* Incoming connection */
+
+               /* Search client ... */
                c = Client_GetFromConn( i );
                assert( c != NULL );
-               strlcpy( My_Connections[i].host, result, sizeof( My_Connections[i].host ));
-               Client_SetHostname( c, result );
-
-#ifdef IDENTAUTH
-               ident = strchr( result, 0 );
-               ident++;
 
-               /* Do we have a result of the IDENT lookup? */
-               if( *ident )
+               /* Only update client information of unregistered clients */
+               if( Client_Type( c ) == CLIENT_UNKNOWN )
                {
-                       Log( LOG_INFO, "IDENT lookup on connection %ld: \"%s\".", i, ident );
-                       Client_SetUser( c, ident, TRUE );
+                       if( s->stage == 0 )
+                       {
+                               /* host name */
+                               strlcpy( My_Connections[i].host, s->buffer, sizeof( My_Connections[i].host ));
+                               Client_SetHostname( c, s->buffer );
+
+#ifdef IDENTAUTH
+                               /* clean up buffer for IDENT result */
+                               len = strlen( s->buffer ) + 1;
+                               memmove( s->buffer, s->buffer + len, sizeof( s->buffer ) - len );
+                               s->bufpos -= len;
+
+                               /* Don't close pipe and clean up, but
+                                * instead wait for IDENT result */
+                               s->stage = 1;
+                               goto try_resolve;
+                       }
+                       else if( s->stage == 1 )
+                       {
+                               /* IDENT user name */
+                               if( s->buffer[0] )
+                               {
+                                       Log( LOG_INFO, "IDENT lookup for connection %ld: \"%s\".", i, s->buffer );
+                                       Client_SetUser( c, s->buffer, TRUE );
+                               }
+                               else Log( LOG_INFO, "IDENT lookup for connection %ld: no result.", i );
+#endif
+                       }
+                       else Log( LOG_ERR, "Resolver: got result for unknown stage %d!?", s->stage );
                }
-               else Log( LOG_INFO, "IDENT lookup on connection %ld: no result.", i );
+#ifdef DEBUG
+               else Log( LOG_DEBUG, "Resolver: discarding result for already registered connection %d.", i );
 #endif
        }
        else
        {
-               /* Outgoing connection (server link!): set IP address */
+               /* Outgoing connection (server link): set the IP address
+                * so that we can connect to it in the main loop. */
+
+               /* Search server ... */
                n = Conf_GetServer( i );
-               if( n > NONE ) strlcpy( Conf_Server[n].ip, result, sizeof( Conf_Server[n].ip ));
-               else Log( LOG_ERR, "Got resolver result for non-configured server!?" );
+               assert( n > NONE );
+
+               strlcpy( Conf_Server[n].ip, s->buffer, sizeof( Conf_Server[n].ip ));
        }
 
+       /* Clean up ... */
+       FD_CLR( r_fd, &Resolver_FDs );
+       close( My_Connections[i].res_stat->pipe[0] );
+       close( My_Connections[i].res_stat->pipe[1] );
+       free( My_Connections[i].res_stat );
+       My_Connections[i].res_stat = NULL;
+
        /* Reset penalty time */
        Conn_ResetPenalty( i );
 } /* Read_Resolver_Result */