Merge branch 'PARSER'

1 files changed, 147 insertions, 0 deletions

diff --git a/src/IOFileClass/FileManIOFile.cpp b/src/IOFileClass/FileManIOFile.cpp
new file mode 100644
index 0000000..ea49201
--- /dev/null
+++ b/src/IOFileClass/FileManIOFile.cpp
@@ -0,0 +1,147 @@
+/**
+ * @file FileManIOFile.cpp
+ * @brief FileManIOFile class definitions
+ * @author manzerbredes
+ * @date 9 Mars 2015
+ *
+ * Contain all definitions of FileManIOFile class.
+ *
+ */
+
+
+#include "FileManIOFile.hpp"
+
+//Constructor with filename
+FileManIOFile::FileManIOFile(std::string filename){
+    this->filename=filename;
+    this->readable=false;
+    this->data="";
+    this->key;
+}
+
+//Destructor
+FileManIOFile::~FileManIOFile(){
+}
+
+
+
+//Read the filename with a key
+void FileManIOFile::read(std::string key){
+
+    //create file object
+    std::ifstream file;
+
+    //Clear data
+    this->data.clear();
+
+    //Open file
+    file.open (this->filename, std::ios::in | std::ios::binary);
+
+    //Get MD5 of decrypted data
+    byte fileMD5[16];
+    file.read((char*) fileMD5, sizeof(fileMD5));
+
+    //Read all data
+    char car;
+    file.read(&car, sizeof(car));
+
+    while(file){
+        this->data+=car,
+        file.read(&car, sizeof(car));
+
+    }
+
+    //Decrypt data
+    this->data=this->aes.decrypt(key, this->data);
+
+    //Get current MD5 of decrypted data
+    byte currentMD5[16];
+    this->hash.getMD5_128(this->data, currentMD5, sizeof(currentMD5));
+
+    //Compare the 2 MD5 to find if file is fully decrypted
+    if(this->hash.compareDigest(fileMD5, currentMD5, sizeof(currentMD5))){
+        //Set readable
+        this->readable=true;
+        //Save the key
+        this->hash.getSHA_256(key, this->key, 32);
+    }
+    else{
+        this->readable=false;
+    }
+
+    //Close file
+    file.close();
+
+
+}
+
+
+
+//Write file with key
+void FileManIOFile::write(std::string key,std::string data){
+
+    std::string dataEncrypted;
+
+    dataEncrypted=this->aes.encrypt(key, data);
+
+    this->writeRoutine(data, dataEncrypted);
+
+
+}
+
+//Write file without key
+void FileManIOFile::write(std::string data){
+    if(not(this->readable)){
+        std::cout << "Can't write data without key (read it before) !" << std::endl;
+        std::exit(EXIT_FAILURE);
+    }
+
+    std::string dataEncrypted;
+
+    dataEncrypted=this->aes.encrypt(this->key, data);
+    this->writeRoutine(data, dataEncrypted);
+
+
+}
+
+
+
+//Get readable attribute
+bool FileManIOFile::isReadable(){
+    return this->readable;
+}
+
+
+
+//Write file
+void FileManIOFile::writeRoutine(std::string data, std::string dataEncrypted){
+
+    //Save MD5 of decrypted data
+    byte digest[16];
+    this->hash.getMD5_128(data, digest, sizeof(digest));
+
+    //Create file instance
+    std::ofstream file;
+
+    //Open it
+    file.open(this->filename, std::ios::out | std::ios::binary);
+
+    //Write MD5 on 16 first bytes
+    file.write((char *) digest,sizeof(digest));
+
+    //Write data
+    file.write(dataEncrypted.c_str(), dataEncrypted.size());
+
+    //Close file
+    file.close();
+
+    //Save data to attribute
+    this->data=data;
+}
+
+
+
+//Get data
+std::string FileManIOFile::getData(){
+    return this->data;
+}