updated validation

Author: shweta3047

server/api/services/code.service.js

@@ -1,6 +1,7 @@
 import fs from "fs";
 import path from "path";
 import { execFile, spawn, exec } from "child_process";
+import ValidationService from "./validation.service";
 const ROOT_DIR = `${process.cwd()}`;
 const SOURCE_DIR = path.join(ROOT_DIR, "executor");
 const TARGET_DIR = `/app/codes`;
@@ -10,8 +11,19 @@ class CodeService {
   async execute(code, input, lang, id) {
     try {
       !input ? (input = "") : null;
+
       //validating code
-      await this.validateCode(code, input, lang, id);
+      const { isValid, message } = await ValidationService.execute(
+        code,
+        input,
+        lang,
+        id
+      );
+      if (!isValid) {
+        throw {
+          message,
+        };
+      }
 
       //writing the code,input  files
       const { file, inputFile } = await this.writeFile(code, lang, input, id);
@@ -42,20 +54,6 @@ class CodeService {
     }
   }
 
-  async validateCode(code, input, lang, id) {
-    switch (lang) {
-      case "javascript": {
-        let words = ["require(", "exports.", "module.exports"];
-        // prevent imports
-        var valid = !words.some((el) => {
-          return code.includes(el);
-        });
-        if (!valid) throw { message: "You have unacceptable libs imported" };
-      }
-    }
-    // throw { message: "You librabry is not accepted : " + lib };
-  }
-
   async writeFile(code, lang, input, id) {
     let fileName = `${id}code`;
     switch (lang) {

server/api/services/validation.service.js

@@ -0,0 +1,48 @@
+class ValidationService {
+  async execute(code, input, lang, id) {
+    switch (lang) {
+      case "javascript": {
+        let words = ["require(", "exports.", "module.exports"];
+        // prevent imports
+        var valid = !words.some((el) => {
+          return code.includes(el);
+        });
+        return {
+          isValid: valid,
+          message: "You have unacceptable libs imported",
+        };
+      }
+      case "python": {
+        let reg1 = RegExp(
+          /\bimport\W+(?:\w+\W+){0,}(?:os|subprocess|importlib)\b/g
+        );
+        words = ["open("];
+
+        if (code.match(reg1)) {
+          return {
+            isValid: false,
+            message: "You have unacceptable libs imported",
+          };
+        } else if (
+          words.every((el) => code.toLowerCase().includes(el.toLowerCase()))
+        ) {
+          return {
+            isValid: false,
+            message: "You have unacceptable libs imported",
+          };
+        }
+        return {
+          isValid: true,
+        };
+      }
+      default: {
+        return {
+          isValid: false,
+          message: "Please select a valid language",
+        };
+      }
+    }
+  }
+}
+
+export default new ValidationService();