plugins/archivezip/zlibstream.h

   1 /*
   2    Copyright (C) 2001-2006, William Joseph.
   3    All Rights Reserved.
   4
   5    This file is part of GtkRadiant.
   6
   7    GtkRadiant is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 2 of the License, or
  10    (at your option) any later version.
  11
  12    GtkRadiant is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with GtkRadiant; if not, write to the Free Software
  19    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  20  */
  21
  22 #if !defined( INCLUDED_ZLIBSTREAM_H )
  23 #define INCLUDED_ZLIBSTREAM_H
  24
  25 #include "zlib.h"
  26 #include "idatastream.h"
  27
  28 /// \brief A wrapper around an InputStream of data compressed with the zlib deflate algorithm.
  29 ///
  30 /// - Uses z_stream to decompress the data stream on the fly.
  31 /// - Uses a buffer to reduce the number of times the wrapped stream must be read.
  32 class DeflatedInputStream : public InputStream
  33 {
  34 InputStream& m_istream;
  35 z_stream m_zipstream;
  36 enum unnamed0 { m_bufsize = 1024 };
  37 unsigned char m_buffer[m_bufsize];
  38
  39 public:
  40 DeflatedInputStream( InputStream& istream )
  41         : m_istream( istream ){
  42         m_zipstream.zalloc = 0;
  43         m_zipstream.zfree = 0;
  44         m_zipstream.opaque = 0;
  45         m_zipstream.avail_in = 0;
  46         inflateInit2( &m_zipstream, -MAX_WBITS );
  47 }
  48 ~DeflatedInputStream(){
  49         inflateEnd( &m_zipstream );
  50 }
  51 size_type read( byte_type* buffer, size_type length ){
  52         m_zipstream.next_out = buffer;
  53         m_zipstream.avail_out = static_cast<uInt>( length );
  54         while ( m_zipstream.avail_out != 0 )
  55         {
  56                 if ( m_zipstream.avail_in == 0 ) {
  57                         m_zipstream.next_in = m_buffer;
  58                         m_zipstream.avail_in = static_cast<uInt>( m_istream.read( m_buffer, m_bufsize ) );
  59                 }
  60                 if ( inflate( &m_zipstream, Z_SYNC_FLUSH ) != Z_OK ) {
  61                         break;
  62                 }
  63         }
  64         return length - m_zipstream.avail_out;
  65 }
  66 };
  67
  68 #endif