Add size limit

wiremod · Mar 26, 2024 · 0b464db · 0b464db
1 parent 37b7cbf
commit 0b464db
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lua/entities/gmod_wire_expression2/core/string.lua b/lua/entities/gmod_wire_expression2/core/string.lua
@@ -508,11 +508,15 @@ local compress = util.Compress
 local decompress = util.Decompress
 
 e2function string compress(string plaintext)
-	self.prf = self.prf + #plaintext * 0.1
+	local len = #plaintext
+	if len > 32768 then return self:throw("Input string is too long!", "") end
+	self.prf = self.prf + len * 0.1
 	return compress(plaintext)
 end
 
 e2function string decompress(string compressed)
-	self.prf = self.prf + #compressed * 0.5
+	local len = #compressed
+	if len > 32768 then return self:throw("Input string is too long!", "") end
+	self.prf = self.prf + len * 0.5
 	return decompress(compressed) or self:throw("Invalid input for decompression!", "")
 end