options: ensure lockmem is set before retrieving probe info

The rawtp provider uses trial-and-error BPF program loading to determine
the number of arguments for a raw tracepoint.  That requires the lockmem
option value to be applied before probe info is retrieved.

Signed-off-by: Kris Van Hees <kris.van.hees@oracle.com>
Reviewed-by: Eugene Loh <eugene.loh@oracle.com>

libdtrace/dt_open.c

@@ -1175,13 +1175,24 @@ dtrace_vopen(int version, int flags, int *errp,
 int
 dtrace_init(dtrace_hdl_t *dtp)
 {
-	int	i;
+	int		i;
+	dtrace_optval_t	lockmem = dtp->dt_options[DTRACEOPT_LOCKMEM];
 
 	/*
 	 * Initialize the BPF library handling.
 	 */
 	dt_dlib_init(dtp);
 
+	/*
+	 * Set the locked-memory limit if so directed by the user.
+	 */
+        if (lockmem != DTRACEOPT_UNSET) {
+		struct rlimit	rl;
+
+		rl.rlim_cur = rl.rlim_max = lockmem;
+		setrlimit(RLIMIT_MEMLOCK, &rl);
+        }
+
 	/*
 	 * Initialize consume handling.
 	 */

libdtrace/dt_work.c

@@ -136,20 +136,10 @@ dtrace_go(dtrace_hdl_t *dtp, uint_t cflags)
 {
 	size_t			size;
 	struct epoll_event	ev;
-	dtrace_optval_t		lockmem = dtp->dt_options[DTRACEOPT_LOCKMEM];
-	struct rlimit		rl;
 
 	if (dtp->dt_active)
 		return dt_set_errno(dtp, EINVAL);
 
-	/*
-	 * Set the locked-memory limit if so directed by the user.
-	 */
-        if (lockmem != DTRACEOPT_UNSET) {
-                rl.rlim_cur = rl.rlim_max = lockmem;
-                setrlimit(RLIMIT_MEMLOCK, &rl);
-        }
-
 	/* Create the BPF programs. */
 	if (dt_bpf_make_progs(dtp, cflags) == -1)
 		return -1;