1 files changed, 91 insertions, 0 deletions
diff --git a/aesxx/utils/encrypt_file.cpp b/aesxx/utils/encrypt_file.cpp
new file mode 100644
index 0000000..570bc60
--- /dev/null
+++ b/aesxx/utils/encrypt_file.cpp
@@ -0,0 +1,91 @@
+// Copyright (c) 2015 Egor Tensin <Egor.Tensin@gmail.com>
+// 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/file.hpp"
+
+#include <aesxx/all.hpp>
+
+#include <boost/program_options.hpp>
+
+#include <exception>
+#include <iostream>
+#include <string>
+#include <vector>
+
+namespace
+{
+    void encrypt_file(
+        aes::Box& box,
+        const std::string& plaintext_path,
+        const std::string& ciphertext_path)
+    {
+        const auto plaintext_buf = file::read_file(plaintext_path);
+        const auto ciphertext_buf = box.encrypt_buffer(
+            plaintext_buf.data(),
+            plaintext_buf.size());
+        file::write_file(ciphertext_path, ciphertext_buf);
+    }
+
+    void encrypt_file(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};
+            encrypt_file(box, settings.input_path, settings.output_path);
+        }
+        else
+        {
+            aes::Box box{algorithm, key};
+            encrypt_file(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;
+        }
+
+        encrypt_file(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;
+}