Files
core/OfficeCryptReader/Test/Test.cpp
2023-02-28 16:29:40 +03:00

88 lines
2.8 KiB
C++

// Test.cpp : Defines the entry point for the console application.
//
#include "tchar.h"
#include "../source/ECMACryptFile.h"
#include "../../Common/OfficeFileFormatChecker.h"
#include "../../DesktopEditor/common/File.h"
#ifndef _QT
#if defined(_WIN64)
#pragma comment(lib, "../../build/bin/icu/win_64/icuuc.lib")
#elif defined (_WIN32)
#if defined(_DEBUG)
//#pragma comment(lib, "../../build/lib/win_32/DEBUG/graphics.lib")
#pragma comment(lib, "../../build/lib/win_32/DEBUG/kernel.lib")
#pragma comment(lib, "../../build/lib/win_32/DEBUG/UnicodeConverter.lib")
#else
//#pragma comment(lib, "../../build/lib/win_32/graphics.lib")
#pragma comment(lib, "../../build/lib/win_32/kernel.lib")
#endif
#pragma comment(lib, "../../build/bin/icu/win_32/icuuc.lib")
#endif
#endif
int _tmain(int argc, _TCHAR* argv[])
{
#define __CRTDBG_MAP_ALLOC
#include <crtdbg.h>
#define DEBUG_NEW new(_NORMAL_BLOCK, __FILE__, __LINE__)
#define new DEBUG_NEW
std::wstring password = L"123";
ECMACryptFile crypt_file;
bool result = false, bDataIntegrity = false;
std::wstring srcFileName = L"D:\\tests\\__67\\test.docx";
std::wstring dstFileName = L"D:\\tests\\__67\\test.docx-crypt.docx";
std::wstring dstFileName2 = dstFileName + L".oox";
//std::wstring dstFileName = srcFileName + L".oox";
//std::wstring dstFileName2 = dstFileName + L"-mycrypt.docx";
//std::wstring srcFileName = L"D:\\test\\_crypted\\test.docx";
//std::wstring dstFileName3 = srcFileName + L"-mycrypt1.docx";
//std::wstring srcFileName1 = L"D:\\test\\_crypted\\test-111.docx-ms.docx";
//std::wstring dstFileName1 = srcFileName1 + L".oox";
//result = crypt_file.DecryptOfficeFile(srcFileName1, dstFileName1, password, bDataIntegrity);
//result = crypt_file.EncryptOfficeFile(srcFileName, dstFileName, password, L"123456789");
////result = crypt_file.DecryptOfficeFile(dstFileName, dstFileName2, password, bDataIntegrity);
//
//std::wstring addit_name = L"11111111111111111111111111111";
//for (size_t i = 0; i < 100; ++i)
//{
// std::string addit_info = crypt_file.ReadAdditional(dstFileName, addit_name);
// std::wstring temp = NSFile::CFileBinary::CreateTempFileWithUniqueName(L"", L"asd");
// //
// addit_info += std::string(temp.begin(), temp.end());
// crypt_file.WriteAdditional(dstFileName, addit_name, addit_info);
//}
std::wstring sFile = srcFileName;
std::wstring sPassword = L"9ddb2443-b29a-1308-a64e-12988c24ded6";
std::wstring sDocinfo = L"9ddb2443-b29a-1308-a64e-12988c24ded6";
bool isCrypt = false;
COfficeFileFormatChecker oChecker;
oChecker.isOfficeFile(sFile);
if (AVS_OFFICESTUDIO_FILE_OTHER_MS_OFFCRYPTO != oChecker.nFileType)
{
ECMACryptFile file;
if (file.EncryptOfficeFile(sFile, sFile, sPassword))
{
file.WriteAdditional(sFile, L"DocumentID", U_TO_UTF8(sDocinfo));
isCrypt = true;
}
}
_CrtDumpMemoryLeaks();
return 0;
}