mirror of
https://github.com/ONLYOFFICE/core.git
synced 2026-07-11 03:42:47 +08:00
120 lines
3.1 KiB
C++
120 lines
3.1 KiB
C++
#include "EncryptDictionary.h"
|
|
#include "Encrypt.h"
|
|
#include "Info.h"
|
|
|
|
#include <ctime>
|
|
|
|
namespace PdfWriter
|
|
{
|
|
//----------------------------------------------------------------------------------------
|
|
// CEncryptDict
|
|
//----------------------------------------------------------------------------------------
|
|
CEncryptDict::CEncryptDict(CXref* pXref)
|
|
{
|
|
m_pEncrypt = new CEncrypt();
|
|
if (m_pEncrypt)
|
|
m_pEncrypt->Init();
|
|
|
|
pXref->Add(this);
|
|
}
|
|
CEncryptDict::~CEncryptDict()
|
|
{
|
|
if (m_pEncrypt)
|
|
delete m_pEncrypt;
|
|
}
|
|
void CEncryptDict::CreateId(CInfoDict* pInfo, CXref* pXref)
|
|
{
|
|
CMd5 oMd5;
|
|
oMd5.Init();
|
|
|
|
std::time_t oTime = time(0);
|
|
oMd5.Update((BYTE*)&oTime, sizeof(oTime));
|
|
|
|
// Создаем идентификатор файла по элементам библиотеки Info.
|
|
if (pInfo)
|
|
{
|
|
const char *sTemp = NULL;
|
|
unsigned int nLen = 0;
|
|
|
|
// Author
|
|
sTemp = pInfo->GetInfo(InfoAuthor);
|
|
if ((nLen = StrLen(sTemp, -1)) > 0)
|
|
oMd5.Update((const BYTE *)sTemp, nLen);
|
|
|
|
// Creator
|
|
sTemp = pInfo->GetInfo(InfoCreator);
|
|
if ((nLen = StrLen(sTemp, -1)) > 0)
|
|
oMd5.Update((const BYTE *)sTemp, nLen);
|
|
|
|
// Producer
|
|
sTemp = pInfo->GetInfo(InfoProducer);
|
|
if ((nLen = StrLen(sTemp, -1)) > 0)
|
|
oMd5.Update((const BYTE *)sTemp, nLen);
|
|
|
|
// Title
|
|
sTemp = pInfo->GetInfo(InfoTitle);
|
|
if ((nLen = StrLen(sTemp, -1)) > 0)
|
|
oMd5.Update((const BYTE *)sTemp, nLen);
|
|
|
|
// Subject
|
|
sTemp = pInfo->GetInfo(InfoSubject);
|
|
if ((nLen = StrLen(sTemp, -1)) > 0)
|
|
oMd5.Update((const BYTE *)sTemp, nLen);
|
|
|
|
// Keywords
|
|
sTemp = pInfo->GetInfo(InfoKeyWords);
|
|
if ((nLen = StrLen(sTemp, -1)) > 0)
|
|
oMd5.Update((const BYTE *)sTemp, nLen);
|
|
|
|
int nXrefEntriesCount = pXref->GetCount();
|
|
oMd5.Update((const BYTE *)&nXrefEntriesCount, sizeof(unsigned int));
|
|
|
|
}
|
|
oMd5.Final(m_pEncrypt->m_anEncryptID);
|
|
}
|
|
void CEncryptDict::SetPassword(const char* sOwnerPassword, const char* sUserPassword)
|
|
{
|
|
if (0 == StrLen(sOwnerPassword, 2))
|
|
return;
|
|
|
|
if (sOwnerPassword && sUserPassword && 0 == StrCmp(sOwnerPassword, sUserPassword))
|
|
return;
|
|
|
|
PadOrTrancatePassword(sOwnerPassword, m_pEncrypt->m_anOwnerPassword);
|
|
PadOrTrancatePassword(sUserPassword, m_pEncrypt->m_anUserPassword);
|
|
}
|
|
void CEncryptDict::Prepare(CInfoDict* pInfo, CXref* pXref)
|
|
{
|
|
CreateId(pInfo, pXref);
|
|
|
|
m_pEncrypt->CreateOwnerKey();
|
|
m_pEncrypt->CreateEncryptionKey();
|
|
m_pEncrypt->CreateUserKey();
|
|
|
|
CBinaryObject* pOwnerKey = new CBinaryObject(m_pEncrypt->m_anOwnerKey, PASSWD_LEN);
|
|
if (!pOwnerKey)
|
|
return;
|
|
|
|
Add("O", pOwnerKey);
|
|
|
|
CBinaryObject* pUserKey = new CBinaryObject(m_pEncrypt->m_anUserKey, PASSWD_LEN);
|
|
if (!pUserKey)
|
|
return;
|
|
|
|
Add("U", pUserKey);
|
|
|
|
Add("Filter", "Standard");
|
|
if (encryptmode_R2 == m_pEncrypt->m_eMode)
|
|
{
|
|
Add("V", 1);
|
|
Add("R", 2);
|
|
}
|
|
else if (encryptmode_R3 == m_pEncrypt->m_eMode)
|
|
{
|
|
Add("V", 2);
|
|
Add("R", 3);
|
|
Add("Length", m_pEncrypt->m_unKeyLen * 8);
|
|
}
|
|
Add("P", m_pEncrypt->m_unPermission);
|
|
}
|
|
} |