From 7100d6af6d22ade0914fa5f8275401e37ca9610f Mon Sep 17 00:00:00 2001
From: Egor Tensin <Egor.Tensin@gmail.com>
Date: Mon, 17 Oct 2016 08:05:40 +0300
Subject: utils: code dedupe

---
 utils/encrypt_file.cpp | 43 +++----------------------------------------
 1 file changed, 3 insertions(+), 40 deletions(-)

(limited to 'utils/encrypt_file.cpp')

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)
-- 
cgit v1.2.3