patch.S

.intel_syntax noprefix

# Unity symbols
.global aArrayIndexOutOfRange
.global aRuntimeUtilitiesFixedArray
.global __clang_call_terminate
.global _ZN6vulkan3ext21vkAcquireNextImageKHRE
.global _ZN6vulkan4fptr15vkWaitForFencesE
.global _ZN6vulkan4fptr16vkDeviceWaitIdleE
.global _Unwind_Resume
.global _Z12profiler_endPN9profiling6MarkerE
.global _Z14profiler_beginPN9profiling6MarkerE
.global _Z18GetVKGfxDeviceCorev
.global _Z20AssertImplementationiPKciiS0_
.global _ZN11GfxDeviceVK27SubmitCurrentCommandBuffersEP13VkSemaphore_Tb
.global _ZN15GfxDeviceVKBase26EnsureCurrentCommandBufferEN2vk13CommandBuffer4TypeEb
.global _ZN18FrameTimingManager29OnRenderThreadOverheadEndImplEv
.global _ZN18FrameTimingManager31OnRenderThreadOverheadBeginImplEv
.global _ZN18VKImmediateContext17BackbufferChangedERN2vk12RenderPassesEt
.global _ZN2vk12TaskExecutor22AddSubmitWaitSemaphoreEP13VkSemaphore_Tj
.global _ZN2vk12TaskExecutor4SyncEv
.global _ZN2vk13FrameTracking11ReturnFenceEP9VkFence_T
.global _ZN2vk13FrameTracking12GetSemaphoreEv
.global _ZN2vk13FrameTracking15ReturnSemaphoreEP13VkSemaphore_Tm
.global _ZN2vk13FrameTracking15UpdateSafeFrameEv
.global _ZN2vk13FrameTracking21ReturnSemaphoreUnusedEP13VkSemaphore_T
.global _ZN2vkL17gAcquireNextFrameE

# Redirects GfxDeviceVK::AdvanceImage(vk::SwapChain*) to custom version
.section .text.patch1,"ax"
.global thunk
.type thunk, @function
thunk:
    jmp _ZN11GfxDeviceVK12AdvanceImageEPN2vk9SwapChainE
    .align 16, 0xCC

# Custom version of GfxDeviceVK::AdvanceImage(vk::SwapChain*)
.section .text.patch2,"ax"
.global _ZN11GfxDeviceVK12AdvanceImageEPN2vk9SwapChainE
.type _ZN11GfxDeviceVK12AdvanceImageEPN2vk9SwapChainE, @function
_ZN11GfxDeviceVK12AdvanceImageEPN2vk9SwapChainE:
    push   rbp
    push   r15
    push   r14
    push   rbx
    sub    rsp,0x28
    mov    r14,rsi
    mov    rbx,rdi
    mov    DWORD PTR [rsp+0xc],0xffffffff
    mov    BYTE PTR [rsp+0xb],0x0
    xorps  xmm0,xmm0
    movaps XMMWORD PTR [rsp+0x10],xmm0
    mov    rdi,QWORD PTR [rdi+0x1e88]
    cmp    BYTE PTR [rdi+0x8],0x0
    je     lbl0
    call   _ZN18FrameTimingManager31OnRenderThreadOverheadBeginImplEv
lbl0:
    lea    r15,[rbx+0xf680]
    mov    rdx,QWORD PTR [rbx+0xf470]
    mov    DWORD PTR [rsp],0x1
    lea    rcx,[rsp+0xc]
    lea    r8,[rsp+0xb]
    lea    r9,[rsp+0x10]
    mov    rdi,r14
    mov    rsi,r15
    call   _ZN2vk9SwapChain16AcquireNextImageERNS_13FrameTrackingEmPjPbP11fixed_arrayIP13VkSemaphore_TLm2EEb
    mov    ebp,eax
    mov    rdi,QWORD PTR [rbx+0x1e88]
    cmp    BYTE PTR [rdi+0x8],0x0
    je     lbl1
    call   _ZN18FrameTimingManager29OnRenderThreadOverheadEndImplEv
