X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/1a6944fd74cfb70ace96d60bde31a7c8e0a5896d..bbee1b4abc13379569607df8f6985ec68dc12231:/src/common/zstream.cpp diff --git a/src/common/zstream.cpp b/src/common/zstream.cpp index 2db13338a3..2636efd14b 100644 --- a/src/common/zstream.cpp +++ b/src/common/zstream.cpp @@ -8,20 +8,37 @@ // Copyright: (c) Guilhem Lavaux // Licence: wxWindows licence ///////////////////////////////////////////////////////////////////////////// + #ifdef __GNUG__ #pragma implementation "zstream.h" #endif // For compilers that support precompilation, includes "wx.h". #include "wx/wxprec.h" -#include -#include -#include #ifdef __BORLANDC__ -#pragma hdrstop + #pragma hdrstop +#endif + +#if wxUSE_ZLIB && wxUSE_STREAMS + +#include "wx/zstream.h" +#include "wx/utils.h" +#include "wx/intl.h" +#include "wx/log.h" + +// normally, the compiler options should contain -I../zlib, but it is +// apparently not the case for all MSW makefiles and so, unless we use +// configure (which defines __WX_SETUP_H__) or it is explicitly overridden by +// the user (who can define wxUSE_ZLIB_H_IN_PATH), we hardcode the path here +#if defined(__WXMSW__) && !defined(__WX_SETUP_H__) && !defined(wxUSE_ZLIB_H_IN_PATH) + #include "../zlib/zlib.h" +#else + #include "zlib.h" #endif +#define ZSTREAM_BUFFER_SIZE 1024 + ////////////////////// // wxZlibInputStream ////////////////////// @@ -29,166 +46,178 @@ wxZlibInputStream::wxZlibInputStream(wxInputStream& stream) : wxFilterInputStream(stream) { - int err; + // I need a private stream buffer. + m_inflate = new z_stream_s; - m_inflate.zalloc = (alloc_func)0; - m_inflate.zfree = (free_func)0; - m_inflate.opaque = (voidpf)0; + m_inflate->zalloc = (alloc_func)0; + m_inflate->zfree = (free_func)0; + m_inflate->opaque = (voidpf)0; - err = inflateInit(&m_inflate); + int err = inflateInit(m_inflate); if (err != Z_OK) { - inflateEnd(&m_inflate); + inflateEnd(m_inflate); + delete m_inflate; return; } - m_inflate.avail_in = 0; + m_z_buffer = new unsigned char[ZSTREAM_BUFFER_SIZE]; + m_z_size = ZSTREAM_BUFFER_SIZE; + + m_inflate->avail_in = 0; + m_inflate->next_in = NULL; } wxZlibInputStream::~wxZlibInputStream() { - inflateEnd(&m_inflate); + inflateEnd(m_inflate); + delete m_inflate; + + delete [] m_z_buffer; } -wxInputStream& wxZlibInputStream::Read(void *buffer, size_t size) +size_t wxZlibInputStream::OnSysRead(void *buffer, size_t size) { int err; - m_inflate.next_out = (unsigned char *)buffer; - m_inflate.avail_out = size; - m_eof = FALSE; - - while (m_inflate.avail_out > 0) { - if (m_inflate.avail_in == 0) { - wxFilterInputStream::Read(m_z_buffer, m_z_size); - m_inflate.next_in = m_z_buffer; - m_inflate.avail_in = wxFilterInputStream::LastRead(); - if (wxFilterInputStream::Eof()) { - m_lastread = size - m_inflate.avail_out; - return *this; - } - } - err = inflate(&m_inflate, Z_FINISH); - if (err == Z_STREAM_END) { - m_lastread = size - m_inflate.avail_out; - m_eof = TRUE; - return *this; - } - } + m_inflate->next_out = (unsigned char *)buffer; + m_inflate->avail_out = size; - m_lastread = size; - return *this; -} + while (m_inflate->avail_out > 0) { + if (m_inflate->avail_in == 0) { -off_t wxZlibInputStream::SeekI(off_t WXUNUSED(pos), wxSeekMode WXUNUSED(mode)) -{ - return 0; -} + m_parent_i_stream->Read(m_z_buffer, wxMin(m_z_size, size)); + m_inflate->next_in = m_z_buffer; + m_inflate->avail_in = m_parent_i_stream->LastRead(); -off_t wxZlibInputStream::TellI() const -{ - return 0; -} + if (m_parent_i_stream->LastError() != wxStream_NOERROR && + m_parent_i_stream->LastError() != wxStream_EOF) + { + m_lasterror = m_parent_i_stream->LastError(); + return 0; // failed to read anything + } -bool wxZlibInputStream::Eof() const -{ - if (!m_eof) - return wxFilterInputStream::Eof(); - return m_eof; + if ( m_inflate->avail_in == 0 ) + { + // EOF + m_lasterror = wxStream_EOF; + break; + } + } + err = inflate(m_inflate, Z_FINISH); + if (err == Z_STREAM_END) + return (size - m_inflate->avail_out); + } + + return size-m_inflate->avail_out; } ////////////////////// // wxZlibOutputStream ////////////////////// -wxZlibOutputStream::wxZlibOutputStream(wxOutputStream& stream) +wxZlibOutputStream::wxZlibOutputStream(wxOutputStream& stream, int level) : wxFilterOutputStream(stream) { int err; - m_deflate.zalloc = (alloc_func)0; - m_deflate.zfree = (free_func)0; - m_deflate.opaque = (voidpf)0; + m_deflate = new z_stream_s; + + m_deflate->zalloc = (alloc_func)0; + m_deflate->zfree = (free_func)0; + m_deflate->opaque = (voidpf)0; - err = deflateInit(&m_deflate, Z_DEFAULT_COMPRESSION); + if (level == -1) level = Z_DEFAULT_COMPRESSION; + wxASSERT_MSG(level >= 0 && level <= 9, wxT("wxZlibOutputStream compression level must be between 0 and 9!")); + + err = deflateInit(m_deflate, level); if (err != Z_OK) { - deflateEnd(&m_deflate); + deflateEnd(m_deflate); return; } - m_deflate.avail_in = 0; - m_deflate.next_out = m_z_buffer; - m_deflate.avail_out = m_z_size; + + m_z_buffer = new unsigned char[ZSTREAM_BUFFER_SIZE]; + m_z_size = ZSTREAM_BUFFER_SIZE; + + m_deflate->avail_in = 0; + m_deflate->next_out = m_z_buffer; + m_deflate->avail_out = m_z_size; } wxZlibOutputStream::~wxZlibOutputStream() { int err; - while (1) { - err = deflate(&m_deflate, Z_FINISH); - if (err == Z_STREAM_END) - break; - if (err < 0) { - wxDebugMsg("wxZlibOutputStream: error during final deflate"); - break; - } - if (m_deflate.avail_out == 0) { - wxFilterOutputStream::Write(m_z_buffer, m_z_size); - if (wxFilterOutputStream::Bad()) { - wxDebugMsg("wxZlibOutputStream: error during final write"); - break; - } - m_deflate.next_out = m_z_buffer; - m_deflate.avail_out = m_z_size; - } + Sync(); + + err = deflate(m_deflate, Z_FINISH); + if (err != Z_STREAM_END) + { + wxLogDebug( wxT("wxZlibOutputStream: an error occured while closing the stream.\n") ); + return; } - wxFilterOutputStream::Write(m_z_buffer, m_z_size-m_deflate.avail_out); - deflateEnd(&m_deflate); + deflateEnd(m_deflate); + delete m_deflate; + + delete[] m_z_buffer; } -wxOutputStream& wxZlibOutputStream::Write(const void *buffer, size_t size) +void wxZlibOutputStream::Sync() { int err; - m_deflate.next_in = (unsigned char *)buffer; - m_deflate.avail_in = size; + m_parent_o_stream->Write(m_z_buffer, m_z_size-m_deflate->avail_out); + m_deflate->next_out = m_z_buffer; + m_deflate->avail_out = m_z_size; - m_bad = FALSE; - while (m_deflate.avail_in > 0) { - if (m_deflate.avail_out == 0) { - wxFilterOutputStream::Write(m_z_buffer, m_z_size); - if (wxFilterOutputStream::Bad()) { - m_lastwrite = size - m_deflate.avail_in; - return *this; - } - - m_deflate.next_out = m_z_buffer; - m_deflate.avail_out = m_z_size; - } - err = deflate(&m_deflate, Z_NO_FLUSH); - if (err < 0) { - m_bad = TRUE; - m_lastwrite = size - m_deflate.avail_in; - return *this; - } + err = deflate(m_deflate, Z_FULL_FLUSH); + if (err != Z_OK) { + return; } - m_lastwrite = size; - return *this; -} -off_t wxZlibOutputStream::SeekO(off_t WXUNUSED(pos), wxSeekMode WXUNUSED(mode)) -{ - return 0; -} + // Fixed by "Stefan Csomor" + while( m_deflate->avail_out == 0 ) + { + m_parent_o_stream->Write(m_z_buffer, m_z_size ); + m_deflate->next_out = m_z_buffer; + m_deflate->avail_out = m_z_size; + err = deflate(m_deflate, Z_FULL_FLUSH); + if (err != Z_OK) { + return; + } + } + // End -off_t wxZlibOutputStream::TellO() const -{ - return 0; + m_parent_o_stream->Write(m_z_buffer, m_z_size-m_deflate->avail_out); + m_deflate->next_out = m_z_buffer; + m_deflate->avail_out = m_z_size; } -bool wxZlibOutputStream::Bad() const +size_t wxZlibOutputStream::OnSysWrite(const void *buffer, size_t size) { - if (!m_bad) - return wxFilterOutputStream::Bad(); - return m_bad; + int err; + + m_deflate->next_in = (unsigned char *)buffer; + m_deflate->avail_in = size; + + while (m_deflate->avail_in > 0) { + + if (m_deflate->avail_out == 0) { + m_parent_o_stream->Write(m_z_buffer, m_z_size); + if (m_parent_o_stream->LastError() != wxStream_NOERROR) + return (size - m_deflate->avail_in); + + m_deflate->next_out = m_z_buffer; + m_deflate->avail_out = m_z_size; + } + + err = deflate(m_deflate, Z_NO_FLUSH); + if (err != Z_OK) + return (size - m_deflate->avail_in); + } + return size; } + +#endif + // wxUSE_ZLIB && wxUSE_STREAMS +