path: root/Cryptor.h

#ifndef CRYPTOR_H
#define CRYPTOR_H

#include <iostream>
#include <string>
#include <fstream>

#ifdef WIN32
#include <windows.h>
#else
#include <termios.h>
#include <unistd.h>
#endif // WIN32

#include "cryptopp562/osrng.h"
#include "cryptopp562/cryptlib.h"
#include "cryptopp562/hex.h"
#include "cryptopp562/filters.h"
#include "cryptopp562/aes.h"
#include "cryptopp562/ccm.h"

#include "Socket.h"

#define DEF_PASSWD_LENGTH 50

#define KEYCHAIN_FILE ".compasskeychain"
#define PASSWORD_PROMPT "ComPASS Password: "
#define PASSWORD_CONF "Confirm Password: "
#define PASSWORD_ERROR "Passwords do not match..."

struct PasswordSpec {
	PasswordSpec() { // Default Values
		ml = DEF_PASSWD_LENGTH;
		ns = nc = nn = false;
	}

	int ml; // max length
	bool ns; // no special chars
	bool nc; // no caps
	bool nn; // no numeric chars
};

class Cryptor {
public:
	static void encryptAndSave(std::string remoteHost, std::string port, std::string directory, std::string payload);
	static std::string loadAndDecrypt(std::string remoteHost, std::string port, std::string directory);
	static std::string createRandomPassword(PasswordSpec spec);
	static void rekey();

private:
	static bool haveKey;
	static unsigned char key[CryptoPP::AES::DEFAULT_KEYLENGTH];

	static void sha256(std::string str);
	static std::string readPassword(bool confirm);
	static std::string readPassword();
	static void assembleKey(bool confirm);
	static void assembleKey();
	static std::string promptPassword(bool confirm);
};

#endif // CRYPTOR_H