lbl1:
    cmp    ebp,0x4
    jne    lbl3
    mov    rdi,rbx
    xor    esi,esi
    xor    edx,edx
    call   _ZN11GfxDeviceVK27SubmitCurrentCommandBuffersEP13VkSemaphore_Tb
    mov    rdi,QWORD PTR [rbx+0xf9a8]
    call   _ZN2vk12TaskExecutor4SyncEv
    mov    rdi,r15
    call   _ZN2vk13FrameTracking15UpdateSafeFrameEv
    mov    rdi,rbx
    mov    esi,0x2
    mov    edx,0x1
    call   _ZN15GfxDeviceVKBase26EnsureCurrentCommandBufferEN2vk13CommandBuffer4TypeEb
    mov    rdi,QWORD PTR [rbx+0x1e88]
    cmp    BYTE PTR [rdi+0x8],0x0
    je     lbl2
    call   _ZN18FrameTimingManager31OnRenderThreadOverheadBeginImplEv
lbl2:
    mov    rdx,QWORD PTR [rbx+0xf470]
    mov    DWORD PTR [rsp],0x0
    lea    rcx,[rsp+0xc]
    lea    r8,[rsp+0xb]
    lea    r9,[rsp+0x10]
    mov    rdi,r14
    mov    rsi,r15
    call   _ZN2vk9SwapChain16AcquireNextImageERNS_13FrameTrackingEmPjPbP11fixed_arrayIP13VkSemaphore_TLm2EEb
    mov    rdi,QWORD PTR [rbx+0x1e88]
    cmp    BYTE PTR [rdi+0x8],0x0
    je     lbl3
    call   _ZN18FrameTimingManager29OnRenderThreadOverheadEndImplEv
lbl3:
    mov    rsi,QWORD PTR [rsp+0x10]
    test   rsi,rsi
    je     lbl4
    mov    rdi,QWORD PTR [rbx+0xf9a8]
    mov    edx,0x400
    call   _ZN2vk12TaskExecutor22AddSubmitWaitSemaphoreEP13VkSemaphore_Tj
lbl4:
    mov    rsi,QWORD PTR [rsp+0x18]
    test   rsi,rsi
    je     lbl5
    mov    rdi,QWORD PTR [rbx+0xf9a8]
    mov    edx,0x400
    call   _ZN2vk12TaskExecutor22AddSubmitWaitSemaphoreEP13VkSemaphore_Tj
lbl5:
    cmp    BYTE PTR [rsp+0xb],0x0
    je     lbl6
    movzx  edx,WORD PTR [rbx+0x1e08]
    mov    rsi,QWORD PTR [rbx+0xf460]
    add    rbx,0x2620
    mov    rdi,rbx
    call   _ZN18VKImmediateContext17BackbufferChangedERN2vk12RenderPassesEt
lbl6:
    mov    eax,DWORD PTR [rsp+0xc]
    mov    DWORD PTR [r14+0x134],eax
    add    rsp,0x28
    pop    rbx
    pop    r14
    pop    r15
    pop    rbp
    ret
    .align 16, 0xCC

