]> arthur.barton.de Git - ngircd-alex.git/blobdiff - src/ngircd/conn.c
Added optional support for IDENT lookups (configure switch "--with-ident").
[ngircd-alex.git] / src / ngircd / conn.c
index 2c65f26e1f52204cffd87e16b60407d482326c6c..7b9dcdb3c0a053a43b6c7be7396e225f326b1473 100644 (file)
@@ -16,7 +16,7 @@
 
 #include "portab.h"
 
-static char UNUSED id[] = "$Id: conn.c,v 1.126 2003/11/05 21:41:02 alex Exp $";
+static char UNUSED id[] = "$Id: conn.c,v 1.129 2003/12/27 13:01:12 alex Exp $";
 
 #include "imp.h"
 #include <assert.h>
@@ -43,7 +43,7 @@ static char UNUSED id[] = "$Id: conn.c,v 1.126 2003/11/05 21:41:02 alex Exp $";
 #include <stdint.h>                    /* e.g. for Mac OS X */
 #endif
 
-#ifdef USE_TCPWRAP
+#ifdef TCPWRAP
 #include <tcpd.h>                      /* for TCP Wrappers */
 #endif
 
@@ -93,7 +93,7 @@ LOCAL fd_set My_Listeners;
 LOCAL fd_set My_Sockets;
 LOCAL fd_set My_Connects;
 
-#ifdef USE_TCPWRAP
+#ifdef TCPWRAP
 INT allow_severity = LOG_INFO;
 INT deny_severity = LOG_ERR;
 #endif
@@ -120,7 +120,9 @@ Conn_Init( VOID )
                Log( LOG_EMERG, "Can't allocate memory! [Conn_Init]" );
                exit( 1 );
        }
+#ifdef DEBUG
        Log( LOG_DEBUG, "Allocated connection pool for %d items (%ld bytes).", Pool_Size, sizeof( CONNECTION ) * Pool_Size );
+#endif
 
        /* zu Beginn haben wir keine Verbindungen */
        FD_ZERO( &My_Listeners );
@@ -147,7 +149,9 @@ Conn_Exit( VOID )
        CONN_ID idx;
        INT i;
 
+#ifdef DEBUG
        Log( LOG_DEBUG, "Shutting down all connections ..." );
+#endif
 
 #ifdef RENDEZVOUS
        Rendezvous_UnregisterListeners( );
@@ -165,12 +169,16 @@ Conn_Exit( VOID )
                        if( FD_ISSET( i, &My_Listeners ))
                        {
                                close( i );
+#ifdef DEBUG
                                Log( LOG_DEBUG, "Listening socket %d closed.", i );
+#endif
                        }
                        else if( FD_ISSET( i, &My_Connects ))
                        {
                                close( i );
+#ifdef DEBUG
                                Log( LOG_DEBUG, "Connection %d closed during creation (socket %d).", idx, i );
+#endif
                        }
                        else if( idx < Pool_Size )
                        {
@@ -225,7 +233,9 @@ Conn_ExitListeners( VOID )
                if( FD_ISSET( i, &My_Sockets ) && FD_ISSET( i, &My_Listeners ))
                {
                        close( i );
+#ifdef DEBUG
                        Log( LOG_DEBUG, "Listening socket %d closed.", i );
+#endif
                }
        }
 } /* Conn_ExitListeners */
@@ -368,10 +378,13 @@ Conn_Handler( VOID )
                Check_Servers( );
                Check_Connections( );
 
