objc4-818.2.tar.gz

[apple/objc4.git] / runtime / Messengers.subproj / objc-msg-x86_64.s

diff --git a/runtime/Messengers.subproj/objc-msg-x86_64.s b/runtime/Messengers.subproj/objc-msg-x86_64.s
index d0909958ac6830467927954eacf117f2d0fe3228..0b8eff7aff416fe01f363227c6b237a37382035f 100644 (file)
--- a/runtime/Messengers.subproj/objc-msg-x86_64.s
+++ b/runtime/Messengers.subproj/objc-msg-x86_64.s
@@ -22,7 +22,7 @@
  */
 
 #include <TargetConditionals.h>
-#if __x86_64__  &&  !(TARGET_OS_SIMULATOR && !TARGET_OS_IOSMAC)
+#if __x86_64__  &&  !(TARGET_OS_SIMULATOR && !TARGET_OS_MACCATALYST)
 
 #include "isa.h"
 
@@ -139,6 +139,10 @@ _objc_restartableRanges:
 #define GETIMP 101
 #define LOOKUP 102
 
+#define MSGSEND 200
+#define METHOD_INVOKE 201
+#define METHOD_INVOKE_STRET 202
+
 
 /********************************************************************
  *
@@ -227,21 +231,28 @@ LExit$0:
 // for a function call.
 //////////////////////////////////////////////////////////////////////
 
-.macro SAVE_REGS
+.macro SAVE_REGS kind
 
+.if \kind != MSGSEND && \kind != METHOD_INVOKE && \kind != METHOD_INVOKE_STRET
+.abort Unknown kind.
+.endif
        push    %rbp
        mov     %rsp, %rbp
 
-       sub     $$0x80+8, %rsp          // +8 for alignment
+       sub     $0x80, %rsp
 
        movdqa  %xmm0, -0x80(%rbp)
        push    %rax                    // might be xmm parameter count
        movdqa  %xmm1, -0x70(%rbp)
        push    %a1
        movdqa  %xmm2, -0x60(%rbp)
+.if \kind == MSGSEND || \kind == METHOD_INVOKE_STRET
        push    %a2
+.endif
        movdqa  %xmm3, -0x50(%rbp)
+.if \kind == MSGSEND || \kind == METHOD_INVOKE
        push    %a3
+.endif
        movdqa  %xmm4, -0x40(%rbp)
        push    %a4
        movdqa  %xmm5, -0x30(%rbp)
@@ -249,6 +260,9 @@ LExit$0:
        movdqa  %xmm6, -0x20(%rbp)
        push    %a6
        movdqa  %xmm7, -0x10(%rbp)
+.if \kind == MSGSEND
+       push    %r10
+.endif
 
 .endmacro
 
@@ -261,8 +275,12 @@ LExit$0:
 // SAVE_REGS.
 //////////////////////////////////////////////////////////////////////
 
-.macro RESTORE_REGS
+.macro RESTORE_REGS kind
 
+.if \kind == MSGSEND
+       pop     %r10
+       orq     $2, %r10 // for the sake of instrumentations, remember it was the slowpath
+.endif
        movdqa  -0x80(%rbp), %xmm0
        pop     %a6
        movdqa  -0x70(%rbp), %xmm1
@@ -270,9 +288,13 @@ LExit$0:
        movdqa  -0x60(%rbp), %xmm2
        pop     %a4
        movdqa  -0x50(%rbp), %xmm3
+.if \kind == MSGSEND || \kind == METHOD_INVOKE
        pop     %a3
+.endif
        movdqa  -0x40(%rbp), %xmm4
+.if \kind == MSGSEND || \kind == METHOD_INVOKE_STRET
        pop     %a2
+.endif
        movdqa  -0x30(%rbp), %xmm5
        pop     %a1
        movdqa  -0x20(%rbp), %xmm6
@@ -447,7 +469,7 @@ LLookupEnd$2:
 
 .macro MethodTableLookup
 
-       SAVE_REGS
+       SAVE_REGS MSGSEND
 
        // lookUpImpOrForward(obj, sel, cls, LOOKUP_INITIALIZE | LOOKUP_RESOLVER)
 .if $0 == NORMAL
@@ -464,7 +486,7 @@ LLookupEnd$2:
        // IMP is now in %rax
        movq    %rax, %r11
 
-       RESTORE_REGS
+       RESTORE_REGS MSGSEND
 
 .if $0 == NORMAL
        test    %r11, %r11              // set ne for nonstret forwarding
@@ -1257,13 +1279,12 @@ LCacheMiss_objc_msgLookupSuper2_stret:
 
 L_method_invoke_small:
        // Small methods require a call to handle swizzling.
-       SAVE_REGS
+       SAVE_REGS METHOD_INVOKE
        movq    %a2, %a1
        call    __method_getImplementationAndName
-       movq    %rdx, %r10
+       movq    %rdx, %a2
        movq    %rax, %r11
-       RESTORE_REGS
-       movq    %r10, %a2
+       RESTORE_REGS METHOD_INVOKE
        jmp     *%r11
 
        END_ENTRY _method_invoke
@@ -1282,13 +1303,12 @@ L_method_invoke_small:
 
 L_method_invoke_stret_small:
        // Small methods require a call to handle swizzling.
-       SAVE_REGS
+       SAVE_REGS METHOD_INVOKE_STRET
        movq    %a3, %a1
        call    __method_getImplementationAndName
-       movq    %rdx, %r10
+       movq    %rdx, %a3
        movq    %rax, %r11
-       RESTORE_REGS
-       movq    %r10, %a3
+       RESTORE_REGS METHOD_INVOKE_STRET
        jmp     *%r11
 
        END_ENTRY _method_invoke_stret