From 4c0eb783006c136ac060c25a4c77a3eb1b700c72 Mon Sep 17 00:00:00 2001
From: Arne Keller <arne.keller@student.kit.edu>
Date: Wed, 10 Mar 2021 16:07:20 +0100
Subject: [PATCH] Limit memory usage of LatexCreatorType

---
 .../latexcreator/LatexCreatorType.java                   | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/main/java/edu/kit/typicalc/view/content/typeinferencecontent/latexcreator/LatexCreatorType.java b/src/main/java/edu/kit/typicalc/view/content/typeinferencecontent/latexcreator/LatexCreatorType.java
index 81bcda1..8f0bef3 100644
--- a/src/main/java/edu/kit/typicalc/view/content/typeinferencecontent/latexcreator/LatexCreatorType.java
+++ b/src/main/java/edu/kit/typicalc/view/content/typeinferencecontent/latexcreator/LatexCreatorType.java
@@ -15,6 +15,8 @@ import static edu.kit.typicalc.view.content.typeinferencecontent.latexcreator.La
  * @see Type
  */
 public class LatexCreatorType implements TypeVisitor {
+    private static final int MAX_LENGTH = 50000;
+
     private final StringBuilder latex = new StringBuilder();
     private boolean needsParentheses = false;
 
@@ -80,6 +82,13 @@ public class LatexCreatorType implements TypeVisitor {
         latex.append(SPACE);
 
         function.getOutput().accept(this);
+        checkMemoryUsage();
         needsParentheses = true;
     }
+
+    private void checkMemoryUsage() {
+        if (latex.length() > MAX_LENGTH) {
+            throw new IllegalStateException("type too large!");
+        }
+    }
 }