From 3304264990b96c09b174716ecb8da63d24457ae8 Mon Sep 17 00:00:00 2001 From: Egor Tensin Date: Sat, 21 Dec 2019 14:50:03 +0300 Subject: utils/ -> aesxx/utils/ --- aesxx/utils/decrypt_bmp.cpp | 93 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 93 insertions(+) create mode 100644 aesxx/utils/decrypt_bmp.cpp (limited to 'aesxx/utils/decrypt_bmp.cpp') diff --git a/aesxx/utils/decrypt_bmp.cpp b/aesxx/utils/decrypt_bmp.cpp new file mode 100644 index 0000000..9f9cd52 --- /dev/null +++ b/aesxx/utils/decrypt_bmp.cpp @@ -0,0 +1,93 @@ +// Copyright (c) 2015 Egor Tensin +// This file is part of the "AES tools" project. +// For details, see https://github.com/egor-tensin/aes-tools. +// Distributed under the MIT License. + +#include "file_cmd_parser.hpp" +#include "helpers/bmp.hpp" +#include "helpers/file.hpp" + +#include + +#include + +#include + +#include +#include +#include + +namespace +{ + void decrypt_bmp( + aes::Box& box, + const std::string& ciphertext_path, + const std::string& plaintext_path) + { + bmp::BmpFile bmp{file::read_file(ciphertext_path)}; + bmp.replace_pixels(box.decrypt_buffer( + bmp.get_pixels(), + bmp.get_pixels_size())); + file::write_file(plaintext_path, bmp.get_buffer(), bmp.get_size()); + } + + void decrypt_bmp(const FileSettings& settings) + { + const auto& algorithm = settings.algorithm; + const auto& mode = settings.mode; + + aes::Box::Key key; + aes::Box::parse_key(key, algorithm, settings.key); + + if (aes::mode_requires_init_vector(mode)) + { + aes::Box::Block iv; + aes::Box::parse_block(iv, algorithm, settings.iv); + + aes::Box box{algorithm, key, mode, iv}; + decrypt_bmp(box, settings.input_path, settings.output_path); + } + else + { + aes::Box box{algorithm, key}; + decrypt_bmp(box, settings.input_path, settings.output_path); + } + } +} + +int main(int argc, char** argv) +{ + try + { + FileSettings settings{argv[0]}; + + try + { + settings.parse(argc, argv); + } + catch (const boost::program_options::error& e) + { + settings.usage_error(e); + return 1; + } + + if (settings.exit_with_usage) + { + settings.usage(); + return 0; + } + + decrypt_bmp(settings); + } + catch (const aes::Error& e) + { + std::cerr << e; + return 1; + } + catch (const std::exception& e) + { + std::cerr << e.what() << "\n"; + return 1; + } + return 0; +} -- cgit v1.2.3