2 * ngIRCd -- The Next Generation IRC Daemon
3 * Copyright (c)2001-2006 Alexander Barton (alex@barton.de)
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
9 * Please read the file COPYING, README and AUTHORS for more information.
11 * Connection compression using ZLIB
22 /* enable more zlib related debug messages: */
23 /* #define DEBUG_ZLIB */
25 static char UNUSED id[] = "$Id: conn-zip.c,v 1.11 2006/07/23 15:19:20 alex Exp $";
33 #include "conn-func.h"
42 Zip_InitConn( CONN_ID Idx )
44 /* Kompression fuer Link initialisieren */
48 My_Connections[Idx].zip.in.avail_in = 0;
49 My_Connections[Idx].zip.in.total_in = 0;
50 My_Connections[Idx].zip.in.total_out = 0;
51 My_Connections[Idx].zip.in.zalloc = NULL;
52 My_Connections[Idx].zip.in.zfree = NULL;
53 My_Connections[Idx].zip.in.data_type = Z_ASCII;
55 if( inflateInit( &My_Connections[Idx].zip.in ) != Z_OK )
58 Log( LOG_ALERT, "Can't initialize compression on connection %d (zlib inflate)!", Idx );
62 My_Connections[Idx].zip.out.total_in = 0;
63 My_Connections[Idx].zip.out.total_in = 0;
64 My_Connections[Idx].zip.out.zalloc = NULL;
65 My_Connections[Idx].zip.out.zfree = NULL;
66 My_Connections[Idx].zip.out.data_type = Z_ASCII;
68 if( deflateInit( &My_Connections[Idx].zip.out, Z_DEFAULT_COMPRESSION ) != Z_OK )
71 Log( LOG_ALERT, "Can't initialize compression on connection %d (zlib deflate)!", Idx );
75 My_Connections[Idx].zip.bytes_in = My_Connections[Idx].bytes_in;
76 My_Connections[Idx].zip.bytes_out = My_Connections[Idx].bytes_out;
78 Log( LOG_INFO, "Enabled link compression (zlib) on connection %d.", Idx );
79 Conn_OPTION_ADD( &My_Connections[Idx], CONN_ZIP );
86 Zip_Buffer( CONN_ID Idx, char *Data, size_t Len )
88 /* Daten zum Komprimieren im "Kompressions-Puffer" sammeln.
89 * Es wird true bei Erfolg, sonst false geliefert. */
92 assert( Data != NULL );
94 assert( Len <= ZWRITEBUFFER_LEN );
96 if (Len > ZWRITEBUFFER_LEN)
99 if ( array_bytes( &My_Connections[Idx].zip.wbuf ) >= ZWRITEBUFFER_LEN ) {
100 /* compression buffer is full, flush */
101 if( ! Zip_Flush( Idx )) return false;
104 return array_catb(&My_Connections[Idx].zip.wbuf, Data, Len);
109 Zip_Flush( CONN_ID Idx )
111 /* Daten komprimieren und in Schreibpuffer kopieren.
112 * Es wird true bei Erfolg, sonst false geliefert. */
115 unsigned char zipbuf[WRITEBUFFER_LEN];
119 out = &My_Connections[Idx].zip.out;
121 out->next_in = array_start(&My_Connections[Idx].zip.wbuf);
125 out->avail_in = (uInt)array_bytes(&My_Connections[Idx].zip.wbuf);
127 out->next_out = zipbuf;
128 out->avail_out = (uInt)sizeof zipbuf;
131 Log(LOG_DEBUG, "out->avail_in %d, out->avail_out %d",
132 out->avail_in, out->avail_out);
134 result = deflate( out, Z_SYNC_FLUSH );
135 if(( result != Z_OK ) || ( out->avail_in > 0 ))
137 Log( LOG_ALERT, "Compression error: code %d!?", result );
138 Conn_Close( Idx, "Compression error!", NULL, false );
142 assert(out->avail_out <= WRITEBUFFER_LEN);
143 zipbuf_used = WRITEBUFFER_LEN - out->avail_out;
145 Log(LOG_DEBUG, "zipbuf_used: %d", zipbuf_used);
147 if (!array_catb(&My_Connections[Idx].wbuf,
148 (char *)zipbuf, (size_t) zipbuf_used))
151 My_Connections[Idx].bytes_out += zipbuf_used;
152 My_Connections[Idx].zip.bytes_out += array_bytes(&My_Connections[Idx].zip.wbuf);
153 array_trunc(&My_Connections[Idx].zip.wbuf);
160 Unzip_Buffer( CONN_ID Idx )
162 /* Daten entpacken und in Lesepuffer kopieren. Bei Fehlern
163 * wird false geliefert, ansonsten true. Der Fall, dass keine
164 * Daten mehr zu entpacken sind, ist _kein_ Fehler! */
167 unsigned char unzipbuf[READBUFFER_LEN];
168 int unzipbuf_used = 0;
169 unsigned int z_rdatalen;
174 assert( Idx > NONE );
176 z_rdatalen = (unsigned int)array_bytes(&My_Connections[Idx].zip.rbuf);
180 in = &My_Connections[Idx].zip.in;
182 in->next_in = array_start(&My_Connections[Idx].zip.rbuf);
186 in->avail_in = z_rdatalen;
187 in->next_out = unzipbuf;
188 in->avail_out = (uInt)sizeof unzipbuf;
191 Log(LOG_DEBUG, "in->avail_in %d, in->avail_out %d",
192 in->avail_in, in->avail_out);
194 result = inflate( in, Z_SYNC_FLUSH );
197 Log( LOG_ALERT, "Decompression error: %s (code=%d, ni=%d, ai=%d, no=%d, ao=%d)!?", in->msg, result, in->next_in, in->avail_in, in->next_out, in->avail_out );
198 Conn_Close( Idx, "Decompression error!", NULL, false );
202 assert(z_rdatalen >= in->avail_in);
203 in_len = z_rdatalen - in->avail_in;
204 unzipbuf_used = READBUFFER_LEN - in->avail_out;
206 Log(LOG_DEBUG, "unzipbuf_used: %d - %d = %d", READBUFFER_LEN,
207 in->avail_out, unzipbuf_used);
209 assert(unzipbuf_used <= READBUFFER_LEN);
210 if (!array_catb(&My_Connections[Idx].rbuf, (char*) unzipbuf,
211 (size_t)unzipbuf_used))
214 if( in->avail_in > 0 ) {
215 array_moveleft(&My_Connections[Idx].zip.rbuf, 1, in_len );
217 array_trunc( &My_Connections[Idx].zip.rbuf );
218 My_Connections[Idx].zip.bytes_in += unzipbuf_used;
226 Zip_SendBytes( CONN_ID Idx )
228 /* Anzahl gesendeter Bytes (komprimiert!) liefern */
230 assert( Idx > NONE );
231 return My_Connections[Idx].zip.bytes_out;
232 } /* Zip_SendBytes */
236 Zip_RecvBytes( CONN_ID Idx )
238 /* Anzahl gesendeter Bytes (komprimiert!) liefern */
240 assert( Idx > NONE );
241 return My_Connections[Idx].zip.bytes_in;
242 } /* Zip_RecvBytes */