examples/example_encrypt.cpp

   1 /******************************************************************************/
   2 /* SlunkCrypt, by LoRd_MuldeR <MuldeR2@GMX.de>                                */
   3 /* This work has been released under the CC0 1.0 Universal license!           */
   4 /******************************************************************************/
   5
   6 //SlunkCrypt API
   7 #include <slunkcrypt.hpp>
   8
   9 //CRT
  10 #include <cstdlib>
  11 #include <iostream>
  12 #include <fstream>
  13
  14 //Example password
  15 static std::string EXAMPLE_PASSWORD("cMRe5E5D)'!2?5]QDlCQ4tBb");
  16
  17 //Const
  18 #define BUFF_SIZE 4096U
  19
  20 static int encrypt_main(int argc, char *argv[])
  21 {
  22         std::cerr << "SlunkCrypt encrypt sample [" << __DATE__ << "]" << std::endl;
  23         std::cerr << "using libSlunkCrypt v" << SLUNKCRYPT_VERSION_MAJOR << '.' << SLUNKCRYPT_VERSION_MINOR << '.' << SLUNKCRYPT_VERSION_PATCH << '\n' << std::endl;
  24
  25         if (argc < 3)
  26         {
  27                 std::cerr << "Usage:\n  encrypt.exe <plaintext.txt> <ciphertext.enc>\n" << std::endl;
  28                 return -1;
  29         }
  30
  31         // -----------------------------------------------------------
  32         // Open input/output files
  33         // -----------------------------------------------------------
  34
  35         std::ifstream file_src(argv[1], std::ios::binary);
  36         if (!file_src.is_open())
  37         {
  38                 std::cerr << "Error: Failed to open input file for reading!" << std::endl;
  39                 return -1;
  40         }
  41
  42         std::ofstream file_dst(argv[2], std::ios::binary);
  43         if (!file_dst.is_open())
  44         {
  45                 std::cerr << "Error: Failed to open output file for writing!" << std::endl;
  46                 return -1;
  47         }
  48
  49         // -----------------------------------------------------------
  50         // Initialize the SlunkCryptEncr instance
  51         // -----------------------------------------------------------
  52
  53         std::cerr << "Initializing key, please wait... " << std::flush;
  54
  55         uint8_t buffer[BUFF_SIZE];
  56         slunkcrypt::Encryptor slunk_encrypt(EXAMPLE_PASSWORD);
  57
  58         std::cerr << "done.\nSlunk-encrypting the file contents, please wait... " << std::flush;
  59
  60         // -----------------------------------------------------------
  61         // Encryption loop
  62         // -----------------------------------------------------------
  63
  64         file_src.exceptions(std::ifstream::badbit);
  65         file_dst.exceptions(std::ifstream::failbit | std::ifstream::badbit);
  66
  67         try
  68         {
  69                 while (file_src.good())
  70                 {
  71                         file_src.read(reinterpret_cast<char*>(buffer), BUFF_SIZE);
  72                         const std::streamsize count = file_src.gcount();
  73                         if (count > 0)
  74                         {
  75                                 if (!slunk_encrypt.inplace(buffer, (size_t)count))
  76                                 {
  77                                         std::cerr << "failed!\n\nError: SlunkCrypt encryption has failed!" << std::endl;
  78                                         return -1;
  79                                 }
  80                                 file_dst.write(reinterpret_cast<char*>(buffer), count);
  81                         }
  82                 }
  83         }
  84         catch (std::ios_base::failure e)
  85         {
  86                 std::cerr << "failed!\n\nI/O Error: \"" << e.code().message() << "\" [Code: " << e.code().value() << "]\n" << std::endl;
  87                 return -1;
  88         }
  89
  90         // -----------------------------------------------------------
  91         // Clean up
  92         // -----------------------------------------------------------
  93
  94         std::cerr << "done.\n\nNonce: " << std::hex << std::uppercase << slunk_encrypt.get_nonce() << '\n' << std::endl;
  95
  96         file_src.close();
  97         file_dst.close();
  98
  99         return 0;
 100 }
 101
 102 int main(int argc, char *argv[])
 103 {
 104         try
 105         {
 106                 return encrypt_main(argc, argv);
 107         }
 108         catch (std::exception e)
 109         {
 110                 std::cerr << "\n\nException: \"" << e.what() << "\"\n" << std::endl;
 111                 return -1;
 112         }
 113 }