114 lines
3 KiB
C++
114 lines
3 KiB
C++
////////////////////////////////////////////////////////////////////////////////
|
|
// This source file is part of the ZipArchive library source distribution and
|
|
// is Copyrighted 2000 - 2012 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 "ZipMemFile.h"
|
|
#include "ZipException.h"
|
|
|
|
//////////////////////////////////////////////////////////////////////
|
|
// Construction/Destruction
|
|
//////////////////////////////////////////////////////////////////////
|
|
|
|
#if defined _ZIP_IMPL_MFC && (_MSC_VER >= 1300 || _ZIP_FILE_IMPLEMENTATION != ZIP_ZFI_WIN)
|
|
IMPLEMENT_DYNAMIC(CZipMemFile, CFile)
|
|
#endif
|
|
|
|
void CZipMemFile::Grow(size_t nGrowTo)
|
|
{
|
|
if (m_nBufSize < (UINT)nGrowTo)
|
|
{
|
|
if (m_nGrowBy == 0)
|
|
CZipException::Throw(CZipException::memError);
|
|
size_t nNewSize = m_nBufSize;
|
|
while (nNewSize < nGrowTo)
|
|
nNewSize += m_nGrowBy;
|
|
BYTE* lpNew;
|
|
if (m_lpBuf)
|
|
lpNew = (BYTE*)realloc((void*) m_lpBuf, nNewSize);
|
|
else
|
|
lpNew = (BYTE*)malloc(nNewSize);
|
|
|
|
if (!lpNew)
|
|
CZipException::Throw(CZipException::memError);
|
|
m_nBufSize = nNewSize;
|
|
m_lpBuf = lpNew;
|
|
}
|
|
}
|
|
|
|
void CZipMemFile::SetLength(ZIP_FILE_USIZE nNewLen)
|
|
{
|
|
if (m_nBufSize < (size_t)nNewLen)
|
|
Grow((size_t)nNewLen);
|
|
else
|
|
m_nPos = (size_t)nNewLen;
|
|
m_nDataSize = (size_t)nNewLen;
|
|
}
|
|
|
|
UINT CZipMemFile::Read(void *lpBuf, UINT nCount)
|
|
{
|
|
if (m_nPos >= m_nDataSize)
|
|
return 0;
|
|
UINT nToRead = (m_nPos + nCount > m_nDataSize) ? (UINT)(m_nDataSize - m_nPos) : nCount;
|
|
memcpy(lpBuf, m_lpBuf + m_nPos, nToRead);
|
|
m_nPos += nToRead;
|
|
return nToRead;
|
|
|
|
}
|
|
|
|
void CZipMemFile::Write(const void *lpBuf, UINT nCount)
|
|
{
|
|
if (!nCount)
|
|
return;
|
|
|
|
if (m_nPos + nCount > m_nBufSize)
|
|
Grow(m_nPos + nCount);
|
|
memcpy(m_lpBuf + m_nPos, lpBuf, nCount);
|
|
m_nPos += nCount;
|
|
if (m_nPos > m_nDataSize)
|
|
m_nDataSize = m_nPos;
|
|
}
|
|
ZIP_FILE_USIZE CZipMemFile::Seek(ZIP_FILE_SIZE lOff, int nFrom)
|
|
{
|
|
ZIP_FILE_USIZE lNew = m_nPos;
|
|
|
|
if (nFrom == CZipAbstractFile::begin)
|
|
{
|
|
if (lOff < 0)
|
|
CZipException::Throw(CZipException::memError);
|
|
lNew = lOff;
|
|
}
|
|
else if (nFrom == CZipAbstractFile::current)
|
|
{
|
|
if (lOff < 0 && (ZIP_FILE_USIZE)(-lOff) > lNew)
|
|
CZipException::Throw(CZipException::memError);
|
|
lNew += lOff;
|
|
}
|
|
else if (nFrom == CZipAbstractFile::end)
|
|
{
|
|
if (lOff < 0 && ZIP_FILE_USIZE(-lOff) > m_nDataSize)
|
|
CZipException::Throw(CZipException::memError);
|
|
lNew = m_nDataSize + lOff;
|
|
}
|
|
else
|
|
return lNew;
|
|
|
|
// assumption that size_t is always signed
|
|
if (lNew > (size_t)(-1)) // max of size_t
|
|
CZipException::Throw(CZipException::memError);
|
|
if (lNew > m_nDataSize)
|
|
Grow((size_t)lNew);
|
|
|
|
m_nPos = (size_t)lNew;
|
|
return lNew;
|
|
}
|