X-Git-Url: https://arthur.barton.de/cgi-bin/gitweb.cgi?p=ngircd-alex.git;a=blobdiff_plain;f=src%2Fngircd%2Fconn.c;h=ab48e8506d114d006dc96d2e4e3127b5756145f3;hp=85a9966b9859283926d36603b374040ab793ca84;hb=39219230b2ce9b1c1478f2cea4ed9269a0fc111f;hpb=a0c032b2ff96c04d8a2a5215f4183b99296826c3 diff --git a/src/ngircd/conn.c b/src/ngircd/conn.c index 85a9966b..ab48e850 100644 --- a/src/ngircd/conn.c +++ b/src/ngircd/conn.c @@ -9,7 +9,7 @@ * Naehere Informationen entnehmen Sie bitter der Datei COPYING. Eine Liste * der an ngIRCd beteiligten Autoren finden Sie in der Datei AUTHORS. * - * $Id: conn.c,v 1.93 2002/11/22 17:58:19 alex Exp $ + * $Id: conn.c,v 1.101 2002/12/02 13:19:37 alex Exp $ * * connect.h: Verwaltung aller Netz-Verbindungen ("connections") */ @@ -42,6 +42,10 @@ #include /* u.a. fuer Mac OS X */ #endif +#ifdef USE_ZLIB +#include +#endif + #include "exp.h" #include "conn.h" @@ -60,6 +64,20 @@ #define SERVER_WAIT (NONE - 1) +#ifdef USE_ZLIB +typedef struct _ZipData +{ + z_stream in; /* "Handle" fuer Input-Stream */ + z_stream out; /* "Handle" fuer Output-Stream */ + CHAR rbuf[READBUFFER_LEN]; /* Lesepuffer */ + INT rdatalen; /* Laenge der Daten im Lesepuffer (komprimiert) */ + CHAR wbuf[WRITEBUFFER_LEN]; /* Schreibpuffer */ + INT wdatalen; /* Laenge der Daten im Schreibpuffer (unkomprimiert) */ + LONG bytes_in, bytes_out; /* Counter fuer Statistik (unkomprimiert!) */ +} ZIPDATA; +#endif + + typedef struct _Connection { INT sock; /* Socket Handle */ @@ -71,12 +89,18 @@ typedef struct _Connection CHAR wbuf[WRITEBUFFER_LEN]; /* Schreibpuffer */ INT wdatalen; /* Laenge der Daten im Schreibpuffer */ INT our_server; /* wenn von uns zu connectender Server: ID */ + time_t starttime; /* Startzeit des Links */ 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) */ + LONG bytes_in, bytes_out; /* Empfangene uns gesendete Bytes */ + LONG msg_in, msg_out; /* Empfangene uns gesendete Nachtichten */ + INT flag; /* "Markierungs-Flag" (vgl. "irc-write"-Modul) */ + INT options; /* Link-Optionen */ +#ifdef USE_ZLIB + ZIPDATA zip; /* Kompressionsinformationen */ +#endif } CONNECTION; @@ -86,7 +110,7 @@ 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 )); @@ -94,6 +118,12 @@ 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 )); +#ifdef USE_ZLIB +LOCAL BOOLEAN Zip_Buffer PARAMS(( CONN_ID Idx, CHAR *Data, INT Len )); +LOCAL BOOLEAN Zip_Flush PARAMS(( CONN_ID Idx )); +LOCAL BOOLEAN Unzip_Buffer PARAMS(( CONN_ID Idx )); +#endif + LOCAL fd_set My_Listeners; LOCAL fd_set My_Sockets; @@ -297,10 +327,13 @@ Conn_Handler( VOID ) struct timeval tv; time_t start, t; LONG i, idx; + BOOLEAN timeout; start = time( NULL ); while(( ! NGIRCd_Quit ) && ( ! NGIRCd_Restart )) { + timeout = TRUE; + Check_Servers( ); Check_Connections( ); @@ -311,7 +344,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,7 +352,11 @@ 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 ); @@ -363,8 +400,9 @@ Conn_Handler( VOID ) } /* Timeout initialisieren */ - tv.tv_sec = 1; 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 ); @@ -451,6 +489,7 @@ va_dcl strcat( buffer, "\r\n" ); ok = Conn_Write( Idx, buffer, strlen( buffer )); + My_Connections[Idx].msg_out++; va_end( ap ); return ok; @@ -467,41 +506,47 @@ 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; } return TRUE; @@ -515,6 +560,11 @@ Conn_Close( CONN_ID Idx, CHAR *LogMsg, CHAR *FwdMsg, BOOLEAN InformClient ) * Sub-Prozessen offene Pipes werden geschlossen. */ 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 ); @@ -523,11 +573,13 @@ Conn_Close( CONN_ID Idx, CHAR *LogMsg, CHAR *FwdMsg, BOOLEAN InformClient ) if( InformClient ) { +#ifndef STRICT_RFC /* Statistik an Client melden, wenn 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!) */ if( FwdMsg ) Conn_WriteStr( Idx, "ERROR :%s", FwdMsg ); @@ -535,13 +587,32 @@ Conn_Close( CONN_ID Idx, CHAR *LogMsg, CHAR *FwdMsg, BOOLEAN InformClient ) if( My_Connections[Idx].sock == NONE ) return; } + /* zunaechst versuchen, noch im Schreibpuffer vorhandene + * Daten auf den Socket zu schreiben ... */ + Try_Write( Idx ); + 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 ); + 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 (socket %d) with %s:%d closed (in: %.1fk/%.1fk/%d%%, out: %.1fk/%.1fk/%d%%).", Idx, My_Connections[Idx].sock, 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 (socket %d) with %s:%d closed (in: %.1fk, out: %.1fk).", Idx, My_Connections[Idx].sock, My_Connections[Idx].host, ntohs( My_Connections[Idx].addr.sin_port ), in_k, out_k ); + } } /* Socket als "ungueltig" markieren */ @@ -570,6 +641,15 @@ Conn_Close( CONN_ID Idx, CHAR *LogMsg, CHAR *FwdMsg, BOOLEAN InformClient ) Conf_Server[My_Connections[Idx].our_server].lasttry = time( NULL ) - Conf_ConnectRetry + RECONNECT_DELAY; } +#ifdef USE_ZLIB + /* Ggf. zlib abmelden */ + if( Conn_Options( Idx ) & CONN_ZIP ) + { + inflateEnd( &My_Connections[Idx].zip.in ); + deflateEnd( &My_Connections[Idx].zip.out ); + } +#endif + /* Connection-Struktur loeschen (=freigeben) */ Init_Conn_Struct( Idx ); } /* Conn_Close */ @@ -709,21 +789,213 @@ Conn_SetServer( CONN_ID Idx, INT ConfServer ) } /* Conn_SetServer */ +GLOBAL VOID +Conn_SetOption( CONN_ID Idx, INT Option ) +{ + /* Option fuer Verbindung setzen. + * Initial sind alle Optionen _nicht_ gesetzt. */ + + assert( Idx > NONE ); + assert( Option != 0 ); + + My_Connections[Idx].options |= Option; +} /* Conn_SetOption */ + + +GLOBAL VOID +Conn_UnsetOption( CONN_ID Idx, INT Option ) +{ + /* Option fuer Verbindung loeschen */ + + assert( Idx > NONE ); + assert( Option != 0 ); + + My_Connections[Idx].options &= ~Option; +} /* Conn_UnsetOption */ + + +GLOBAL INT +Conn_Options( CONN_ID Idx ) +{ + assert( Idx > NONE ); + return My_Connections[Idx].options; +} /* Conn_Options */ + + +#ifdef USE_ZLIB + +GLOBAL BOOLEAN +Conn_InitZip( CONN_ID Idx ) +{ + /* Kompression fuer Link initialisieren */ + + assert( Idx > NONE ); + + My_Connections[Idx].zip.in.avail_in = 0; + My_Connections[Idx].zip.in.total_in = 0; + My_Connections[Idx].zip.in.total_out = 0; + My_Connections[Idx].zip.in.zalloc = NULL; + My_Connections[Idx].zip.in.zfree = NULL; + My_Connections[Idx].zip.in.data_type = Z_ASCII; + + if( inflateInit( &My_Connections[Idx].zip.in ) != Z_OK ) + { + /* Fehler! */ + Log( LOG_ALERT, "Can't initialize compression on connection %d (zlib inflate)!", Idx ); + return FALSE; + } + + My_Connections[Idx].zip.out.total_in = 0; + My_Connections[Idx].zip.out.total_in = 0; + My_Connections[Idx].zip.out.zalloc = NULL; + My_Connections[Idx].zip.out.zfree = NULL; + My_Connections[Idx].zip.out.data_type = Z_ASCII; + + if( deflateInit( &My_Connections[Idx].zip.out, Z_DEFAULT_COMPRESSION ) != Z_OK ) + { + /* Fehler! */ + Log( LOG_ALERT, "Can't initialize compression on connection %d (zlib deflate)!", Idx ); + return FALSE; + } + + My_Connections[Idx].zip.bytes_in = My_Connections[Idx].bytes_in; + My_Connections[Idx].zip.bytes_out = My_Connections[Idx].bytes_out; + + Log( LOG_INFO, "Enabled link compression (zlib) on connection %d.", Idx ); + Conn_SetOption( Idx, CONN_ZIP ); + + return TRUE; +} /* Conn_InitZip */ + + +GLOBAL LONG +Conn_SendBytesZip( CONN_ID Idx ) +{ + /* Anzahl gesendeter Bytes (komprimiert!) liefern */ + + assert( Idx > NONE ); + return My_Connections[Idx].zip.bytes_out; +} /* Conn_SendBytesZip */ + + +GLOBAL LONG +Conn_RecvBytesZip( CONN_ID Idx ) +{ + /* Anzahl gesendeter Bytes (komprimiert!) liefern */ + + assert( Idx > NONE ); + return My_Connections[Idx].zip.bytes_in; +} /* Conn_RecvBytesZip */ + +#endif + + +GLOBAL time_t +Conn_StartTime( CONN_ID Idx ) +{ + /* Zeitpunkt des Link-Starts liefern (in Sekunden) */ + + assert( Idx > NONE ); + return My_Connections[Idx].starttime; +} /* Conn_Uptime */ + + +GLOBAL INT +Conn_SendQ( CONN_ID Idx ) +{ + /* Laenge der Daten im Schreibbuffer liefern */ + + assert( Idx > NONE ); +#ifdef USE_ZLIB + return My_Connections[Idx].zip.wdatalen; +#else + return My_Connections[Idx].wdatalen; +#endif +} /* Conn_SendQ */ + + +GLOBAL LONG +Conn_SendMsg( CONN_ID Idx ) +{ + /* Anzahl gesendeter Nachrichten liefern */ + + assert( Idx > NONE ); + return My_Connections[Idx].msg_out; +} /* Conn_SendMsg */ + + +GLOBAL LONG +Conn_SendBytes( CONN_ID Idx ) +{ + /* Anzahl gesendeter Bytes (unkomprimiert) liefern */ + + assert( Idx > NONE ); + return My_Connections[Idx].bytes_out; +} /* Conn_SendBytes */ + + +GLOBAL INT +Conn_RecvQ( CONN_ID Idx ) +{ + /* Laenge der Daten im Lesebuffer liefern */ + + assert( Idx > NONE ); +#ifdef USE_ZLIB + return My_Connections[Idx].zip.rdatalen; +#else + return My_Connections[Idx].rdatalen; +#endif +} /* Conn_RecvQ */ + + +GLOBAL LONG +Conn_RecvMsg( CONN_ID Idx ) +{ + /* Anzahl empfangener Nachrichten liefern */ + + assert( Idx > NONE ); + return My_Connections[Idx].msg_in; +} /* Conn_RecvMsg */ + + +GLOBAL LONG +Conn_RecvBytes( CONN_ID Idx ) +{ + /* Anzahl empfangener Bytes (unkomprimiert) liefern */ + + assert( Idx > NONE ); + return My_Connections[Idx].bytes_in; +} /* Conn_RecvBytes */ + + 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; FD_ZERO( &write_socket ); FD_SET( My_Connections[Idx].sock, &write_socket ); - if( select( My_Connections[Idx].sock + 1, NULL, &write_socket, NULL, 0 ) == -1 ) + if( select( My_Connections[Idx].sock + 1, NULL, &write_socket, NULL, &tv ) == -1 ) { /* Fehler! */ if( errno != EINTR ) @@ -820,6 +1092,12 @@ Handle_Write( CONN_ID Idx ) 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 */ @@ -835,9 +1113,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 ); @@ -950,7 +1225,6 @@ New_Connection( INT Sock ) if( s ) { /* Sub-Prozess wurde asyncron gestartet */ - Conn_WriteStr( idx, "NOTICE AUTH :%sLooking up your hostname ...", NOTICE_TXTPREFIX ); My_Connections[idx].res_stat = s; } @@ -987,12 +1261,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 ); @@ -1000,7 +1289,18 @@ 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 ) { @@ -1024,11 +1324,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 ); @@ -1036,61 +1331,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 */ @@ -1320,13 +1669,26 @@ Init_Conn_Struct( LONG Idx ) 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 */ @@ -1407,8 +1769,6 @@ Read_Resolver_Result( INT r_fd ) assert( c != NULL ); strcpy( My_Connections[i].host, result ); Client_SetHostname( c, result ); - - Conn_WriteStr( i, "NOTICE AUTH :%sGot your hostname.", NOTICE_TXTPREFIX ); } else { @@ -1422,4 +1782,115 @@ Read_Resolver_Result( INT r_fd ) } /* Read_Resolver_Result */ +#ifdef USE_ZLIB + +LOCAL BOOLEAN +Zip_Buffer( CONN_ID Idx, CHAR *Data, INT Len ) +{ + /* Daten zum Komprimieren im "Kompressions-Puffer" sammeln. + * Es wird TRUE bei Erfolg, sonst FALSE geliefert. */ + + assert( Idx > NONE ); + assert( Data != NULL ); + assert( Len > 0 ); + + /* Ist noch Platz im Kompressions-Puffer? */ + if( ZWRITEBUFFER_LEN - My_Connections[Idx].zip.wdatalen < Len + 50 ) + { + /* Nein! Puffer zunaechst leeren ...*/ + if( ! Zip_Flush( Idx )) return FALSE; + } + + /* Daten kopieren */ + memmove( My_Connections[Idx].zip.wbuf + My_Connections[Idx].zip.wdatalen, Data, Len ); + My_Connections[Idx].zip.wdatalen += Len; + + return TRUE; +} /* Zip_Buffer */ + + +LOCAL BOOLEAN +Zip_Flush( CONN_ID Idx ) +{ + /* Daten komprimieren und in Schreibpuffer kopieren. + * Es wird TRUE bei Erfolg, sonst FALSE geliefert. */ + + INT result, out_len; + z_stream *out; + + out = &My_Connections[Idx].zip.out; + + out->next_in = My_Connections[Idx].zip.wbuf; + out->avail_in = My_Connections[Idx].zip.wdatalen; + out->next_out = My_Connections[Idx].wbuf + My_Connections[Idx].wdatalen; + out->avail_out = WRITEBUFFER_LEN - My_Connections[Idx].wdatalen; + + result = deflate( out, Z_SYNC_FLUSH ); + if(( result != Z_OK ) || ( out->avail_in > 0 )) + { + Log( LOG_ALERT, "Compression error: code %d!?", result ); + Conn_Close( Idx, "Compression error!", NULL, FALSE ); + return FALSE; + } + + out_len = WRITEBUFFER_LEN - My_Connections[Idx].wdatalen - out->avail_out; + My_Connections[Idx].wdatalen += out_len; + My_Connections[Idx].bytes_out += out_len; + My_Connections[Idx].zip.bytes_out += My_Connections[Idx].zip.wdatalen; + My_Connections[Idx].zip.wdatalen = 0; + + return TRUE; +} /* Zip_Flush */ + + +LOCAL BOOLEAN +Unzip_Buffer( CONN_ID Idx ) +{ + /* Daten entpacken und in Lesepuffer kopieren. Bei Fehlern + * wird FALSE geliefert, ansonsten TRUE. Der Fall, dass keine + * Daten mehr zu entpacken sind, ist _kein_ Fehler! */ + + INT result, in_len, out_len; + z_stream *in; + + assert( Idx > NONE ); + + if( My_Connections[Idx].zip.rdatalen <= 0 ) return TRUE; + + in = &My_Connections[Idx].zip.in; + + in->next_in = My_Connections[Idx].zip.rbuf; + in->avail_in = My_Connections[Idx].zip.rdatalen; + in->next_out = My_Connections[Idx].rbuf + My_Connections[Idx].rdatalen; + in->avail_out = READBUFFER_LEN - My_Connections[Idx].rdatalen - 1; + + result = inflate( in, Z_SYNC_FLUSH ); + if( result != Z_OK ) + { + Log( LOG_ALERT, "Decompression error: code %d (ni=%d, ai=%d, no=%d, ao=%d)!?", result, in->next_in, in->avail_in, in->next_out, in->avail_out ); + Conn_Close( Idx, "Decompression error!", NULL, FALSE ); + return FALSE; + } + + in_len = My_Connections[Idx].zip.rdatalen - in->avail_in; + out_len = READBUFFER_LEN - My_Connections[Idx].rdatalen - 1 - in->avail_out; + My_Connections[Idx].rdatalen += out_len; + + if( in->avail_in > 0 ) + { + /* es konnten nicht alle Daten entpackt werden, vermutlich war + * im Ziel-Puffer kein Platz mehr. Umkopieren ... */ + My_Connections[Idx].zip.rdatalen -= in_len; + memmove( My_Connections[Idx].zip.rbuf, My_Connections[Idx].zip.rbuf + in_len, My_Connections[Idx].zip.rdatalen ); + } + else My_Connections[Idx].zip.rdatalen = 0; + My_Connections[Idx].zip.bytes_in += out_len; + + return TRUE; +} /* Unzip_Buffer */ + + +#endif + + /* -eof- */