X-Git-Url: https://git.saurik.com/apple/xnu.git/blobdiff_plain/3e170ce000f1506b7b5d2c5c7faec85ceabb573d..eb6b6ca394357805f2bdba989abae309f718b4d8:/osfmk/x86_64/machine_routines_asm.s?ds=sidebyside

diff --git a/osfmk/x86_64/machine_routines_asm.s b/osfmk/x86_64/machine_routines_asm.s
index af962f2f4..c0a1bdc90 100644
--- a/osfmk/x86_64/machine_routines_asm.s
+++ b/osfmk/x86_64/machine_routines_asm.s
@@ -175,18 +175,40 @@ ENTRY(_rtc_tsc_to_nanoseconds)
 	shrdq   $32,%rdx,%rax			/* %rdx:%rax >>= 32 */
 	ret
     
-    
+
+/*
+ *  typedef void (*thread_continue_t)(void *param, wait_result_t)
+ *
+ *	void call_continuation( thread_continue_t continuation,
+ *	            			void *param,
+ *				            wait_result_t wresult,
+ *                          bool enable interrupts)
+ */
 
 Entry(call_continuation)
-	movq	%rdi,%rcx			/* get continuation */
-	movq	%rsi,%rdi			/* continuation param */
-	movq	%rdx,%rsi			/* wait result */
+
+	movq    %rdi, %r12  /* continuation */
+    movq    %rsi, %r13  /* continuation param */
+    movq    %rdx, %r14  /* wait result */
+
 	movq	%gs:CPU_KERNEL_STACK,%rsp	/* set the stack */
 	xorq	%rbp,%rbp			/* zero frame pointer */
+
+    test    %ecx, %ecx
+    jz 1f
+    mov     $1, %edi
+    call   _ml_set_interrupts_enabled
+1:
+
+	movq	%r12,%rcx			/* continuation */
+	movq	%r13,%rdi			/* continuation param */
+	movq	%r14,%rsi			/* wait result */
+
 	call	*%rcx				/* call continuation */
 	movq	%gs:CPU_ACTIVE_THREAD,%rdi
 	call	EXT(thread_terminate)
 
+
 Entry(x86_init_wrapper)
 	xor	%rbp, %rbp
 	movq	%rsi, %rsp