+               t = time( NULL );
+
                /* noch volle Lese-Buffer suchen */
                for( i = 0; i < Pool_Size; i++ )
                {
-                       if(( My_Connections[i].sock > NONE ) && ( My_Connections[i].rdatalen > 0 ))
+                       if(( My_Connections[i].sock > NONE ) && ( My_Connections[i].rdatalen > 0 ) &&
+                        ( My_Connections[i].delaytime < t ))
                        {
                                /* Kann aus dem Buffer noch ein Befehl extrahiert werden? */
                                if( Handle_Buffer( i )) timeout = FALSE;
@@ -382,7 +395,7 @@ Conn_Handler( VOID )
                FD_ZERO( &write_sockets );
                for( i = 0; i < Pool_Size; i++ )
                {
-#ifdef USE_ZLIB
+#ifdef ZLIB
                        if(( My_Connections[i].sock > NONE ) && (( My_Connections[i].wdatalen > 0 ) || ( My_Connections[i].zip.wdatalen > 0 )))
 #else
                        if(( My_Connections[i].sock > NONE ) && ( My_Connections[i].wdatalen > 0 ))
@@ -400,7 +413,6 @@ Conn_Handler( VOID )
                }
 
                /* von welchen Sockets koennte gelesen werden? */
-               t = time( NULL );
                read_sockets = My_Sockets;
                for( i = 0; i < Pool_Size; i++ )
                {
@@ -431,7 +443,7 @@ Conn_Handler( VOID )
 
                /* Timeout initialisieren */
                tv.tv_usec = 0;
-               if( timeout ) tv.tv_sec = TIME_RES;
+               if( timeout ) tv.tv_sec = 1;
                else tv.tv_sec = 0;
 
                /* Auf Aktivitaet warten */
@@ -545,7 +557,9 @@ Conn_Write( CONN_ID Idx, CHAR *Data, INT Len )
         * In diesem Fall wird hier einfach ein Fehler geliefert. */
        if( My_Connections[Idx].sock <= NONE )
        {
+#ifdef DEBUG
                Log( LOG_DEBUG, "Skipped write on closed socket (connection %d).", Idx );
+#endif
                return FALSE;
        }
 
@@ -567,7 +581,7 @@ Conn_Write( CONN_ID Idx, CHAR *Data, INT Len )
                }
        }
 
-#ifdef USE_ZLIB
+#ifdef ZLIB
        if( My_Connections[Idx].options & CONN_ZIP )
        {
                /* Daten komprimieren und in Puffer kopieren */
@@ -597,7 +611,7 @@ Conn_Close( CONN_ID Idx, CHAR *LogMsg, CHAR *FwdMsg, BOOLEAN InformClient )
 
        CLIENT *c;
        DOUBLE in_k, out_k;
-#ifdef USE_ZLIB
+#ifdef ZLIB
        DOUBLE in_z_k, out_z_k;
        INT in_p, out_p;
 #endif
@@ -659,7 +673,7 @@ Conn_Close( CONN_ID Idx, CHAR *LogMsg, CHAR *FwdMsg, BOOLEAN InformClient )
        /* Calculate statistics and log information */
        in_k = (DOUBLE)My_Connections[Idx].bytes_in / 1024;
        out_k = (DOUBLE)My_Connections[Idx].bytes_out / 1024;
-#ifdef USE_ZLIB
+#ifdef ZLIB
        if( My_Connections[Idx].options & CONN_ZIP )
        {
                in_z_k = (DOUBLE)My_Connections[Idx].zip.bytes_in / 1024;
@@ -687,7 +701,7 @@ Conn_Close( CONN_ID Idx, CHAR *LogMsg, CHAR *FwdMsg, BOOLEAN InformClient )
        /* Servers: Modify time of next connect attempt? */
        Conf_UnsetServer( Idx );
 
-#ifdef USE_ZLIB
+#ifdef ZLIB
        /* Clean up zlib, if link was compressed */
        if( Conn_Options( Idx ) & CONN_ZIP )
        {
@@ -748,7 +762,7 @@ Try_Write( CONN_ID Idx )
        assert( My_Connections[Idx].sock > NONE );
 
        /* sind ueberhaupt Daten vorhanden? */
-#ifdef USE_ZLIB
+#ifdef ZLIB
        if(( ! My_Connections[Idx].wdatalen > 0 ) && ( ! My_Connections[Idx].zip.wdatalen )) return TRUE;
 #else
        if( ! My_Connections[Idx].wdatalen > 0 ) return TRUE;
@@ -854,14 +868,16 @@ 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
 
                /* PASS und SERVER verschicken */
                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 );
        }
 
-#ifdef USE_ZLIB
+#ifdef ZLIB
        /* Schreibpuffer leer, aber noch Daten im Kompressionsbuffer?
         * Dann muss dieser nun geflushed werden! */
        if( My_Connections[Idx].wdatalen == 0 ) Zip_Flush( Idx );
@@ -896,7 +912,7 @@ New_Connection( INT Sock )
        /* Neue Client-Verbindung von Listen-Socket annehmen und
         * CLIENT-Struktur anlegen. */
 
-#ifdef USE_TCPWRAP
+#ifdef TCPWRAP
        struct request_info req;
 #endif
        struct sockaddr_in new_addr;
@@ -918,7 +934,7 @@ New_Connection( INT Sock )
                return;
        }
 
-#ifdef USE_TCPWRAP
+#ifdef TCPWRAP
        /* Validate socket using TCP Wrappers */
        request_init( &req, RQ_DAEMON, PACKAGE_NAME, RQ_FILE, new_sock, RQ_CLIENT_SIN, &new_addr, NULL );
        if( ! hosts_access( &req ))
@@ -995,9 +1011,13 @@ New_Connection( INT Sock )
                        /* 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
                }
+#ifdef DEBUG
                else 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;
@@ -1034,7 +1054,11 @@ New_Connection( INT Sock )
        /* Hostnamen ermitteln */
        strlcpy( My_Connections[idx].host, inet_ntoa( new_addr.sin_addr ), sizeof( My_Connections[idx].host ));
        Client_SetHostname( c, My_Connections[idx].host );
+#ifdef IDENTAUTH
+       s = Resolve_Addr( &new_addr, My_Connections[idx].sock );
+#else
        s = Resolve_Addr( &new_addr );
+#endif
        if( s )
        {
                /* Sub-Prozess wurde asyncron gestartet */
@@ -1061,7 +1085,9 @@ Socket2Index( INT Sock )
        {
                /* die Connection wurde vermutlich (wegen eines
                 * Fehlers) bereits wieder abgebaut ... */
+#ifdef DEBUG
                Log( LOG_DEBUG, "Socket2Index: can't get connection for socket %d!", Sock );
+#endif
                return NONE;
        }
        else return idx;
@@ -1075,7 +1101,7 @@ Read_Request( CONN_ID Idx )
         * Tritt ein Fehler auf, so wird der Socket geschlossen. */
 
        INT len, bsize;
-#ifdef USE_ZLIB
+#ifdef ZLIB
        CLIENT *c;
 #endif
 
@@ -1085,12 +1111,12 @@ Read_Request( CONN_ID Idx )
        /* wenn noch nicht registriert: maximal mit ZREADBUFFER_LEN arbeiten,
         * ansonsten koennen Daten ggf. nicht umkopiert werden. */
        bsize = READBUFFER_LEN;
-#ifdef USE_ZLIB
+#ifdef ZLIB
        c = Client_GetFromConn( Idx );
        if(( Client_Type( c ) != CLIENT_USER ) && ( Client_Type( c ) != CLIENT_SERVER ) && ( Client_Type( c ) != CLIENT_SERVICE ) && ( bsize > ZREADBUFFER_LEN )) bsize = ZREADBUFFER_LEN;
 #endif
 
-#ifdef USE_ZLIB
+#ifdef ZLIB
        if(( bsize - My_Connections[Idx].rdatalen - 1 < 1 ) || ( ZREADBUFFER_LEN - My_Connections[Idx].zip.rdatalen < 1 ))
 #else
        if( bsize - My_Connections[Idx].rdatalen - 1 < 1 )
@@ -1102,7 +1128,7 @@ Read_Request( CONN_ID Idx )
                return;
        }
 
-#ifdef USE_ZLIB
+#ifdef ZLIB
        if( My_Connections[Idx].options & CONN_ZIP )
        {
                len = recv( My_Connections[Idx].sock, My_Connections[Idx].zip.rbuf + My_Connections[Idx].zip.rdatalen, ( ZREADBUFFER_LEN - My_Connections[Idx].zip.rdatalen ), 0 );
@@ -1157,14 +1183,17 @@ Handle_Buffer( CONN_ID Idx )
        CHAR *ptr;
        INT len, delta;
        BOOLEAN action, result;
-#ifdef USE_ZLIB
+#ifdef ZLIB
        BOOLEAN old_z;
 #endif
 
        result = FALSE;
        do
        {
-#ifdef USE_ZLIB
+               /* Check penalty */
+               if( My_Connections[Idx].delaytime > time( NULL )) return result;
+               
+#ifdef ZLIB
                /* ggf. noch unkomprimiete Daten weiter entpacken */
                if( My_Connections[Idx].options & CONN_ZIP )
                {
@@ -1210,7 +1239,7 @@ Handle_Buffer( CONN_ID Idx )
                                return FALSE;
                        }
 
-#ifdef USE_ZLIB
+#ifdef ZLIB
                        /* merken, ob Stream bereits komprimiert wird */
                        old_z = My_Connections[Idx].options & CONN_ZIP;
 #endif
@@ -1227,7 +1256,7 @@ Handle_Buffer( CONN_ID Idx )
                        My_Connections[Idx].rdatalen -= len;
                        memmove( My_Connections[Idx].rbuf, My_Connections[Idx].rbuf + len, My_Connections[Idx].rdatalen );
 
-#ifdef USE_ZLIB
+#ifdef ZLIB
                        if(( ! old_z ) && ( My_Connections[Idx].options & CONN_ZIP ) && ( My_Connections[Idx].rdatalen > 0 ))
                        {
                                /* Mit dem letzten Befehl wurde Socket-Kompression aktiviert.
@@ -1243,7 +1272,9 @@ Handle_Buffer( CONN_ID Idx )
                                        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
                                }
                        }
 #endif
@@ -1280,14 +1311,18 @@ Check_Connections( VOID )
                                if( My_Connections[i].lastping < time( NULL ) - Conf_PongTimeout )
                                {
                                        /* Timeout */
+#ifdef DEBUG
                                        Log( LOG_DEBUG, "Connection %d: Ping timeout: %d seconds.", i, Conf_PongTimeout );
+#endif
                                        Conn_Close( i, NULL, "Ping timeout", TRUE );
                                }
                        }
                        else if( My_Connections[i].lastdata < time( NULL ) - Conf_PingTimeout )
                        {
                                /* es muss ein PING gesendet werden */
+#ifdef DEBUG
                                Log( LOG_DEBUG, "Connection %d: sending PING ...", i );
+#endif
                                My_Connections[i].lastping = time( NULL );
                                Conn_WriteStr( i, "PING :%s", Client_ID( Client_ThisServer( )));
                        }
@@ -1298,7 +1333,9 @@ Check_Connections( VOID )
                        if( My_Connections[i].lastdata < time( NULL ) - Conf_PingTimeout )
                        {
                                /* Timeout */
+#ifdef DEBUG
                                Log( LOG_DEBUG, "Connection %d timed out ...", i );
+#endif
                                Conn_Close( i, NULL, "Timeout", FALSE );
                        }
                }
@@ -1354,7 +1391,9 @@ Check_Servers( VOID )
                        Log( LOG_ALERT, "Can't establist server connection: connection limit reached (%d)!", Pool_Size );
                        return;
                }
