AES.cpp 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758
  1. /*
  2. This file is part of cpp-ethereum.
  3. cpp-ethereum is free software: you can redistribute it and/or modify
  4. it under the terms of the GNU General Public License as published by
  5. the Free Software Foundation, either version 3 of the License, or
  6. (at your option) any later version.
  7. cpp-ethereum is distributed in the hope that it will be useful,
  8. but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. GNU General Public License for more details.
  11. You should have received a copy of the GNU General Public License
  12. along with cpp-ethereum. If not, see <http://www.gnu.org/licenses/>.
  13. */
  14. /** @file AES.cpp
  15. * @author Alex Leverington <nessence@gmail.com>
  16. * @date 2014
  17. */
  18. #include "AES.h"
  19. #include <libdevcore/Common.h>
  20. #include "CryptoPP.h"
  21. using namespace std;
  22. using namespace dev;
  23. using namespace dev::crypto;
  24. using namespace CryptoPP;
  25. bytes dev::aesDecrypt(bytesConstRef _ivCipher, std::string const& _password, unsigned _rounds, bytesConstRef _salt)
  26. {
  27. bytes pw = asBytes(_password);
  28. if (!_salt.size())
  29. _salt = &pw;
  30. bytes target(64);
  31. CryptoPP::PKCS5_PBKDF2_HMAC<CryptoPP::SHA256>().DeriveKey(target.data(), target.size(), 0, pw.data(), pw.size(), _salt.data(), _salt.size(), _rounds);
  32. try
  33. {
  34. CryptoPP::AES::Decryption aesDecryption(target.data(), 16);
  35. auto cipher = _ivCipher.cropped(16);
  36. auto iv = _ivCipher.cropped(0, 16);
  37. CryptoPP::CBC_Mode_ExternalCipher::Decryption cbcDecryption(aesDecryption, iv.data());
  38. std::string decrypted;
  39. CryptoPP::StreamTransformationFilter stfDecryptor(cbcDecryption, new CryptoPP::StringSink(decrypted));
  40. stfDecryptor.Put(cipher.data(), cipher.size());
  41. stfDecryptor.MessageEnd();
  42. return asBytes(decrypted);
  43. }
  44. catch (exception const& e)
  45. {
  46. cerr << e.what() << endl;
  47. return bytes();
  48. }
  49. }