Merge pull request #19 from SahilK-027/main

Loop Iteration Limit to Handle Infinite While Loops

BackEnd/Interpreter/Statements/Loop/While_statement.ts

@@ -20,10 +20,19 @@ export const evaluate_boolean_while_statement = (
   stmt: WhileStatement,
   env: Environment,
 ): RuntimeVal => {
+  let iterationCnt = 0;
   let breakLoop = false;
   let evaluatedCondition = evaluate(stmt.condition, env) as BooleanVal;
+  const startTime = performance.now();
 
   while (evaluatedCondition.value && !breakLoop) {
+    iterationCnt++;
+
+    // Check for infinite loop
+    if (env.checkInfiniteLoop(iterationCnt)) {
+      break;
+    }
+
     // Create a new environment for each iteration
     const whileEnv = new Environment(env);
     for (const statement of stmt.body) {
@@ -59,10 +68,18 @@ export const evaluate_numeric_while_statement = (
   stmt: WhileStatement,
   env: Environment,
 ): RuntimeVal => {
+  let iterationCnt = 0;
   let breakLoop = false;
   let evaluatedCondition = evaluate(stmt.condition, env) as NumberVal;
 
   while (evaluatedCondition.value && !breakLoop) {
+    iterationCnt++;
+
+    // Check for infinite loop
+    if (env.checkInfiniteLoop(iterationCnt)) {
+      break;
+    }
+
     // Create a new environment for each iteration
     const whileEnv = new Environment(env);
     for (const statement of stmt.body) {

BackEnd/Scope/environment.ts

@@ -7,7 +7,7 @@ export default class Environment {
   private parent?: Environment;
   private variables: Map<string, RuntimeVal>;
   private constants: Set<string>;
-
+  private MAX_ALLOWED_ITERATIONS: number;
   /**
    * Creates an instance of the Environment class.
    * @param parentENV - The parent environment (optional).
@@ -17,6 +17,16 @@ export default class Environment {
     this.parent = parentENV;
     this.variables = new Map();
     this.constants = new Set();
+    this.MAX_ALLOWED_ITERATIONS = 102702;
+  }
+
+  public checkInfiniteLoop(
+    iterationCnt: number,
+  ): Boolean {
+    if (iterationCnt > this.MAX_ALLOWED_ITERATIONS) {
+      throw `TIME LIMIT EXCEEDED...`;
+    }
+    return false;
   }
 
   /**