Encryption for Insert and Truncate complete

Author: rnpsmun2022

REPL.js

@@ -22,7 +22,11 @@ function run(command) {
         console.table(util.inspect(result[0], false, null, true));
       } else {
         console.log(result[0].info);
-        printTable(result[0].data);
+        if (result[0].data.length !== 0) {
+          printTable(result[0].data);
+        } else {
+          console.log("Table Empty");
+        }
       }
     }
     console.log("\n");

classes.js

@@ -133,6 +133,11 @@ class BodytoDetailsResolver {
       details.values = this.body[0];
       details.table = this.body[2];
       details.columns = this.body[4];
+      if (this.body[this.body.length - 2] === "p") {
+        details.password = this.body[this.body.length - 1];
+      } else {
+        details.password = false;
+      }
     } else if (this.action === "delete") {
       if (this.body[0] === "database") {
         details.object = "database";
@@ -145,6 +150,11 @@ class BodytoDetailsResolver {
       }
     } else if (this.action === "truncate") {
       details.table = this.body[0];
+      if (this.body[this.body.length - 2] === "p") {
+        details.password = this.body[this.body.length - 1];
+      } else {
+        details.password = false;
+      }
     } else if (this.action === "use") {
       details.database = this.body[0];
     } else if (this.action === "default") {

parser/actions/insert/insert.js

@@ -8,6 +8,10 @@ class hashPassword {
     this.index = index;
   }
 }
+const crypto = require("crypto");
+
+const ALGORITHM = "aes-256-ctr";
+const SALT_ROUNDS = 10;
 
 function valueChecker(columns, values) {
   /*console.log('in valueChecker');
@@ -58,14 +62,23 @@ function actionInsert(code, preRunData) {
       `${preRunData[1]}`,
       `${code.details.table}.json`
     );
-    let table = require(storagePath);
+    let Decrypter = new classes.Decrypter(preRunData[1], code.details.table);
+    //console.log(Decrypter.check());
+    console.log(code);
+    let table = null;
+    if (!Decrypter.check()) {
+      table = require(storagePath);
+    } else {
+      table = JSON.parse(Decrypter.decrypt(code.details.password));
+    }
     if (code.details.columns === "*") {
       code.details.columns = "all";
     }
     let fullColumnData = readColumns(
       new classes.ActionDetails("select", {
         object: "columns",
         table: code.details.table,
+        password: code.details.password,
       }),
       preRunData
     ).data;
@@ -105,7 +118,22 @@ function actionInsert(code, preRunData) {
       if (USING_DATABASE === "") {
         return "KeyError: Not using any database currently";
       }
-      fs.writeFileSync(storagePath, JSON.stringify(table, null, 4));
+      if (!Decrypter.check()) {
+        fs.writeFileSync(storagePath, JSON.stringify(table, null, 4));
+      } else {
+        const passwordHash = bcrypt.hashSync(
+          code.details.password,
+          SALT_ROUNDS
+        );
+        const cipher = crypto.createCipher(ALGORITHM, code.details.password);
+        let encryptedData = cipher.update(JSON.stringify(table), "utf8", "hex");
+        encryptedData += cipher.final("hex");
+        const encryptedObject = {
+          passwordHash,
+          data: encryptedData,
+        };
+        fs.writeFileSync(storagePath, JSON.stringify(encryptedObject), "utf-8");
+      }
       if (preRunData[0])
         return new classes.Success(
           `Successfully inserted the values into the table ${table.name}`

parser/actions/select/table.js

@@ -15,8 +15,8 @@ function selectTable(code, preRunData) {
     code.details.columns = "all";
   }
   let Decrypter = new classes.Decrypter(preRunData[1], code.details.name);
-  console.log(Decrypter.check());
-  console.log(code);
+  //console.log(Decrypter.check());
+  //console.log(code);
   let table = null;
   if (!Decrypter.check()) {
     table = require(storagePath);

parser/actions/truncate/truncate.js

@@ -1,19 +1,59 @@
-const path = require('path');
-const fs = require('fs');
-const classes = require(path.join(__dirname, '../../../', 'classes.js'));
+const path = require("path");
+const fs = require("fs");
+const classes = require(path.join(__dirname, "../../../", "classes.js"));
+const crypto = require("crypto");
+const bcrypt = require("bcrypt");
 
-function actionTruncate(code, preRunData){
-    let storagePath = path.join(__dirname, '../../../', 'storage', `${preRunData[1]}`, `${code.details.table}.json`);
-    try{
-        let table = require(storagePath);
-        table.values = [];
-        let USING_DATABASE = preRunData[1];
-        if(USING_DATABASE===''){return "KeyError: Not using any database currently"};
-        fs.writeFileSync(storagePath, JSON.stringify(table, null, 4));
-        if(preRunData[0]) return new classes.Success(`Successfully truncated the table ${table.name}`);
-    } catch(e){
-        return new classes.Error("StorageError", `Cannot find table ${code.details.table}`);
+const ALGORITHM = "aes-256-ctr";
+const SALT_ROUNDS = 10;
+
+function actionTruncate(code, preRunData) {
+  let storagePath = path.join(
+    __dirname,
+    "../../../",
+    "storage",
+    `${preRunData[1]}`,
+    `${code.details.table}.json`
+  );
+  try {
+    let Decrypter = new classes.Decrypter(preRunData[1], code.details.table);
+    //console.log(Decrypter.check());
+    console.log(code);
+    let table = null;
+    if (!Decrypter.check()) {
+      table = require(storagePath);
+    } else {
+      table = JSON.parse(Decrypter.decrypt(code.details.password));
+    }
+    table.values = [];
+    let USING_DATABASE = preRunData[1];
+    if (USING_DATABASE === "") {
+      return "KeyError: Not using any database currently";
+    }
+    if (!Decrypter.check()) {
+      fs.writeFileSync(storagePath, JSON.stringify(table, null, 4));
+    } else {
+      const passwordHash = bcrypt.hashSync(code.details.password, SALT_ROUNDS);
+      const cipher = crypto.createCipher(ALGORITHM, code.details.password);
+      let encryptedData = cipher.update(JSON.stringify(table), "utf8", "hex");
+      encryptedData += cipher.final("hex");
+      const encryptedObject = {
+        passwordHash,
+        data: encryptedData,
+      };
+      fs.writeFileSync(storagePath, JSON.stringify(encryptedObject), "utf-8");
     }
+    if (preRunData[0])
+      return new classes.Success(
+        `Successfully truncated the table ${table.name}`
+      );
+  } catch (e) {
+    console.log(e);
+    return new classes.Error(
+      "StorageError",
+      `Cannot find table ${code.details.table}`
+    );
+  }
 }
 
-module.exports = actionTruncate;
+module.exports = actionTruncate;

storage/TestingAfterYears/testing2.json

@@ -1 +1 @@
-{"passwordHash":"$2b$10$dtd4wRRGdQfUFgaP9LZCuOpp9MqfWpwdM.4.8YoPzqeoaYXsU0YNi","data":"7a5a1afadb051a68d942afaee6226bd05b0020d1d17dc180cba4d600996e948f0dec8d25abd0a540f867f7a4252cfd5702c8a7a929b368dd089cf273de8426fa1c9f9b5f29292258222ad73fb22604fc48ce05396b9c17648b44ec539fabc0c4803d853c87f3b730b11e0198d4d1492b02ca6e9134afa06d52725cf16e810ab340d85085072dfc279005df2103ca6c5275a9503fe650253f5be2d63bec8c379cf87cdb142d77cd9fc44967bcf08205743ac10d4d073e7d8eb642fd08e620fc2b2e3265da027273e6629d7b581d3413b022bae350501b07a637bead9a121479205564243ac108b0308dddd993dbec42721466b0aa853ff6fd1b572c5d4e520ff8246195be8a5307413481c704c96c4a60e45bb99dcd0feb67a863cd1755688ce8a7c0889198b610f64e81d8d6c221bf9b7b4b1a800246381ed1b11882e7a17c840d0c48e9b83a469bcba81f997d88659c67ec445a"}
+{"passwordHash":"$2b$10$kNBnyrvL25H7goSOgPq.Tu3Ui7J4nJ7ORH.5lLhEoPq3T6GgjYkOe","data":"7a7254bb940e593f9e14f0fdb03f60c41d4b6294803081ff86ea8741cc599ac00afb8d25a9e5da0dba22a5ff5f7eb41a439afee025ae689a04c5be68d28b26aa06cd9a4236213e466b3282669a7551be02d64a4769861537843ca51dd8a99db98e7192308ea4e863b1572eb18c8f1e252aba3cd076b3e706003312f146ea58f216991ec44b7ca567ad0fce3e5298230024800e0db907361959909f78faeb78c58503b0596c3bcf9f8a10208297f7720d6dbc014f430011cfe30ea95ea77db97e711a7f9c433e20a33d"}