fix(unchunker): initial allowed message size to env var (#158)

* initial allowed message size to env var
Kitware · May 20, 2024 · 488d637 · 488d637
1 parent 3508390
commit 488d637
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/src/wslink/chunking.py b/python/src/wslink/chunking.py
@@ -1,3 +1,4 @@
+import os
 import sys
 import secrets
 import msgpack
@@ -87,7 +88,7 @@ class UnChunker:
 
     def __init__(self):
         self.pending_messages = {}
-        self.max_message_size = 512
+        self.max_message_size = int(os.environ.get("WSLINK_AUTH_MSG_SIZE", 512))
 
     def set_max_message_size(self, size):
         self.max_message_size = size