Mirror/zeek
1
0
Fork 0
mirror of https://github.com/zeek/zeek.git synced 2025-10-10 10:38:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Initial import of svn+ssh:://svn.icir.org/bro/trunk/bro as of r7088

Browse source
This commit is contained in:
Robin Sommer 2010-09-27 20:42:30 -07:00
commit 61757ac78b
1383 changed files with 380824 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

94
src/ZIP.cc Normal file
View file

@ -0,0 +1,94 @@
// $Id: ZIP.cc,v 1.1.4.2 2006/05/31 21:49:29 sommer Exp $
//
// See the file "COPYING" in the main distribution directory for copyright.
#include "ZIP.h"
#ifdef HAVE_LIBZ
ZIP_Analyzer::ZIP_Analyzer(Connection* conn, bool orig, Method arg_method)
: TCP_SupportAnalyzer(AnalyzerTag::Zip, conn, orig)
{
zip = 0;
zip_status = Z_OK;
method = arg_method;
zip = new z_stream;
zip->zalloc = 0;
zip->zfree = 0;
zip->opaque = 0;
zip->next_out = 0;
zip->avail_out = 0;
zip->next_in = 0;
zip->avail_in = 0;
// "15" here means maximum compression. "32" is a gross overload
// hack that means "check it for whether it's a gzip file". Sheesh.
zip_status = inflateInit2(zip, 15 + 32);
if ( zip_status != Z_OK )
{
Weird("inflate_init_failed");
delete zip;
zip = 0;
}
}
ZIP_Analyzer::~ZIP_Analyzer()
{
delete zip;
}
void ZIP_Analyzer::Done()
{
Analyzer::Done();
if ( zip )
inflateEnd(zip);
}
void ZIP_Analyzer::DeliverStream(int len, const u_char* data, bool orig)
{
TCP_SupportAnalyzer::DeliverStream(len, data, orig);
if ( ! len || zip_status != Z_OK )
return;
static unsigned int unzip_size = 4096;
Bytef unzipbuf[unzip_size];
zip->next_in = (Bytef*) data;
zip->avail_in = len;
do
{
zip->next_out = unzipbuf;
zip->avail_out = unzip_size;
zip_status = inflate(zip, Z_SYNC_FLUSH);
if ( zip_status != Z_STREAM_END &&
zip_status != Z_OK &&
zip_status != Z_BUF_ERROR )
{
Weird("inflate_failed");
inflateEnd(zip);
break;
}
int have = unzip_size - zip->avail_out;
if ( have )
ForwardStream(have, unzipbuf, IsOrig());
if ( zip_status == Z_STREAM_END )
{
inflateEnd(zip);
delete zip;
zip = 0;
break;
}
zip_status = Z_OK;
}
while ( zip->avail_out == 0 );
}
#endif