+#ifdef DEBUG
                Log( LOG_DEBUG, "Preparing connection %d for \"%s\" ...", idx, Conf_Server[i].host );
+#endif
 
                /* Verbindungs-Struktur initialisieren */
                Init_Conn_Struct( idx );
@@ -1466,7 +1505,9 @@ New_Server( INT Server, CONN_ID Idx )
        FD_SET( new_sock, &My_Connects );
        if( new_sock > Conn_MaxFD ) Conn_MaxFD = new_sock;
 
+#ifdef DEBUG
        Log( LOG_DEBUG, "Registered new connection %d on socket %d.", Idx, My_Connections[Idx].sock );
+#endif
 } /* New_Server */
 
 
@@ -1494,7 +1535,7 @@ Init_Conn_Struct( CONN_ID Idx )
        My_Connections[Idx].flag = 0;
        My_Connections[Idx].options = 0;
 
-#ifdef USE_ZLIB
+#ifdef ZLIB
        My_Connections[Idx].zip.rbuf[0] = '\0';
        My_Connections[Idx].zip.rdatalen = 0;
        My_Connections[Idx].zip.wbuf[0] = '\0';
@@ -1542,34 +1583,38 @@ Read_Resolver_Result( INT r_fd )
 
        FD_CLR( r_fd, &Resolver_FDs );
 
