Change file architecture

1 files changed, 133 insertions, 0 deletions

diff --git a/src/IOFileClass/FileManIOFile.hpp b/src/IOFileClass/FileManIOFile.hpp
new file mode 100644
index 0000000..221cf13
--- /dev/null
+++ b/src/IOFileClass/FileManIOFile.hpp
@@ -0,0 +1,133 @@
+/**
+ * @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