diff options
| -rw-r--r-- | snekbox/nsjail.py | 18 |
1 files changed, 16 insertions, 2 deletions
diff --git a/snekbox/nsjail.py b/snekbox/nsjail.py index 6972a40..d5659e5 100644 --- a/snekbox/nsjail.py +++ b/snekbox/nsjail.py @@ -47,8 +47,22 @@ class NsJail: def _read_config() -> NsJailConfig: """Read the NsJail config at `NSJAIL_CFG` and return a protobuf Message object.""" config = NsJailConfig() - with open(NSJAIL_CFG, encoding="utf-8") as f: - text_format.Parse(f.read(), config) + + try: + with open(NSJAIL_CFG, encoding="utf-8") as f: + config_text = f.read() + except FileNotFoundError: + log.fatal(f"The NsJail config at {NSJAIL_CFG!r} could not be found.") + sys.exit(1) + except OSError as e: + log.fatal(f"The NsJail config at {NSJAIL_CFG!r} could not read.", exc_info=e) + sys.exit(1) + + try: + text_format.Parse(config_text, config) + except text_format.ParseError as e: + log.fatal(f"The NsJail config at {NSJAIL_CFG!r} could not be parsed.", exc_info=e) + sys.exit(1) return config |