2012-01-08 05:08:02 +01:00
|
|
|
/*
|
|
|
|
* This file is a part of Tito - a cryptography library
|
2022-07-03 10:31:40 +02:00
|
|
|
* and is distributed under the 2-Clause BSD licence.
|
2012-01-08 05:08:02 +01:00
|
|
|
* Author: Tomasz Sowa <t.sowa@ttmath.org>
|
|
|
|
*/
|
|
|
|
|
|
|
|
/*
|
2022-08-14 12:53:19 +02:00
|
|
|
* Copyright (c) 2012-2022, Tomasz Sowa
|
2012-01-08 05:08:02 +01:00
|
|
|
* All rights reserved.
|
|
|
|
*
|
|
|
|
* Redistribution and use in source and binary forms, with or without
|
|
|
|
* modification, are permitted provided that the following conditions are met:
|
2022-07-03 10:31:40 +02:00
|
|
|
*
|
|
|
|
* 1. Redistributions of source code must retain the above copyright notice,
|
|
|
|
* this list of conditions and the following disclaimer.
|
|
|
|
*
|
|
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
|
|
* documentation and/or other materials provided with the distribution.
|
2012-01-08 05:08:02 +01:00
|
|
|
*
|
|
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
|
|
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
2022-07-03 10:31:40 +02:00
|
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
2012-01-08 05:08:02 +01:00
|
|
|
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
|
|
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
|
|
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
|
|
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
|
|
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
2022-07-03 10:31:40 +02:00
|
|
|
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
|
|
* POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
*
|
2012-01-08 05:08:02 +01:00
|
|
|
*/
|
|
|
|
|
|
|
|
#ifndef headerfile_tito_crypto
|
|
|
|
#define headerfile_tito_crypto
|
|
|
|
|
|
|
|
#include "aes.h"
|
|
|
|
#include "base64.h"
|
2022-08-14 12:53:19 +02:00
|
|
|
#include "log/log.h"
|
2012-01-08 05:08:02 +01:00
|
|
|
|
|
|
|
|
|
|
|
namespace Tito
|
|
|
|
{
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class Crypto
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
|
2012-03-30 06:24:52 +02:00
|
|
|
Crypto();
|
2012-01-08 05:08:02 +01:00
|
|
|
|
|
|
|
/*
|
2022-08-14 12:53:19 +02:00
|
|
|
setting AES key (key length can be: 16, 24 or 32 bytes)
|
2012-01-08 05:08:02 +01:00
|
|
|
*/
|
2022-08-14 12:53:19 +02:00
|
|
|
bool SetAESKey(unsigned char * aes_key, int key_length);
|
2012-01-08 05:08:02 +01:00
|
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
crypting AES256 and base64 then
|
|
|
|
so the 'out' string is ready to save as a normal text
|
|
|
|
*/
|
2012-03-08 14:54:21 +01:00
|
|
|
void Crypt(const std::wstring & in, std::wstring & out);
|
|
|
|
void Crypt(const std::string & in, std::string & out);
|
|
|
|
void Crypt(const std::vector<char> & in, std::string & out);
|
2012-01-08 05:08:02 +01:00
|
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
decrypting base64 and AES256 then
|
|
|
|
decrypt can return false if the control sums are incorrect
|
|
|
|
in such a case out is empty
|
|
|
|
if 'in' is empty the 'out' will be empty too
|
|
|
|
*/
|
2012-03-08 14:54:21 +01:00
|
|
|
bool Decrypt(const std::wstring & in, std::wstring & out);
|
|
|
|
bool Decrypt(const std::string & in, std::string & out);
|
|
|
|
bool Decrypt(const std::string & in, std::vector<char> & out);
|
2012-01-08 05:08:02 +01:00
|
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
clearing a string or a vector
|
|
|
|
first it puts some characters and then calls clear() method
|
|
|
|
*/
|
|
|
|
template<class BufferType>
|
|
|
|
void Clear(BufferType & str);
|
|
|
|
|
|
|
|
|
2022-08-14 12:53:19 +02:00
|
|
|
/*
|
|
|
|
* pikotools logger
|
|
|
|
* default: null (no logging)
|
|
|
|
*/
|
|
|
|
void set_logger(pt::Log * plog);
|
|
|
|
|
|
|
|
|
|
|
|
|
2012-01-08 05:08:02 +01:00
|
|
|
private:
|
|
|
|
|
|
|
|
// AES cryptography
|
|
|
|
AES aes;
|
|
|
|
|
|
|
|
// base64 encoding/decoding
|
|
|
|
Base64 base64;
|
|
|
|
|
|
|
|
// a table used with AES
|
|
|
|
std::vector<AES::uint8> uint8_tab;
|
|
|
|
|
|
|
|
// AES name (always empty)
|
|
|
|
std::string aes_name;
|
|
|
|
std::string aes_tmp_name;
|
|
|
|
|
|
|
|
std::string utf8_str;
|
|
|
|
std::string base64_str;
|
|
|
|
std::string aes_str;
|
|
|
|
|
2012-03-30 06:24:52 +02:00
|
|
|
// a dummy character used when an input string is empty
|
|
|
|
unsigned char dummy_char;
|
|
|
|
|
2022-08-14 12:53:19 +02:00
|
|
|
// logger, default null
|
|
|
|
pt::Log * plog;
|
|
|
|
|
|
|
|
|
2012-01-08 05:08:02 +01:00
|
|
|
void Clear();
|
|
|
|
|
2012-02-15 08:45:56 +01:00
|
|
|
|
|
|
|
// temporarily fix for Visual Studio
|
2012-03-08 14:54:21 +01:00
|
|
|
// crypto.obj : error LNK2001: unresolved external symbol "void __cdecl Tito::AssignString(char const *,unsigned int,class std::basic_string<wchar_t,struct std::char_traits<wchar_t>,class std::allocator<wchar_t> > &,bool)" (?AssignString@Tito@@YAXPBDIAAV?$basic_string@_WU?$char_traits@_W@std@@V?$allocator@_W@2@@std@@_N@Z)
|
|
|
|
void AssignString(const char * src, size_t len, std::wstring & dst, bool clear = true);
|
|
|
|
void AssignString(const wchar_t * src, size_t len, std::string & dst, bool clear = true);
|
2012-02-15 08:45:56 +01:00
|
|
|
void AssignString(const std::string & src, std::wstring & dst, bool clear = true);
|
|
|
|
void AssignString(const std::wstring & src, std::string & dst, bool clear = true);
|
|
|
|
|
|
|
|
|
2012-01-08 05:08:02 +01:00
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
template<class BufferType>
|
|
|
|
void Crypto::Clear(BufferType & str)
|
|
|
|
{
|
|
|
|
for(size_t i=0 ; i<str.size() ; ++i)
|
|
|
|
str[i] = 0x5c;
|
|
|
|
|
|
|
|
str.clear();
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
} // namespace Tito
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|