# Custom version of vk::SwapChain::AcquireNextImage(vk::FrameTracking&, unsigned long, unsigned int*, bool*, fixed_array<VkSemaphore_T*, 2ul>*, bool)
# Now with a extra bool parameter at the end
.section .text.patch3,"ax"
.global _ZN2vk9SwapChain16AcquireNextImageERNS_13FrameTrackingEmPjPbP11fixed_arrayIP13VkSemaphore_TLm2EEb
.type _ZN2vk9SwapChain16AcquireNextImageERNS_13FrameTrackingEmPjPbP11fixed_arrayIP13VkSemaphore_TLm2EEb, @function
_ZN2vk9SwapChain16AcquireNextImageERNS_13FrameTrackingEmPjPbP11fixed_arrayIP13VkSemaphore_TLm2EEb:
    push   rbp
    push   r15
    push   r14
    push   r13
    push   r12
    push   rbx
    sub    rsp,0x28
    mov    r15,r9
    mov    r12,r8
    mov    r14,rcx
    mov    rbx,rdx
    mov    rbp,rsi
    mov    r13,rdi
    lea    rdi,[rip+_ZN2vkL17gAcquireNextFrameE]
    call   _Z14profiler_beginPN9profiling6MarkerE
    xorps  xmm0,xmm0
    movups XMMWORD PTR [r15],xmm0
    cmp    BYTE PTR [r13+0x14c],0x0
    je     lbl8
    mov    al,BYTE PTR [r13+0x14d]
    mov    BYTE PTR [r12],al
    mov    BYTE PTR [r13+0x14d],0x0
    mov    eax,DWORD PTR [r13+0x148]
    mov    DWORD PTR [r14],eax
    mov    eax,DWORD PTR [r13+0x148]
    add    eax,0x1
    xor    ecx,ecx
    cmp    QWORD PTR [r13+0x90],rax
    cmova  ecx,eax
    mov    DWORD PTR [r13+0x148],ecx
    mov    rax,QWORD PTR [r13+0x150]
    mov    rdx,QWORD PTR [rax+rcx*8]
    mov    QWORD PTR [rsp+0x8],rdx
    mov    QWORD PTR [rax+rcx*8],0x0
    cmp    QWORD PTR [rsp+0x8],0x0
    je     lbl7
    mov    rdi,QWORD PTR [r13+0xb8]
    lea    rdx,[rsp+0x8]
    mov    esi,0x1
    mov    ecx,0x1
    mov    r8,0xffffffffffffffff
    call   QWORD PTR [rip+_ZN6vulkan4fptr15vkWaitForFencesE]
    call   _Z18GetVKGfxDeviceCorev
    add    rax,0xf680
    mov    rsi,QWORD PTR [rsp+0x8]
    mov    rdi,rax
    call   _ZN2vk13FrameTracking11ReturnFenceEP9VkFence_T
lbl7:
    add    DWORD PTR [r13+0x13c],0x1
    mov    ebp,0x1
    jmp    lbl19
lbl8:
    mov    QWORD PTR [rsp+0x20],rbx
    mov    BYTE PTR [r12],0x0
    cmp    QWORD PTR [r13+0x68],0x0
    mov    QWORD PTR [rsp+0x10],rbp
    je     lbl10
    mov    rdi,rbp
    call   _ZN2vk13FrameTracking12GetSemaphoreEv
    mov    rbp,rax
    mov    rsi,QWORD PTR [r13+0x68]
    mov    rdi,QWORD PTR [r13+0xb8]
    mov    rdx,0xffffffffffffffff
    mov    rcx,rax
    xor    r8d,r8d
    mov    r9,r14
    call   QWORD PTR [rip+_ZN6vulkan3ext21vkAcquireNextImageKHRE]
    mov    ebx,eax
    test   eax,eax
    js     lbl14
    mov    rdi,QWORD PTR [rsp+0x10]
    mov    rsi,rbp
    mov    rdx,QWORD PTR [rsp+0x20]
    call   _ZN2vk13FrameTracking15ReturnSemaphoreEP13VkSemaphore_Tm
    mov    QWORD PTR [r15],rbp
    mov    eax,0x1
    cmp    ebx,0xc4653214
    je     lbl11
lbl9:
    cmp    ebx,0x3b9acdeb
    je     lbl11
    cmp    ebx,0xc4653600
    je     lbl11
    jmp    lbl13
lbl10:
    mov    ebx,0xc4653214
    xor    eax,eax
lbl11:
    mov    QWORD PTR [rsp+0x18],rax
    mov    ebp,0x4
    cmp    BYTE PTR [rsp+0x60],0x0
    jne    lbl19
    mov    rdi,QWORD PTR [r13+0xb8]
    call   QWORD PTR [rip+_ZN6vulkan4fptr16vkDeviceWaitIdleE]
    lea    rsi,[r13+0xf8]
    mov    rax,QWORD PTR [r13+0x0]
    mov    rdi,r13
    xor    edx,edx
    xor    ecx,ecx
    call   QWORD PTR [rax+0x18]
    mov    BYTE PTR [r12],0x1
    cmp    BYTE PTR [r13+0x14c],0x0
    je     lbl15
    mov    eax,DWORD PTR [r13+0x148]
    mov    DWORD PTR [r14],eax
    mov    eax,DWORD PTR [r13+0x148]
    add    eax,0x1
    xor    ecx,ecx
    cmp    QWORD PTR [r13+0x90],rax
    cmova  ecx,eax
    mov    DWORD PTR [r13+0x148],ecx
    mov    rax,QWORD PTR [r13+0x150]
    mov    rdx,QWORD PTR [rax+rcx*8]
    mov    QWORD PTR [rsp+0x8],rdx
    mov    QWORD PTR [rax+rcx*8],0x0
    cmp    QWORD PTR [rsp+0x8],0x0
    je     lbl12
    mov    rdi,QWORD PTR [r13+0xb8]
    lea    rdx,[rsp+0x8]
    mov    esi,0x1
    mov    ecx,0x1
    mov    r8,0xffffffffffffffff
    call   QWORD PTR [rip+_ZN6vulkan4fptr15vkWaitForFencesE]
    call   _Z18GetVKGfxDeviceCorev
    add    rax,0xf680
    mov    rsi,QWORD PTR [rsp+0x8]
    mov    rdi,rax
    call   _ZN2vk13FrameTracking11ReturnFenceEP9VkFence_T
