From 46ce371d9c550ea840cb946b06c5614569ca98fa Mon Sep 17 00:00:00 2001 From: Shivani Bhardwaj Date: Thu, 22 Jun 2023 17:51:05 +0530 Subject: [PATCH] suricatasc: handle exceptions in caller --- python/bin/suricatasc | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/python/bin/suricatasc b/python/bin/suricatasc index dc108da4af..d090f856f5 100755 --- a/python/bin/suricatasc +++ b/python/bin/suricatasc @@ -1,6 +1,6 @@ #! /usr/bin/env python # -# Copyright(C) 2013-2022 Open Information Security Foundation +# Copyright(C) 2013-2023 Open Information Security Foundation # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by @@ -69,8 +69,16 @@ except SuricataReturnException as err: sys.exit(1) if args.command: - (command, arguments) = sc.parse_command(args.command) - res = sc.send_command(command, arguments) + try: + (command, arguments) = sc.parse_command(args.command) + except SuricataCommandException as err: + print(err.value) + sys.exit(1) + try: + res = sc.send_command(command, arguments) + except (SuricataCommandException, SuricataReturnException) as err: + print(err.value) + sys.exit(1) print(json.dumps(res)) sc.close() if res['return'] == 'OK':