pengjunjie/core
1
0
Fork 1
mirror of https://github.com/ONLYOFFICE/core.git synced 2026-02-10 18:05:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
Elena Subbotina
2024-07-29 21:58:34 +03:00
parent 0eb8c93cf9
commit af01306997
3 changed files with 68 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
OfficeCryptReader/source/CryptTransform.cpp
View File

@ -305,8 +305,15 @@ _buf GenerateOdfKey(_buf & pSalt, _buf & pPassword, int keySize, int spin, CRYPT
{
_buf pKey (keySize);
_buf empty (NULL, 0, false);
_buf pPassword_hash;
_buf pPassword_hash = HashAppend(pPassword, empty, algorithm);
if (algorithm == CRYPT_METHOD::None)
{
pPassword_hash = pPassword; }
else
{
pPassword_hash = HashAppend(pPassword, empty, algorithm);
}
PKCS5_PBKDF2_HMAC<SHA1> pbkdf;
@ -819,36 +826,40 @@ void ECMADecryptor::Decrypt(unsigned char* data_inp, int size, unsigned char*& d
}
}
//-----------------------------------------------------------------------------------------------------------
void odfWriteProtect::SetPassword (const std::wstring &password_)
void ODFWriteProtect::SetPassword (const std::wstring &password_)
{
password = password_;
}
void odfWriteProtect::SetProtectKey (const std::string &key)
{
protect_key = key;
}
void odfWriteProtect::SetProtectAlgorithm (const CRYPT_METHOD::_hashAlgorithm &alg)
{
hash = alg;
}
void odfWriteProtect::Generate()
{
_buf pPassword (password);
_buf empty (NULL, 0, false);
_buf pHashTest = HashAppend(empty, pPassword, hash);
protect_key = std::string((char*)pHashTest.ptr, pHashTest.size);
}
bool odfWriteProtect::Verify()
void ODFWriteProtect::SetCryptData(_odfWriteProtectData &_data)
{
_buf pPassword (password);
_buf empty (NULL, 0, false);
_buf pHash (protect_key);
data = _data;
}
void ODFWriteProtect::GetCryptData(_odfWriteProtectData &_data)
{
_data = data;
}
_buf pHashTest = HashAppend(empty, pPassword, hash);
return (pHashTest == pHash);
void ODFWriteProtect::Generate()
{
std::string passw_ansi = std::string(password.begin(), password.end());
_buf pPassword (passw_ansi);
_buf pSalt (data.saltValue);
_buf pHash = GenerateOdfKey(pSalt, pPassword, 16, data.spinCount, CRYPT_METHOD::None);
data.hashValue = std::string((char*)pHash.ptr, pHash.size);
}
bool ODFWriteProtect::Verify()
{
std::string passw_ansi = std::string(password.begin(), password.end());
_buf pPassword (passw_ansi);
_buf pSalt (data.saltValue);
_buf pHash (data.hashValue);
_buf pHashTest = GenerateOdfKey(pSalt, pPassword, 16, data.spinCount, CRYPT_METHOD::None);
return (pHashTest == pHash);
}
//----------------------------------------------------------------------------------------------------------
void ECMAWriteProtect::SetPassword (const std::wstring &password_)

22
OfficeCryptReader/source/CryptTransform.h
View File

@ -40,12 +40,14 @@ namespace CRYPT_METHOD
{
enum _hashAlgorithm
{
None,
MD5,
SHA1,
SHA224,
SHA256,
SHA384,
SHA512
SHA512,
PBKDF2
};
enum _cipherAlgorithm
@ -81,6 +83,7 @@ struct _ecmaWriteProtectData
std::string saltValue;
std::string hashValue;
};
typedef _ecmaWriteProtectData _odfWriteProtectData;
struct _ecmaCryptData
{
//default ms2010
@ -162,21 +165,22 @@ private:
};
//---------------------------------------------------------------------------------------------------
class odfWriteProtect
class ODFWriteProtect
{
public:
odfWriteProtect() : hash(CRYPT_METHOD::SHA1) {}
ODFWriteProtect(){}
virtual ~ODFWriteProtect(){}
void SetPassword (const std::wstring & password);
void SetProtectKey (const std::string & protect_key);
void SetProtectAlgorithm (const CRYPT_METHOD::_hashAlgorithm & hash);
void SetPassword (const std::wstring &password);
void SetCryptData(_odfWriteProtectData &data);
void GetCryptData(_odfWriteProtectData &data);
void Generate();
bool Verify();
private:
std::wstring password;
CRYPT_METHOD::_hashAlgorithm hash;
std::string protect_key;
std::wstring password;
_odfWriteProtectData data;
};
//---------------------------------------------------------------------------------------------------
class ECMAEncryptor

19
OfficeCryptReader/source/ECMACryptFile.cpp
View File

@ -1093,3 +1093,22 @@ bool ECMACryptFile::WriteAdditional(const std::wstring &file_name, const std::ws
return true;
}
bool TestOdfProtect(const std::wstring &passward, const std::string &salt, const std::string &hash, int count)
{
CRYPT::ODFWriteProtect test;
CRYPT::_odfWriteProtectData ver1;
ver1.hashAlgorithm = CRYPT_METHOD::PBKDF2;
ver1.hashValue = DecodeBase64(hash);
ver1.saltValue = DecodeBase64(salt);
ver1.spinCount = count;
test.SetCryptData(ver1);
test.SetPassword(passward);
bool result_ver1 = test.Verify();
return result_ver1;
}