diff options
Diffstat (limited to 'utils/encrypt_file.cpp')
-rw-r--r-- | utils/encrypt_file.cpp | 43 |
1 files changed, 3 insertions, 40 deletions
diff --git a/utils/encrypt_file.cpp b/utils/encrypt_file.cpp index 8741f54..721b735 100644 --- a/utils/encrypt_file.cpp +++ b/utils/encrypt_file.cpp @@ -4,65 +4,28 @@ // Distributed under the MIT License. #include "file_cmd_parser.hpp" +#include "helpers/file.hpp" #include <aesxx/all.hpp> #include <boost/program_options.hpp> -#include <cstdlib> - #include <exception> -#include <fstream> #include <iostream> -#include <iterator> #include <string> #include <vector> 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<char> read_file(const std::string& path) - { - const auto size = static_cast<std::size_t>(get_file_size(path)); - - std::ifstream ifs; - ifs.exceptions(std::ifstream::badbit | std::ifstream::failbit); - ifs.open(path, std::ifstream::binary); - - std::vector<char> plaintext_buf; - plaintext_buf.reserve(size); - plaintext_buf.assign( - std::istreambuf_iterator<char>(ifs), - std::istreambuf_iterator<char>()); - return plaintext_buf; - } - - void write_file( - const std::string& path, - const std::vector<unsigned char>& src) - { - std::ofstream ofs; - ofs.exceptions(std::ofstream::badbit | std::ofstream::failbit); - ofs.open(path, std::ofstream::binary); - ofs.write(reinterpret_cast<const char*>(src.data()), src.size()); - } - void encrypt_file( aes::Box& box, const std::string& plaintext_path, const std::string& ciphertext_path) { - const auto plaintext_buf = read_file(plaintext_path); + const auto plaintext_buf = file::read_file(plaintext_path); const auto ciphertext_buf = box.encrypt_buffer( plaintext_buf.data(), plaintext_buf.size()); - write_file(ciphertext_path, ciphertext_buf); + file::write_file(ciphertext_path, ciphertext_buf); } void encrypt_file(const Settings& settings) |