[DTrace-devel] [PATCH 03/04] rawtp: report lockmem issues when determining rawtp argument count

[Kris Van Hees]

Signed-off-by: Kris Van Hees <kris.van.hees at oracle.com>
---
 libdtrace/dt_prov_rawtp.c | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/libdtrace/dt_prov_rawtp.c b/libdtrace/dt_prov_rawtp.c
index 1ff0a311..c2d75150 100644
--- a/libdtrace/dt_prov_rawtp.c
+++ b/libdtrace/dt_prov_rawtp.c
@@ -181,7 +181,9 @@ static int probe_info(dtrace_hdl_t *dtp, const dt_probe_t *prp,
 		dif.dtdo_len = 2;
 
 		bpf_fd = dt_bpf_prog_load(dt_rawtp.prog_type, &dif, 0, NULL, 0);
-		if (bpf_fd == -1)
+		if (bpf_fd == -EPERM)
+			return dt_bpf_lockmem_error(dtp, "Cannot retrieve argument count");
+		else if (bpf_fd < 0)
 			continue;
 		rtp_fd = dt_bpf_raw_tracepoint_open(prp->desc->prb, bpf_fd);
 		close(bpf_fd);
-- 
2.39.3