131 lines
3.1 KiB
C
131 lines
3.1 KiB
C
|
/*
|
||
|
* This file is a part of Tito - a cryptography library
|
||
|
* and is distributed under the (new) BSD licence.
|
||
|
* Author: Tomasz Sowa <t.sowa@ttmath.org>
|
||
|
*/
|
||
|
|
||
|
/*
|
||
|
* Copyright (c) 2012, Tomasz Sowa
|
||
|
* All rights reserved.
|
||
|
*
|
||
|
* Redistribution and use in source and binary forms, with or without
|
||
|
* modification, are permitted provided that the following conditions are met:
|
||
|
*
|
||
|
* * Redistributions of source code must retain the above copyright notice,
|
||
|
* this list of conditions and the following disclaimer.
|
||
|
*
|
||
|
* * 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.
|
||
|
*
|
||
|
* * Neither the name Tomasz Sowa nor the names of contributors to this
|
||
|
* project may be used to endorse or promote products derived
|
||
|
* from this software without specific prior written permission.
|
||
|
*
|
||
|
* 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
|
||
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
||
|
* 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)
|
||
|
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
|
||
|
* THE POSSIBILITY OF SUCH DAMAGE.
|
||
|
*/
|
||
|
|
||
|
#ifndef headerfile_tito_crypto
|
||
|
#define headerfile_tito_crypto
|
||
|
|
||
|
#include "aes.h"
|
||
|
#include "base64.h"
|
||
|
|
||
|
|
||
|
namespace Tito
|
||
|
{
|
||
|
|
||
|
|
||
|
|
||
|
class Crypto
|
||
|
{
|
||
|
public:
|
||
|
|
||
|
|
||
|
/*
|
||
|
setting AES256 key
|
||
|
the table should consists of 32 bytes
|
||
|
*/
|
||
|
bool SetAESKey(unsigned char aes_key[32]);
|
||
|
|
||
|
|
||
|
/*
|
||
|
crypting AES256 and base64 then
|
||
|
so the 'out' string is ready to save as a normal text
|
||
|
*/
|
||
|
void Crypt(const std::wstring & in, std::wstring & out);
|
||
|
void Crypt(const std::vector<char> & in, std::string & out);
|
||
|
|
||
|
|
||
|
/*
|
||
|
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
|
||
|
*/
|
||
|
bool Decrypt(const std::wstring & in, std::wstring & out);
|
||
|
bool Decrypt(const std::string & in, std::vector<char> & out);
|
||
|
|
||
|
|
||
|
/*
|
||
|
clearing a string or a vector
|
||
|
first it puts some characters and then calls clear() method
|
||
|
*/
|
||
|
template<class BufferType>
|
||
|
void Clear(BufferType & str);
|
||
|
|
||
|
|
||
|
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;
|
||
|
|
||
|
void Clear();
|
||
|
|
||
|
};
|
||
|
|
||
|
|
||
|
|
||
|
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
|
||
|
|
||
|
|