1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
|
////////////////////////////////////////////////////////////////////////////////
// This source file is part of the ZipArchive library source distribution and
// is Copyrighted 2000 - 2007 by Artpol Software - Tadeusz Dracz
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2
// of the License, or (at your option) any later version.
//
// For the licensing details refer to the License.txt file.
//
// Web Site: http://www.artpol-software.com
////////////////////////////////////////////////////////////////////////////////
#include "stdafx.h"
#include "ZipCompressor.h"
#include "BytesWriter.h"
#include "DeflateCompressor.h"
using namespace ZipArchiveLib;
CZipCompressor* CZipCompressor::CreateCompressor(WORD uMethod, CZipStorage* pStorage)
{
if (uMethod == methodStore || uMethod == methodDeflate)
return new CDeflateCompressor(pStorage);
return NULL;
}
void CZipCompressor::UpdateFileCrc(const void *pBuffer, DWORD uSize)
{
m_pFile->m_uCrc32 = zarch_crc32(m_pFile->m_uCrc32, (zarch_Bytef*)pBuffer, uSize);
}
void CZipCompressor::UpdateCrc(const void *pBuffer, DWORD uSize)
{
m_uCrc32 = zarch_crc32(m_uCrc32, (zarch_Bytef*)pBuffer, uSize);
}
void CZipCompressor::UpdateOptions(const COptionsMap& optionsMap)
{
const COptions* pOptions = GetOptions();
if (pOptions == NULL)
return;
const COptions* pNewOptions = optionsMap.Get(pOptions->GetType());
if (pNewOptions != NULL)
UpdateOptions(pNewOptions);
}
void CZipCompressor::InitBuffer()
{
// This should be greated that 64k for deflate when creating offsets pairs is enabled
// otherwise deflate will not be able to write one block in one go and will never report
// a flushed block for low-compressable data
const COptions* pOptions = GetOptions();
DWORD bufferSize = 0;
if (pOptions != NULL)
bufferSize = pOptions->m_iBufferSize;
if (bufferSize == 0)
bufferSize = COptions::cDefaultBufferSize;
m_pBuffer.Allocate(bufferSize);
}
void CZipCompressor::COptionsMap::Set(const CZipCompressor::COptions* pOptions)
{
if (pOptions == NULL)
return;
int iType = pOptions->GetType();
Remove(iType);
SetAt(iType, pOptions->Clone());
}
CZipCompressor::COptions* CZipCompressor::COptionsMap::Get(int iType) const
{
COptions* pTemp = NULL;
if (Lookup(iType, pTemp))
return pTemp;
else
return NULL;
}
void CZipCompressor::COptionsMap::Remove(int iType)
{
COptions* pTemp = Get(iType);
if (pTemp != NULL)
{
delete pTemp;
RemoveKey(iType);
}
}
CZipCompressor::COptionsMap::~COptionsMap()
{
COptionsMap::iterator iter = GetStartPosition();
while (IteratorValid(iter))
{
COptions* pOptions = NULL;
int iType = 0;
GetNextAssoc(iter, iType, pOptions);
delete pOptions;
}
RemoveAll();
}
|