X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=config.py;h=81bd7d113021b3b66324836804004fdd38beb5aa;hb=4c74ef100038a239bae9b4fd4d61fb04f5b391b4;hp=b344631003bb29dc186c0d7db8f2631b8c7ac4bc;hpb=2966ecdb8c49266d491a1f75791868920d68a510;p=python_utils.git

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}')