Files
core/OfficeCryptReader/source/CryptTransform.h
2017-03-29 12:35:09 +03:00

146 lines
3.8 KiB
C++

/*
* (c) Copyright Ascensio System SIA 2010-2017
*
* 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 Lubanas st. 125a-25, Riga, Latvia,
* EU, LV-1021.
*
* 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
*
*/
#pragma once
#include <string>
#include <vector>
#include <boost/shared_ptr.hpp>
namespace CRYPT_METHOD
{
enum _hashAlgorithm
{
SHA1,
SHA224,
SHA256,
SHA384,
SHA512
};
enum _cipherAlgorithm
{
XOR,
RC4,
AES_CBC,
AES_CFB,
AES_ECB
};
}
namespace CRYPT
{
class Decryptor
{
public:
virtual void Decrypt(char* data, const size_t size, const unsigned long stream_pos) = 0;
virtual bool SetPassword(std::wstring password) = 0;
virtual bool IsVerify() = 0;
};
typedef boost::shared_ptr<Decryptor> DecryptorPtr;
struct _ecmaCryptData
{
//default ms2010
_ecmaCryptData() : cipherAlgorithm(CRYPT_METHOD::AES_CBC), hashAlgorithm(CRYPT_METHOD::SHA1), spinCount(100000),
keySize(0x10), hashSize(0x14), blockSize(0x10), saltSize(0x10), bAgile(true)
//default ms2013/ms2016
//_cryptData(): cipherAlgorithm(CRYPT_METHOD::AES_CBC), hashAlgorithm(CRYPT_METHOD::SHA256), spinCount(100000),
// keySize(0x20), hashSize(0x40), blockSize(0x10), saltSize(0x10), bAgile(true)
{
}
CRYPT_METHOD::_cipherAlgorithm cipherAlgorithm;
CRYPT_METHOD::_hashAlgorithm hashAlgorithm;
int spinCount;
int keySize;
int hashSize;
int blockSize;
int saltSize;
std::string dataSaltValue;
std::string saltValue;
std::string encryptedKeyValue;
std::string encryptedVerifierInput;
std::string encryptedVerifierValue;
std::string encryptedHmacKey;
std::string encryptedHmacValue;
bool bAgile;
//..........
};
class ECMAEncryptor
{
public:
ECMAEncryptor();
virtual ~ECMAEncryptor(){}
void SetPassword (std::wstring password);
void SetCryptData(_ecmaCryptData &data);
void GetCryptData(_ecmaCryptData &data);
int Encrypt (unsigned char* data, int size, unsigned char*& data_out);
void UpdateDataIntegrity(unsigned char* data, int size);
private:
std::wstring password;
_ecmaCryptData cryptData;
};
class ECMADecryptor : public Decryptor
{
public:
ECMADecryptor();
virtual ~ECMADecryptor(){}
void Decrypt (unsigned char* data, int size, unsigned char*& data_out);
virtual void Decrypt (char* data, const size_t size, const unsigned long stream_pos);
virtual bool SetPassword (std::wstring password);
virtual bool IsVerify();
void SetCryptData(_ecmaCryptData &data);
private:
bool IsDataIntegrity(unsigned char* data, int size);
std::wstring password;
_ecmaCryptData cryptData;
bool bVerify;
};
}