protect jit pages

1 files changed, 14 insertions, 2 deletions

diff --git a/src/ejit.c b/src/ejit.c
index 660d476..265acea 100644
--- a/src/ejit.c
+++ b/src/ejit.c
@@ -363,17 +363,29 @@ struct ejit_func *ejit_create_func(enum ejit_type rtype, size_t argc,
 	return f;
 }
 
+void ejit_compile_func_unsafe(struct ejit_func *f)
+{
+	ejit_select_compile_func(f,
+			gpr_stats_len(&f->gpr),
+			fpr_stats_len(&f->fpr),
+			f->use_64,
+			true,
+			false);
+}
+
 void ejit_compile_func(struct ejit_func *f)
 {
+	/* might make sense to use flags instead of bools... */
 	ejit_select_compile_func(f,
 			gpr_stats_len(&f->gpr),
 			fpr_stats_len(&f->fpr),
 			f->use_64,
+			true,
 			true);
 }
 
 void ejit_select_compile_func(struct ejit_func *f, size_t gpr, size_t fpr,
-                              bool use_64, bool try_jit)
+                              bool use_64, bool try_jit, bool im_scawed)
 {
 	/* emit a final end instruction in case user didn't do a return */
 	emit_insn_o(f, END);
@@ -384,7 +396,7 @@ void ejit_select_compile_func(struct ejit_func *f, size_t gpr, size_t fpr,
 	assert(!f->use_64 || (use_64 == f->use_64));
 
 	/* try to jit compile if possible */
-	if (try_jit && ejit_compile(f, use_64))
+	if (try_jit && ejit_compile(f, use_64, im_scawed))
 		return;
 
 	/* otherwise, convert opcodes to address labels */