mirror of
https://github.com/avinal/Krypto.git
synced 2026-01-09 22:58:36 +05:30
util implementation added
This commit is contained in:
103
src/Utils.cpp
Normal file
103
src/Utils.cpp
Normal file
@@ -0,0 +1,103 @@
|
||||
/**
|
||||
* /mnt/z/my_git/RCRS-CSD-327/src/Utils.cpp
|
||||
* @file Utils.cpp
|
||||
* @author Avinal Kumar
|
||||
* @since March 04, 2021
|
||||
*
|
||||
* File utilities
|
||||
*/
|
||||
|
||||
#include "Utils.hpp"
|
||||
|
||||
namespace MCPS {
|
||||
|
||||
/**
|
||||
* Generates file stats for a given file name.
|
||||
*
|
||||
* @param file_name name of the file
|
||||
*/
|
||||
filestat::filestat(std::string const &file_name) : filename(file_name) {
|
||||
filepath = fs::absolute(fs::current_path() / file_name);
|
||||
m_time = to_time_t(fs::last_write_time(filepath));
|
||||
filetype = fs::status(filepath).type();
|
||||
filesize = fs::file_size(filepath);
|
||||
is_regular = fs::is_regular_file(filepath);
|
||||
}
|
||||
|
||||
void filestat::print() {
|
||||
std::cout << "File name: " << filename << std::endl
|
||||
<< "File path: " << filepath << std::endl
|
||||
<< "File Size: " << filesize << " bytes." << std::endl
|
||||
<< "Last modification time: "
|
||||
<< std::asctime(std::localtime(&m_time)) << std::endl;
|
||||
}
|
||||
|
||||
DWORD fileop::pad_input(std::vector<char> &input_buf, DWORD in_size) {
|
||||
int dv = sizeof(WORD) * 2, rem;
|
||||
rem = ((in_size >= dv) ? (((in_size / dv) + 1) * dv - in_size)
|
||||
: (dv - in_size));
|
||||
if (rem < dv) {
|
||||
input_buf.resize(rem + in_size + 1);
|
||||
in_size += rem;
|
||||
}
|
||||
return in_size;
|
||||
}
|
||||
|
||||
DWORD fileop::attch_key(std::vector<char> &input_buf, std::string const &key,
|
||||
DWORD in_size) {
|
||||
input_buf.resize(in_size + MAXKEYBYTES + 3);
|
||||
std::for_each(key.begin(), key.end(), [&](char k) {
|
||||
input_buf[in_size] = k;
|
||||
in_size++;
|
||||
});
|
||||
return in_size;
|
||||
}
|
||||
|
||||
DWORD fileop::read_file(std::string const &infile, std::vector<char> &input_buf,
|
||||
filestat const &stat) {
|
||||
WORD in_size = 0, readsize = stat.filesize + 1;
|
||||
std::ifstream ifile(infile, std::ios::in | std::ios::binary);
|
||||
if (!ifile) {
|
||||
std::cerr << "Unable to open file: " << infile << std::endl;
|
||||
return -1;
|
||||
}
|
||||
|
||||
input_buf.clear();
|
||||
input_buf.reserve(readsize + in_size + 1);
|
||||
if (ifile.is_open()) {
|
||||
input_buf.insert(input_buf.begin(), std::istreambuf_iterator<char>(ifile),
|
||||
std::istreambuf_iterator<char>());
|
||||
}
|
||||
in_size = input_buf.size();
|
||||
ifile.close();
|
||||
return in_size;
|
||||
}
|
||||
|
||||
DWORD fileop::write_file(std::string const &outfile,
|
||||
std::vector<char> &output_buf, DWORD out_size) {
|
||||
std::ofstream ofile(outfile,
|
||||
std::ios::out | std::ios::binary | std::ios::trunc);
|
||||
if (!ofile) {
|
||||
std::cerr << "Unable to create file: " << outfile << std::endl;
|
||||
}
|
||||
if (ofile.is_open()) {
|
||||
ofile.write(reinterpret_cast<char *>(&output_buf[0]),
|
||||
out_size * sizeof(char));
|
||||
}
|
||||
ofile.close();
|
||||
return 0;
|
||||
}
|
||||
|
||||
std::unordered_map<std::string, filestat> fileop::scan_current_directory() {
|
||||
std::string ext(".cpp");
|
||||
fs::path current = fs::current_path();
|
||||
for (const auto &file : fs::directory_iterator(current)) {
|
||||
if (fs::is_regular_file(file)) {
|
||||
std::string _filename(file.path().filename());
|
||||
filestat fstat(_filename);
|
||||
_files.insert({_filename, fstat});
|
||||
}
|
||||
}
|
||||
return _files;
|
||||
}
|
||||
} // namespace MCPS
|
||||
Reference in New Issue
Block a user