aboutsummaryrefslogblamecommitdiffstatshomepage
path: root/utils/encrypt_file.cpp
blob: 570bc6087803f077c0b3a85fd9117f96d7fcd87f (plain) (tree)
1
2
3
4
5
6
7
8
9



                                                             
 
                              
                           
 
                        


                                    
                    

                   



                 
                      
                      

                                           
     
                                                                   
                                                       

                                  
                                                          

     
                                                   
     

                                                   
 
                          
                                                          
 
                                                 
         
                               
                                                              
 

                                                                         


            
                                         
                                                                         







                               

                                       
           
         
                                       

                                                      
         
                                    

                     

                                     
         

                             
         






                               





                                      
             
 
// 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;
}