lbl12:
    add    DWORD PTR [r13+0x13c],0x1
lbl13:
    test   ebx,ebx
    jns    lbl17
    jmp    lbl18
lbl14:
    mov    rdi,QWORD PTR [rsp+0x10]
    mov    rsi,rbp
    call   _ZN2vk13FrameTracking21ReturnSemaphoreUnusedEP13VkSemaphore_T
    xor    eax,eax
    cmp    ebx,0xc4653214
    jne    lbl9
    jmp    lbl11
lbl15:
    mov    rdi,QWORD PTR [rsp+0x10]
    call   _ZN2vk13FrameTracking12GetSemaphoreEv
    mov    r12,rax
    mov    rsi,QWORD PTR [r13+0x68]
    mov    rdi,QWORD PTR [r13+0xb8]
    mov    rdx,0xffffffffffffffff
    mov    rcx,rax
    xor    r8d,r8d
    mov    r9,r14
    call   QWORD PTR [rip+_ZN6vulkan3ext21vkAcquireNextImageKHRE]
    mov    ebx,eax
    test   eax,eax
    js     lbl20
    mov    rdi,QWORD PTR [rsp+0x10]
    mov    rsi,r12
    mov    rdx,QWORD PTR [rsp+0x20]
    call   _ZN2vk13FrameTracking15ReturnSemaphoreEP13VkSemaphore_Tm
    cmp    QWORD PTR [rsp+0x18],0x2
    jb     lbl16
    lea    rsi,[rip+aRuntimeUtilitiesFixedArray]
    lea    r8,[rip+aArrayIndexOutOfRange]
    xor    edi,edi
    mov    edx,0x9d
    mov    ecx,0xffffffff
    call   _Z20AssertImplementationiPKciiS0_
lbl16:
    mov    rax,QWORD PTR [rsp+0x18]
    mov    QWORD PTR [r15+rax*8],r12
lbl17:
    add    DWORD PTR [r13+0x13c],0x1
lbl18:
    not    ebx
    shr    ebx,0x1f
    mov    ebp,ebx
lbl19:
    lea    rdi,[rip+_ZN2vkL17gAcquireNextFrameE]
    call   _Z12profiler_endPN9profiling6MarkerE
    mov    eax,ebp
    add    rsp,0x28
    pop    rbx
    pop    r12
    pop    r13
    pop    r14
    pop    r15
    pop    rbp
    ret
lbl20:
    mov    rdi,QWORD PTR [rsp+0x10]
    mov    rsi,r12
    call   _ZN2vk13FrameTracking21ReturnSemaphoreUnusedEP13VkSemaphore_T
    jmp    lbl18
    jmp    lbl21
    jmp    lbl21
    jmp    lbl21
    mov    rdi,rax
    call   __clang_call_terminate
    mov    rbx,rax
    lea    rdi,[rip+_ZN2vkL17gAcquireNextFrameE]
    call   _Z12profiler_endPN9profiling6MarkerE
    jmp    lbl22
    mov    rdi,rax
    call   __clang_call_terminate
lbl21:
    mov    rbx,rax
    lea    rdi,[rip+_ZN2vkL17gAcquireNextFrameE]
    call   _Z12profiler_endPN9profiling6MarkerE
lbl22:
    mov    rdi,rbx
    call   _Unwind_Resume
    mov    rdi,rax
    call   __clang_call_terminate
    .align 16, 0xCC