From 3d992e76ef7991cdff413ae58c07066f1a19a5d3 Mon Sep 17 00:00:00 2001 From: Egor Tensin Date: Mon, 3 Aug 2015 14:23:27 +0300 Subject: rename the utilities --- utils/aes_encrypt_bmp.cpp | 204 ---------------------------------------------- 1 file changed, 204 deletions(-) delete mode 100644 utils/aes_encrypt_bmp.cpp (limited to 'utils/aes_encrypt_bmp.cpp') diff --git a/utils/aes_encrypt_bmp.cpp b/utils/aes_encrypt_bmp.cpp deleted file mode 100644 index 2615889..0000000 --- a/utils/aes_encrypt_bmp.cpp +++ /dev/null @@ -1,204 +0,0 @@ -/** - * \file - * \author Egor Tensin - * \date 2015 - * \copyright This file is licensed under the terms of the MIT License. - * See LICENSE.txt for details. - */ - -#include "aes_file_common.hpp" - -#include - -#include - -#include - -#include -#include - -#include -#include -#include -#include -#include -#include -#include - -#include - -namespace -{ - std::ifstream::pos_type get_file_size(const std::string& path) - { - std::ifstream ifs; - ifs.exceptions(std::ifstream::badbit | std::ifstream::failbit); - ifs.open(path, std::ifstream::binary | std::ifstream::ate); - return ifs.tellg(); - } - - std::vector read_file(const std::string& path) - { - const auto size = static_cast(get_file_size(path)); - - std::ifstream ifs; - ifs.exceptions(std::ifstream::badbit | std::ifstream::failbit); - ifs.open(path, std::ifstream::binary); - - std::vector src_buf; - src_buf.reserve(size); - src_buf.assign(std::istreambuf_iterator(ifs), - std::istreambuf_iterator()); - return src_buf; - } - - void write_file(const std::string& path, const std::vector& src) - { - std::ofstream ofs; - ofs.exceptions(std::ofstream::badbit | std::ofstream::failbit); - ofs.open(path, std::ofstream::binary); - ofs.write(src.data(), src.size()); - } - - template - bool encrypt_bmp_with_algorithm( - const AesNI_BoxAlgorithmParams& algorithm_params, - aesni::Mode mode, - std::deque& args) - { - AesNI_BoxBlock iv; - AesNI_BoxBlock* iv_ptr = nullptr; - - if (aesni::mode_requires_initialization_vector(mode)) - { - if (args.empty()) - return false; - - aesni::from_string(iv.aes_block, args.front()); - iv_ptr = &iv; - args.pop_front(); - } - - if (args.size() != 2) - return false; - - const auto src_path = args[0]; - const auto dest_path = args[1]; - - const auto src_buf = read_file(src_path); - - const auto bmp_header = reinterpret_cast(src_buf.data()); - - const auto header_size = bmp_header->bfOffBits; - const auto pixels = src_buf.data() + header_size; - const auto pixels_size = src_buf.size() - header_size; - - AesNI_Box box; - - aesni_box_init( - &box, - algorithm, - &algorithm_params, - mode, - iv_ptr, - aesni::ErrorDetailsThrowsInDestructor()); - - std::size_t cipherpixels_size; - - aesni_box_encrypt_buffer( - &box, - pixels, - pixels_size, - nullptr, - &cipherpixels_size, - aesni::ErrorDetailsThrowsInDestructor()); - - std::vector dest_buf; - dest_buf.resize(header_size + cipherpixels_size); - std::memcpy(dest_buf.data(), src_buf.data(), header_size); - - aesni_box_encrypt_buffer( - &box, - pixels, - pixels_size, - dest_buf.data() + header_size, - &cipherpixels_size, - aesni::ErrorDetailsThrowsInDestructor()); - - write_file(dest_path, dest_buf); - - return true; - } - - bool encrypt_bmp( - aesni::Algorithm algorithm, - aesni::Mode mode, - std::deque& args) - { - if (args.empty()) - return false; - - AesNI_BoxAlgorithmParams algorithm_params; - - switch (algorithm) - { - case AESNI_AES128: - aesni::from_string( - algorithm_params.aes128_key, args.front()); - args.pop_front(); - return encrypt_bmp_with_algorithm( - algorithm_params, mode, args); - - case AESNI_AES192: - aesni::from_string( - algorithm_params.aes192_key, args.front()); - args.pop_front(); - return encrypt_bmp_with_algorithm( - algorithm_params, mode, args); - - case AESNI_AES256: - aesni::from_string( - algorithm_params.aes256_key, args.front()); - args.pop_front(); - return encrypt_bmp_with_algorithm( - algorithm_params, mode, args); - - default: - return false; - } - } -} - -int main(int argc, char** argv) -{ - try - { - CommandLineParser cmd_parser("aes_encrypt_bmp.exe"); - - if (!cmd_parser.parse_options(argc, argv)) - return 0; - - if (!encrypt_bmp(cmd_parser.get_algorithm(), cmd_parser.get_mode(), cmd_parser.get_args())) - { - cmd_parser.print_usage(); - return 1; - } - - return 0; - } - catch (const boost::program_options::error& e) - { - std::cerr << "Usage error: " << e.what() << "\n"; - return 1; - } - catch (const aesni::Error& e) - { - std::cerr << e; - return 1; - } - catch (const std::exception& e) - { - std::cerr << e.what() << "\n"; - return 1; - } -} -- cgit v1.2.3