From: Scott Gasch <scott@gasch.org>
Date: Wed, 17 Aug 2022 19:01:26 +0000 (-0700)
Subject: Do not allow data that is too large in zookeeper.
X-Git-Url: https://wannabe.guru.org/gitweb/?a=commitdiff_plain;h=4c74ef100038a239bae9b4fd4d61fb04f5b391b4;p=python_utils.git

Do not allow data that is too large in zookeeper.
---

diff --git a/config.py b/config.py
index b344631..81bd7d1 100644
--- a/config.py
+++ b/config.py
@@ -393,6 +393,9 @@ def _augment_sys_argv_from_loadfile():
                 newargs = [
                     arg.strip('\n') for arg in contents.split('\n') if 'config_savefile' not in arg
                 ]
+                size = sys.getsizeof(newargs)
+                if size > 1024 * 1024:
+                    raise Exception(f'Saved args are too large!  ({size} bytes)')
             except Exception as e:
                 raise Exception(f'Error reading {zkpath} from zookeeper.') from e
             SAVED_MESSAGES.append(f'Loaded config from zookeeper from {zkpath}')