forgetIt/IOFileClass/FileManIOFile.hpp

/**
 * @file FileManIOFile.hpp
 * @brief FileManIOFile class definitions
 * @author manzerbredes
 * @date 9 Mars 2015
 *
 * Contain all definitions of FileManIOFile class.
 *
 */


#ifndef __FileManIOFile__
#define __FileManIOFile__


//----- std -----
#include <iostream>
#include <string>
#include <fstream>


//----- class -----
#include "HASHCrypt.hpp"
#include "AESCrypt.hpp"




/**
 * @class FileManIOFile FileManIOFile.hpp "/CryptClass/FileManIOFile.hpp"
 * @brief Class for quick open and close encrypted files.
 * @author manzerbredes
 *
 * -----File organisation-----
 *
 * 16 first bytes : md5 of decrypted data
 * Rest of the file : data encrypted (ASE for now)
 *
 * \bug Need check if file exist and can be opened
 */
class FileManIOFile {

    public:

        //Constructor
        FileManIOFile(std::string filename);

        //Destructor
        ~FileManIOFile();


        /**
        * @brief Read encrypted file.
        *
        * @param key : key to encrypt data
        *
        * Read data from "filename" attribute.
        * If file fully decrypted, readable var switch to true,
        * and key saved in key attribute (before been crypted with SHA-256 algorithm).
        *
        */
        void read(std::string key);


        /**
        * @brief Write data in encrypted file.
        *
        * @param key : key to encrypt data
        * @param data : data to write
        *
        * Write the file with or without key.
        * To write data without key, you need to read it before (to save the key
        * in attribute key).
        *
        */
        void write(std::string key, std::string data);
        void write(std::string data);


        /**
        * @brief True if file fully decrypted.
        *
        * @return readable attribute
        *
        * Return "readable" attribute.
        *
        */
        bool isReadable();


        /**
        * @brief Get data attribute.
        *
        * @return data attribute.
        *
        * **Warning** if data not fully decrypted (readable!=true),
        * data will be unreadable (unparsable).
        *
        */
        std::string getData();


    private:

        /**
        * @brief Write data in encrypted file.
        *
        * @param data : data to write (for MD5)
        * @param dataEncrypted : data to write
        *
        * Write encryptedData to filename
        *
        */
        void writeRoutine(std::string data, std::string dataEncrypted);


        //Attributes:

        AESCrypt aes; ///< AES instance

        HASHCrypt hash; ///< HASH instance

        std::string filename; ///< Filename attribute

        std::string data; ///< Data attribute

        bool readable; ///< Readable attribute

        byte key[32]; ///< Key in SHA-256

};

#endif