X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/4c68a10264347cc04b1a6634000595f160792231..74a8f67d96591cec101def2a7d47c64072aff7fd:/src/common/zstream.cpp diff --git a/src/common/zstream.cpp b/src/common/zstream.cpp index 09d616bab2..739a58e836 100644 --- a/src/common/zstream.cpp +++ b/src/common/zstream.cpp @@ -1,5 +1,5 @@ ////////////////////////////////////////////////////////////////////////////// -// Name: zstream.cpp +// Name: src/common/zstream.cpp // Purpose: Compressed stream classes // Author: Guilhem Lavaux // Modified by: Mike Wetherell @@ -9,32 +9,33 @@ // Licence: wxWindows licence ///////////////////////////////////////////////////////////////////////////// -#if defined(__GNUG__) && !defined(NO_GCC_PRAGMA) -#pragma implementation "zstream.h" -#endif - // For compilers that support precompilation, includes "wx.h". #include "wx/wxprec.h" #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" +#include "wx/versioninfo.h" + +#ifndef WX_PRECOMP + #include "wx/intl.h" + #include "wx/log.h" + #include "wx/utils.h" +#endif + // 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" + #include "../zlib/zlib.h" #else - #include "zlib.h" + #include "zlib.h" #endif enum { @@ -43,25 +44,103 @@ enum { ZSTREAM_AUTO = 0x20 // auto detect between gzip and zlib }; + +wxVersionInfo wxGetZlibVersionInfo() +{ + return wxVersionInfo("zlib", + ZLIB_VERNUM >> 12, + (ZLIB_VERNUM >> 8) & 0x0F, + (ZLIB_VERNUM & 0xFF) / 0x10); +} + +///////////////////////////////////////////////////////////////////////////// +// Zlib Class factory + +IMPLEMENT_DYNAMIC_CLASS(wxZlibClassFactory, wxFilterClassFactory) + +static wxZlibClassFactory g_wxZlibClassFactory; + +wxZlibClassFactory::wxZlibClassFactory() +{ + if (this == &g_wxZlibClassFactory) + PushFront(); +} + +const wxChar * const * +wxZlibClassFactory::GetProtocols(wxStreamProtocolType type) const +{ + static const wxChar *mimes[] = { wxT("application/x-deflate"), NULL }; + static const wxChar *encs[] = { wxT("deflate"), NULL }; + static const wxChar *empty[] = { NULL }; + + switch (type) { + case wxSTREAM_MIMETYPE: return mimes; + case wxSTREAM_ENCODING: return encs; + default: return empty; + } +} + + +///////////////////////////////////////////////////////////////////////////// +// Gzip Class factory + +IMPLEMENT_DYNAMIC_CLASS(wxGzipClassFactory, wxFilterClassFactory) + +static wxGzipClassFactory g_wxGzipClassFactory; + +wxGzipClassFactory::wxGzipClassFactory() +{ + if (this == &g_wxGzipClassFactory && wxZlibInputStream::CanHandleGZip()) + PushFront(); +} + +const wxChar * const * +wxGzipClassFactory::GetProtocols(wxStreamProtocolType type) const +{ + static const wxChar *protos[] = + { wxT("gzip"), NULL }; + static const wxChar *mimes[] = + { wxT("application/gzip"), wxT("application/x-gzip"), NULL }; + static const wxChar *encs[] = + { wxT("gzip"), NULL }; + static const wxChar *exts[] = + { wxT(".gz"), wxT(".gzip"), NULL }; + static const wxChar *empty[] = + { NULL }; + + switch (type) { + case wxSTREAM_PROTOCOL: return protos; + case wxSTREAM_MIMETYPE: return mimes; + case wxSTREAM_ENCODING: return encs; + case wxSTREAM_FILEEXT: return exts; + default: return empty; + } +} + + ////////////////////// // wxZlibInputStream ////////////////////// wxZlibInputStream::wxZlibInputStream(wxInputStream& stream, int flags) : wxFilterInputStream(stream) +{ + Init(flags); +} + +wxZlibInputStream::wxZlibInputStream(wxInputStream *stream, int flags) + : wxFilterInputStream(stream) +{ + Init(flags); +} + +void wxZlibInputStream::Init(int flags) { m_inflate = NULL; m_z_buffer = new unsigned char[ZSTREAM_BUFFER_SIZE]; m_z_size = ZSTREAM_BUFFER_SIZE; m_pos = 0; -#if WXWIN_COMPATIBILITY_2_4 - // treat compatibilty mode as auto - m_24compatibilty = flags == wxZLIB_24COMPATIBLE; - if (m_24compatibilty) - flags = wxZLIB_AUTO; -#endif - // if gzip is asked for but not supported... if ((flags == wxZLIB_GZIP || flags == wxZLIB_AUTO) && !CanHandleGZip()) { if (flags == wxZLIB_AUTO) { @@ -135,14 +214,17 @@ size_t wxZlibInputStream::OnSysRead(void *buffer, size_t size) break; case Z_STREAM_END: - // Unread any data taken from past the end of the deflate stream, so that - // any additional data can be read from the underlying stream (the crc - // in a gzip for example) - if (m_inflate->avail_in) { - m_parent_i_stream->Ungetch(m_inflate->next_in, m_inflate->avail_in); - m_inflate->avail_in = 0; + if (m_inflate->avail_out) { + // Unread any data taken from past the end of the deflate stream, so that + // any additional data can be read from the underlying stream (the crc + // in a gzip for example) + if (m_inflate->avail_in) { + m_parent_i_stream->Reset(); + m_parent_i_stream->Ungetch(m_inflate->next_in, m_inflate->avail_in); + m_inflate->avail_in = 0; + } + m_lasterror = wxSTREAM_EOF; } - m_lasterror = wxSTREAM_EOF; break; case Z_BUF_ERROR: @@ -151,18 +233,15 @@ size_t wxZlibInputStream::OnSysRead(void *buffer, size_t size) // by the parent strean, m_lasterror = wxSTREAM_READ_ERROR; if (m_parent_i_stream->Eof()) -#if WXWIN_COMPATIBILITY_2_4 - if (m_24compatibilty) - m_lasterror = wxSTREAM_EOF; - else -#endif + { wxLogError(_("Can't read inflate stream: unexpected EOF in underlying stream.")); + } break; default: wxString msg(m_inflate->msg, *wxConvCurrent); if (!msg) - msg.Format(_("zlib error %d"), err); + msg = wxString::Format(_("zlib error %d"), err); wxLogError(_("Can't read from inflate stream: %s"), msg.c_str()); m_lasterror = wxSTREAM_READ_ERROR; } @@ -180,6 +259,16 @@ size_t wxZlibInputStream::OnSysRead(void *buffer, size_t size) return major > 1 || (major == 1 && minor >= 2); } +bool wxZlibInputStream::SetDictionary(const char *data, const size_t datalen) +{ + return (inflateSetDictionary(m_inflate, (Bytef*)data, datalen) == Z_OK); +} + +bool wxZlibInputStream::SetDictionary(const wxMemoryBuffer &buf) +{ + return SetDictionary((char*)buf.GetData(), buf.GetDataLen()); +} + ////////////////////// // wxZlibOutputStream @@ -189,6 +278,19 @@ wxZlibOutputStream::wxZlibOutputStream(wxOutputStream& stream, int level, int flags) : wxFilterOutputStream(stream) +{ + Init(level, flags); +} + +wxZlibOutputStream::wxZlibOutputStream(wxOutputStream *stream, + int level, + int flags) + : wxFilterOutputStream(stream) +{ + Init(level, flags); +} + +void wxZlibOutputStream::Init(int level, int flags) { m_deflate = NULL; m_z_buffer = new unsigned char[ZSTREAM_BUFFER_SIZE]; @@ -218,7 +320,7 @@ wxZlibOutputStream::wxZlibOutputStream(wxOutputStream& stream, memset(m_deflate, 0, sizeof(z_stream_s)); m_deflate->next_out = m_z_buffer; m_deflate->avail_out = m_z_size; - + // see zlib.h for documentation on windowBits int windowBits = MAX_WBITS; switch (flags) { @@ -228,7 +330,7 @@ wxZlibOutputStream::wxZlibOutputStream(wxOutputStream& stream, default: wxFAIL_MSG(wxT("Invalid zlib flag")); } - if (deflateInit2(m_deflate, level, Z_DEFLATED, windowBits, + if (deflateInit2(m_deflate, level, Z_DEFLATED, windowBits, 8, Z_DEFAULT_STRATEGY) == Z_OK) return; } @@ -238,20 +340,18 @@ wxZlibOutputStream::wxZlibOutputStream(wxOutputStream& stream, m_lasterror = wxSTREAM_WRITE_ERROR; } -wxZlibOutputStream::~wxZlibOutputStream() -{ - if (m_deflate && m_z_buffer) - DoFlush(true); - deflateEnd(m_deflate); - delete m_deflate; +bool wxZlibOutputStream::Close() + { + DoFlush(true); + deflateEnd(m_deflate); + wxDELETE(m_deflate); + wxDELETEA(m_z_buffer); - delete[] m_z_buffer; -} + return wxFilterOutputStream::Close() && IsOk(); + } void wxZlibOutputStream::DoFlush(bool final) { - wxASSERT_MSG(m_deflate && m_z_buffer, wxT("Deflate stream not open")); - if (!m_deflate || !m_z_buffer) m_lasterror = wxSTREAM_WRITE_ERROR; if (!IsOk()) @@ -284,7 +384,11 @@ size_t wxZlibOutputStream::OnSysWrite(const void *buffer, size_t size) wxASSERT_MSG(m_deflate && m_z_buffer, wxT("Deflate stream not open")); if (!m_deflate || !m_z_buffer) + { + // notice that this will make IsOk() test just below return false m_lasterror = wxSTREAM_WRITE_ERROR; + } + if (!IsOk() || !size) return 0; @@ -312,7 +416,7 @@ size_t wxZlibOutputStream::OnSysWrite(const void *buffer, size_t size) m_lasterror = wxSTREAM_WRITE_ERROR; wxString msg(m_deflate->msg, *wxConvCurrent); if (!msg) - msg.Format(_("zlib error %d"), err); + msg = wxString::Format(_("zlib error %d"), err); wxLogError(_("Can't write to deflate stream: %s"), msg.c_str()); } @@ -322,10 +426,19 @@ size_t wxZlibOutputStream::OnSysWrite(const void *buffer, size_t size) } /* static */ bool wxZlibOutputStream::CanHandleGZip() -{ +{ return wxZlibInputStream::CanHandleGZip(); } +bool wxZlibOutputStream::SetDictionary(const char *data, const size_t datalen) +{ + return (deflateSetDictionary(m_deflate, (Bytef*)data, datalen) == Z_OK); +} + +bool wxZlibOutputStream::SetDictionary(const wxMemoryBuffer &buf) +{ + return SetDictionary((char*)buf.GetData(), buf.GetDataLen()); +} + #endif // wxUSE_ZLIB && wxUSE_STREAMS -