diff --git a/coredump/coredump.py b/coredump/coredump.py index 5e63d2138..88a1b374c 100644 --- a/coredump/coredump.py +++ b/coredump/coredump.py @@ -1,5 +1,6 @@ import argparse import os +import sys import criu_coredump @@ -34,7 +35,12 @@ def main(): opts = vars(parser.parse_args()) - coredump(opts) + try: + coredump(opts) + except SystemExit as error: + print('ERROR: %s' % error) + print('Exiting') + sys.exit(1) if __name__ == '__main__': diff --git a/coredump/criu_coredump/coredump.py b/coredump/criu_coredump/coredump.py index e2c56c01c..8ee402676 100644 --- a/coredump/criu_coredump/coredump.py +++ b/coredump/criu_coredump/coredump.py @@ -692,7 +692,11 @@ class coredump_generator: files = self.reg_files fname = next(filter(lambda x: x["id"] == shmid, files))["name"] - f = open(fname, 'rb') + try: + f = open(fname, 'rb') + except FileNotFoundError: + sys.exit('Required file %s not found.' % fname) + f.seek(off) start = vma["start"]