Enhance the bpftrace function to raise an exception if the underlying
bpftrace command returns a non-zero exit code. This helps detect and
surface errors from bpftrace execution in test scripts, improving
robustness and debugging capabilities.
Signed-off-by: Breno Leitao <leitao@xxxxxxxxxx>
---
tools/testing/selftests/net/lib/py/utils.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/tools/testing/selftests/net/lib/py/utils.py b/tools/testing/selftests/net/lib/py/utils.py
index 760ccf6fccccc..c4e26567ee6fb 100644
--- a/tools/testing/selftests/net/lib/py/utils.py
+++ b/tools/testing/selftests/net/lib/py/utils.py
@@ -203,6 +203,9 @@ def bpftrace(expr, json=None, ns=None, host=None, timeout=None):
expr += ' interval:s:' + str(timeout) + ' { exit(); }'
cmd_arr += ['-e', expr]
cmd_obj = cmd(cmd_arr, ns=ns, host=host, shell=False)
+ if cmd_obj.ret != 0:
+ raise Exception("Warning: bpftrace command returned a non-zero exit code.")
+
if json:
# bpftrace prints objects as lines
ret = {}
--
2.47.1
