From 4c74ef100038a239bae9b4fd4d61fb04f5b391b4 Mon Sep 17 00:00:00 2001 From: Scott Gasch Date: Wed, 17 Aug 2022 12:01:26 -0700 Subject: [PATCH] Do not allow data that is too large in zookeeper. --- config.py | 3 +++ 1 file changed, 3 insertions(+) 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}') -- 2.47.1