//////////////////////////////////////////////////////////////////////////////
-// Name: zstream.cpp
+// Name: src/common/zstream.cpp
// Purpose: Compressed stream classes
// Author: Guilhem Lavaux
// Modified by: Mike Wetherell
// Created: 11/07/98
// RCS-ID: $Id$
// Copyright: (c) Guilhem Lavaux
-// Licence: wxWidgets licence
+// 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"
+
+#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 {
ZSTREAM_AUTO = 0x20 // auto detect between gzip and zlib
};
+
+/////////////////////////////////////////////////////////////////////////////
+// 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[] = { _T("application/x-deflate"), NULL };
+ static const wxChar *encs[] = { _T("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[] =
+ { _T("gzip"), NULL };
+ static const wxChar *mimes[] =
+ { _T("application/gzip"), _T("application/x-gzip"), NULL };
+ static const wxChar *encs[] =
+ { _T("gzip"), NULL };
+ static const wxChar *exts[] =
+ { _T(".gz"), _T(".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) {
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:
// 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;
}
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
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];
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) {
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;
}
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);
+ delete m_deflate;
- delete[] m_z_buffer;
-}
+ m_deflate = NULL;
+ delete[] m_z_buffer;
+ m_z_buffer = NULL;
+
+ 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())
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;
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());
}
}
/* 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
-
projects / wxWidgets.git / blobdiff