src/common/zipstrm.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        zipstrm.cpp
   3 // Purpose:     Streams for Zip files
   4 // Author:      Mike Wetherell
   5 // RCS-ID:      $Id$
   6 // Copyright:   (c) Mike Wetherell
   7 // Licence:     wxWindows licence
   8 /////////////////////////////////////////////////////////////////////////////
   9
  10 // For compilers that support precompilation, includes "wx.h".
  11 #include "wx/wxprec.h"
  12
  13 #ifdef __BORLANDC__
  14   #pragma hdrstop
  15 #endif
  16
  17 #ifndef WX_PRECOMP
  18   #include "wx/defs.h"
  19 #endif
  20
  21 #if wxUSE_ZLIB && wxUSE_STREAMS && wxUSE_ZIPSTREAM
  22
  23 #include "wx/zipstrm.h"
  24 #include "wx/log.h"
  25 #include "wx/intl.h"
  26 #include "wx/datstrm.h"
  27 #include "wx/zstream.h"
  28 #include "wx/mstream.h"
  29 #include "wx/utils.h"
  30 #include "wx/buffer.h"
  31 #include "wx/ptr_scpd.h"
  32 #include "wx/wfstream.h"
  33 #include "wx/link.h"
  34 #include "zlib.h"
  35
  36 // value for the 'version needed to extract' field (20 means 2.0)
  37 enum {
  38     VERSION_NEEDED_TO_EXTRACT = 20
  39 };
  40
  41 // signatures for the various records (PKxx)
  42 enum {
  43     CENTRAL_MAGIC = 0x02014b50,     // central directory record
  44     LOCAL_MAGIC   = 0x04034b50,     // local header
  45     END_MAGIC     = 0x06054b50,     // end of central directory record
  46     SUMS_MAGIC    = 0x08074b50      // data descriptor (info-zip)
  47 };
  48
  49 // unix file attributes. zip stores them in the high 16 bits of the
  50 // 'external attributes' field, hence the extra zeros.
  51 enum {
  52     wxZIP_S_IFMT  = 0xF0000000,
  53     wxZIP_S_IFDIR = 0x40000000,
  54     wxZIP_S_IFREG = 0x80000000
  55 };
  56
  57 // minimum sizes for the various records
  58 enum {
  59     CENTRAL_SIZE  = 46,
  60     LOCAL_SIZE    = 30,
  61     END_SIZE      = 22,
  62     SUMS_SIZE     = 12
  63 };
  64
  65 // The number of bytes that must be written to an wxZipOutputStream before
  66 // a zip entry is created. The purpose of this latency is so that
  67 // OpenCompressor() can see a little data before deciding which compressor
  68 // it should use.
  69 enum {
  70     OUTPUT_LATENCY = 4096
  71 };
  72
  73 // Some offsets into the local header
  74 enum {
  75     SUMS_OFFSET  = 14
  76 };
  77
  78 IMPLEMENT_DYNAMIC_CLASS(wxZipEntry, wxArchiveEntry)
  79 IMPLEMENT_DYNAMIC_CLASS(wxZipClassFactory, wxArchiveClassFactory)
  80
  81 wxFORCE_LINK_THIS_MODULE(zipstrm)
  82
  83
  84 /////////////////////////////////////////////////////////////////////////////
  85 // Helpers
  86
  87 // read a string of a given length
  88 //
  89 static wxString ReadString(wxInputStream& stream, wxUint16 len, wxMBConv& conv)
  90 {
  91 #if wxUSE_UNICODE
  92     wxCharBuffer buf(len);
  93     stream.Read(buf.data(), len);
  94     wxString str(buf, conv);
  95 #else
  96     wxString str;
  97     (void)conv;
  98     {
  99         wxStringBuffer buf(str, len);
 100         stream.Read(buf, len);
 101     }
 102 #endif
 103
 104     return str;
 105 }
 106
 107 // Decode a little endian wxUint32 number from a character array
 108 //
 109 static inline wxUint32 CrackUint32(const char *m)
 110 {
 111     const unsigned char *n = (const unsigned char*)m;
 112     return (n[3] << 24) | (n[2] << 16) | (n[1] << 8) | n[0];
 113 }
 114
 115 // Temporarily lower the logging level in debug mode to avoid a warning
 116 // from SeekI about seeking on a stream with data written back to it.
 117 //
 118 static wxFileOffset QuietSeek(wxInputStream& stream, wxFileOffset pos)
 119 {
 120 #ifdef __WXDEBUG__
 121     wxLogLevel level = wxLog::GetLogLevel();
 122     wxLog::SetLogLevel(wxLOG_Debug - 1);
 123     wxFileOffset result = stream.SeekI(pos);
 124     wxLog::SetLogLevel(level);
 125     return result;
 126 #else
 127     return stream.SeekI(pos);
 128 #endif
 129 }
 130
 131
 132 /////////////////////////////////////////////////////////////////////////////
 133 // Read a zip header
 134
 135 class wxZipHeader
 136 {
 137 public:
 138     wxZipHeader(wxInputStream& stream, size_t size);
 139
 140     inline wxUint8 Read8();
 141     inline wxUint16 Read16();
 142     inline wxUint32 Read32();
 143
 144     const char *GetData() const             { return m_data; }
 145     size_t GetSize() const                  { return m_size; }
 146     operator bool() const                   { return m_ok; }
 147
 148     size_t Seek(size_t pos)                 { m_pos = pos; return m_pos; }
 149     size_t Skip(size_t size)                { m_pos += size; return m_pos; }
 150
 151     wxZipHeader& operator>>(wxUint8& n)     { n = Read8();  return *this; }
 152     wxZipHeader& operator>>(wxUint16& n)    { n = Read16(); return *this; }
 153     wxZipHeader& operator>>(wxUint32& n)    { n = Read32(); return *this; }
 154
 155 private:
 156     char m_data[64];
 157     size_t m_size;
 158     size_t m_pos;
 159     bool m_ok;
 160 };
 161
 162 wxZipHeader::wxZipHeader(wxInputStream& stream, size_t size)
 163   : m_size(0),
 164     m_pos(0),
 165     m_ok(false)
 166 {
 167     wxCHECK_RET(size <= sizeof(m_data), _T("buffer too small"));
 168     m_size = stream.Read(m_data, size).LastRead();
 169     m_ok = m_size == size;
 170 }
 171
 172 wxUint8 wxZipHeader::Read8()
 173 {
 174     wxASSERT(m_pos < m_size);
 175     return *wx_reinterpret_cast(wxUint8*, m_data + m_pos++);
 176 }
 177
 178 wxUint16 wxZipHeader::Read16()
 179 {
 180     wxASSERT(m_pos + 2 <= m_size);
 181     wxUint16 n = *wx_reinterpret_cast(wxUint16*, m_data + m_pos);
 182     m_pos += 2;
 183     return wxUINT16_SWAP_ON_BE(n);
 184 }
 185
 186 wxUint32 wxZipHeader::Read32()
 187 {
 188     wxASSERT(m_pos + 4 <= m_size);
 189     wxUint32 n = *wx_reinterpret_cast(wxUint32*, m_data + m_pos);
 190     m_pos += 4;
 191     return wxUINT32_SWAP_ON_BE(n);
 192 }
 193
 194
 195 /////////////////////////////////////////////////////////////////////////////
 196 // Stored input stream
 197 // Trival decompressor for files which are 'stored' in the zip file.
 198
 199 class wxStoredInputStream : public wxFilterInputStream
 200 {
 201 public:
 202     wxStoredInputStream(wxInputStream& stream);
 203
 204     void Open(wxFileOffset len) { Close(); m_len = len; }
 205     void Close() { m_pos = 0; m_lasterror = wxSTREAM_NO_ERROR; }
 206
 207     virtual char Peek() { return wxInputStream::Peek(); }
 208     virtual wxFileOffset GetLength() const { return m_len; }
 209
 210 protected:
 211     virtual size_t OnSysRead(void *buffer, size_t size);
 212     virtual wxFileOffset OnSysTell() const { return m_pos; }
 213
 214 private:
 215     wxFileOffset m_pos;
 216     wxFileOffset m_len;
 217
 218     DECLARE_NO_COPY_CLASS(wxStoredInputStream)
 219 };
 220
 221 wxStoredInputStream::wxStoredInputStream(wxInputStream& stream)
 222   : wxFilterInputStream(stream),
 223     m_pos(0),
 224     m_len(0)
 225 {
 226 }
 227
 228 size_t wxStoredInputStream::OnSysRead(void *buffer, size_t size)
 229 {
 230     size_t count = wx_truncate_cast(size_t,
 231                 wxMin(size + wxFileOffset(0), m_len - m_pos + size_t(0)));
 232     count = m_parent_i_stream->Read(buffer, count).LastRead();
 233     m_pos += count;
 234
 235     if (count < size)
 236         m_lasterror = m_pos == m_len ? wxSTREAM_EOF : wxSTREAM_READ_ERROR;
 237
 238     return count;
 239 }
 240
 241
 242 /////////////////////////////////////////////////////////////////////////////
 243 // Stored output stream
 244 // Trival compressor for files which are 'stored' in the zip file.
 245
 246 class wxStoredOutputStream : public wxFilterOutputStream
 247 {
 248 public:
 249     wxStoredOutputStream(wxOutputStream& stream) :
 250         wxFilterOutputStream(stream), m_pos(0) { }
 251
 252     bool Close() {
 253         m_pos = 0;
 254         m_lasterror = wxSTREAM_NO_ERROR;
 255         return true;
 256     }
 257
 258 protected:
 259     virtual size_t OnSysWrite(const void *buffer, size_t size);
 260     virtual wxFileOffset OnSysTell() const { return m_pos; }
 261
 262 private:
 263     wxFileOffset m_pos;
 264     DECLARE_NO_COPY_CLASS(wxStoredOutputStream)
 265 };
 266
 267 size_t wxStoredOutputStream::OnSysWrite(const void *buffer, size_t size)
 268 {
 269     if (!IsOk() || !size)
 270         return 0;
 271     size_t count = m_parent_o_stream->Write(buffer, size).LastWrite();
 272     if (count != size)
 273         m_lasterror = wxSTREAM_WRITE_ERROR;
 274     m_pos += count;
 275     return count;
 276 }
 277
 278
 279 /////////////////////////////////////////////////////////////////////////////
 280 // wxRawInputStream
 281 //
 282 // Used to handle the unusal case of raw copying an entry of unknown
 283 // length. This can only happen when the zip being copied from is being
 284 // read from a non-seekable stream, and also was original written to a
 285 // non-seekable stream.
 286 //
 287 // In this case there's no option but to decompress the stream to find
 288 // it's length, but we can still write the raw compressed data to avoid the
 289 // compression overhead (which is the greater one).
 290 //
 291 // Usage is like this:
 292 //  m_rawin = new wxRawInputStream(*m_parent_i_stream);
 293 //  m_decomp = m_rawin->Open(OpenDecompressor(m_rawin->GetTee()));
 294 //
 295 // The wxRawInputStream owns a wxTeeInputStream object, the role of which
 296 // is something like the unix 'tee' command; it is a transparent filter, but
 297 // allows the data read to be read a second time via an extra method 'GetData'.
 298 //
 299 // The wxRawInputStream then draws data through the tee using a decompressor
 300 // then instead of returning the decompressed data, retuns the raw data
 301 // from wxTeeInputStream::GetData().
 302
 303 class wxTeeInputStream : public wxFilterInputStream
 304 {
 305 public:
 306     wxTeeInputStream(wxInputStream& stream);
 307
 308     size_t GetCount() const { return m_end - m_start; }
 309     size_t GetData(char *buffer, size_t size);
 310
 311     void Open();
 312     bool Final();
 313
 314     wxInputStream& Read(void *buffer, size_t size);
 315
 316 protected:
 317     virtual size_t OnSysRead(void *buffer, size_t size);
 318     virtual wxFileOffset OnSysTell() const { return m_pos; }
 319
 320 private:
 321     wxFileOffset m_pos;
 322     wxMemoryBuffer m_buf;
 323     size_t m_start;
 324     size_t m_end;
 325
 326     DECLARE_NO_COPY_CLASS(wxTeeInputStream)
 327 };
 328
 329 wxTeeInputStream::wxTeeInputStream(wxInputStream& stream)
 330   : wxFilterInputStream(stream),
 331     m_pos(0), m_buf(8192), m_start(0), m_end(0)
 332 {
 333 }
 334
 335 void wxTeeInputStream::Open()
 336 {
 337     m_pos = m_start = m_end = 0;
 338     m_lasterror = wxSTREAM_NO_ERROR;
 339 }
 340
 341 bool wxTeeInputStream::Final()
 342 {
 343     bool final = m_end == m_buf.GetDataLen();
 344     m_end = m_buf.GetDataLen();
 345     return final;
 346 }
 347
 348 wxInputStream& wxTeeInputStream::Read(void *buffer, size_t size)
 349 {
 350     size_t count = wxInputStream::Read(buffer, size).LastRead();
 351     m_end = m_buf.GetDataLen();
 352     m_buf.AppendData(buffer, count);
 353     return *this;
 354 }
 355
 356 size_t wxTeeInputStream::OnSysRead(void *buffer, size_t size)
 357 {
 358     size_t count = m_parent_i_stream->Read(buffer, size).LastRead();
 359     if (count < size)
 360         m_lasterror = m_parent_i_stream->GetLastError();
 361     return count;
 362 }
 363
 364 size_t wxTeeInputStream::GetData(char *buffer, size_t size)
 365 {
 366     if (m_wbacksize) {
 367         size_t len = m_buf.GetDataLen();
 368         len = len > m_wbacksize ? len - m_wbacksize : 0;
 369         m_buf.SetDataLen(len);
 370         if (m_end > len) {
 371             wxFAIL; // we've already returned data that's now being ungot
 372             m_end = len;
 373         }
 374         m_parent_i_stream->Reset();
 375         m_parent_i_stream->Ungetch(m_wback, m_wbacksize);
 376         free(m_wback);
 377         m_wback = NULL;
 378         m_wbacksize = 0;
 379         m_wbackcur = 0;
 380     }
 381
 382     if (size > GetCount())
 383         size = GetCount();
 384     if (size) {
 385         memcpy(buffer, m_buf + m_start, size);
 386         m_start += size;
 387         wxASSERT(m_start <= m_end);
 388     }
 389
 390     if (m_start == m_end && m_start > 0 && m_buf.GetDataLen() > 0) {
 391         size_t len = m_buf.GetDataLen();
 392         char *buf = (char*)m_buf.GetWriteBuf(len);
 393         len -= m_end;
 394         memmove(buf, buf + m_end, len);
 395         m_buf.UngetWriteBuf(len);
 396         m_start = m_end = 0;
 397     }
 398
 399     return size;
 400 }
 401
 402 class wxRawInputStream : public wxFilterInputStream
 403 {
 404 public:
 405     wxRawInputStream(wxInputStream& stream);
 406     virtual ~wxRawInputStream() { delete m_tee; }
 407
 408     wxInputStream* Open(wxInputStream *decomp);
 409     wxInputStream& GetTee() const { return *m_tee; }
 410
 411 protected:
 412     virtual size_t OnSysRead(void *buffer, size_t size);
 413     virtual wxFileOffset OnSysTell() const { return m_pos; }
 414
 415 private:
 416     wxFileOffset m_pos;
 417     wxTeeInputStream *m_tee;
 418
 419     enum { BUFSIZE = 8192 };
 420     wxCharBuffer m_dummy;
 421
 422     DECLARE_NO_COPY_CLASS(wxRawInputStream)
 423 };
 424
 425 wxRawInputStream::wxRawInputStream(wxInputStream& stream)
 426   : wxFilterInputStream(stream),
 427     m_pos(0),
 428     m_tee(new wxTeeInputStream(stream)),
 429     m_dummy(BUFSIZE)
 430 {
 431 }
 432
 433 wxInputStream *wxRawInputStream::Open(wxInputStream *decomp)
 434 {
 435     if (decomp) {
 436         m_parent_i_stream = decomp;
 437         m_pos = 0;
 438         m_lasterror = wxSTREAM_NO_ERROR;
 439         m_tee->Open();
 440         return this;
 441     } else {
 442         return NULL;
 443     }
 444 }
 445
 446 size_t wxRawInputStream::OnSysRead(void *buffer, size_t size)
 447 {
 448     char *buf = (char*)buffer;
 449     size_t count = 0;
 450
 451     while (count < size && IsOk())
 452     {
 453         while (m_parent_i_stream->IsOk() && m_tee->GetCount() == 0)
 454             m_parent_i_stream->Read(m_dummy.data(), BUFSIZE);
 455
 456         size_t n = m_tee->GetData(buf + count, size - count);
 457         count += n;
 458
 459         if (n == 0 && m_tee->Final())
 460             m_lasterror = m_parent_i_stream->GetLastError();
 461     }
 462
 463     m_pos += count;
 464     return count;
 465 }
 466
 467
 468 /////////////////////////////////////////////////////////////////////////////
 469 // Zlib streams than can be reused without recreating.
 470
 471 class wxZlibOutputStream2 : public wxZlibOutputStream
 472 {
 473 public:
 474     wxZlibOutputStream2(wxOutputStream& stream, int level) :
 475         wxZlibOutputStream(stream, level, wxZLIB_NO_HEADER) { }
 476
 477     bool Open(wxOutputStream& stream);
 478     bool Close() { DoFlush(true); m_pos = wxInvalidOffset; return IsOk(); }
 479 };
 480
 481 bool wxZlibOutputStream2::Open(wxOutputStream& stream)
 482 {
 483     wxCHECK(m_pos == wxInvalidOffset, false);
 484
 485     m_deflate->next_out = m_z_buffer;
 486     m_deflate->avail_out = m_z_size;
 487     m_pos = 0;
 488     m_lasterror = wxSTREAM_NO_ERROR;
 489     m_parent_o_stream = &stream;
 490
 491     if (deflateReset(m_deflate) != Z_OK) {
 492         wxLogError(_("can't re-initialize zlib deflate stream"));
 493         m_lasterror = wxSTREAM_WRITE_ERROR;
 494         return false;
 495     }
 496
 497     return true;
 498 }
 499
 500 class wxZlibInputStream2 : public wxZlibInputStream
 501 {
 502 public:
 503     wxZlibInputStream2(wxInputStream& stream) :
 504         wxZlibInputStream(stream, wxZLIB_NO_HEADER) { }
 505
 506     bool Open(wxInputStream& stream);
 507 };
 508
 509 bool wxZlibInputStream2::Open(wxInputStream& stream)
 510 {
 511     m_inflate->avail_in = 0;
 512     m_pos = 0;
 513     m_lasterror = wxSTREAM_NO_ERROR;
 514     m_parent_i_stream = &stream;
 515
 516     if (inflateReset(m_inflate) != Z_OK) {
 517         wxLogError(_("can't re-initialize zlib inflate stream"));
 518         m_lasterror = wxSTREAM_READ_ERROR;
 519         return false;
 520     }
 521
 522     return true;
 523 }
 524
 525
 526 /////////////////////////////////////////////////////////////////////////////
 527 // Class to hold wxZipEntry's Extra and LocalExtra fields
 528
 529 class wxZipMemory
 530 {
 531 public:
 532     wxZipMemory() : m_data(NULL), m_size(0), m_capacity(0), m_ref(1) { }
 533
 534     wxZipMemory *AddRef() { m_ref++; return this; }
 535     void Release() { if (--m_ref == 0) delete this; }
 536
 537     char *GetData() const { return m_data; }
 538     size_t GetSize() const { return m_size; }
 539     size_t GetCapacity() const { return m_capacity; }
 540
 541     wxZipMemory *Unique(size_t size);
 542
 543 private:
 544     ~wxZipMemory() { delete [] m_data; }
 545
 546     char *m_data;
 547     size_t m_size;
 548     size_t m_capacity;
 549     int m_ref;
 550
 551     wxSUPPRESS_GCC_PRIVATE_DTOR_WARNING(wxZipMemory)
 552 };
 553
 554 wxZipMemory *wxZipMemory::Unique(size_t size)
 555 {
 556     wxZipMemory *zm;
 557
 558     if (m_ref > 1) {
 559         --m_ref;
 560         zm = new wxZipMemory;
 561     } else {
 562         zm = this;
 563     }
 564
 565     if (zm->m_capacity < size) {
 566         delete [] zm->m_data;
 567         zm->m_data = new char[size];
 568         zm->m_capacity = size;
 569     }
 570
 571     zm->m_size = size;
 572     return zm;
 573 }
 574
 575 static inline wxZipMemory *AddRef(wxZipMemory *zm)
 576 {
 577     if (zm)
 578         zm->AddRef();
 579     return zm;
 580 }
 581
 582 static inline void Release(wxZipMemory *zm)
 583 {
 584     if (zm)
 585         zm->Release();
 586 }
 587
 588 static void Copy(wxZipMemory*& dest, wxZipMemory *src)
 589 {
 590     Release(dest);
 591     dest = AddRef(src);
 592 }
 593
 594 static void Unique(wxZipMemory*& zm, size_t size)
 595 {
 596     if (!zm && size)
 597         zm = new wxZipMemory;
 598     if (zm)
 599         zm = zm->Unique(size);
 600 }
 601
 602
 603 /////////////////////////////////////////////////////////////////////////////
 604 // Collection of weak references to entries
 605
 606 WX_DECLARE_HASH_MAP(long, wxZipEntry*, wxIntegerHash,
 607                     wxIntegerEqual, wx__OffsetZipEntryMap);
 608
 609 class wxZipWeakLinks
 610 {
 611 public:
 612     wxZipWeakLinks() : m_ref(1) { }
 613
 614     void Release(const wxZipInputStream* WXUNUSED(x))
 615         { if (--m_ref == 0) delete this; }
 616     void Release(wxFileOffset key)
 617         { RemoveEntry(key); if (--m_ref == 0) delete this; }
 618
 619     wxZipWeakLinks *AddEntry(wxZipEntry *entry, wxFileOffset key);
 620     void RemoveEntry(wxFileOffset key)
 621         { m_entries.erase(wx_truncate_cast(key_type, key)); }
 622     wxZipEntry *GetEntry(wxFileOffset key) const;
 623     bool IsEmpty() const { return m_entries.empty(); }
 624
 625 private:
 626     ~wxZipWeakLinks() { wxASSERT(IsEmpty()); }
 627
 628     typedef wx__OffsetZipEntryMap::key_type key_type;
 629
 630     int m_ref;
 631     wx__OffsetZipEntryMap m_entries;
 632
 633     wxSUPPRESS_GCC_PRIVATE_DTOR_WARNING(wxZipWeakLinks)
 634 };
 635
 636 wxZipWeakLinks *wxZipWeakLinks::AddEntry(wxZipEntry *entry, wxFileOffset key)
 637 {
 638     m_entries[wx_truncate_cast(key_type, key)] = entry;
 639     m_ref++;
 640     return this;
 641 }
 642
 643 wxZipEntry *wxZipWeakLinks::GetEntry(wxFileOffset key) const
 644 {
 645     wx__OffsetZipEntryMap::const_iterator it =
 646         m_entries.find(wx_truncate_cast(key_type, key));
 647     return it != m_entries.end() ?  it->second : NULL;
 648 }
 649
 650
 651 /////////////////////////////////////////////////////////////////////////////
 652 // ZipEntry
 653
 654 wxZipEntry::wxZipEntry(
 655     const wxString& name /*=wxEmptyString*/,
 656     const wxDateTime& dt /*=wxDateTime::Now()*/,
 657     wxFileOffset size    /*=wxInvalidOffset*/)
 658   :
 659     m_SystemMadeBy(wxZIP_SYSTEM_MSDOS),
 660     m_VersionMadeBy(wxMAJOR_VERSION * 10 + wxMINOR_VERSION),
 661     m_VersionNeeded(VERSION_NEEDED_TO_EXTRACT),
 662     m_Flags(0),
 663     m_Method(wxZIP_METHOD_DEFAULT),
 664     m_DateTime(dt),
 665     m_Crc(0),
 666     m_CompressedSize(wxInvalidOffset),
 667     m_Size(size),
 668     m_Key(wxInvalidOffset),
 669     m_Offset(wxInvalidOffset),
 670     m_DiskStart(0),
 671     m_InternalAttributes(0),
 672     m_ExternalAttributes(0),
 673     m_Extra(NULL),
 674     m_LocalExtra(NULL),
 675     m_zipnotifier(NULL),
 676     m_backlink(NULL)
 677 {
 678     if (!name.empty())
 679         SetName(name);
 680 }
 681
 682 wxZipEntry::~wxZipEntry()
 683 {
 684     if (m_backlink)
 685         m_backlink->Release(m_Key);
 686     Release(m_Extra);
 687     Release(m_LocalExtra);
 688 }
 689
 690 wxZipEntry::wxZipEntry(const wxZipEntry& e)
 691   : wxArchiveEntry(e),
 692     m_SystemMadeBy(e.m_SystemMadeBy),
 693     m_VersionMadeBy(e.m_VersionMadeBy),
 694     m_VersionNeeded(e.m_VersionNeeded),
 695     m_Flags(e.m_Flags),
 696     m_Method(e.m_Method),
 697     m_DateTime(e.m_DateTime),
 698     m_Crc(e.m_Crc),
 699     m_CompressedSize(e.m_CompressedSize),
 700     m_Size(e.m_Size),
 701     m_Name(e.m_Name),
 702     m_Key(e.m_Key),
 703     m_Offset(e.m_Offset),
 704     m_Comment(e.m_Comment),
 705     m_DiskStart(e.m_DiskStart),
 706     m_InternalAttributes(e.m_InternalAttributes),
 707     m_ExternalAttributes(e.m_ExternalAttributes),
 708     m_Extra(AddRef(e.m_Extra)),
 709     m_LocalExtra(AddRef(e.m_LocalExtra)),
 710     m_zipnotifier(NULL),
 711     m_backlink(NULL)
 712 {
 713 }
 714
 715 wxZipEntry& wxZipEntry::operator=(const wxZipEntry& e)
 716 {
 717     if (&e != this) {
 718         m_SystemMadeBy = e.m_SystemMadeBy;
 719         m_VersionMadeBy = e.m_VersionMadeBy;
 720         m_VersionNeeded = e.m_VersionNeeded;
 721         m_Flags = e.m_Flags;
 722         m_Method = e.m_Method;
 723         m_DateTime = e.m_DateTime;
 724         m_Crc = e.m_Crc;
 725         m_CompressedSize = e.m_CompressedSize;
 726         m_Size = e.m_Size;
 727         m_Name = e.m_Name;
 728         m_Key = e.m_Key;
 729         m_Offset = e.m_Offset;
 730         m_Comment = e.m_Comment;
 731         m_DiskStart = e.m_DiskStart;
 732         m_InternalAttributes = e.m_InternalAttributes;
 733         m_ExternalAttributes = e.m_ExternalAttributes;
 734         Copy(m_Extra, e.m_Extra);
 735         Copy(m_LocalExtra, e.m_LocalExtra);
 736         m_zipnotifier = NULL;
 737         if (m_backlink) {
 738             m_backlink->Release(m_Key);
 739             m_backlink = NULL;
 740         }
 741     }
 742     return *this;
 743 }
 744
 745 wxString wxZipEntry::GetName(wxPathFormat format /*=wxPATH_NATIVE*/) const
 746 {
 747     bool isDir = IsDir() && !m_Name.empty();
 748
 749     // optimisations for common (and easy) cases
 750     switch (wxFileName::GetFormat(format)) {
 751         case wxPATH_DOS:
 752         {
 753             wxString name(isDir ? m_Name + _T("\\") : m_Name);
 754             for (size_t i = name.length() - 1; i > 0; --i)
 755                 if (name[i] == _T('/'))
 756                     name[i] = _T('\\');
 757             return name;
 758         }
 759
 760         case wxPATH_UNIX:
 761             return isDir ? m_Name + _T("/") : m_Name;
 762
 763         default:
 764             ;
 765     }
 766
 767     wxFileName fn;
 768
 769     if (isDir)
 770         fn.AssignDir(m_Name, wxPATH_UNIX);
 771     else
 772         fn.Assign(m_Name, wxPATH_UNIX);
 773
 774     return fn.GetFullPath(format);
 775 }
 776
 777 // Static - Internally tars and zips use forward slashes for the path
 778 // separator, absolute paths aren't allowed, and directory names have a
 779 // trailing slash.  This function converts a path into this internal format,
 780 // but without a trailing slash for a directory.
 781 //
 782 wxString wxZipEntry::GetInternalName(const wxString& name,
 783                                      wxPathFormat format /*=wxPATH_NATIVE*/,
 784                                      bool *pIsDir        /*=NULL*/)
 785 {
 786     wxString internal;
 787
 788     if (wxFileName::GetFormat(format) != wxPATH_UNIX)
 789         internal = wxFileName(name, format).GetFullPath(wxPATH_UNIX);
 790     else
 791         internal = name;
 792
 793     bool isDir = !internal.empty() && internal.Last() == '/';
 794     if (pIsDir)
 795         *pIsDir = isDir;
 796     if (isDir)
 797         internal.erase(internal.length() - 1);
 798
 799     while (!internal.empty() && *internal.begin() == '/')
 800         internal.erase(0, 1);
 801     while (!internal.empty() && internal.compare(0, 2, _T("./")) == 0)
 802         internal.erase(0, 2);
 803     if (internal == _T(".") || internal == _T(".."))
 804         internal = wxEmptyString;
 805
 806     return internal;
 807 }
 808
 809 void wxZipEntry::SetSystemMadeBy(int system)
 810 {
 811     int mode = GetMode();
 812     bool wasUnix = IsMadeByUnix();
 813
 814     m_SystemMadeBy = (wxUint8)system;
 815
 816     if (!wasUnix && IsMadeByUnix()) {
 817         SetIsDir(IsDir());
 818         SetMode(mode);
 819     } else if (wasUnix && !IsMadeByUnix()) {
 820         m_ExternalAttributes &= 0xffff;
 821     }
 822 }
 823
 824 void wxZipEntry::SetIsDir(bool isDir /*=true*/)
 825 {
 826     if (isDir)
 827         m_ExternalAttributes |= wxZIP_A_SUBDIR;
 828     else
 829         m_ExternalAttributes &= ~wxZIP_A_SUBDIR;
 830
 831     if (IsMadeByUnix()) {
 832         m_ExternalAttributes &= ~wxZIP_S_IFMT;
 833         if (isDir)
 834             m_ExternalAttributes |= wxZIP_S_IFDIR;
 835         else
 836             m_ExternalAttributes |= wxZIP_S_IFREG;
 837     }
 838 }
 839
 840 // Return unix style permission bits
 841 //
 842 int wxZipEntry::GetMode() const
 843 {
 844     // return unix permissions if present
 845     if (IsMadeByUnix())
 846         return (m_ExternalAttributes >> 16) & 0777;
 847
 848     // otherwise synthesize from the dos attribs
 849     int mode = 0644;
 850     if (m_ExternalAttributes & wxZIP_A_RDONLY)
 851         mode &= ~0200;
 852     if (m_ExternalAttributes & wxZIP_A_SUBDIR)
 853         mode |= 0111;
 854
 855     return mode;
 856 }
 857
 858 // Set unix permissions
 859 //
 860 void wxZipEntry::SetMode(int mode)
 861 {
 862     // Set dos attrib bits to be compatible
 863     if (mode & 0222)
 864         m_ExternalAttributes &= ~wxZIP_A_RDONLY;
 865     else
 866         m_ExternalAttributes |= wxZIP_A_RDONLY;
 867
 868     // set the actual unix permission bits if the system type allows
 869     if (IsMadeByUnix()) {
 870         m_ExternalAttributes &= ~(0777L << 16);
 871         m_ExternalAttributes |= (mode & 0777L) << 16;
 872     }
 873 }
 874
 875 const char *wxZipEntry::GetExtra() const
 876 {
 877     return m_Extra ? m_Extra->GetData() : NULL;
 878 }
 879
 880 size_t wxZipEntry::GetExtraLen() const
 881 {
 882     return m_Extra ? m_Extra->GetSize() : 0;
 883 }
 884
 885 void wxZipEntry::SetExtra(const char *extra, size_t len)
 886 {
 887     Unique(m_Extra, len);
 888     if (len)
 889         memcpy(m_Extra->GetData(), extra, len);
 890 }
 891
 892 const char *wxZipEntry::GetLocalExtra() const
 893 {
 894     return m_LocalExtra ? m_LocalExtra->GetData() : NULL;
 895 }
 896
 897 size_t  wxZipEntry::GetLocalExtraLen() const
 898 {
 899     return m_LocalExtra ? m_LocalExtra->GetSize() : 0;
 900 }
 901
 902 void wxZipEntry::SetLocalExtra(const char *extra, size_t len)
 903 {
 904     Unique(m_LocalExtra, len);
 905     if (len)
 906         memcpy(m_LocalExtra->GetData(), extra, len);
 907 }
 908
 909 void wxZipEntry::SetNotifier(wxZipNotifier& notifier)
 910 {
 911     wxArchiveEntry::UnsetNotifier();
 912     m_zipnotifier = &notifier;
 913     m_zipnotifier->OnEntryUpdated(*this);
 914 }
 915
 916 void wxZipEntry::Notify()
 917 {
 918     if (m_zipnotifier)
 919         m_zipnotifier->OnEntryUpdated(*this);
 920     else if (GetNotifier())
 921         GetNotifier()->OnEntryUpdated(*this);
 922 }
 923
 924 void wxZipEntry::UnsetNotifier()
 925 {
 926     wxArchiveEntry::UnsetNotifier();
 927     m_zipnotifier = NULL;
 928 }
 929
 930 size_t wxZipEntry::ReadLocal(wxInputStream& stream, wxMBConv& conv)
 931 {
 932     wxUint16 nameLen, extraLen;
 933     wxUint32 compressedSize, size, crc;
 934
 935     wxZipHeader ds(stream, LOCAL_SIZE - 4);
 936     if (!ds)
 937         return 0;
 938
 939     ds >> m_VersionNeeded >> m_Flags >> m_Method;
 940     SetDateTime(wxDateTime().SetFromDOS(ds.Read32()));
 941     ds >> crc >> compressedSize >> size >> nameLen >> extraLen;
 942
 943     bool sumsValid = (m_Flags & wxZIP_SUMS_FOLLOW) == 0;
 944
 945     if (sumsValid || crc)
 946         m_Crc = crc;
 947     if ((sumsValid || compressedSize) || m_Method == wxZIP_METHOD_STORE)
 948         m_CompressedSize = compressedSize;
 949     if ((sumsValid || size) || m_Method == wxZIP_METHOD_STORE)
 950         m_Size = size;
 951
 952     SetName(ReadString(stream, nameLen, conv), wxPATH_UNIX);
 953     if (stream.LastRead() != nameLen + 0u)
 954         return 0;
 955
 956     if (extraLen || GetLocalExtraLen()) {
 957         Unique(m_LocalExtra, extraLen);
 958         if (extraLen) {
 959             stream.Read(m_LocalExtra->GetData(), extraLen);
 960             if (stream.LastRead() != extraLen + 0u)
 961                 return 0;
 962         }
 963     }
 964
 965     return LOCAL_SIZE + nameLen + extraLen;
 966 }
 967
 968 size_t wxZipEntry::WriteLocal(wxOutputStream& stream, wxMBConv& conv) const
 969 {
 970     wxString unixName = GetName(wxPATH_UNIX);
 971     const wxWX2MBbuf name_buf = conv.cWX2MB(unixName);
 972     const char *name = name_buf;
 973     if (!name) name = "";
 974     wxUint16 nameLen = wx_truncate_cast(wxUint16, strlen(name));
 975
 976     wxDataOutputStream ds(stream);
 977
 978     ds << m_VersionNeeded << m_Flags << m_Method;
 979     ds.Write32(GetDateTime().GetAsDOS());
 980
 981     ds.Write32(m_Crc);
 982     ds.Write32(m_CompressedSize != wxInvalidOffset ?
 983                wx_truncate_cast(wxUint32, m_CompressedSize) : 0);
 984     ds.Write32(m_Size != wxInvalidOffset ?
 985                wx_truncate_cast(wxUint32, m_Size) : 0);
 986
 987     ds << nameLen;
 988     wxUint16 extraLen = wx_truncate_cast(wxUint16, GetLocalExtraLen());
 989     ds.Write16(extraLen);
 990
 991     stream.Write(name, nameLen);
 992     if (extraLen)
 993         stream.Write(m_LocalExtra->GetData(), extraLen);
 994
 995     return LOCAL_SIZE + nameLen + extraLen;
 996 }
 997
 998 size_t wxZipEntry::ReadCentral(wxInputStream& stream, wxMBConv& conv)
 999 {
1000     wxUint16 nameLen, extraLen, commentLen;
1001
1002     wxZipHeader ds(stream, CENTRAL_SIZE - 4);
1003     if (!ds)
1004         return 0;
1005
1006     ds >> m_VersionMadeBy >> m_SystemMadeBy;
1007
1008     SetVersionNeeded(ds.Read16());
1009     SetFlags(ds.Read16());
1010     SetMethod(ds.Read16());
1011     SetDateTime(wxDateTime().SetFromDOS(ds.Read32()));
1012     SetCrc(ds.Read32());
1013     SetCompressedSize(ds.Read32());
1014     SetSize(ds.Read32());
1015
1016     ds >> nameLen >> extraLen >> commentLen
1017        >> m_DiskStart >> m_InternalAttributes >> m_ExternalAttributes;
1018     SetOffset(ds.Read32());
1019
1020     SetName(ReadString(stream, nameLen, conv), wxPATH_UNIX);
1021     if (stream.LastRead() != nameLen + 0u)
1022         return 0;
1023
1024     if (extraLen || GetExtraLen()) {
1025         Unique(m_Extra, extraLen);
1026         if (extraLen) {
1027             stream.Read(m_Extra->GetData(), extraLen);
1028             if (stream.LastRead() != extraLen + 0u)
1029                 return 0;
1030         }
1031     }
1032
1033     if (commentLen) {
1034         m_Comment = ReadString(stream, commentLen, conv);
1035         if (stream.LastRead() != commentLen + 0u)
1036             return 0;
1037     } else {
1038         m_Comment.clear();
1039     }
1040
1041     return CENTRAL_SIZE + nameLen + extraLen + commentLen;
1042 }
1043
1044 size_t wxZipEntry::WriteCentral(wxOutputStream& stream, wxMBConv& conv) const
1045 {
1046     wxString unixName = GetName(wxPATH_UNIX);
1047     const wxWX2MBbuf name_buf = conv.cWX2MB(unixName);
1048     const char *name = name_buf;
1049     if (!name) name = "";
1050     wxUint16 nameLen = wx_truncate_cast(wxUint16, strlen(name));
1051
1052     const wxWX2MBbuf comment_buf = conv.cWX2MB(m_Comment);
1053     const char *comment = comment_buf;
1054     if (!comment) comment = "";
1055     wxUint16 commentLen = wx_truncate_cast(wxUint16, strlen(comment));
1056
1057     wxUint16 extraLen = wx_truncate_cast(wxUint16, GetExtraLen());
1058
1059     wxDataOutputStream ds(stream);
1060
1061     ds << CENTRAL_MAGIC << m_VersionMadeBy << m_SystemMadeBy;
1062
1063     ds.Write16(wx_truncate_cast(wxUint16, GetVersionNeeded()));
1064     ds.Write16(wx_truncate_cast(wxUint16, GetFlags()));
1065     ds.Write16(wx_truncate_cast(wxUint16, GetMethod()));
1066     ds.Write32(GetDateTime().GetAsDOS());
1067     ds.Write32(GetCrc());
1068     ds.Write32(wx_truncate_cast(wxUint32, GetCompressedSize()));
1069     ds.Write32(wx_truncate_cast(wxUint32, GetSize()));
1070     ds.Write16(nameLen);
1071     ds.Write16(extraLen);
1072
1073     ds << commentLen << m_DiskStart << m_InternalAttributes
1074        << m_ExternalAttributes << wx_truncate_cast(wxUint32, GetOffset());
1075
1076     stream.Write(name, nameLen);
1077     if (extraLen)
1078         stream.Write(GetExtra(), extraLen);
1079     stream.Write(comment, commentLen);
1080
1081     return CENTRAL_SIZE + nameLen + extraLen + commentLen;
1082 }
1083
1084 // Info-zip prefixes this record with a signature, but pkzip doesn't. So if
1085 // the 1st value is the signature then it is probably an info-zip record,
1086 // though there is a small chance that it is in fact a pkzip record which
1087 // happens to have the signature as it's CRC.
1088 //
1089 size_t wxZipEntry::ReadDescriptor(wxInputStream& stream)
1090 {
1091     wxZipHeader ds(stream, SUMS_SIZE);
1092     if (!ds)
1093         return 0;
1094
1095     m_Crc = ds.Read32();
1096     m_CompressedSize = ds.Read32();
1097     m_Size = ds.Read32();
1098
1099     // if 1st value is the signature then this is probably an info-zip record
1100     if (m_Crc == SUMS_MAGIC)
1101     {
1102         wxZipHeader buf(stream, 8);
1103         wxUint32 u1 = buf.GetSize() >= 4 ? buf.Read32() : LOCAL_MAGIC;
1104         wxUint32 u2 = buf.GetSize() == 8 ? buf.Read32() : 0;
1105
1106         // look for the signature of the following record to decide which
1107         if ((u1 == LOCAL_MAGIC || u1 == CENTRAL_MAGIC) &&
1108             (u2 != LOCAL_MAGIC && u2 != CENTRAL_MAGIC))
1109         {
1110             // it's a pkzip style record after all!
1111             if (buf.GetSize() > 0)
1112                 stream.Ungetch(buf.GetData(), buf.GetSize());
1113         }
1114         else
1115         {
1116             // it's an info-zip record as expected
1117             if (buf.GetSize() > 4)
1118                 stream.Ungetch(buf.GetData() + 4, buf.GetSize() - 4);
1119             m_Crc = wx_truncate_cast(wxUint32, m_CompressedSize);
1120             m_CompressedSize = m_Size;
1121             m_Size = u1;
1122             return SUMS_SIZE + 4;
1123         }
1124     }
1125
1126     return SUMS_SIZE;
1127 }
1128
1129 size_t wxZipEntry::WriteDescriptor(wxOutputStream& stream, wxUint32 crc,
1130                                    wxFileOffset compressedSize, wxFileOffset size)
1131 {
1132     m_Crc = crc;
1133     m_CompressedSize = compressedSize;
1134     m_Size = size;
1135
1136     wxDataOutputStream ds(stream);
1137
1138     ds.Write32(crc);
1139     ds.Write32(wx_truncate_cast(wxUint32, compressedSize));
1140     ds.Write32(wx_truncate_cast(wxUint32, size));
1141
1142     return SUMS_SIZE;
1143 }
1144
1145
1146 /////////////////////////////////////////////////////////////////////////////
1147 // wxZipEndRec - holds the end of central directory record
1148
1149 class wxZipEndRec
1150 {
1151 public:
1152     wxZipEndRec();
1153
1154     int GetDiskNumber() const                   { return m_DiskNumber; }
1155     int GetStartDisk() const                    { return m_StartDisk; }
1156     int GetEntriesHere() const                  { return m_EntriesHere; }
1157     int GetTotalEntries() const                 { return m_TotalEntries; }
1158     wxFileOffset GetSize() const                { return m_Size; }
1159     wxFileOffset GetOffset() const              { return m_Offset; }
1160     wxString GetComment() const                 { return m_Comment; }
1161
1162     void SetDiskNumber(int num)
1163         { m_DiskNumber = wx_truncate_cast(wxUint16, num); }
1164     void SetStartDisk(int num)
1165         { m_StartDisk = wx_truncate_cast(wxUint16, num); }
1166     void SetEntriesHere(int num)
1167         { m_EntriesHere = wx_truncate_cast(wxUint16, num); }
1168     void SetTotalEntries(int num)
1169         { m_TotalEntries = wx_truncate_cast(wxUint16, num); }
1170     void SetSize(wxFileOffset size)
1171         { m_Size = wx_truncate_cast(wxUint32, size); }
1172     void SetOffset(wxFileOffset offset)
1173         { m_Offset = wx_truncate_cast(wxUint32, offset); }
1174     void SetComment(const wxString& comment)
1175         { m_Comment = comment; }
1176
1177     bool Read(wxInputStream& stream, wxMBConv& conv);
1178     bool Write(wxOutputStream& stream, wxMBConv& conv) const;
1179
1180 private:
1181     wxUint16 m_DiskNumber;
1182     wxUint16 m_StartDisk;
1183     wxUint16 m_EntriesHere;
1184     wxUint16 m_TotalEntries;
1185     wxUint32 m_Size;
1186     wxUint32 m_Offset;
1187     wxString m_Comment;
1188 };
1189
1190 wxZipEndRec::wxZipEndRec()
1191   : m_DiskNumber(0),
1192     m_StartDisk(0),
1193     m_EntriesHere(0),
1194     m_TotalEntries(0),
1195     m_Size(0),
1196     m_Offset(0)
1197 {
1198 }
1199
1200 bool wxZipEndRec::Write(wxOutputStream& stream, wxMBConv& conv) const
1201 {
1202     const wxWX2MBbuf comment_buf = conv.cWX2MB(m_Comment);
1203     const char *comment = comment_buf;
1204     if (!comment) comment = "";
1205     wxUint16 commentLen = (wxUint16)strlen(comment);
1206
1207     wxDataOutputStream ds(stream);
1208
1209     ds << END_MAGIC << m_DiskNumber << m_StartDisk << m_EntriesHere
1210        << m_TotalEntries << m_Size << m_Offset << commentLen;
1211
1212     stream.Write(comment, commentLen);
1213
1214     return stream.IsOk();
1215 }
1216
1217 bool wxZipEndRec::Read(wxInputStream& stream, wxMBConv& conv)
1218 {
1219     wxZipHeader ds(stream, END_SIZE - 4);
1220     if (!ds)
1221         return false;
1222
1223     wxUint16 commentLen;
1224
1225     ds >> m_DiskNumber >> m_StartDisk >> m_EntriesHere
1226        >> m_TotalEntries >> m_Size >> m_Offset >> commentLen;
1227
1228     if (commentLen) {
1229         m_Comment = ReadString(stream, commentLen, conv);
1230         if (stream.LastRead() != commentLen + 0u)
1231             return false;
1232     }
1233
1234     if (m_DiskNumber != 0 || m_StartDisk != 0 ||
1235             m_EntriesHere != m_TotalEntries)
1236         wxLogWarning(_("assuming this is a multi-part zip concatenated"));
1237
1238     return true;
1239 }
1240
1241
1242 /////////////////////////////////////////////////////////////////////////////
1243 // A weak link from an input stream to an output stream
1244
1245 class wxZipStreamLink
1246 {
1247 public:
1248     wxZipStreamLink(wxZipOutputStream *stream) : m_ref(1), m_stream(stream) { }
1249
1250     wxZipStreamLink *AddRef() { m_ref++; return this; }
1251     wxZipOutputStream *GetOutputStream() const { return m_stream; }
1252
1253     void Release(class wxZipInputStream *WXUNUSED(s))
1254         { if (--m_ref == 0) delete this; }
1255     void Release(class wxZipOutputStream *WXUNUSED(s))
1256         { m_stream = NULL; if (--m_ref == 0) delete this; }
1257
1258 private:
1259     ~wxZipStreamLink() { }
1260
1261     int m_ref;
1262     wxZipOutputStream *m_stream;
1263
1264     wxSUPPRESS_GCC_PRIVATE_DTOR_WARNING(wxZipStreamLink)
1265 };
1266
1267
1268 /////////////////////////////////////////////////////////////////////////////
1269 // Input stream
1270
1271 // leave the default wxZipEntryPtr free for users
1272 wxDECLARE_SCOPED_PTR(wxZipEntry, wx__ZipEntryPtr)
1273 wxDEFINE_SCOPED_PTR (wxZipEntry, wx__ZipEntryPtr)
1274
1275 // constructor
1276 //
1277 wxZipInputStream::wxZipInputStream(wxInputStream& stream,
1278                                    wxMBConv& conv /*=wxConvLocal*/)
1279   : wxArchiveInputStream(stream, conv)
1280 {
1281     Init();
1282 }
1283
1284 #if 1 //WXWIN_COMPATIBILITY_2_6
1285
1286 // Part of the compatibility constructor, which has been made inline to
1287 // avoid a problem with it not being exported by mingw 3.2.3
1288 //
1289 void wxZipInputStream::Init(const wxString& file)
1290 {
1291     // no error messages
1292     wxLogNull nolog;
1293     Init();
1294     m_allowSeeking = true;
1295     m_ffile = wx_static_cast(wxFFileInputStream*, m_parent_i_stream);
1296     wx__ZipEntryPtr entry;
1297
1298     if (m_ffile->Ok()) {
1299         do {
1300             entry.reset(GetNextEntry());
1301         }
1302         while (entry.get() != NULL && entry->GetInternalName() != file);
1303     }
1304
1305     if (entry.get() == NULL)
1306         m_lasterror = wxSTREAM_READ_ERROR;
1307 }
1308
1309 wxInputStream& wxZipInputStream::OpenFile(const wxString& archive)
1310 {
1311     wxLogNull nolog;
1312     return *new wxFFileInputStream(archive);
1313 }
1314
1315 #endif // WXWIN_COMPATIBILITY_2_6
1316
1317 void wxZipInputStream::Init()
1318 {
1319     m_store = new wxStoredInputStream(*m_parent_i_stream);
1320     m_inflate = NULL;
1321     m_rawin = NULL;
1322     m_raw = false;
1323     m_headerSize = 0;
1324     m_decomp = NULL;
1325     m_parentSeekable = false;
1326     m_weaklinks = new wxZipWeakLinks;
1327     m_streamlink = NULL;
1328     m_offsetAdjustment = 0;
1329     m_position = wxInvalidOffset;
1330     m_signature = 0;
1331     m_TotalEntries = 0;
1332     m_lasterror = m_parent_i_stream->GetLastError();
1333     m_ffile = NULL;
1334 #if 1 //WXWIN_COMPATIBILITY_2_6
1335     m_allowSeeking = false;
1336 #endif
1337 }
1338
1339 wxZipInputStream::~wxZipInputStream()
1340 {
1341     CloseDecompressor(m_decomp);
1342
1343     delete m_store;
1344     delete m_inflate;
1345     delete m_rawin;
1346     delete m_ffile;
1347
1348     m_weaklinks->Release(this);
1349
1350     if (m_streamlink)
1351         m_streamlink->Release(this);
1352 }
1353
1354 wxString wxZipInputStream::GetComment()
1355 {
1356     if (m_position == wxInvalidOffset)
1357         if (!LoadEndRecord())
1358             return wxEmptyString;
1359
1360     if (!m_parentSeekable && Eof() && m_signature) {
1361         m_lasterror = wxSTREAM_NO_ERROR;
1362         m_lasterror = ReadLocal(true);
1363     }
1364
1365     return m_Comment;
1366 }
1367
1368 int wxZipInputStream::GetTotalEntries()
1369 {
1370     if (m_position == wxInvalidOffset)
1371         LoadEndRecord();
1372     return m_TotalEntries;
1373 }
1374
1375 wxZipStreamLink *wxZipInputStream::MakeLink(wxZipOutputStream *out)
1376 {
1377     wxZipStreamLink *link = NULL;
1378
1379     if (!m_parentSeekable && (IsOpened() || !Eof())) {
1380         link = new wxZipStreamLink(out);
1381         if (m_streamlink)
1382             m_streamlink->Release(this);
1383         m_streamlink = link->AddRef();
1384     }
1385
1386     return link;
1387 }
1388
1389 bool wxZipInputStream::LoadEndRecord()
1390 {
1391     wxCHECK(m_position == wxInvalidOffset, false);
1392     if (!IsOk())
1393         return false;
1394
1395     m_position = 0;
1396
1397     // First find the end-of-central-directory record.
1398     if (!FindEndRecord()) {
1399         // failed, so either this is a non-seekable stream (ok), or not a zip
1400         if (m_parentSeekable) {
1401             m_lasterror = wxSTREAM_READ_ERROR;
1402             wxLogError(_("invalid zip file"));
1403             return false;
1404         }
1405         else {
1406             wxLogNull nolog;
1407             wxFileOffset pos = m_parent_i_stream->TellI();
1408             // FIXME
1409             //if (pos != wxInvalidOffset)
1410             if (pos >= 0 && pos <= LONG_MAX)
1411                 m_offsetAdjustment = m_position = pos;
1412             return true;
1413         }
1414     }
1415
1416     wxZipEndRec endrec;
1417
1418     // Read in the end record
1419     wxFileOffset endPos = m_parent_i_stream->TellI() - 4;
1420     if (!endrec.Read(*m_parent_i_stream, GetConv()))
1421         return false;
1422
1423     m_TotalEntries = endrec.GetTotalEntries();
1424     m_Comment = endrec.GetComment();
1425
1426     // Now find the central-directory. we have the file offset of
1427     // the CD, so look there first.
1428     if (m_parent_i_stream->SeekI(endrec.GetOffset()) != wxInvalidOffset &&
1429             ReadSignature() == CENTRAL_MAGIC) {
1430         m_signature = CENTRAL_MAGIC;
1431         m_position = endrec.GetOffset();
1432         m_offsetAdjustment = 0;
1433         return true;
1434     }
1435
1436     // If it's not there, then it could be that the zip has been appended
1437     // to a self extractor, so take the CD size (also in endrec), subtract
1438     // it from the file offset of the end-central-directory and look there.
1439     if (m_parent_i_stream->SeekI(endPos - endrec.GetSize())
1440             != wxInvalidOffset && ReadSignature() == CENTRAL_MAGIC) {
1441         m_signature = CENTRAL_MAGIC;
1442         m_position = endPos - endrec.GetSize();
1443         m_offsetAdjustment = m_position - endrec.GetOffset();
1444         return true;
1445     }
1446
1447     wxLogError(_("can't find central directory in zip"));
1448     m_lasterror = wxSTREAM_READ_ERROR;
1449     return false;
1450 }
1451
1452 // Find the end-of-central-directory record.
1453 // If found the stream will be positioned just past the 4 signature bytes.
1454 //
1455 bool wxZipInputStream::FindEndRecord()
1456 {
1457     if (!m_parent_i_stream->IsSeekable())
1458         return false;
1459
1460     // usually it's 22 bytes in size and the last thing in the file
1461     {
1462         wxLogNull nolog;
1463         if (m_parent_i_stream->SeekI(-END_SIZE, wxFromEnd) == wxInvalidOffset)
1464             return false;
1465     }
1466
1467     m_parentSeekable = true;
1468     m_signature = 0;
1469     char magic[4];
1470     if (m_parent_i_stream->Read(magic, 4).LastRead() != 4)
1471         return false;
1472     if ((m_signature = CrackUint32(magic)) == END_MAGIC)
1473         return true;
1474
1475     // unfortunately, the record has a comment field that can be up to 65535
1476     // bytes in length, so if the signature not found then search backwards.
1477     wxFileOffset pos = m_parent_i_stream->TellI();
1478     const int BUFSIZE = 1024;
1479     wxCharBuffer buf(BUFSIZE);
1480
1481     memcpy(buf.data(), magic, 3);
1482     wxFileOffset minpos = wxMax(pos - 65535L, 0);
1483
1484     while (pos > minpos) {
1485         size_t len = wx_truncate_cast(size_t,
1486                         pos - wxMax(pos - (BUFSIZE - 3), minpos));
1487         memcpy(buf.data() + len, buf, 3);
1488         pos -= len;
1489
1490         if (m_parent_i_stream->SeekI(pos, wxFromStart) == wxInvalidOffset ||
1491                 m_parent_i_stream->Read(buf.data(), len).LastRead() != len)
1492             return false;
1493
1494         char *p = buf.data() + len;
1495
1496         while (p-- > buf.data()) {
1497             if ((m_signature = CrackUint32(p)) == END_MAGIC) {
1498                 size_t remainder = buf.data() + len - p;
1499                 if (remainder > 4)
1500                     m_parent_i_stream->Ungetch(p + 4, remainder - 4);
1501                 return true;
1502             }
1503         }
1504     }
1505
1506     return false;
1507 }
1508
1509 wxZipEntry *wxZipInputStream::GetNextEntry()
1510 {
1511     if (m_position == wxInvalidOffset)
1512         if (!LoadEndRecord())
1513             return NULL;
1514
1515     m_lasterror = m_parentSeekable ? ReadCentral() : ReadLocal();
1516     if (!IsOk())
1517         return NULL;
1518
1519     wx__ZipEntryPtr entry(new wxZipEntry(m_entry));
1520     entry->m_backlink = m_weaklinks->AddEntry(entry.get(), entry->GetKey());
1521     return entry.release();
1522 }
1523
1524 wxStreamError wxZipInputStream::ReadCentral()
1525 {
1526     if (!AtHeader())
1527         CloseEntry();
1528
1529     if (m_signature == END_MAGIC)
1530         return wxSTREAM_EOF;
1531
1532     if (m_signature != CENTRAL_MAGIC) {
1533         wxLogError(_("error reading zip central directory"));
1534         return wxSTREAM_READ_ERROR;
1535     }
1536
1537     if (QuietSeek(*m_parent_i_stream, m_position + 4) == wxInvalidOffset)
1538         return wxSTREAM_READ_ERROR;
1539
1540     size_t size = m_entry.ReadCentral(*m_parent_i_stream, GetConv());
1541     if (!size) {
1542         m_signature = 0;
1543         return wxSTREAM_READ_ERROR;
1544     }
1545
1546     m_position += size;
1547     m_signature = ReadSignature();
1548
1549     if (m_offsetAdjustment)
1550         m_entry.SetOffset(m_entry.GetOffset() + m_offsetAdjustment);
1551     m_entry.SetKey(m_entry.GetOffset());
1552
1553     return wxSTREAM_NO_ERROR;
1554 }
1555
1556 wxStreamError wxZipInputStream::ReadLocal(bool readEndRec /*=false*/)
1557 {
1558     if (!AtHeader())
1559         CloseEntry();
1560
1561     if (!m_signature)
1562         m_signature = ReadSignature();
1563
1564     if (m_signature == CENTRAL_MAGIC || m_signature == END_MAGIC) {
1565         if (m_streamlink && !m_streamlink->GetOutputStream()) {
1566             m_streamlink->Release(this);
1567             m_streamlink = NULL;
1568         }
1569     }
1570
1571     while (m_signature == CENTRAL_MAGIC) {
1572         if (m_weaklinks->IsEmpty() && m_streamlink == NULL)
1573             return wxSTREAM_EOF;
1574
1575         size_t size = m_entry.ReadCentral(*m_parent_i_stream, GetConv());
1576         m_position += size;
1577         m_signature = 0;
1578         if (!size)
1579             return wxSTREAM_READ_ERROR;
1580
1581         wxZipEntry *entry = m_weaklinks->GetEntry(m_entry.GetOffset());
1582         if (entry) {
1583             entry->SetSystemMadeBy(m_entry.GetSystemMadeBy());
1584             entry->SetVersionMadeBy(m_entry.GetVersionMadeBy());
1585             entry->SetComment(m_entry.GetComment());
1586             entry->SetDiskStart(m_entry.GetDiskStart());
1587             entry->SetInternalAttributes(m_entry.GetInternalAttributes());
1588             entry->SetExternalAttributes(m_entry.GetExternalAttributes());
1589             Copy(entry->m_Extra, m_entry.m_Extra);
1590             entry->Notify();
1591             m_weaklinks->RemoveEntry(entry->GetOffset());
1592         }
1593
1594         m_signature = ReadSignature();
1595     }
1596
1597     if (m_signature == END_MAGIC) {
1598         if (readEndRec || m_streamlink) {
1599             wxZipEndRec endrec;
1600             endrec.Read(*m_parent_i_stream, GetConv());
1601             m_Comment = endrec.GetComment();
1602             m_signature = 0;
1603             if (m_streamlink) {
1604                 m_streamlink->GetOutputStream()->SetComment(endrec.GetComment());
1605                 m_streamlink->Release(this);
1606                 m_streamlink = NULL;
1607             }
1608         }
1609         return wxSTREAM_EOF;
1610     }
1611
1612     if (m_signature != LOCAL_MAGIC) {
1613         wxLogError(_("error reading zip local header"));
1614         return wxSTREAM_READ_ERROR;
1615     }
1616
1617     m_headerSize = m_entry.ReadLocal(*m_parent_i_stream, GetConv());
1618     m_signature = 0;
1619     m_entry.SetOffset(m_position);
1620     m_entry.SetKey(m_position);
1621
1622     if (!m_headerSize) {
1623         return wxSTREAM_READ_ERROR;
1624     } else {
1625         m_TotalEntries++;
1626         return wxSTREAM_NO_ERROR;
1627     }
1628 }
1629
1630 wxUint32 wxZipInputStream::ReadSignature()
1631 {
1632     char magic[4];
1633     m_parent_i_stream->Read(magic, 4);
1634     return m_parent_i_stream->LastRead() == 4 ? CrackUint32(magic) : 0;
1635 }
1636
1637 bool wxZipInputStream::OpenEntry(wxArchiveEntry& entry)
1638 {
1639     wxZipEntry *zipEntry = wxStaticCast(&entry, wxZipEntry);
1640     return zipEntry ? OpenEntry(*zipEntry) : false;
1641 }
1642
1643 // Open an entry
1644 //
1645 bool wxZipInputStream::DoOpen(wxZipEntry *entry, bool raw)
1646 {
1647     if (m_position == wxInvalidOffset)
1648         if (!LoadEndRecord())
1649             return false;
1650     if (m_lasterror == wxSTREAM_READ_ERROR)
1651         return false;
1652     if (IsOpened())
1653         CloseEntry();
1654
1655     m_raw = raw;
1656
1657     if (entry) {
1658         if (AfterHeader() && entry->GetKey() == m_entry.GetOffset())
1659             return true;
1660         // can only open the current entry on a non-seekable stream
1661         wxCHECK(m_parentSeekable, false);
1662     }
1663
1664     m_lasterror = wxSTREAM_READ_ERROR;
1665
1666     if (entry)
1667         m_entry = *entry;
1668
1669     if (m_parentSeekable) {
1670         if (QuietSeek(*m_parent_i_stream, m_entry.GetOffset())
1671                 == wxInvalidOffset)
1672             return false;
1673         if (ReadSignature() != LOCAL_MAGIC) {
1674             wxLogError(_("bad zipfile offset to entry"));
1675             return false;
1676         }
1677     }
1678
1679     if (m_parentSeekable || AtHeader()) {
1680         m_headerSize = m_entry.ReadLocal(*m_parent_i_stream, GetConv());
1681         if (m_headerSize && m_parentSeekable) {
1682             wxZipEntry *ref = m_weaklinks->GetEntry(m_entry.GetKey());
1683             if (ref) {
1684                 Copy(ref->m_LocalExtra, m_entry.m_LocalExtra);
1685                 ref->Notify();
1686                 m_weaklinks->RemoveEntry(ref->GetKey());
1687             }
1688             if (entry && entry != ref) {
1689                 Copy(entry->m_LocalExtra, m_entry.m_LocalExtra);
1690                 entry->Notify();
1691             }
1692         }
1693     }
1694
1695     if (m_headerSize)
1696         m_lasterror = wxSTREAM_NO_ERROR;
1697     return IsOk();
1698 }
1699
1700 bool wxZipInputStream::OpenDecompressor(bool raw /*=false*/)
1701 {
1702     wxASSERT(AfterHeader());
1703
1704     wxFileOffset compressedSize = m_entry.GetCompressedSize();
1705
1706     if (raw)
1707         m_raw = true;
1708
1709     if (m_raw) {
1710         if (compressedSize != wxInvalidOffset) {
1711             m_store->Open(compressedSize);
1712             m_decomp = m_store;
1713         } else {
1714             if (!m_rawin)
1715                 m_rawin = new wxRawInputStream(*m_parent_i_stream);
1716             m_decomp = m_rawin->Open(OpenDecompressor(m_rawin->GetTee()));
1717         }
1718     } else {
1719         if (compressedSize != wxInvalidOffset &&
1720                 (m_entry.GetMethod() != wxZIP_METHOD_DEFLATE ||
1721                  wxZlibInputStream::CanHandleGZip())) {
1722             m_store->Open(compressedSize);
1723             m_decomp = OpenDecompressor(*m_store);
1724         } else {
1725             m_decomp = OpenDecompressor(*m_parent_i_stream);
1726         }
1727     }
1728
1729     m_crcAccumulator = crc32(0, Z_NULL, 0);
1730     m_lasterror = m_decomp ? m_decomp->GetLastError() : wxSTREAM_READ_ERROR;
1731     return IsOk();
1732 }
1733
1734 // Can be overriden to add support for additional decompression methods
1735 //
1736 wxInputStream *wxZipInputStream::OpenDecompressor(wxInputStream& stream)
1737 {
1738     switch (m_entry.GetMethod()) {
1739         case wxZIP_METHOD_STORE:
1740             if (m_entry.GetSize() == wxInvalidOffset) {
1741                 wxLogError(_("stored file length not in Zip header"));
1742                 break;
1743             }
1744             m_store->Open(m_entry.GetSize());
1745             return m_store;
1746
1747         case wxZIP_METHOD_DEFLATE:
1748             if (!m_inflate)
1749                 m_inflate = new wxZlibInputStream2(stream);
1750             else
1751                 m_inflate->Open(stream);
1752             return m_inflate;
1753
1754         default:
1755             wxLogError(_("unsupported Zip compression method"));
1756     }
1757
1758     return NULL;
1759 }
1760
1761 bool wxZipInputStream::CloseDecompressor(wxInputStream *decomp)
1762 {
1763     if (decomp && decomp == m_rawin)
1764         return CloseDecompressor(m_rawin->GetFilterInputStream());
1765     if (decomp != m_store && decomp != m_inflate)
1766         delete decomp;
1767     return true;
1768 }
1769
1770 // Closes the current entry and positions the underlying stream at the start
1771 // of the next entry
1772 //
1773 bool wxZipInputStream::CloseEntry()
1774 {
1775     if (AtHeader())
1776         return true;
1777     if (m_lasterror == wxSTREAM_READ_ERROR)
1778         return false;
1779
1780     if (!m_parentSeekable) {
1781         if (!IsOpened() && !OpenDecompressor(true))
1782             return false;
1783
1784         const int BUFSIZE = 8192;
1785         wxCharBuffer buf(BUFSIZE);
1786         while (IsOk())
1787             Read(buf.data(), BUFSIZE);
1788
1789         m_position += m_headerSize + m_entry.GetCompressedSize();
1790     }
1791
1792     if (m_lasterror == wxSTREAM_EOF)
1793         m_lasterror = wxSTREAM_NO_ERROR;
1794
1795     CloseDecompressor(m_decomp);
1796     m_decomp = NULL;
1797     m_entry = wxZipEntry();
1798     m_headerSize = 0;
1799     m_raw = false;
1800
1801     return IsOk();
1802 }
1803
1804 size_t wxZipInputStream::OnSysRead(void *buffer, size_t size)
1805 {
1806     if (!IsOpened())
1807         if ((AtHeader() && !DoOpen()) || !OpenDecompressor())
1808             m_lasterror = wxSTREAM_READ_ERROR;
1809     if (!IsOk() || !size)
1810         return 0;
1811
1812     size_t count = m_decomp->Read(buffer, size).LastRead();
1813     if (!m_raw)
1814         m_crcAccumulator = crc32(m_crcAccumulator, (Byte*)buffer, count);
1815     if (count < size)
1816         m_lasterror = m_decomp->GetLastError();
1817
1818     if (Eof()) {
1819         if ((m_entry.GetFlags() & wxZIP_SUMS_FOLLOW) != 0) {
1820             m_headerSize += m_entry.ReadDescriptor(*m_parent_i_stream);
1821             wxZipEntry *entry = m_weaklinks->GetEntry(m_entry.GetKey());
1822
1823             if (entry) {
1824                 entry->SetCrc(m_entry.GetCrc());
1825                 entry->SetCompressedSize(m_entry.GetCompressedSize());
1826                 entry->SetSize(m_entry.GetSize());
1827                 entry->Notify();
1828             }
1829         }
1830
1831         if (!m_raw) {
1832             m_lasterror = wxSTREAM_READ_ERROR;
1833
1834             if (m_entry.GetSize() != TellI())
1835                 wxLogError(_("reading zip stream (entry %s): bad length"),
1836                            m_entry.GetName().c_str());
1837             else if (m_crcAccumulator != m_entry.GetCrc())
1838                 wxLogError(_("reading zip stream (entry %s): bad crc"),
1839                            m_entry.GetName().c_str());
1840             else
1841                 m_lasterror = wxSTREAM_EOF;
1842         }
1843     }
1844
1845     return count;
1846 }
1847
1848 #if 1 //WXWIN_COMPATIBILITY_2_6
1849
1850 // Borrowed from VS's zip stream (c) 1999 Vaclav Slavik
1851 //
1852 wxFileOffset wxZipInputStream::OnSysSeek(wxFileOffset seek, wxSeekMode mode)
1853 {
1854     // seeking works when the stream is created with the compatibility
1855     // constructor
1856     if (!m_allowSeeking)
1857         return wxInvalidOffset;
1858     if (!IsOpened())
1859         if ((AtHeader() && !DoOpen()) || !OpenDecompressor())
1860             m_lasterror = wxSTREAM_READ_ERROR;
1861     if (!IsOk())
1862         return wxInvalidOffset;
1863
1864     // NB: since ZIP files don't natively support seeking, we have to
1865     //     implement a brute force workaround -- reading all the data
1866     //     between current and the new position (or between beginning of
1867     //     the file and new position...)
1868
1869     wxFileOffset nextpos;
1870     wxFileOffset pos = TellI();
1871
1872     switch ( mode )
1873     {
1874         case wxFromCurrent : nextpos = seek + pos; break;
1875         case wxFromStart : nextpos = seek; break;
1876         case wxFromEnd : nextpos = GetLength() + seek; break;
1877         default : nextpos = pos; break; /* just to fool compiler, never happens */
1878     }
1879
1880     wxFileOffset toskip wxDUMMY_INITIALIZE(0);
1881     if ( nextpos >= pos )
1882     {
1883         toskip = nextpos - pos;
1884     }
1885     else
1886     {
1887         wxZipEntry current(m_entry);
1888         if (!OpenEntry(current))
1889         {
1890             m_lasterror = wxSTREAM_READ_ERROR;
1891             return pos;
1892         }
1893         toskip = nextpos;
1894     }
1895
1896     if ( toskip > 0 )
1897     {
1898         const int BUFSIZE = 4096;
1899         size_t sz;
1900         char buffer[BUFSIZE];
1901         while ( toskip > 0 )
1902         {
1903             sz = wx_truncate_cast(size_t, wxMin(toskip, BUFSIZE));
1904             Read(buffer, sz);
1905             toskip -= sz;
1906         }
1907     }
1908
1909     pos = nextpos;
1910     return pos;
1911 }
1912
1913 #endif // WXWIN_COMPATIBILITY_2_6
1914
1915
1916 /////////////////////////////////////////////////////////////////////////////
1917 // Output stream
1918
1919 #include "wx/listimpl.cpp"
1920 WX_DEFINE_LIST(wx__ZipEntryList)
1921
1922 wxZipOutputStream::wxZipOutputStream(wxOutputStream& stream,
1923                                      int level      /*=-1*/,
1924                                      wxMBConv& conv /*=wxConvLocal*/)
1925   : wxArchiveOutputStream(stream, conv),
1926     m_store(new wxStoredOutputStream(stream)),
1927     m_deflate(NULL),
1928     m_backlink(NULL),
1929     m_initialData(new char[OUTPUT_LATENCY]),
1930     m_initialSize(0),
1931     m_pending(NULL),
1932     m_raw(false),
1933     m_headerOffset(0),
1934     m_headerSize(0),
1935     m_entrySize(0),
1936     m_comp(NULL),
1937     m_level(level),
1938     m_offsetAdjustment(wxInvalidOffset)
1939 {
1940 }
1941
1942 wxZipOutputStream::~wxZipOutputStream()
1943 {
1944     Close();
1945     WX_CLEAR_LIST(wx__ZipEntryList, m_entries);
1946     delete m_store;
1947     delete m_deflate;
1948     delete m_pending;
1949     delete [] m_initialData;
1950     if (m_backlink)
1951         m_backlink->Release(this);
1952 }
1953
1954 bool wxZipOutputStream::PutNextEntry(
1955     const wxString& name,
1956     const wxDateTime& dt /*=wxDateTime::Now()*/,
1957     wxFileOffset size    /*=wxInvalidOffset*/)
1958 {
1959     return PutNextEntry(new wxZipEntry(name, dt, size));
1960 }
1961
1962 bool wxZipOutputStream::PutNextDirEntry(
1963     const wxString& name,
1964     const wxDateTime& dt /*=wxDateTime::Now()*/)
1965 {
1966     wxZipEntry *entry = new wxZipEntry(name, dt);
1967     entry->SetIsDir();
1968     return PutNextEntry(entry);
1969 }
1970
1971 bool wxZipOutputStream::CopyEntry(wxZipEntry *entry,
1972                                   wxZipInputStream& inputStream)
1973 {
1974     wx__ZipEntryPtr e(entry);
1975
1976     return
1977         inputStream.DoOpen(e.get(), true) &&
1978         DoCreate(e.release(), true) &&
1979         Write(inputStream).IsOk() && inputStream.Eof();
1980 }
1981
1982 bool wxZipOutputStream::PutNextEntry(wxArchiveEntry *entry)
1983 {
1984     wxZipEntry *zipEntry = wxStaticCast(entry, wxZipEntry);
1985     if (!zipEntry)
1986         delete entry;
1987     return PutNextEntry(zipEntry);
1988 }
1989
1990 bool wxZipOutputStream::CopyEntry(wxArchiveEntry *entry,
1991                                   wxArchiveInputStream& stream)
1992 {
1993     wxZipEntry *zipEntry = wxStaticCast(entry, wxZipEntry);
1994
1995     if (!zipEntry || !stream.OpenEntry(*zipEntry)) {
1996         delete entry;
1997         return false;
1998     }
1999
2000     return CopyEntry(zipEntry, wx_static_cast(wxZipInputStream&, stream));
2001 }
2002
2003 bool wxZipOutputStream::CopyArchiveMetaData(wxZipInputStream& inputStream)
2004 {
2005     m_Comment = inputStream.GetComment();
2006     if (m_backlink)
2007         m_backlink->Release(this);
2008     m_backlink = inputStream.MakeLink(this);
2009     return true;
2010 }
2011
2012 bool wxZipOutputStream::CopyArchiveMetaData(wxArchiveInputStream& stream)
2013 {
2014     return CopyArchiveMetaData(wx_static_cast(wxZipInputStream&, stream));
2015 }
2016
2017 void wxZipOutputStream::SetLevel(int level)
2018 {
2019     if (level != m_level) {
2020         if (m_comp != m_deflate)
2021             delete m_deflate;
2022         m_deflate = NULL;
2023         m_level = level;
2024     }
2025 }
2026
2027 bool wxZipOutputStream::DoCreate(wxZipEntry *entry, bool raw /*=false*/)
2028 {
2029     CloseEntry();
2030
2031     m_pending = entry;
2032     if (!m_pending)
2033         return false;
2034
2035     // write the signature bytes right away
2036     wxDataOutputStream ds(*m_parent_o_stream);
2037     ds << LOCAL_MAGIC;
2038
2039     // and if this is the first entry test for seekability
2040     if (m_headerOffset == 0 && m_parent_o_stream->IsSeekable()) {
2041 #if wxUSE_LOG
2042         bool logging = wxLog::IsEnabled();
2043         wxLogNull nolog;
2044 #endif // wxUSE_LOG
2045         wxFileOffset here = m_parent_o_stream->TellO();
2046
2047         if (here != wxInvalidOffset && here >= 4) {
2048             if (m_parent_o_stream->SeekO(here - 4) == here - 4) {
2049                 m_offsetAdjustment = here - 4;
2050 #if wxUSE_LOG
2051                 wxLog::EnableLogging(logging);
2052 #endif // wxUSE_LOG
2053                 m_parent_o_stream->SeekO(here);
2054             }
2055         }
2056     }
2057
2058     m_pending->SetOffset(m_headerOffset);
2059
2060     m_crcAccumulator = crc32(0, Z_NULL, 0);
2061
2062     if (raw)
2063         m_raw = true;
2064
2065     m_lasterror = wxSTREAM_NO_ERROR;
2066     return true;
2067 }
2068
2069 // Can be overriden to add support for additional compression methods
2070 //
2071 wxOutputStream *wxZipOutputStream::OpenCompressor(
2072     wxOutputStream& stream,
2073     wxZipEntry& entry,
2074     const Buffer bufs[])
2075 {
2076     if (entry.GetMethod() == wxZIP_METHOD_DEFAULT) {
2077         if (GetLevel() == 0
2078                 && (IsParentSeekable()
2079                     || entry.GetCompressedSize() != wxInvalidOffset
2080                     || entry.GetSize() != wxInvalidOffset)) {
2081             entry.SetMethod(wxZIP_METHOD_STORE);
2082         } else {
2083             int size = 0;
2084             for (int i = 0; bufs[i].m_data; ++i)
2085                 size += bufs[i].m_size;
2086             entry.SetMethod(size <= 6 ?
2087                             wxZIP_METHOD_STORE : wxZIP_METHOD_DEFLATE);
2088         }
2089     }
2090
2091     switch (entry.GetMethod()) {
2092         case wxZIP_METHOD_STORE:
2093             if (entry.GetCompressedSize() == wxInvalidOffset)
2094                 entry.SetCompressedSize(entry.GetSize());
2095             return m_store;
2096
2097         case wxZIP_METHOD_DEFLATE:
2098         {
2099             int defbits = wxZIP_DEFLATE_NORMAL;
2100             switch (GetLevel()) {
2101                 case 0: case 1:
2102                     defbits = wxZIP_DEFLATE_SUPERFAST;
2103                     break;
2104                 case 2: case 3: case 4:
2105                     defbits = wxZIP_DEFLATE_FAST;
2106                     break;
2107                 case 8: case 9:
2108                     defbits = wxZIP_DEFLATE_EXTRA;
2109                     break;
2110             }
2111             entry.SetFlags((entry.GetFlags() & ~wxZIP_DEFLATE_MASK) |
2112                             defbits | wxZIP_SUMS_FOLLOW);
2113
2114             if (!m_deflate)
2115                 m_deflate = new wxZlibOutputStream2(stream, GetLevel());
2116             else
2117                 m_deflate->Open(stream);
2118
2119             return m_deflate;
2120         }
2121
2122         default:
2123             wxLogError(_("unsupported Zip compression method"));
2124     }
2125
2126     return NULL;
2127 }
2128
2129 bool wxZipOutputStream::CloseCompressor(wxOutputStream *comp)
2130 {
2131     if (comp == m_deflate)
2132         m_deflate->Close();
2133     else if (comp != m_store)
2134         delete comp;
2135     return true;
2136 }
2137
2138 // This is called when OUPUT_LATENCY bytes has been written to the
2139 // wxZipOutputStream to actually create the zip entry.
2140 //
2141 void wxZipOutputStream::CreatePendingEntry(const void *buffer, size_t size)
2142 {
2143     wxASSERT(IsOk() && m_pending && !m_comp);
2144     wx__ZipEntryPtr spPending(m_pending);
2145     m_pending = NULL;
2146
2147     Buffer bufs[] = {
2148         { m_initialData, m_initialSize },
2149         { (const char*)buffer, size },
2150         { NULL, 0 }
2151     };
2152
2153     if (m_raw)
2154         m_comp = m_store;
2155     else
2156         m_comp = OpenCompressor(*m_store, *spPending,
2157                                 m_initialSize ? bufs : bufs + 1);
2158
2159     if (IsParentSeekable()
2160         || (spPending->m_Crc
2161             && spPending->m_CompressedSize != wxInvalidOffset
2162             && spPending->m_Size != wxInvalidOffset))
2163         spPending->m_Flags &= ~wxZIP_SUMS_FOLLOW;
2164     else
2165         if (spPending->m_CompressedSize != wxInvalidOffset)
2166             spPending->m_Flags |= wxZIP_SUMS_FOLLOW;
2167
2168     m_headerSize = spPending->WriteLocal(*m_parent_o_stream, GetConv());
2169     m_lasterror = m_parent_o_stream->GetLastError();
2170
2171     if (IsOk()) {
2172         m_entries.push_back(spPending.release());
2173         OnSysWrite(m_initialData, m_initialSize);
2174     }
2175
2176     m_initialSize = 0;
2177 }
2178
2179 // This is called to write out the zip entry when Close has been called
2180 // before OUTPUT_LATENCY bytes has been written to the wxZipOutputStream.
2181 //
2182 void wxZipOutputStream::CreatePendingEntry()
2183 {
2184     wxASSERT(IsOk() && m_pending && !m_comp);
2185     wx__ZipEntryPtr spPending(m_pending);
2186     m_pending = NULL;
2187     m_lasterror = wxSTREAM_WRITE_ERROR;
2188
2189     if (!m_raw) {
2190         // Initially compresses the data to memory, then fall back to 'store'
2191         // if the compressor makes the data larger rather than smaller.
2192         wxMemoryOutputStream mem;
2193         Buffer bufs[] = { { m_initialData, m_initialSize }, { NULL, 0 } };
2194         wxOutputStream *comp = OpenCompressor(mem, *spPending, bufs);
2195
2196         if (!comp)
2197             return;
2198         if (comp != m_store) {
2199             bool ok = comp->Write(m_initialData, m_initialSize).IsOk();
2200             CloseCompressor(comp);
2201             if (!ok)
2202                 return;
2203         }
2204
2205         m_entrySize = m_initialSize;
2206         m_crcAccumulator = crc32(0, (Byte*)m_initialData, m_initialSize);
2207
2208         if (mem.GetSize() > 0 && mem.GetSize() < m_initialSize) {
2209             m_initialSize = mem.GetSize();
2210             mem.CopyTo(m_initialData, m_initialSize);
2211         } else {
2212             spPending->SetMethod(wxZIP_METHOD_STORE);
2213         }
2214
2215         spPending->SetSize(m_entrySize);
2216         spPending->SetCrc(m_crcAccumulator);
2217         spPending->SetCompressedSize(m_initialSize);
2218     }
2219
2220     spPending->m_Flags &= ~wxZIP_SUMS_FOLLOW;
2221     m_headerSize = spPending->WriteLocal(*m_parent_o_stream, GetConv());
2222
2223     if (m_parent_o_stream->IsOk()) {
2224         m_entries.push_back(spPending.release());
2225         m_comp = m_store;
2226         m_store->Write(m_initialData, m_initialSize);
2227     }
2228
2229     m_initialSize = 0;
2230     m_lasterror = m_parent_o_stream->GetLastError();
2231 }
2232
2233 // Write the 'central directory' and the 'end-central-directory' records.
2234 //
2235 bool wxZipOutputStream::Close()
2236 {
2237     CloseEntry();
2238
2239     if (m_lasterror == wxSTREAM_WRITE_ERROR || m_entries.size() == 0)
2240         return false;
2241
2242     wxZipEndRec endrec;
2243
2244     endrec.SetEntriesHere(m_entries.size());
2245     endrec.SetTotalEntries(m_entries.size());
2246     endrec.SetOffset(m_headerOffset);
2247     endrec.SetComment(m_Comment);
2248
2249     wx__ZipEntryList::iterator it;
2250     wxFileOffset size = 0;
2251
2252     for (it = m_entries.begin(); it != m_entries.end(); ++it) {
2253         size += (*it)->WriteCentral(*m_parent_o_stream, GetConv());
2254         delete *it;
2255     }
2256     m_entries.clear();
2257
2258     endrec.SetSize(size);
2259     endrec.Write(*m_parent_o_stream, GetConv());
2260
2261     m_lasterror = m_parent_o_stream->GetLastError();
2262     if (!IsOk())
2263         return false;
2264     m_lasterror = wxSTREAM_EOF;
2265     return true;
2266 }
2267
2268 // Finish writing the current entry
2269 //
2270 bool wxZipOutputStream::CloseEntry()
2271 {
2272     if (IsOk() && m_pending)
2273         CreatePendingEntry();
2274     if (!IsOk())
2275         return false;
2276     if (!m_comp)
2277         return true;
2278
2279     CloseCompressor(m_comp);
2280     m_comp = NULL;
2281
2282     wxFileOffset compressedSize = m_store->TellO();
2283
2284     wxZipEntry& entry = *m_entries.back();
2285
2286     // When writing raw the crc and size can't be checked
2287     if (m_raw) {
2288         m_crcAccumulator = entry.GetCrc();
2289         m_entrySize = entry.GetSize();
2290     }
2291
2292     // Write the sums in the trailing 'data descriptor' if necessary
2293     if (entry.m_Flags & wxZIP_SUMS_FOLLOW) {
2294         wxASSERT(!IsParentSeekable());
2295         m_headerOffset +=
2296             entry.WriteDescriptor(*m_parent_o_stream, m_crcAccumulator,
2297                                   compressedSize, m_entrySize);
2298         m_lasterror = m_parent_o_stream->GetLastError();
2299     }
2300
2301     // If the local header didn't have the correct crc and size written to
2302     // it then seek back and fix it
2303     else if (m_crcAccumulator != entry.GetCrc()
2304             || m_entrySize != entry.GetSize()
2305             || compressedSize != entry.GetCompressedSize())
2306     {
2307         if (IsParentSeekable()) {
2308             wxFileOffset here = m_parent_o_stream->TellO();
2309             wxFileOffset headerOffset = m_headerOffset + m_offsetAdjustment;
2310             m_parent_o_stream->SeekO(headerOffset + SUMS_OFFSET);
2311             entry.WriteDescriptor(*m_parent_o_stream, m_crcAccumulator,
2312                                   compressedSize, m_entrySize);
2313             m_parent_o_stream->SeekO(here);
2314             m_lasterror = m_parent_o_stream->GetLastError();
2315         } else {
2316             m_lasterror = wxSTREAM_WRITE_ERROR;
2317         }
2318     }
2319
2320     m_headerOffset += m_headerSize + compressedSize;
2321     m_headerSize = 0;
2322     m_entrySize = 0;
2323     m_store->Close();
2324     m_raw = false;
2325
2326     if (IsOk())
2327         m_lasterror = m_parent_o_stream->GetLastError();
2328     else
2329         wxLogError(_("error writing zip entry '%s': bad crc or length"),
2330                    entry.GetName().c_str());
2331     return IsOk();
2332 }
2333
2334 void wxZipOutputStream::Sync()
2335 {
2336     if (IsOk() && m_pending)
2337         CreatePendingEntry(NULL, 0);
2338     if (!m_comp)
2339         m_lasterror = wxSTREAM_WRITE_ERROR;
2340     if (IsOk()) {
2341         m_comp->Sync();
2342         m_lasterror = m_comp->GetLastError();
2343     }
2344 }
2345
2346 size_t wxZipOutputStream::OnSysWrite(const void *buffer, size_t size)
2347 {
2348     if (IsOk() && m_pending) {
2349         if (m_initialSize + size < OUTPUT_LATENCY) {
2350             memcpy(m_initialData + m_initialSize, buffer, size);
2351             m_initialSize += size;
2352             return size;
2353         } else {
2354             CreatePendingEntry(buffer, size);
2355         }
2356     }
2357
2358     if (!m_comp)
2359         m_lasterror = wxSTREAM_WRITE_ERROR;
2360     if (!IsOk() || !size)
2361         return 0;
2362
2363     if (m_comp->Write(buffer, size).LastWrite() != size)
2364         m_lasterror = wxSTREAM_WRITE_ERROR;
2365     m_crcAccumulator = crc32(m_crcAccumulator, (Byte*)buffer, size);
2366     m_entrySize += m_comp->LastWrite();
2367
2368     return m_comp->LastWrite();
2369 }
2370
2371 #endif // wxUSE_ZLIB && wxUSE_STREAMS && wxUSE_ZIPSTREAM