amazing-username
87 lines
2.6 KiB
C++
87 lines
2.6 KiB
C++
#include<fstream>
|
|
#include"Encryption.h"
|
|
#include"GeneratePasswordFileName.h"
|
|
#include"KeyRetrieval.h"
|
|
#include"FileNameRetrieval.h"
|
|
#include"FolderStructure.h"
|
|
|
|
Encryption::Encryption(const string& message, const string& keyPath)
|
|
{
|
|
this->message = message;
|
|
setupMap(keyPath);
|
|
configurePasswordFileName();
|
|
//encryptMessage();
|
|
}
|
|
Encryption::Encryption(const Password<>& pass, const Key<>& ky)
|
|
{
|
|
message.assign(pass.retrieveDecryptedMessage());
|
|
setupMap(ky.path());
|
|
encryptMessage(pass);
|
|
}
|
|
|
|
void Encryption::setEncryptedMessage(const string& encryptedMessage) { this->encryptedMessage = encryptedMessage; }
|
|
void Encryption::setMessage(const string& message) { this->message = message; }
|
|
void Encryption::encryptMessage(const Password<>& pass)
|
|
{
|
|
fstream ioEvent{};
|
|
ioEvent.open(pass.passwordPath(), std::ios::out);
|
|
auto p = pass.passwordPath();
|
|
|
|
addMetadata(pass, ioEvent);
|
|
|
|
for (auto indexOfString = 0u; indexOfString!=message.size(); ++indexOfString)
|
|
{
|
|
ioEvent << encryptedCharacters[message.at(indexOfString)];
|
|
encryptedMessage += encryptedCharacters[message.at(indexOfString)];
|
|
}
|
|
this->encryptedMessage = encryptedMessage;
|
|
ioEvent.close();
|
|
}
|
|
void Encryption::addMetadata(const Password<>& pass, fstream& ioEvent)
|
|
{
|
|
auto yearLocal = pass.year;
|
|
auto monthLocal = pass.month;
|
|
auto dayOfMonthLocal = pass.dayOfMonth;
|
|
|
|
ioEvent << yearLocal << '\n' << monthLocal << '\n' << dayOfMonthLocal << '\n';
|
|
}
|
|
|
|
|
|
void Encryption::setupMap(const string& keyPath)
|
|
{
|
|
KeyRetrieval kr{keyPath};
|
|
vector<string> k{kr.keyStructure()};
|
|
vector<int> c{kr.codeCharacterStructure()};
|
|
|
|
for (auto index = 0u; index!=k.size(); ++index)
|
|
encryptedCharacters[c[index]] = k[index];
|
|
}
|
|
void Encryption::configurePasswordFileName()
|
|
{
|
|
for (auto passwordNameDoesNotExist = true; passwordNameDoesNotExist; passwordNameDoesNotExist = repetitive())
|
|
{
|
|
GeneratePasswordFileName gp{};
|
|
passwordFileName.assign(gp.passwordFileNameString());
|
|
}
|
|
string tmpString{passwordFileName};
|
|
passwordFileName.assign(FolderStructure::passwordDirectory);
|
|
passwordFileName.append(tmpString);
|
|
}
|
|
bool Encryption::repetitive()
|
|
{
|
|
passwordFileName.append(".txt");
|
|
FileNameRetrieval fnr;
|
|
fnr.retrievePasswordNames();
|
|
vector<string> createdPasswordNames{fnr.passwordNameContainer()};
|
|
auto count = 0;
|
|
for (auto cPNBegin = createdPasswordNames.begin(); cPNBegin!=createdPasswordNames.end(); ++cPNBegin)
|
|
{
|
|
if (passwordFileName.compare(*cPNBegin)==0) ++count;
|
|
if (count>0) return true;
|
|
}
|
|
return false;
|
|
}
|
|
string Encryption::getEncryptedMessage() const { return encryptedMessage; }
|
|
string Encryption::getMessage() const { return message; }
|
|
map<char, string> Encryption::encryptedCharactersStructure() { return encryptedCharacters; }
|