diff options
Diffstat (limited to 'Untracked/Doxygen/CryptClass/AESCrypt.cpp')
| -rw-r--r-- | Untracked/Doxygen/CryptClass/AESCrypt.cpp | 110 |
1 files changed, 110 insertions, 0 deletions
diff --git a/Untracked/Doxygen/CryptClass/AESCrypt.cpp b/Untracked/Doxygen/CryptClass/AESCrypt.cpp new file mode 100644 index 0000000..002cb60 --- /dev/null +++ b/Untracked/Doxygen/CryptClass/AESCrypt.cpp @@ -0,0 +1,110 @@ +/** + * @file AESCrypt.cpp + * @brief AESCrypt class definitions + * @author manzerbredes + * @date 8 Mars 2015 + * + * Contain all definitions of AESCrypt class. + * + */ + +#include "AESCrypt.hpp" + +//Constructor +AESCrypt::AESCrypt(){ + this->hash=HASHCrypt(); //Init hash attribute +} + +//Destructor +AESCrypt::~AESCrypt(){ +} + + +//Encrypt string +std::string AESCrypt::encrypt(std::string key, std::string data){ + + //Generate SHA-256 + byte digest[32]; + hash.getSHA_256(key, digest, (int)sizeof(digest)); + + + + + //Add padding for AES + int pad=0; + int tmpL=data.length(); + while(tmpL % 128 != 0){ + tmpL++; + pad++; + + } + std::cout <<pad; + while(data.length() % 128 != 0){ + data+=pad; + } + + + //Contain data encrypted + std::string cipher; + + //Use try, catch to be ensure no problems happening + try{ + //Create encoder to encrypt data + CryptoPP::ECB_Mode<CryptoPP::AES>::Encryption encoder; + encoder.SetKey( digest, sizeof(digest) ); + + //Encrypt data with StreamTransformationFilter with NO PADDING + CryptoPP::StringSource ss1(data, true, + new CryptoPP::StreamTransformationFilter( encoder, + new CryptoPP::StringSink( cipher ), + CryptoPP::StreamTransformationFilter::PKCS_PADDING + + ) + ); + } + catch( CryptoPP::Exception& e ) + { + std::cerr << e.what() << std::endl; + exit(EXIT_FAILURE); + } + + //return encrypted data + return cipher; + +} + +//Decrypt string +std::string AESCrypt::decrypt(std::string key, std::string data){ + + + byte digest[32]; + hash.getSHA_256(key, digest, (int)sizeof(digest)); + + + //Contain data decrypted + std::string cipher; + + //Use try, catch to be ensure no problems happening + try { + + //Create decoder to encrypt data + CryptoPP::ECB_Mode< CryptoPP::AES >::Decryption decoder; + decoder.SetKey( digest, sizeof(digest) ); + + //Decrypt data with StreamTransformationFilter with NO PADDING + CryptoPP::StringSource ss3( data, true, + new CryptoPP::StreamTransformationFilter( decoder, + new CryptoPP::StringSink( cipher ), + CryptoPP::StreamTransformationFilter::PKCS_PADDING + ) + ); + } + catch( CryptoPP::Exception& e ) + { + std::cerr << e.what() << std::endl; + exit(1); + } + + //return decrypted data + return cipher; +} |
