From f077ebf68bee355216d0be2a03675f433d06fe69 Mon Sep 17 00:00:00 2001 From: Nathaniel Mitchell Date: Mon, 16 Dec 2024 10:41:23 -0800 Subject: [PATCH] Add error message on missing data when using the replayhelper Signed-off-by: Nathaniel Mitchell --- chipsec/helper/replay/replayhelper.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/chipsec/helper/replay/replayhelper.py b/chipsec/helper/replay/replayhelper.py index de05fb83..c37b6fa1 100644 --- a/chipsec/helper/replay/replayhelper.py +++ b/chipsec/helper/replay/replayhelper.py @@ -86,7 +86,12 @@ class ReplayHelper(Helper): targs = str(args) if str(cmd) in self._data: if targs in self._data[str(cmd)]: - return self._data[cmd][targs].pop() + try: + return self._data[cmd][targs].pop() + except IndexError as err: + logger().log_error(f'Ran out of entries for {str(cmd)} {targs}') + err.args = (err.args[0] + f': {str(cmd)} {targs}',) + raise err logger().log_error(f"Missing entry for {str(cmd)} {targs}") return None