fix: Avoid trying to store NUL char in DB

Haidra-Org · Mar 9, 2024 · 35b7396 · 35b7396
1 parent c6d6aa4
commit 35b7396
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/horde/classes/base/processing_generation.py b/horde/classes/base/processing_generation.py
@@ -84,7 +84,8 @@ def set_generation(self, generation, things_per_sec, **kwargs):
         # We return -1 to know to send a different error
         if self.is_faulted():
             return -1
-        self.generation = generation
+        # Sanitize NUL char away from string literal we store in the DB
+        self.generation = generation.replace("\x00", "\uFFFD")
         # Support for two typical properties
         self.seed = kwargs.get("seed", None)
         self.gen_metadata = kwargs.get("gen_metadata", None)