X-Git-Url: https://arthur.barton.de/cgi-bin/gitweb.cgi?p=ngircd-alex.git;a=blobdiff_plain;f=src%2Fngircd%2Fconn.c;h=37c5c3819a49390a298bfcb80a33b7919595d1f8;hp=d54c382c17a69309621e37a1319344f277e5d2f3;hb=cf4ad8d6f7f9adffda596d67551811ac26f6a35a;hpb=22b7b07932e282b5edd90d25d6da7262bfb3feda diff --git a/src/ngircd/conn.c b/src/ngircd/conn.c index d54c382c..37c5c381 100644 --- a/src/ngircd/conn.c +++ b/src/ngircd/conn.c @@ -2,21 +2,22 @@ * ngIRCd -- The Next Generation IRC Daemon * Copyright (c)2001,2002 by Alexander Barton (alex@barton.de) * - * Dieses Programm ist freie Software. Sie koennen es unter den Bedingungen - * der GNU General Public License (GPL), wie von der Free Software Foundation - * herausgegeben, weitergeben und/oder modifizieren, entweder unter Version 2 - * der Lizenz oder (wenn Sie es wuenschen) jeder spaeteren Version. - * Naehere Informationen entnehmen Sie bitter der Datei COPYING. Eine Liste - * der an ngIRCd beteiligten Autoren finden Sie in der Datei AUTHORS. + * 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 + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * Please read the file COPYING, README and AUTHORS for more information. * - * $Id: conn.c,v 1.94 2002/11/23 16:09:57 alex Exp $ - * - * connect.h: Verwaltung aller Netz-Verbindungen ("connections") + * Connection management */ +#define CONN_MODULE + #include "portab.h" +static char UNUSED id[] = "$Id: conn.c,v 1.116 2003/02/21 19:19:27 alex Exp $"; + #include "imp.h" #include #include @@ -42,14 +43,18 @@ #include /* u.a. fuer Mac OS X */ #endif +#include "defines.h" +#include "resolve.h" + #include "exp.h" #include "conn.h" #include "imp.h" #include "ngircd.h" #include "client.h" -#include "resolve.h" #include "conf.h" +#include "conn-zip.h" +#include "conn-func.h" #include "log.h" #include "parse.h" #include "tool.h" @@ -60,48 +65,24 @@ #define SERVER_WAIT (NONE - 1) -typedef struct _Connection -{ - INT sock; /* Socket Handle */ - struct sockaddr_in addr; /* Adresse des Client */ - RES_STAT *res_stat; /* "Resolver-Status", s.o. */ - CHAR host[HOST_LEN]; /* Hostname */ - CHAR rbuf[READBUFFER_LEN]; /* Lesepuffer */ - INT rdatalen; /* Laenge der Daten im Lesepuffer */ - CHAR wbuf[WRITEBUFFER_LEN]; /* Schreibpuffer */ - INT wdatalen; /* Laenge der Daten im Schreibpuffer */ - INT our_server; /* wenn von uns zu connectender Server: ID */ - time_t lastdata; /* Letzte Aktivitaet */ - time_t lastping; /* Letzter PING */ - time_t lastprivmsg; /* Letzte PRIVMSG */ - time_t delaytime; /* Nicht beachten bis ("penalty") */ - LONG bytes_in, bytes_out; /* Counter fuer Statistik */ - INT flag; /* Channel-Flag (vgl. "irc-write"-Modul) */ -} CONNECTION; - - LOCAL VOID Handle_Read PARAMS(( INT sock )); LOCAL BOOLEAN Handle_Write PARAMS(( CONN_ID Idx )); LOCAL VOID New_Connection PARAMS(( INT Sock )); LOCAL CONN_ID Socket2Index PARAMS(( INT Sock )); LOCAL VOID Read_Request PARAMS(( CONN_ID Idx )); LOCAL BOOLEAN Try_Write PARAMS(( CONN_ID Idx )); -LOCAL VOID Handle_Buffer PARAMS(( CONN_ID Idx )); +LOCAL BOOLEAN Handle_Buffer PARAMS(( CONN_ID Idx )); LOCAL VOID Check_Connections PARAMS(( VOID )); LOCAL VOID Check_Servers PARAMS(( VOID )); -LOCAL VOID Init_Conn_Struct PARAMS(( LONG Idx )); +LOCAL VOID Init_Conn_Struct PARAMS(( CONN_ID Idx )); LOCAL BOOLEAN Init_Socket PARAMS(( INT Sock )); LOCAL VOID New_Server PARAMS(( INT Server, CONN_ID Idx )); LOCAL VOID Read_Resolver_Result PARAMS(( INT r_fd )); - LOCAL fd_set My_Listeners; LOCAL fd_set My_Sockets; LOCAL fd_set My_Connects; -LOCAL CONNECTION *My_Connections; -LOCAL LONG Pool_Size; - GLOBAL VOID Conn_Init( VOID ) @@ -124,7 +105,7 @@ Conn_Init( VOID ) Log( LOG_EMERG, "Can't allocate memory! [Conn_Init]" ); exit( 1 ); } - Log( LOG_DEBUG, "Allocted connection pool for %ld items.", Pool_Size ); + Log( LOG_DEBUG, "Allocted connection pool for %d items (%ld bytes).", Pool_Size, sizeof( CONNECTION ) * Pool_Size ); /* zu Beginn haben wir keine Verbindungen */ FD_ZERO( &My_Listeners ); @@ -136,6 +117,9 @@ Conn_Init( VOID ) /* Connection-Struktur initialisieren */ for( i = 0; i < Pool_Size; i++ ) Init_Conn_Struct( i ); + + /* Global write counter */ + WCounter = 0; } /* Conn_Init */ @@ -170,7 +154,7 @@ Conn_Exit( VOID ) } else if( idx < Pool_Size ) { - if( NGIRCd_Restart ) Conn_Close( idx, NULL, "Server going down (restarting)", TRUE ); + if( NGIRCd_SignalRestart ) Conn_Close( idx, NULL, "Server going down (restarting)", TRUE ); else Conn_Close( idx, NULL, "Server going down", TRUE ); } else @@ -190,8 +174,7 @@ Conn_Exit( VOID ) GLOBAL INT Conn_InitListeners( VOID ) { - /* Ports, auf denen der Server Verbindungen entgegennehmen - * soll, initialisieren */ + /* Initialize ports on which the server should accept connections */ INT created, i; @@ -208,7 +191,7 @@ Conn_InitListeners( VOID ) GLOBAL VOID Conn_ExitListeners( VOID ) { - /* Alle "Listen-Sockets" schliessen */ + /* Close down all listening sockets */ INT i; @@ -227,9 +210,7 @@ Conn_ExitListeners( VOID ) GLOBAL BOOLEAN Conn_NewListener( CONST UINT Port ) { - /* Neuen Listen-Socket erzeugen: der Server wartet dann auf - * dem angegebenen Port auf Verbindungen. Kann der Listen- - * Socket nicht erteugt werden, so wird NULL geliefert.*/ + /* Create new listening socket on specified port */ struct sockaddr_in addr; INT sock; @@ -296,13 +277,19 @@ Conn_Handler( VOID ) fd_set read_sockets, write_sockets; struct timeval tv; time_t start, t; - LONG i, idx; + CONN_ID i, idx; + BOOLEAN timeout; start = time( NULL ); - while(( ! NGIRCd_Quit ) && ( ! NGIRCd_Restart )) + while(( ! NGIRCd_SignalQuit ) && ( ! NGIRCd_SignalRestart )) { - Check_Servers( ); + timeout = TRUE; + /* Should the configuration be reloaded? */ + if( NGIRCd_SignalRehash ) NGIRCd_Rehash( ); + + /* Check configured servers and established links */ + Check_Servers( ); Check_Connections( ); /* noch volle Lese-Buffer suchen */ @@ -311,7 +298,7 @@ Conn_Handler( VOID ) if(( My_Connections[i].sock > NONE ) && ( My_Connections[i].rdatalen > 0 )) { /* Kann aus dem Buffer noch ein Befehl extrahiert werden? */ - Handle_Buffer( i ); + if( Handle_Buffer( i )) timeout = FALSE; } } @@ -319,12 +306,17 @@ Conn_Handler( VOID ) FD_ZERO( &write_sockets ); for( i = 0; i < Pool_Size; i++ ) { +#ifdef USE_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 )) +#endif { /* Socket der Verbindung in Set aufnehmen */ FD_SET( My_Connections[i].sock, &write_sockets ); } } + /* Sockets mit im Aufbau befindlichen ausgehenden Verbindungen suchen */ for( i = 0; i < Pool_Size; i++ ) { @@ -350,7 +342,6 @@ Conn_Handler( VOID ) { /* Fuer die Verbindung ist eine "Penalty-Zeit" gesetzt */ FD_CLR( My_Connections[i].sock, &read_sockets ); - FD_CLR( My_Connections[i].sock, &write_sockets ); } } for( i = 0; i < Conn_MaxFD + 1; i++ ) @@ -363,7 +354,9 @@ Conn_Handler( VOID ) } /* Timeout initialisieren */ - tv.tv_sec = TIME_RES; tv.tv_usec = 0; + tv.tv_usec = 0; + if( timeout ) tv.tv_sec = TIME_RES; + else tv.tv_sec = 0; /* Auf Aktivitaet warten */ i = select( Conn_MaxFD + 1, &read_sockets, &write_sockets, NULL, &tv ); @@ -407,6 +400,9 @@ Conn_Handler( VOID ) if( FD_ISSET( i, &read_sockets )) Handle_Read( i ); } } + + if( NGIRCd_SignalQuit ) Log( LOG_NOTICE|LOG_snotice, "Server going down NOW!" ); + else if( NGIRCd_SignalRestart ) Log( LOG_NOTICE|LOG_snotice, "Server restarting NOW!" ); } /* Conn_Handler */ @@ -448,8 +444,9 @@ va_dcl if( NGIRCd_Sniffer ) Log( LOG_DEBUG, " -> connection %d: '%s'.", Idx, buffer ); #endif - strcat( buffer, "\r\n" ); + strlcat( buffer, "\r\n", sizeof( buffer )); ok = Conn_Write( Idx, buffer, strlen( buffer )); + My_Connections[Idx].msg_out++; va_end( ap ); return ok; @@ -466,43 +463,52 @@ Conn_Write( CONN_ID Idx, CHAR *Data, INT Len ) assert( Data != NULL ); assert( Len > 0 ); - /* Ist der entsprechende Socket ueberhaupt noch offen? - * In einem "Handler-Durchlauf" kann es passieren, dass - * dem nicht mehr so ist, wenn einer von mehreren - * Conn_Write()'s fehlgeschlagen ist. In diesem Fall - * wird hier einfach ein Fehler geliefert. */ + /* Ist der entsprechende Socket ueberhaupt noch offen? In einem + * "Handler-Durchlauf" kann es passieren, dass dem nicht mehr so + * ist, wenn einer von mehreren Conn_Write()'s fehlgeschlagen ist. + * In diesem Fall wird hier einfach ein Fehler geliefert. */ if( My_Connections[Idx].sock <= NONE ) { Log( LOG_DEBUG, "Skipped write on closed socket (connection %d).", Idx ); return FALSE; } - /* pruefen, ob Daten im Schreibpuffer sind. Wenn ja, zunaechst - * pruefen, ob diese gesendet werden koennen */ - if( My_Connections[Idx].wdatalen > 0 ) + /* Pruefen, ob im Schreibpuffer genuegend Platz ist. Ziel ist es, + * moeglichts viel im Puffer zu haben und _nicht_ gleich alles auf den + * Socket zu schreiben (u.a. wg. Komprimierung). */ + if( WRITEBUFFER_LEN - My_Connections[Idx].wdatalen - Len <= 0 ) { + /* Der Puffer ist dummerweise voll. Jetzt versuchen, den Puffer + * zu schreiben, wenn das nicht klappt, haben wir ein Problem ... */ if( ! Try_Write( Idx )) return FALSE; + + /* nun neu pruefen: */ + if( WRITEBUFFER_LEN - My_Connections[Idx].wdatalen - Len <= 0 ) + { + Log( LOG_NOTICE, "Write buffer overflow (connection %d)!", Idx ); + Conn_Close( Idx, "Write buffer overflow!", NULL, FALSE ); + return FALSE; + } } - /* pruefen, ob im Schreibpuffer genuegend Platz ist */ - if( WRITEBUFFER_LEN - My_Connections[Idx].wdatalen - Len <= 0 ) +#ifdef USE_ZLIB + if( My_Connections[Idx].options & CONN_ZIP ) { - /* der Puffer ist dummerweise voll ... */ - Log( LOG_NOTICE, "Write buffer overflow (connection %d)!", Idx ); - Conn_Close( Idx, "Write buffer overflow!", NULL, FALSE ); - return FALSE; + /* Daten komprimieren und in Puffer kopieren */ + if( ! Zip_Buffer( Idx, Data, Len )) return FALSE; } - - /* Daten in Puffer kopieren */ - memcpy( My_Connections[Idx].wbuf + My_Connections[Idx].wdatalen, Data, Len ); - My_Connections[Idx].wdatalen += Len; - - /* pruefen, on Daten vorhanden sind und geschrieben werden koennen */ - if( My_Connections[Idx].wdatalen > 0 ) + else +#endif { - if( ! Try_Write( Idx )) return FALSE; + /* Daten in Puffer kopieren */ + memcpy( My_Connections[Idx].wbuf + My_Connections[Idx].wdatalen, Data, Len ); + My_Connections[Idx].wdatalen += Len; + My_Connections[Idx].bytes_out += Len; } + /* Adjust global write counter */ + WCounter += Len; + return TRUE; } /* Conn_Write */ @@ -510,216 +516,136 @@ Conn_Write( CONN_ID Idx, CHAR *Data, INT Len ) GLOBAL VOID Conn_Close( CONN_ID Idx, CHAR *LogMsg, CHAR *FwdMsg, BOOLEAN InformClient ) { - /* Verbindung schliessen. Evtl. noch von Resolver - * Sub-Prozessen offene Pipes werden geschlossen. */ + /* Close connection. Open pipes of asyncronous resolver + * sub-processes are closed down. */ CLIENT *c; + DOUBLE in_k, out_k; +#ifdef USE_ZLIB + DOUBLE in_z_k, out_z_k; + INT in_p, out_p; +#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. */ + return; + } + + /* Mark link as "closing" */ + My_Connections[Idx].options |= CONN_ISCLOSING; + + /* Search client, if any */ c = Client_GetFromConn( Idx ); + /* Should the client be informed? */ if( InformClient ) { - /* Statistik an Client melden, wenn User */ +#ifndef STRICT_RFC + /* Send statistics to client if registered as user: */ if(( c != NULL ) && ( Client_Type( c ) == CLIENT_USER )) { Conn_WriteStr( Idx, "NOTICE %s :%sConnection statistics: client %.1f kb, server %.1f kb.", Client_ThisServer( ), NOTICE_TXTPREFIX, (DOUBLE)My_Connections[Idx].bytes_in / 1024, (DOUBLE)My_Connections[Idx].bytes_out / 1024 ); } +#endif - /* ERROR an Client schicken (von RFC so vorgesehen!) */ + /* 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 */ + (VOID)Try_Write( Idx ); + + /* Shut down socket */ if( close( My_Connections[Idx].sock ) != 0 ) { - Log( LOG_ERR, "Error closing connection %d (socket %d) with %s:%d - %s!", Idx, My_Connections[Idx].sock, My_Connections[Idx].host, ntohs( My_Connections[Idx].addr.sin_port), strerror( errno )); - } - else - { - Log( LOG_INFO, "Connection %d (socket %d) with %s:%d closed (%.1fK in/%.1fK out).", Idx, My_Connections[Idx].sock, My_Connections[Idx].host, ntohs( My_Connections[Idx].addr.sin_port ), (DOUBLE)My_Connections[Idx].bytes_in / 1024, (DOUBLE)My_Connections[Idx].bytes_out / 1024 ); + /* Oops, we can't close the socket!? This is fatal! */ + Log( LOG_EMERG, "Error closing connection %d (socket %d) with %s:%d - %s!", Idx, My_Connections[Idx].sock, My_Connections[Idx].host, ntohs( My_Connections[Idx].addr.sin_port), strerror( errno )); + Log( LOG_ALERT, "%s exiting due to fatal errors!", PACKAGE ); + exit( 1 ); } - - /* Socket als "ungueltig" markieren */ + + /* Mark socket as invalid: */ FD_CLR( My_Connections[Idx].sock, &My_Sockets ); FD_CLR( My_Connections[Idx].sock, &My_Connects ); My_Connections[Idx].sock = NONE; + /* If there is still a client, unregister it now */ if( c ) Client_Destroy( c, LogMsg, FwdMsg, TRUE ); + /* 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 + if( My_Connections[Idx].options & CONN_ZIP ) + { + in_z_k = (DOUBLE)My_Connections[Idx].zip.bytes_in / 1024; + out_z_k = (DOUBLE)My_Connections[Idx].zip.bytes_out / 1024; + in_p = (INT)(( in_k * 100 ) / in_z_k ); + out_p = (INT)(( out_k * 100 ) / out_z_k ); + Log( LOG_INFO, "Connection %d with %s:%d closed (in: %.1fk/%.1fk/%d%%, out: %.1fk/%.1fk/%d%%).", Idx, My_Connections[Idx].host, ntohs( My_Connections[Idx].addr.sin_port ), in_k, in_z_k, in_p, out_k, out_z_k, out_p ); + } + else +#endif + { + Log( LOG_INFO, "Connection %d with %s:%d closed (in: %.1fk, out: %.1fk).", Idx, My_Connections[Idx].host, ntohs( My_Connections[Idx].addr.sin_port ), in_k, out_k ); + } + + /* Is there a resolver sub-process running? */ if( My_Connections[Idx].res_stat ) { - /* Resolver-Strukturen freigeben, wenn noch nicht geschehen */ + /* Free resolver structures */ FD_CLR( My_Connections[Idx].res_stat->pipe[0], &Resolver_FDs ); close( My_Connections[Idx].res_stat->pipe[0] ); close( My_Connections[Idx].res_stat->pipe[1] ); free( My_Connections[Idx].res_stat ); } - /* Startzeit des naechsten Connect-Versuchs modifizieren? */ - if(( My_Connections[Idx].our_server > NONE ) && ( Conf_Server[My_Connections[Idx].our_server].lasttry < time( NULL ) - Conf_ConnectRetry )) + /* Servers: Modify time of next connect attempt? */ + Conf_UnsetServer( Idx ); + +#ifdef USE_ZLIB + /* Clean up zlib, if link was compressed */ + if( Conn_Options( Idx ) & CONN_ZIP ) { - /* Okay, die Verbindung stand schon "genuegend lange": - * lasttry-Zeitpunkt so setzen, dass der naechste - * Verbindungsversuch in RECONNECT_DELAY Sekunden - * gestartet wird. */ - Conf_Server[My_Connections[Idx].our_server].lasttry = time( NULL ) - Conf_ConnectRetry + RECONNECT_DELAY; + inflateEnd( &My_Connections[Idx].zip.in ); + deflateEnd( &My_Connections[Idx].zip.out ); } +#endif - /* Connection-Struktur loeschen (=freigeben) */ + /* Clean up connection structure (=free it) */ Init_Conn_Struct( Idx ); } /* Conn_Close */ -GLOBAL VOID -Conn_UpdateIdle( CONN_ID Idx ) -{ - /* Idle-Timer zuruecksetzen */ - - assert( Idx > NONE ); - My_Connections[Idx].lastprivmsg = time( NULL ); -} - - -GLOBAL time_t -Conn_GetIdle( CONN_ID Idx ) -{ - /* Idle-Time einer Verbindung liefern (in Sekunden) */ - - assert( Idx > NONE ); - return time( NULL ) - My_Connections[Idx].lastprivmsg; -} /* Conn_GetIdle */ - - -GLOBAL time_t -Conn_LastPing( CONN_ID Idx ) -{ - /* Zeitpunkt des letzten PING liefern */ - - assert( Idx > NONE ); - return My_Connections[Idx].lastping; -} /* Conn_LastPing */ - - -GLOBAL VOID -Conn_SetPenalty( CONN_ID Idx, time_t Seconds ) -{ - /* Penalty-Delay fuer eine Verbindung (in Sekunden) setzen; - * waehrend dieser Zeit wird der entsprechende Socket vom Server - * bei Lese-Operationen komplett ignoriert. Der Delay kann mit - * dieser Funktion nur erhoeht, nicht aber verringert werden. */ - - time_t t; - - assert( Idx > NONE ); - assert( Seconds >= 0 ); - - t = time( NULL ) + Seconds; - if( t > My_Connections[Idx].delaytime ) My_Connections[Idx].delaytime = t; -} /* Conn_SetPenalty */ - - -GLOBAL VOID -Conn_ResetPenalty( CONN_ID Idx ) -{ - assert( Idx > NONE ); - My_Connections[Idx].delaytime = 0; -} /* Conn_ResetPenalty */ - - -GLOBAL VOID -Conn_ClearFlags( VOID ) -{ - /* Alle Connection auf "nicht-markiert" setzen */ - - LONG i; - - for( i = 0; i < Pool_Size; i++ ) My_Connections[i].flag = 0; -} /* Conn_ClearFlags */ - - -GLOBAL INT -Conn_Flag( CONN_ID Idx ) -{ - /* Ist eine Connection markiert (TRUE) oder nicht? */ - - assert( Idx > NONE ); - return My_Connections[Idx].flag; -} /* Conn_Flag */ - - -GLOBAL VOID -Conn_SetFlag( CONN_ID Idx, INT Flag ) -{ - /* Connection markieren */ - - assert( Idx > NONE ); - My_Connections[Idx].flag = Flag; -} /* Conn_SetFlag */ - - -GLOBAL CONN_ID -Conn_First( VOID ) -{ - /* Connection-Struktur der ersten Verbindung liefern; - * Ist keine Verbindung vorhanden, wird NONE geliefert. */ - - LONG i; - - for( i = 0; i < Pool_Size; i++ ) - { - if( My_Connections[i].sock != NONE ) return i; - } - return NONE; -} /* Conn_First */ - - -GLOBAL CONN_ID -Conn_Next( CONN_ID Idx ) -{ - /* Naechste Verbindungs-Struktur liefern; existiert keine - * weitere, so wird NONE geliefert. */ - - LONG i = NONE; - - assert( Idx > NONE ); - - for( i = Idx + 1; i < Pool_Size; i++ ) - { - if( My_Connections[i].sock != NONE ) return i; - } - return NONE; -} /* Conn_Next */ - - -GLOBAL VOID -Conn_SetServer( CONN_ID Idx, INT ConfServer ) -{ - /* Connection als Server markieren: Index des konfigurierten - * Servers speichern. Verbindung muss bereits bestehen! */ - - assert( Idx > NONE ); - assert( My_Connections[Idx].sock > NONE ); - - My_Connections[Idx].our_server = ConfServer; -} /* Conn_SetServer */ - - LOCAL BOOLEAN Try_Write( CONN_ID Idx ) { - /* Versuchen, Daten aus dem Schreib-Puffer in den - * Socket zu schreiben. */ + /* Versuchen, Daten aus dem Schreib-Puffer in den Socket zu + * schreiben. TRUE wird geliefert, wenn entweder keine Daten + * zum Versenden vorhanden sind oder erfolgreich bearbeitet + * werden konnten. Im Fehlerfall wird FALSE geliefert und + * die Verbindung geschlossen. */ fd_set write_socket; struct timeval tv; assert( Idx > NONE ); assert( My_Connections[Idx].sock > NONE ); - assert( My_Connections[Idx].wdatalen > 0 ); + + /* sind ueberhaupt Daten vorhanden? */ +#ifdef USE_ZLIB + if(( ! My_Connections[Idx].wdatalen > 0 ) && ( ! My_Connections[Idx].zip.wdatalen )) return TRUE; +#else + if( ! My_Connections[Idx].wdatalen > 0 ) return TRUE; +#endif /* Timeout initialisieren: 0 Sekunden, also nicht blockieren */ tv.tv_sec = 0; tv.tv_usec = 0; @@ -783,6 +709,7 @@ Handle_Write( CONN_ID Idx ) /* Daten aus Schreibpuffer versenden bzw. Connection aufbauen */ INT len, res, err; + CLIENT *c; assert( Idx > NONE ); assert( My_Connections[Idx].sock > NONE ); @@ -804,25 +731,34 @@ Handle_Write( CONN_ID Idx ) { /* Fehler! */ if( res != 0 ) Log( LOG_CRIT, "getsockopt (connection %d): %s!", Idx, strerror( errno )); - else Log( LOG_CRIT, "Can't connect socket to \"%s:%d\" (connection %d): %s!", My_Connections[Idx].host, Conf_Server[My_Connections[Idx].our_server].port, Idx, strerror( err )); + else Log( LOG_CRIT, "Can't connect socket to \"%s:%d\" (connection %d): %s!", My_Connections[Idx].host, Conf_Server[Conf_GetServer( Idx )].port, Idx, strerror( err )); - /* Socket etc. pp. aufraeumen */ + /* Clean up socket, connection and client structures */ FD_CLR( My_Connections[Idx].sock, &My_Sockets ); + c = Client_GetFromConn( Idx ); + if( c ) Client_DestroyNow( c ); close( My_Connections[Idx].sock ); Init_Conn_Struct( Idx ); /* Bei Server-Verbindungen lasttry-Zeitpunkt auf "jetzt" setzen */ - Conf_Server[My_Connections[Idx].our_server].lasttry = time( NULL ); + Conf_Server[Conf_GetServer( Idx )].lasttry = time( NULL ); + Conf_UnsetServer( Idx ); return FALSE; } - Log( LOG_DEBUG, "Connection %d with \"%s:%d\" established, now sendig PASS and SERVER ...", Idx, My_Connections[Idx].host, Conf_Server[My_Connections[Idx].our_server].port ); + 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 ); /* PASS und SERVER verschicken */ - Conn_WriteStr( Idx, "PASS %s %s", Conf_Server[My_Connections[Idx].our_server].pwd_out, NGIRCd_ProtoID ); + 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 + /* Schreibpuffer leer, aber noch Daten im Kompressionsbuffer? + * Dann muss dieser nun geflushed werden! */ + if( My_Connections[Idx].wdatalen == 0 ) Zip_Flush( Idx ); +#endif + assert( My_Connections[Idx].wdatalen > 0 ); /* Daten schreiben */ @@ -838,9 +774,6 @@ Handle_Write( CONN_ID Idx ) return FALSE; } - /* Connection-Statistik aktualisieren */ - My_Connections[Idx].bytes_out += len; - /* Puffer anpassen */ My_Connections[Idx].wdatalen -= len; memmove( My_Connections[Idx].wbuf, My_Connections[Idx].wbuf + len, My_Connections[Idx].wdatalen ); @@ -898,6 +831,12 @@ New_Connection( INT Sock ) } if( new_size > Conf_MaxConnections ) new_size = Conf_MaxConnections; } + if( new_size < Pool_Size ) + { + Log( LOG_ALERT, "Can't accespt connection: limit (%d) reached -- overflow!", Pool_Size ); + close( new_sock ); + return; + } /* zunaechst realloc() versuchen; wenn das scheitert, malloc() versuchen * und Daten ggf. "haendisch" umkopieren. (Haesslich! Eine wirklich @@ -918,11 +857,18 @@ New_Connection( INT Sock ) /* Struktur umkopieren ... */ memcpy( ptr, My_Connections, sizeof( CONNECTION ) * Pool_Size ); - Log( LOG_DEBUG, "Allocated new connection pool for %ld items. [malloc()/memcpy()]", new_size ); + Log( LOG_DEBUG, "Allocated new connection pool for %ld items (%ld bytes). [malloc()/memcpy()]", new_size, sizeof( CONNECTION ) * new_size ); } - else Log( LOG_DEBUG, "Allocated new connection pool for %ld items. [realloc()]", new_size ); + else Log( LOG_DEBUG, "Allocated new connection pool for %ld items (%ld bytes). [realloc()]", new_size, sizeof( CONNECTION ) * new_size ); + /* Adjust pointer to new block */ My_Connections = ptr; + + /* Initialize new items */ + for( idx = Pool_Size; idx < new_size; idx++ ) Init_Conn_Struct( idx ); + idx = Pool_Size; + + /* Adjust new pool size */ Pool_Size = new_size; } @@ -947,13 +893,12 @@ New_Connection( INT Sock ) Log( LOG_INFO, "Accepted connection %d from %s:%d on socket %d.", idx, inet_ntoa( new_addr.sin_addr ), ntohs( new_addr.sin_port), Sock ); /* Hostnamen ermitteln */ - strcpy( My_Connections[idx].host, inet_ntoa( new_addr.sin_addr )); + strlcpy( My_Connections[idx].host, inet_ntoa( new_addr.sin_addr ), sizeof( My_Connections[idx].host )); Client_SetHostname( c, My_Connections[idx].host ); s = Resolve_Addr( &new_addr ); if( s ) { /* Sub-Prozess wurde asyncron gestartet */ - Conn_WriteStr( idx, "NOTICE AUTH :%sLooking up your hostname ...", NOTICE_TXTPREFIX ); My_Connections[idx].res_stat = s; } @@ -990,12 +935,27 @@ Read_Request( CONN_ID Idx ) /* Daten von Socket einlesen und entsprechend behandeln. * Tritt ein Fehler auf, so wird der Socket geschlossen. */ - INT len; + INT len, bsize; +#ifdef USE_ZLIB + CLIENT *c; +#endif assert( Idx > NONE ); assert( My_Connections[Idx].sock > NONE ); - if( READBUFFER_LEN - My_Connections[Idx].rdatalen - 2 < 0 ) + /* wenn noch nicht registriert: maximal mit ZREADBUFFER_LEN arbeiten, + * ansonsten koennen Daten ggf. nicht umkopiert werden. */ + bsize = READBUFFER_LEN; +#ifdef USE_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 + 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 ) +#endif { /* Der Lesepuffer ist voll */ Log( LOG_ERR, "Read buffer overflow (connection %d): %d bytes!", Idx, My_Connections[Idx].rdatalen ); @@ -1003,12 +963,23 @@ Read_Request( CONN_ID Idx ) return; } - len = recv( My_Connections[Idx].sock, My_Connections[Idx].rbuf + My_Connections[Idx].rdatalen, READBUFFER_LEN - My_Connections[Idx].rdatalen - 2, 0 ); +#ifdef USE_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 ); + if( len > 0 ) My_Connections[Idx].zip.rdatalen += len; + } + else +#endif + { + len = recv( My_Connections[Idx].sock, My_Connections[Idx].rbuf + My_Connections[Idx].rdatalen, bsize - My_Connections[Idx].rdatalen - 1, 0 ); + if( len > 0 ) My_Connections[Idx].rdatalen += len; + } if( len == 0 ) { /* Socket wurde geschlossen */ - Log( LOG_INFO, "%s:%d is closing the connection ...", inet_ntoa( My_Connections[Idx].addr.sin_addr ), ntohs( My_Connections[Idx].addr.sin_port)); + Log( LOG_INFO, "%s:%d (%s) is closing the connection ...", My_Connections[Idx].host, ntohs( My_Connections[Idx].addr.sin_port), inet_ntoa( My_Connections[Idx].addr.sin_addr )); Conn_Close( Idx, "Socket closed!", "Client closed connection", FALSE ); return; } @@ -1027,11 +998,6 @@ Read_Request( CONN_ID Idx ) /* Connection-Statistik aktualisieren */ My_Connections[Idx].bytes_in += len; - /* Lesebuffer updaten */ - My_Connections[Idx].rdatalen += len; - assert( My_Connections[Idx].rdatalen < READBUFFER_LEN ); - My_Connections[Idx].rbuf[My_Connections[Idx].rdatalen] = '\0'; - /* Timestamp aktualisieren */ My_Connections[Idx].lastdata = time( NULL ); @@ -1039,61 +1005,115 @@ Read_Request( CONN_ID Idx ) } /* Read_Request */ -LOCAL VOID +LOCAL BOOLEAN Handle_Buffer( CONN_ID Idx ) { - /* Daten im Lese-Puffer einer Verbindung verarbeiten. */ + /* Daten im Lese-Puffer einer Verbindung verarbeiten. + * Wurde ein Request verarbeitet, so wird TRUE geliefert, + * ansonsten FALSE (auch bei Fehlern). */ #ifndef STRICT_RFC CHAR *ptr1, *ptr2; #endif CHAR *ptr; INT len, delta; - - /* Eine komplette Anfrage muss mit CR+LF enden, vgl. - * RFC 2812. Haben wir eine? */ - ptr = strstr( My_Connections[Idx].rbuf, "\r\n" ); - - if( ptr ) delta = 2; -#ifndef STRICT_RFC - else - { - /* Nicht RFC-konforme Anfrage mit nur CR oder LF? Leider - * machen soetwas viele Clients, u.a. "mIRC" :-( */ - ptr1 = strchr( My_Connections[Idx].rbuf, '\r' ); - ptr2 = strchr( My_Connections[Idx].rbuf, '\n' ); - delta = 1; - if( ptr1 && ptr2 ) ptr = ptr1 > ptr2 ? ptr2 : ptr1; - else if( ptr1 ) ptr = ptr1; - else if( ptr2 ) ptr = ptr2; - } + BOOLEAN action, result; +#ifdef USE_ZLIB + BOOLEAN old_z; #endif - if( ptr ) + result = FALSE; + do { - /* Ende der Anfrage wurde gefunden */ - *ptr = '\0'; - len = ( ptr - My_Connections[Idx].rbuf ) + delta; - if( len > ( COMMAND_LEN - 1 )) +#ifdef USE_ZLIB + /* ggf. noch unkomprimiete Daten weiter entpacken */ + if( My_Connections[Idx].options & CONN_ZIP ) { - /* Eine Anfrage darf(!) nicht laenger als 512 Zeichen - * (incl. CR+LF!) werden; vgl. RFC 2812. Wenn soetwas - * empfangen wird, wird der Client disconnectiert. */ - Log( LOG_ERR, "Request too long (connection %d): %d bytes (max. %d expected)!", Idx, My_Connections[Idx].rdatalen, COMMAND_LEN - 1 ); - Conn_Close( Idx, NULL, "Request too long", TRUE ); - return; + if( ! Unzip_Buffer( Idx )) return FALSE; } +#endif + + if( My_Connections[Idx].rdatalen < 1 ) break; - if( len > delta ) + /* Eine komplette Anfrage muss mit CR+LF enden, vgl. + * RFC 2812. Haben wir eine? */ + My_Connections[Idx].rbuf[My_Connections[Idx].rdatalen] = '\0'; + ptr = strstr( My_Connections[Idx].rbuf, "\r\n" ); + + if( ptr ) delta = 2; +#ifndef STRICT_RFC + else { - /* Es wurde ein Request gelesen */ - if( ! Parse_Request( Idx, My_Connections[Idx].rbuf )) return; + /* Nicht RFC-konforme Anfrage mit nur CR oder LF? Leider + * machen soetwas viele Clients, u.a. "mIRC" :-( */ + ptr1 = strchr( My_Connections[Idx].rbuf, '\r' ); + ptr2 = strchr( My_Connections[Idx].rbuf, '\n' ); + delta = 1; + if( ptr1 && ptr2 ) ptr = ptr1 > ptr2 ? ptr2 : ptr1; + else if( ptr1 ) ptr = ptr1; + else if( ptr2 ) ptr = ptr2; } +#endif + + action = FALSE; + if( ptr ) + { + /* Ende der Anfrage wurde gefunden */ + *ptr = '\0'; + len = ( ptr - My_Connections[Idx].rbuf ) + delta; + if( len > ( COMMAND_LEN - 1 )) + { + /* Eine Anfrage darf(!) nicht laenger als 512 Zeichen + * (incl. CR+LF!) werden; vgl. RFC 2812. Wenn soetwas + * empfangen wird, wird der Client disconnectiert. */ + Log( LOG_ERR, "Request too long (connection %d): %d bytes (max. %d expected)!", Idx, My_Connections[Idx].rdatalen, COMMAND_LEN - 1 ); + Conn_Close( Idx, NULL, "Request too long", TRUE ); + return FALSE; + } - /* Puffer anpassen */ - My_Connections[Idx].rdatalen -= len; - memmove( My_Connections[Idx].rbuf, My_Connections[Idx].rbuf + len, My_Connections[Idx].rdatalen ); - } +#ifdef USE_ZLIB + /* merken, ob Stream bereits komprimiert wird */ + old_z = My_Connections[Idx].options & CONN_ZIP; +#endif + + if( len > delta ) + { + /* Es wurde ein Request gelesen */ + My_Connections[Idx].msg_in++; + if( ! Parse_Request( Idx, My_Connections[Idx].rbuf )) return FALSE; + else action = TRUE; + } + + /* Puffer anpassen */ + My_Connections[Idx].rdatalen -= len; + memmove( My_Connections[Idx].rbuf, My_Connections[Idx].rbuf + len, My_Connections[Idx].rdatalen ); + +#ifdef USE_ZLIB + if(( ! old_z ) && ( My_Connections[Idx].options & CONN_ZIP ) && ( My_Connections[Idx].rdatalen > 0 )) + { + /* 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 ) + { + /* 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; + Log( LOG_DEBUG, "Moved already received data (%d bytes) to uncompression buffer.", My_Connections[Idx].zip.rdatalen ); + } + } +#endif + } + + if( action ) result = TRUE; + } while( action ); + + return result; } /* Handle_Buffer */ @@ -1105,7 +1125,7 @@ Check_Connections( VOID ) * auch das nicht "hilft", Client disconnectieren. */ CLIENT *c; - LONG i; + CONN_ID i; for( i = 0; i < Pool_Size; i++ ) { @@ -1150,51 +1170,45 @@ Check_Connections( VOID ) LOCAL VOID Check_Servers( VOID ) { - /* Pruefen, ob Server-Verbindungen aufgebaut werden - * muessen bzw. koennen */ + /* Check if we can establish further server links */ RES_STAT *s; - LONG idx, n; - INT i; - - /* Wenn "Passive-Mode" aktiv: nicht verbinden */ - if( NGIRCd_Passive ) return; + CONN_ID idx; + INT i, n; - for( i = 0; i < Conf_Server_Count; i++ ) + /* Serach all connections, are there results from the resolver? */ + for( idx = 0; idx < Pool_Size; idx++ ) { - /* Ist ein Hostname und Port definiert? */ - if(( ! Conf_Server[i].host[0] ) || ( ! Conf_Server[i].port > 0 )) continue; + if( My_Connections[idx].sock != SERVER_WAIT ) continue; - /* Haben wir schon eine Verbindung? */ - for( n = 0; n < Pool_Size; n++ ) - { - if( My_Connections[n].sock == NONE ) continue; - - /* Verbindung zu diesem Server? */ - if( My_Connections[n].our_server == i ) - { - /* Komplett aufgebaute Verbindung? */ - if( My_Connections[n].sock > NONE ) break; + /* IP resolved? */ + if( My_Connections[idx].res_stat == NULL ) New_Server( Conf_GetServer( idx ), idx ); + } - /* IP schon aufgeloest? */ - if( My_Connections[n].res_stat == NULL ) New_Server( i, n ); - } + /* Check all configured servers */ + for( i = 0; i < MAX_SERVERS; i++ ) + { + /* Valid outgoing server which isn't already connected or disabled? */ + if(( ! Conf_Server[i].host[0] ) || ( ! Conf_Server[i].port > 0 ) || ( Conf_Server[i].conn_id > NONE ) || ( Conf_Server[i].flags & CONF_SFLAG_DISABLED )) continue; - /* Verbindung in dieser Server-Gruppe? */ - if(( My_Connections[n].our_server != NONE ) && ( Conf_Server[i].group != NONE )) + /* Is there already a connection in this group? */ + if( Conf_Server[i].group > NONE ) + { + for( n = 0; n < MAX_SERVERS; n++ ) { - if( Conf_Server[My_Connections[n].our_server].group == Conf_Server[i].group ) break; + if( n == i ) continue; + if(( Conf_Server[n].conn_id > NONE ) && ( Conf_Server[n].group == Conf_Server[i].group )) break; } + if( n < MAX_SERVERS ) continue; } - if( n < Pool_Size ) continue; - /* Wann war der letzte Connect-Versuch? */ + /* Check last connect attempt? */ if( Conf_Server[i].lasttry > time( NULL ) - Conf_ConnectRetry ) continue; - /* Okay, Verbindungsaufbau versuchen */ + /* Okay, try to connect now */ Conf_Server[i].lasttry = time( NULL ); - /* Freie Connection-Struktur suschen */ + /* Search free connection structure */ for( idx = 0; idx < Pool_Size; idx++ ) if( My_Connections[idx].sock == NONE ) break; if( idx >= Pool_Size ) { @@ -1206,12 +1220,11 @@ Check_Servers( VOID ) /* Verbindungs-Struktur initialisieren */ Init_Conn_Struct( idx ); My_Connections[idx].sock = SERVER_WAIT; - My_Connections[idx].our_server = i; + Conf_Server[i].conn_id = idx; - /* Hostnamen in IP aufloesen (Default bzw. im Fehlerfall: versuchen, den - * konfigurierten Text direkt als IP-Adresse zu verwenden ... */ - strcpy( Conf_Server[My_Connections[idx].our_server].ip, Conf_Server[i].host ); - strcpy( My_Connections[idx].host, Conf_Server[i].host ); + /* Resolve Hostname. If this fails, try to use it as an IP address */ + strlcpy( Conf_Server[i].ip, Conf_Server[i].host, sizeof( Conf_Server[i].ip )); + strlcpy( My_Connections[idx].host, Conf_Server[i].host, sizeof( My_Connections[idx].host )); s = Resolve_Name( Conf_Server[i].host ); if( s ) { @@ -1299,7 +1312,7 @@ New_Server( INT Server, CONN_ID Idx ) /* Verbindung registrieren */ My_Connections[Idx].sock = new_sock; My_Connections[Idx].addr = new_addr; - strcpy( My_Connections[Idx].host, Conf_Server[Server].host ); + strlcpy( My_Connections[Idx].host, Conf_Server[Server].host, sizeof( My_Connections[Idx].host )); /* Neuen Socket registrieren */ FD_SET( new_sock, &My_Sockets ); @@ -1311,7 +1324,7 @@ New_Server( INT Server, CONN_ID Idx ) LOCAL VOID -Init_Conn_Struct( LONG Idx ) +Init_Conn_Struct( CONN_ID Idx ) { /* Connection-Struktur initialisieren */ @@ -1322,14 +1335,26 @@ Init_Conn_Struct( LONG Idx ) My_Connections[Idx].rdatalen = 0; My_Connections[Idx].wbuf[0] = '\0'; My_Connections[Idx].wdatalen = 0; - My_Connections[Idx].our_server = NONE; + My_Connections[Idx].starttime = time( NULL ); My_Connections[Idx].lastdata = time( NULL ); My_Connections[Idx].lastping = 0; My_Connections[Idx].lastprivmsg = time( NULL ); My_Connections[Idx].delaytime = 0; My_Connections[Idx].bytes_in = 0; My_Connections[Idx].bytes_out = 0; + My_Connections[Idx].msg_in = 0; + My_Connections[Idx].msg_out = 0; My_Connections[Idx].flag = 0; + My_Connections[Idx].options = 0; + +#ifdef USE_ZLIB + My_Connections[Idx].zip.rbuf[0] = '\0'; + My_Connections[Idx].zip.rdatalen = 0; + My_Connections[Idx].zip.wbuf[0] = '\0'; + My_Connections[Idx].zip.wdatalen = 0; + My_Connections[Idx].zip.bytes_in = 0; + My_Connections[Idx].zip.bytes_out = 0; +#endif } /* Init_Conn_Struct */ @@ -1366,7 +1391,7 @@ Read_Resolver_Result( INT r_fd ) CHAR result[HOST_LEN]; CLIENT *c; - INT len, i; + INT len, i, n; FD_CLR( r_fd, &Resolver_FDs ); @@ -1408,16 +1433,15 @@ Read_Resolver_Result( INT r_fd ) /* Eingehende Verbindung: Hostnamen setzen */ c = Client_GetFromConn( i ); assert( c != NULL ); - strcpy( My_Connections[i].host, result ); + strlcpy( My_Connections[i].host, result, sizeof( My_Connections[i].host )); Client_SetHostname( c, result ); - - Conn_WriteStr( i, "NOTICE AUTH :%sGot your hostname.", NOTICE_TXTPREFIX ); } else { /* Ausgehende Verbindung (=Server): IP setzen */ - assert( My_Connections[i].our_server > NONE ); - strcpy( Conf_Server[My_Connections[i].our_server].ip, result ); + 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 */