-       /* Anfrage vom Parent lesen */
+       /* Read result from pipe */
        len = read( r_fd, result, HOST_LEN - 1 );
        if( len < 0 )
        {
-               /* Fehler beim Lesen aus der Pipe */
+               /* Error! */
                close( r_fd );
                Log( LOG_CRIT, "Resolver: Can't read result: %s!", strerror( errno ));
                return;
        }
        result[len] = '\0';
 
-       /* zugehoerige Connection suchen */
+       /* 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( i >= Pool_Size )
        {
-               /* Opsa! Keine passende Connection gefunden!? Vermutlich
-                * wurde sie schon wieder geschlossen. */
+               /* Ops, none found? Probably the connection has already
+                * been closed. */
                close( r_fd );
+#ifdef DEBUG
                Log( LOG_DEBUG, "Resolver: Got result for unknown connection!?" );
+#endif
                return;
        }
 
+#ifdef DEBUG
        Log( LOG_DEBUG, "Resolver: %s is \"%s\".", My_Connections[i].host, result );
+#endif
 
-       /* Aufraeumen */
+       /* Clean up ... */
        close( My_Connections[i].res_stat->pipe[0] );
        close( My_Connections[i].res_stat->pipe[1] );
        free( My_Connections[i].res_stat );
@@ -1577,21 +1622,37 @@ Read_Resolver_Result( INT r_fd )
 
        if( My_Connections[i].sock > NONE )
        {
-               /* Eingehende Verbindung: Hostnamen setzen */
+#ifdef IDENTAUTH
+               CHAR *ident;
+#endif
+               /* Incoming connection: set hostname */
                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 )
+               {
+                       Log( LOG_INFO, "IDENT lookup on connection %ld: \"%s\".", i, ident );
+                       Client_SetUser( c, ident, TRUE );
+               }
+               else Log( LOG_INFO, "IDENT lookup on connection %ld: no result.", i );
+#endif
        }
        else
        {
-               /* Ausgehende Verbindung (=Server): IP setzen */
+               /* Outgoing connection (server link!): set IP address */
                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!?" );
        }
 
-       /* Penalty-Zeit zurueck setzen */
+       /* Reset penalty time */
        Conn_ResetPenalty( i );
 } /* Read_Resolver_Result */