426 lines
9.6 KiB
C++
426 lines
9.6 KiB
C++
/*
|
|
* (c) Copyright Ascensio System SIA 2010-2023
|
|
*
|
|
* This program is a free software product. You can redistribute it and/or
|
|
* modify it under the terms of the GNU Affero General Public License (AGPL)
|
|
* version 3 as published by the Free Software Foundation. In accordance with
|
|
* Section 7(a) of the GNU AGPL its Section 15 shall be amended to the effect
|
|
* that Ascensio System SIA expressly excludes the warranty of non-infringement
|
|
* of any third-party rights.
|
|
*
|
|
* This program is distributed WITHOUT ANY WARRANTY; without even the implied
|
|
* warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. For
|
|
* details, see the GNU AGPL at: http://www.gnu.org/licenses/agpl-3.0.html
|
|
*
|
|
* You can contact Ascensio System SIA at 20A-6 Ernesta Birznieka-Upish
|
|
* street, Riga, Latvia, EU, LV-1050.
|
|
*
|
|
* The interactive user interfaces in modified source and object code versions
|
|
* of the Program must display Appropriate Legal Notices, as required under
|
|
* Section 5 of the GNU AGPL version 3.
|
|
*
|
|
* Pursuant to Section 7(b) of the License you must retain the original Product
|
|
* logo when distributing the program. Pursuant to Section 7(e) we decline to
|
|
* grant you any rights under trademark law for use of our trademarks.
|
|
*
|
|
* All the Product's GUI elements, including illustrations and icon sets, as
|
|
* well as technical writing content are licensed under the terms of the
|
|
* Creative Commons Attribution-ShareAlike 4.0 International. See the License
|
|
* terms at http://creativecommons.org/licenses/by-sa/4.0/legalcode
|
|
*
|
|
*/
|
|
|
|
#include "Ole1FormatReader.h"
|
|
|
|
CDataStream::CDataStream() : pBuffer(NULL)
|
|
{
|
|
}
|
|
CDataStream::CDataStream(BYTE* pBuf, unsigned int unSize)
|
|
{
|
|
SetStream(pBuf, unSize);
|
|
}
|
|
CDataStream::~CDataStream()
|
|
{
|
|
}
|
|
void CDataStream::SetStream(BYTE* pBuf, unsigned int unSize)
|
|
{
|
|
pBuffer = pBuf;
|
|
pCur = pBuf;
|
|
pEnd = pBuf + unSize + 1;
|
|
};
|
|
BYTE* CDataStream::GetCurPtr()
|
|
{
|
|
return pCur;
|
|
}
|
|
|
|
unsigned char CDataStream::ReadUChar()
|
|
{
|
|
if (pCur + 1 >= pEnd)
|
|
return 0;
|
|
|
|
unsigned char unResult = pCur[0];
|
|
pCur++;
|
|
return unResult;
|
|
};
|
|
void CDataStream::WriteUChar(unsigned char val)
|
|
{
|
|
if (pCur + 1 >= pEnd)
|
|
return;
|
|
|
|
pCur[0] = val;
|
|
pCur++;
|
|
};
|
|
unsigned short CDataStream::ReadUShort()
|
|
{
|
|
if (pCur + 2 >= pEnd)
|
|
return 0;
|
|
|
|
unsigned short ushResult = (pCur[0]) | ((pCur[1]) << 8);
|
|
pCur += 2;
|
|
return ushResult;
|
|
};
|
|
void CDataStream::WriteUShort(unsigned short val)
|
|
{
|
|
if (pCur + 2 >= pEnd)
|
|
return;
|
|
|
|
((unsigned short*) pCur)[0] = val;
|
|
pCur += 2;
|
|
};
|
|
_UINT32 CDataStream::ReadULong()
|
|
{
|
|
if (pCur + 4 >= pEnd)
|
|
return 0;
|
|
|
|
_UINT32 unResult = (_UINT32)((pCur[0] << 0) | ((pCur[1]) << 8) | ((pCur[2]) << 16) | ((pCur[3]) << 24));
|
|
pCur += 4;
|
|
return unResult;
|
|
};
|
|
void CDataStream::WriteULong(_UINT32 val)
|
|
{
|
|
if (pCur + 4 >= pEnd)
|
|
return;
|
|
|
|
((_UINT32*)pCur)[0] = val;
|
|
pCur += 4;
|
|
};
|
|
double CDataStream::ReadDouble()
|
|
{
|
|
if (pCur + 4 >= pEnd)
|
|
return 0;
|
|
|
|
float output;
|
|
|
|
*((unsigned char*)(&output) + 0) = pCur[0];
|
|
*((unsigned char*)(&output) + 1) = pCur[1];
|
|
*((unsigned char*)(&output) + 2) = pCur[2];
|
|
*((unsigned char*)(&output) + 3) = pCur[3];
|
|
|
|
pCur += 4;
|
|
|
|
return output;
|
|
|
|
int lIntValue = (int)((pCur[0] << 16) | ((pCur[1]) << 8) | ((pCur[2]) << 0));
|
|
int lFracValue = (int)(pCur[3]);
|
|
pCur += 4;
|
|
return (double)(lIntValue + (lFracValue / 16.0));
|
|
};
|
|
char CDataStream::ReadChar()
|
|
{
|
|
return (char)ReadUChar();
|
|
}
|
|
void CDataStream::WriteChar(char val)
|
|
{
|
|
WriteUChar((unsigned char)val);
|
|
}
|
|
short CDataStream::ReadShort()
|
|
{
|
|
return (short)ReadUShort();
|
|
}
|
|
void CDataStream::WriteShort(short val)
|
|
{
|
|
WriteUShort((unsigned short) val);
|
|
}
|
|
_INT32 CDataStream::ReadLong()
|
|
{
|
|
return (_INT32)ReadULong();
|
|
}
|
|
void CDataStream::WriteLong(_INT32 val)
|
|
{
|
|
WriteULong((_UINT32)val);
|
|
}
|
|
void CDataStream::ReadBytes(char* pBuffer, unsigned int ulSize)
|
|
{
|
|
size_t ulRemainSize = (pEnd - pCur);
|
|
size_t ulFinalSize = (ulRemainSize > ulSize ? ulSize : ulRemainSize);
|
|
|
|
for (size_t ulIndex = 0; ulIndex < ulFinalSize; ulIndex++)
|
|
{
|
|
pBuffer[ulIndex] = ReadChar();
|
|
}
|
|
}
|
|
void CDataStream::WriteBytes(char* pBuffer, unsigned int ulSize)
|
|
{
|
|
size_t ulRemainSize = (pEnd - pCur);
|
|
size_t ulFinalSize = (ulRemainSize > ulSize ? ulSize : ulRemainSize);
|
|
|
|
for (size_t ulIndex = 0; ulIndex < ulFinalSize; ulIndex++)
|
|
{
|
|
WriteChar(pBuffer[ulIndex]);
|
|
}
|
|
}
|
|
void CDataStream::ReadBytes(unsigned char* pBuffer, unsigned int ulSize)
|
|
{
|
|
size_t ulRemainSize = (pEnd - pCur);
|
|
size_t ulFinalSize = (ulRemainSize > ulSize ? ulSize : ulRemainSize);
|
|
|
|
for (size_t ulIndex = 0; ulIndex < ulFinalSize; ulIndex++)
|
|
{
|
|
pBuffer[ulIndex] = ReadUChar();
|
|
}
|
|
}
|
|
void CDataStream::WriteBytes(unsigned char* pBuffer, unsigned int ulSize)
|
|
{
|
|
size_t ulRemainSize = (pEnd - pCur);
|
|
size_t ulFinalSize = (ulRemainSize > ulSize ? ulSize : ulRemainSize);
|
|
|
|
for (size_t ulIndex = 0; ulIndex < ulFinalSize; ulIndex++)
|
|
{
|
|
WriteUChar(pBuffer[ulIndex]);
|
|
}
|
|
}
|
|
void CDataStream::ReadBytes(unsigned short* pBuffer, unsigned int ulSize)
|
|
{
|
|
size_t ulRemainSize = (pEnd - pCur) / 2;
|
|
size_t ulFinalSize = (ulRemainSize > ulSize ? ulSize : ulRemainSize);
|
|
|
|
for (size_t ulIndex = 0; ulIndex < ulFinalSize; ulIndex++)
|
|
{
|
|
pBuffer[ulIndex] = ReadUShort();
|
|
}
|
|
}
|
|
void CDataStream::ReadBytes(short* pBuffer, unsigned int ulSize)
|
|
{
|
|
size_t ulRemainSize = (pEnd - pCur) / 2;
|
|
size_t ulFinalSize = (ulRemainSize > ulSize ? ulSize : ulRemainSize);
|
|
|
|
for (size_t ulIndex = 0; ulIndex < ulFinalSize; ulIndex++)
|
|
{
|
|
pBuffer[ulIndex] = ReadShort();
|
|
}
|
|
}
|
|
void CDataStream::ReadBytes(_UINT32* pBuffer, unsigned int ulSize)
|
|
{
|
|
size_t ulRemainSize = (pEnd - pCur) / 4;
|
|
size_t ulFinalSize = (ulRemainSize > ulSize ? ulSize : ulRemainSize);
|
|
|
|
for (size_t ulIndex = 0; ulIndex < ulFinalSize; ulIndex++)
|
|
{
|
|
pBuffer[ulIndex] = ReadULong();
|
|
}
|
|
}
|
|
CDataStream& CDataStream::operator>>(unsigned char& nValue)
|
|
{
|
|
nValue = ReadUChar();
|
|
return *this;
|
|
}
|
|
CDataStream& CDataStream::operator<<(unsigned char nValue)
|
|
{
|
|
WriteUChar(nValue);
|
|
return *this;
|
|
}
|
|
CDataStream& CDataStream::operator>>(unsigned short& nValue)
|
|
{
|
|
nValue = ReadUShort();
|
|
return *this;
|
|
}
|
|
CDataStream& CDataStream::operator<<(unsigned short nValue)
|
|
{
|
|
WriteUShort(nValue);
|
|
return *this;
|
|
}
|
|
CDataStream& CDataStream::operator>>(_UINT32& nValue)
|
|
{
|
|
nValue = ReadULong();
|
|
return *this;
|
|
}
|
|
CDataStream& CDataStream::operator<<(_UINT32 nValue)
|
|
{
|
|
WriteULong(nValue);
|
|
return *this;
|
|
}
|
|
CDataStream& CDataStream::operator>>(double& dValue)
|
|
{
|
|
dValue = ReadDouble();
|
|
return *this;
|
|
}
|
|
CDataStream& CDataStream::operator>>(char& nValue)
|
|
{
|
|
nValue = ReadChar();
|
|
return *this;
|
|
}
|
|
CDataStream& CDataStream::operator<<(char nValue)
|
|
{
|
|
WriteChar(nValue);
|
|
return *this;
|
|
}
|
|
CDataStream& CDataStream::operator>>(bool& nValue)
|
|
{
|
|
nValue = !!ReadChar();
|
|
return *this;
|
|
}
|
|
CDataStream& CDataStream::operator>>(short& nValue)
|
|
{
|
|
nValue = ReadShort();
|
|
return *this;
|
|
}
|
|
CDataStream& CDataStream::operator<<(short nValue)
|
|
{
|
|
WriteShort(nValue);
|
|
return *this;
|
|
}
|
|
CDataStream& CDataStream::operator>>(_INT32& nValue)
|
|
{
|
|
nValue = ReadLong();
|
|
return *this;
|
|
}
|
|
CDataStream& CDataStream::operator<<(_INT32 nValue)
|
|
{
|
|
WriteLong(nValue);
|
|
return *this;
|
|
}
|
|
|
|
bool CDataStream::IsValid() const
|
|
{
|
|
if (NULL == pBuffer)
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
bool CDataStream::IsEof() const
|
|
{
|
|
if (pCur >= pEnd)
|
|
return true;
|
|
|
|
return false;
|
|
}
|
|
unsigned int CDataStream::Tell()
|
|
{
|
|
return (unsigned int)(pCur - pBuffer);
|
|
}
|
|
void CDataStream::Skip(unsigned int ulSkip)
|
|
{
|
|
pCur += ulSkip;
|
|
}
|
|
void CDataStream::SeekBack(unsigned int ulSkipBack)
|
|
{
|
|
pCur -= ulSkipBack;
|
|
}
|
|
void CDataStream::SeekToStart()
|
|
{
|
|
pCur = pBuffer;
|
|
}
|
|
unsigned int CDataStream::CanReadWrite()
|
|
{
|
|
return (unsigned int)(pEnd - pCur);
|
|
}
|
|
|
|
static CDataStream & operator >> (CDataStream & strm, LengthPrefixedAnsiString & str)
|
|
{
|
|
strm >> str.size;
|
|
char *s = new char[str.size];
|
|
strm.ReadBytes(s, str.size);
|
|
str.val = std::string(s, str.size);
|
|
delete []s;
|
|
return strm;
|
|
}
|
|
static CDataStream & operator << (CDataStream & strm, LengthPrefixedAnsiString str)
|
|
{
|
|
strm << str.size;
|
|
|
|
strm.WriteBytes((char*)str.val.c_str(), str.size - 1);
|
|
strm.WriteUChar(0);
|
|
return strm;
|
|
}
|
|
|
|
Ole1FormatReaderWriter::Ole1FormatReaderWriter() : NativeData(NULL), NativeDataSize(0), OtherData(NULL), OtherDataSize(0)
|
|
{
|
|
}
|
|
Ole1FormatReaderWriter::Ole1FormatReaderWriter(BYTE *pData, int Size) : NativeData(NULL), NativeDataSize(0), OtherData(NULL), OtherDataSize(0)
|
|
{
|
|
Read(pData, Size);
|
|
}
|
|
Ole1FormatReaderWriter::~Ole1FormatReaderWriter()
|
|
{
|
|
if (OtherData)
|
|
delete []OtherData;
|
|
OtherData = NULL;
|
|
}
|
|
void Ole1FormatReaderWriter::Read(BYTE *pData, int Size)
|
|
{
|
|
NativeDataSize = 0;
|
|
if (!pData || Size < 8) return;
|
|
|
|
CDataStream stream(pData, Size);
|
|
|
|
stream >> Header.OLEVersion >> Header.FormatID;
|
|
|
|
if (Header.FormatID == 2)
|
|
{
|
|
if (Header.OLEVersion & 0x00000500 ||
|
|
Header.OLEVersion & 0x00010001)
|
|
{
|
|
stream >> Header.ClassName;
|
|
}
|
|
stream >> Header.Width >> Header.Height;
|
|
|
|
stream >> NativeDataSize;
|
|
|
|
NativeData = stream.GetCurPtr();
|
|
stream.Skip(NativeDataSize);
|
|
|
|
if (stream.IsEof())
|
|
return;
|
|
|
|
// далее графическое представление
|
|
OtherDataSize = stream.CanReadWrite();
|
|
|
|
OtherData = new BYTE[OtherDataSize];
|
|
stream.ReadBytes(OtherData, OtherDataSize);
|
|
}
|
|
|
|
}
|
|
void Ole1FormatReaderWriter::Write(BYTE *pData, int &Size)
|
|
{
|
|
Size = 0;
|
|
if (!pData) return;
|
|
if (NativeDataSize < 1) return;
|
|
|
|
CDataStream stream(pData, NativeDataSize + 2048);
|
|
|
|
stream << Header.OLEVersion << Header.FormatID;
|
|
|
|
stream << Header.ClassName;
|
|
|
|
stream << Header.Width << Header.Height;
|
|
|
|
stream << NativeDataSize;
|
|
|
|
stream.WriteBytes(NativeData, NativeDataSize);
|
|
|
|
/// далее графическое представление
|
|
BYTE other[9] = {1, 5, 0, 0, 0, 0, 0, 0, 0};
|
|
|
|
Size = stream.Tell() - 1;
|
|
stream.WriteBytes(other, 9);
|
|
Size = stream.Tell() - 1;
|
|
|
|
//padding ???
|
|
|
|
char padding[8];
|
|
memset(padding, 0, 8);
|
|
|
|
stream.WriteBytes(padding, (Size / 8 + 1 ) * 8 - Size);
|
|
Size = stream.Tell() - 1;
|
|
}
|