Files
Krypto/include/Utils.hpp
2021-03-05 01:41:13 +05:30

66 lines
1.5 KiB
C++

/**
* /mnt/z/my_git/RCRS-CSD-327/include/Utils.hpp
* @file Utils.hpp
* @author Avinal Kumar
* @since March 04, 2021
*
* Utilities
*/
// include blowfish library
#include <blowfish/blowfish.hpp>
// include standard headers
#include <filesystem>
#include <fstream>
#include <iostream>
#include <string>
#include <vector>
// defines
namespace fs = std::filesystem;
#if !defined(UTILS_HPP)
#define UTILS_HPP
namespace MCPS {
struct filestat {
std::string filename;
fs::path filepath;
DWORD filesize;
fs::file_type filetype;
std::time_t m_time;
bool is_regular;
filestat(std::string const &file_name);
void print();
private:
template <typename TP> std::time_t to_time_t(TP tp) {
using namespace std::chrono;
auto sctp = time_point_cast<system_clock::duration>(tp - TP::clock::now() +
system_clock::now());
return system_clock::to_time_t(sctp);
}
};
class fileop {
private:
std::unordered_map<std::string, filestat> _files;
public:
DWORD read_file(std::string const &infile, std::vector<char> &input_buf,
filestat const &stat);
DWORD write_file(std::string const &outfile, std::vector<char> &output_buf,
DWORD out_size);
std::unordered_map<std::string, filestat> scan_current_directory();
DWORD pad_input(std::vector<char> &input_buf, DWORD in_size);
DWORD attch_key(std::vector<char> &input_buf, std::string const &key,
DWORD in_size);
};
} // namespace MCPS
#endif // UTILS_HPP