caio/clang-p2996
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
890c4bece26e005cd9fa5511fe0efa7307794de5
clang-p2996/llvm/test/CodeGen/AArch64/ptrauth-call.ll
Daniil Kovalev 3d5e8e4693 [PAC][CodeGen] Do not emit trivial 'mov xN, xN' on tail call (#109100) 
Under some conditions, a trivial `mov xN xN` instruction was emitted on
tail calls. Consider the following code:

```
class Test {
public:
  virtual void f() {}
};

void call_f(Test *t) {
  t->f();
}
```

Correponding assembly:

```
_Z6call_fP4Test:
        ldr     x16, [x0]
        mov     x17, x0
        movk    x17, #6503, lsl #48
        autda   x16, x17
        ldr     x1, [x16]
 =====> mov     x16, x16
        movk    x16, #54167, lsl #48
        braa    x1, x16
```

This patch makes such movs being omitted.

Co-authored-by: Anatoly Trosinenko <atrosinenko@accesssoftek.com>
2024-09-19 12:17:58 +03:00

426 lines
13 KiB
LLVM
Raw Blame History

; RUN: llc -mtriple arm64e-apple-darwin -o - %s -asm-verbose=0 \
; RUN: | FileCheck %s --check-prefixes=CHECK,DARWIN
; RUN: llc -mtriple aarch64 -mattr=+pauth -o - %s -asm-verbose=0 \
; RUN: | FileCheck %s --check-prefixes=CHECK,ELF
; RUN: llc -mtriple arm64e-apple-darwin -o - %s -asm-verbose=0 \
; RUN: -global-isel -global-isel-abort=1 -verify-machineinstrs \
; RUN: | FileCheck %s --check-prefixes=CHECK,DARWIN
; RUN: llc -mtriple aarch64 -mattr=+pauth -o - %s -asm-verbose=0 \
; RUN: -global-isel -global-isel-abort=1 -verify-machineinstrs \
; RUN: | FileCheck %s --check-prefixes=CHECK,ELF
define i32 @test_call_ia_0(ptr %arg0) #0 {
; DARWIN-LABEL: test_call_ia_0:
; DARWIN-NEXT: stp x29, x30, [sp, #-16]!
; DARWIN-NEXT: blraaz x0
; DARWIN-NEXT: ldp x29, x30, [sp], #16
; DARWIN-NEXT: ret
;
; ELF-LABEL: test_call_ia_0:
; ELF-NEXT: str x30, [sp, #-16]!
; ELF-NEXT: blraaz x0
; ELF-NEXT: ldr x30, [sp], #16
; ELF-NEXT: ret
%tmp0 = call i32 %arg0() [ "ptrauth"(i32 0, i64 0) ]
ret i32 %tmp0
}
define i32 @test_call_ib_0(ptr %arg0) #0 {
; DARWIN-LABEL: test_call_ib_0:
; DARWIN-NEXT: stp x29, x30, [sp, #-16]!
; DARWIN-NEXT: blrabz x0
; DARWIN-NEXT: ldp x29, x30, [sp], #16
; DARWIN-NEXT: ret
;
; ELF-LABEL: test_call_ib_0:
; ELF-NEXT: str x30, [sp, #-16]!
; ELF-NEXT: blrabz x0
; ELF-NEXT: ldr x30, [sp], #16
; ELF-NEXT: ret
%tmp0 = call i32 %arg0() [ "ptrauth"(i32 1, i64 0) ]
ret i32 %tmp0
}
define i32 @test_tailcall_ia_0(ptr %arg0) #0 {
; CHECK-LABEL: test_tailcall_ia_0:
; CHECK-NEXT: braaz x0
%tmp0 = tail call i32 %arg0() [ "ptrauth"(i32 0, i64 0) ]
ret i32 %tmp0
}
define i32 @test_tailcall_ib_0(ptr %arg0) #0 {
; CHECK-LABEL: test_tailcall_ib_0:
; CHECK-NEXT: brabz x0
%tmp0 = tail call i32 %arg0() [ "ptrauth"(i32 1, i64 0) ]
ret i32 %tmp0
}
define i32 @test_call_ia_imm(ptr %arg0) #0 {
; DARWIN-LABEL: test_call_ia_imm:
; DARWIN-NEXT: stp x29, x30, [sp, #-16]!
; DARWIN-NEXT: mov x17, #42
; DARWIN-NEXT: blraa x0, x17
; DARWIN-NEXT: ldp x29, x30, [sp], #16
; DARWIN-NEXT: ret
;
; ELF-LABEL: test_call_ia_imm:
; ELF-NEXT: str x30, [sp, #-16]!
; ELF-NEXT: mov x17, #42
; ELF-NEXT: blraa x0, x17
; ELF-NEXT: ldr x30, [sp], #16
; ELF-NEXT: ret
%tmp0 = call i32 %arg0() [ "ptrauth"(i32 0, i64 42) ]
ret i32 %tmp0
}
define i32 @test_call_ib_imm(ptr %arg0) #0 {
; DARWIN-LABEL: test_call_ib_imm:
; DARWIN-NEXT: stp x29, x30, [sp, #-16]!
; DARWIN-NEXT: mov x17, #42
; DARWIN-NEXT: blrab x0, x17
; DARWIN-NEXT: ldp x29, x30, [sp], #16
; DARWIN-NEXT: ret
;
; ELF-LABEL: test_call_ib_imm:
; ELF-NEXT: str x30, [sp, #-16]!
; ELF-NEXT: mov x17, #42
; ELF-NEXT: blrab x0, x17
; ELF-NEXT: ldr x30, [sp], #16
; ELF-NEXT: ret
%tmp0 = call i32 %arg0() [ "ptrauth"(i32 1, i64 42) ]
ret i32 %tmp0
}
define i32 @test_tailcall_ia_imm(ptr %arg0) #0 {
; CHECK-LABEL: test_tailcall_ia_imm:
; CHECK-NEXT: mov x16, #42
; CHECK-NEXT: braa x0, x16
%tmp0 = tail call i32 %arg0() [ "ptrauth"(i32 0, i64 42) ]
ret i32 %tmp0
}
define i32 @test_tailcall_ib_imm(ptr %arg0) #0 {
; CHECK-LABEL: test_tailcall_ib_imm:
; CHECK-NEXT: mov x16, #42
; CHECK-NEXT: brab x0, x16
%tmp0 = tail call i32 %arg0() [ "ptrauth"(i32 1, i64 42) ]
ret i32 %tmp0
}
define i32 @test_call_ia_var(ptr %arg0, ptr %arg1) #0 {
; DARWIN-LABEL: test_call_ia_var:
; DARWIN-NEXT: stp x29, x30, [sp, #-16]!
; DARWIN-NEXT: ldr x8, [x1]
; DARWIN-NEXT: blraa x0, x8
; DARWIN-NEXT: ldp x29, x30, [sp], #16
; DARWIN-NEXT: ret
;
; ELF-LABEL: test_call_ia_var:
; ELF-NEXT: str x30, [sp, #-16]!
; ELF-NEXT: ldr x8, [x1]
; ELF-NEXT: blraa x0, x8
; ELF-NEXT: ldr x30, [sp], #16
; ELF-NEXT: ret
%tmp0 = load i64, ptr %arg1
%tmp1 = call i32 %arg0() [ "ptrauth"(i32 0, i64 %tmp0) ]
ret i32 %tmp1
}
define i32 @test_call_ib_var(ptr %arg0, ptr %arg1) #0 {
; DARWIN-LABEL: test_call_ib_var:
; DARWIN-NEXT: stp x29, x30, [sp, #-16]!
; DARWIN-NEXT: ldr x8, [x1]
; DARWIN-NEXT: blrab x0, x8
; DARWIN-NEXT: ldp x29, x30, [sp], #16
; DARWIN-NEXT: ret
;
; ELF-LABEL: test_call_ib_var:
; ELF-NEXT: str x30, [sp, #-16]!
; ELF-NEXT: ldr x8, [x1]
; ELF-NEXT: blrab x0, x8
; ELF-NEXT: ldr x30, [sp], #16
; ELF-NEXT: ret
%tmp0 = load i64, ptr %arg1
%tmp1 = call i32 %arg0() [ "ptrauth"(i32 1, i64 %tmp0) ]
ret i32 %tmp1
}
define i32 @test_tailcall_ia_var(ptr %arg0, ptr %arg1) #0 {
; CHECK-LABEL: test_tailcall_ia_var:
; CHECK: ldr x1, [x1]
; CHECK: braa x0, x1
%tmp0 = load i64, ptr %arg1
%tmp1 = tail call i32 %arg0() [ "ptrauth"(i32 0, i64 %tmp0) ]
ret i32 %tmp1
}
define i32 @test_tailcall_ib_var(ptr %arg0, ptr %arg1) #0 {
; CHECK-LABEL: test_tailcall_ib_var:
; CHECK: ldr x1, [x1]
; CHECK: brab x0, x1
%tmp0 = load i64, ptr %arg1
%tmp1 = tail call i32 %arg0() [ "ptrauth"(i32 1, i64 %tmp0) ]
ret i32 %tmp1
}
define void @test_tailcall_omit_mov_x16_x16(ptr %objptr) #0 {
; CHECK-LABEL: test_tailcall_omit_mov_x16_x16:
; CHECK: ldr x16, [x0]
; CHECK: mov x17, x0
; CHECK: movk x17, #6503, lsl #48
; CHECK: autda x16, x17
; CHECK: ldr x1, [x16]
; CHECK: movk x16, #54167, lsl #48
; CHECK: braa x1, x16
%vtable.signed = load ptr, ptr %objptr, align 8
%objptr.int = ptrtoint ptr %objptr to i64
%vtable.discr = tail call i64 @llvm.ptrauth.blend(i64 %objptr.int, i64 6503)
%vtable.signed.int = ptrtoint ptr %vtable.signed to i64
%vtable.unsigned.int = tail call i64 @llvm.ptrauth.auth(i64 %vtable.signed.int, i32 2, i64 %vtable.discr)
%vtable.unsigned = inttoptr i64 %vtable.unsigned.int to ptr
%virt.func.signed = load ptr, ptr %vtable.unsigned, align 8
%virt.func.discr = tail call i64 @llvm.ptrauth.blend(i64 %vtable.unsigned.int, i64 54167)
tail call void %virt.func.signed(ptr %objptr) [ "ptrauth"(i32 0, i64 %virt.func.discr) ]
ret void
}
define i32 @test_call_ia_arg(ptr %arg0, i64 %arg1) #0 {
; DARWIN-LABEL: test_call_ia_arg:
; DARWIN-NEXT: stp x29, x30, [sp, #-16]!
; DARWIN-NEXT: blraa x0, x1
; DARWIN-NEXT: ldp x29, x30, [sp], #16
; DARWIN-NEXT: ret
;
; ELF-LABEL: test_call_ia_arg:
; ELF-NEXT: str x30, [sp, #-16]!
; ELF-NEXT: blraa x0, x1
; ELF-NEXT: ldr x30, [sp], #16
; ELF-NEXT: ret
%tmp0 = call i32 %arg0() [ "ptrauth"(i32 0, i64 %arg1) ]
ret i32 %tmp0
}
define i32 @test_call_ib_arg(ptr %arg0, i64 %arg1) #0 {
; DARWIN-LABEL: test_call_ib_arg:
; DARWIN-NEXT: stp x29, x30, [sp, #-16]!
; DARWIN-NEXT: blrab x0, x1
; DARWIN-NEXT: ldp x29, x30, [sp], #16
; DARWIN-NEXT: ret
;
; ELF-LABEL: test_call_ib_arg:
; ELF-NEXT: str x30, [sp, #-16]!
; ELF-NEXT: blrab x0, x1
; ELF-NEXT: ldr x30, [sp], #16
; ELF-NEXT: ret
%tmp0 = call i32 %arg0() [ "ptrauth"(i32 1, i64 %arg1) ]
ret i32 %tmp0
}
define i32 @test_tailcall_ia_arg(ptr %arg0, i64 %arg1) #0 {
; CHECK-LABEL: test_tailcall_ia_arg:
; CHECK: braa x0, x1
%tmp0 = tail call i32 %arg0() [ "ptrauth"(i32 0, i64 %arg1) ]
ret i32 %tmp0
}
define i32 @test_tailcall_ib_arg(ptr %arg0, i64 %arg1) #0 {
; CHECK-LABEL: test_tailcall_ib_arg:
; CHECK: brab x0, x1
%tmp0 = tail call i32 %arg0() [ "ptrauth"(i32 1, i64 %arg1) ]
ret i32 %tmp0
}
define i32 @test_call_ia_arg_ind(ptr %arg0, i64 %arg1) #0 {
; DARWIN-LABEL: test_call_ia_arg_ind:
; DARWIN-NEXT: stp x29, x30, [sp, #-16]!
; DARWIN-NEXT: ldr x8, [x0]
; DARWIN-NEXT: blraa x8, x1
; DARWIN-NEXT: ldp x29, x30, [sp], #16
; DARWIN-NEXT: ret
;
; ELF-LABEL: test_call_ia_arg_ind:
; ELF-NEXT: str x30, [sp, #-16]!
; ELF-NEXT: ldr x8, [x0]
; ELF-NEXT: blraa x8, x1
; ELF-NEXT: ldr x30, [sp], #16
; ELF-NEXT: ret
%tmp0 = load ptr, ptr %arg0
%tmp1 = call i32 %tmp0() [ "ptrauth"(i32 0, i64 %arg1) ]
ret i32 %tmp1
}
define i32 @test_call_ib_arg_ind(ptr %arg0, i64 %arg1) #0 {
; DARWIN-LABEL: test_call_ib_arg_ind:
; DARWIN-NEXT: stp x29, x30, [sp, #-16]!
; DARWIN-NEXT: ldr x8, [x0]
; DARWIN-NEXT: blrab x8, x1
; DARWIN-NEXT: ldp x29, x30, [sp], #16
; DARWIN-NEXT: ret
;
; ELF-LABEL: test_call_ib_arg_ind:
; ELF-NEXT: str x30, [sp, #-16]!
; ELF-NEXT: ldr x8, [x0]
; ELF-NEXT: blrab x8, x1
; ELF-NEXT: ldr x30, [sp], #16
; ELF-NEXT: ret
%tmp0 = load ptr, ptr %arg0
%tmp1 = call i32 %tmp0() [ "ptrauth"(i32 1, i64 %arg1) ]
ret i32 %tmp1
}
define i32 @test_tailcall_ia_arg_ind(ptr %arg0, i64 %arg1) #0 {
; CHECK-LABEL: test_tailcall_ia_arg_ind:
; CHECK: ldr x0, [x0]
; CHECK: braa x0, x1
%tmp0 = load ptr, ptr %arg0
%tmp1 = tail call i32 %tmp0() [ "ptrauth"(i32 0, i64 %arg1) ]
ret i32 %tmp1
}
define i32 @test_tailcall_ib_arg_ind(ptr %arg0, i64 %arg1) #0 {
; CHECK-LABEL: test_tailcall_ib_arg_ind:
; CHECK: ldr x0, [x0]
; CHECK: brab x0, x1
%tmp0 = load ptr, ptr %arg0
%tmp1 = tail call i32 %tmp0() [ "ptrauth"(i32 1, i64 %arg1) ]
ret i32 %tmp1
}
; Test direct calls
define i32 @test_direct_call() #0 {
; DARWIN-LABEL: test_direct_call:
; DARWIN-NEXT: stp x29, x30, [sp, #-16]!
; DARWIN-NEXT: bl _f
; DARWIN-NEXT: ldp x29, x30, [sp], #16
; DARWIN-NEXT: ret
;
; ELF-LABEL: test_direct_call:
; ELF-NEXT: str x30, [sp, #-16]!
; ELF-NEXT: bl f
; ELF-NEXT: ldr x30, [sp], #16
; ELF-NEXT: ret
%tmp0 = call i32 ptrauth(ptr @f, i32 0, i64 42)() [ "ptrauth"(i32 0, i64 42) ]
ret i32 %tmp0
}
define i32 @test_direct_tailcall(ptr %arg0) #0 {
; DARWIN-LABEL: test_direct_tailcall:
; DARWIN: b _f
;
; ELF-LABEL: test_direct_tailcall:
; ELF-NEXT: b f
%tmp0 = tail call i32 ptrauth(ptr @f, i32 0, i64 42)() [ "ptrauth"(i32 0, i64 42) ]
ret i32 %tmp0
}
define i32 @test_direct_call_mismatch() #0 {
; DARWIN-LABEL: test_direct_call_mismatch:
; DARWIN-NEXT: stp x29, x30, [sp, #-16]!
; DARWIN-NEXT: adrp x16, _f@GOTPAGE
; DARWIN-NEXT: ldr x16, [x16, _f@GOTPAGEOFF]
; DARWIN-NEXT: mov x17, #42
; DARWIN-NEXT: pacia x16, x17
; DARWIN-NEXT: mov x8, x16
; DARWIN-NEXT: mov x17, #42
; DARWIN-NEXT: blrab x8, x17
; DARWIN-NEXT: ldp x29, x30, [sp], #16
; DARWIN-NEXT: ret
;
; ELF-LABEL: test_direct_call_mismatch:
; ELF-NEXT: str x30, [sp, #-16]!
; ELF-NEXT: adrp x16, :got:f
; ELF-NEXT: ldr x16, [x16, :got_lo12:f]
; ELF-NEXT: mov x17, #42
; ELF-NEXT: pacia x16, x17
; ELF-NEXT: mov x8, x16
; ELF-NEXT: mov x17, #42
; ELF-NEXT: blrab x8, x17
; ELF-NEXT: ldr x30, [sp], #16
; ELF-NEXT: ret
%tmp0 = call i32 ptrauth(ptr @f, i32 0, i64 42)() [ "ptrauth"(i32 1, i64 42) ]
ret i32 %tmp0
}
define i32 @test_direct_call_addr() #0 {
; DARWIN-LABEL: test_direct_call_addr:
; DARWIN-NEXT: stp x29, x30, [sp, #-16]!
; DARWIN-NEXT: bl _f
; DARWIN-NEXT: ldp x29, x30, [sp], #16
; DARWIN-NEXT: ret
;
; ELF-LABEL: test_direct_call_addr:
; ELF-NEXT: str x30, [sp, #-16]!
; ELF-NEXT: bl f
; ELF-NEXT: ldr x30, [sp], #16
; ELF-NEXT: ret
%tmp0 = call i32 ptrauth(ptr @f, i32 1, i64 0, ptr @f.ref.ib.0.addr)() [ "ptrauth"(i32 1, i64 ptrtoint (ptr @f.ref.ib.0.addr to i64)) ]
ret i32 %tmp0
}
define i32 @test_direct_call_addr_blend() #0 {
; DARWIN-LABEL: test_direct_call_addr_blend:
; DARWIN-NEXT: stp x29, x30, [sp, #-16]!
; DARWIN-NEXT: bl _f
; DARWIN-NEXT: ldp x29, x30, [sp], #16
; DARWIN-NEXT: ret
;
; ELF-LABEL: test_direct_call_addr_blend:
; ELF-NEXT: str x30, [sp, #-16]!
; ELF-NEXT: bl f
; ELF-NEXT: ldr x30, [sp], #16
; ELF-NEXT: ret
%tmp0 = call i64 @llvm.ptrauth.blend(i64 ptrtoint (ptr @f.ref.ib.42.addr to i64), i64 42)
%tmp1 = call i32 ptrauth(ptr @f, i32 1, i64 42, ptr @f.ref.ib.42.addr)() [ "ptrauth"(i32 1, i64 %tmp0) ]
ret i32 %tmp1
}
define i32 @test_direct_call_addr_gep_different_index_types() #0 {
; DARWIN-LABEL: test_direct_call_addr_gep_different_index_types:
; DARWIN-NEXT: stp x29, x30, [sp, #-16]!
; DARWIN-NEXT: bl _f
; DARWIN-NEXT: ldp x29, x30, [sp], #16
; DARWIN-NEXT: ret
;
; ELF-LABEL: test_direct_call_addr_gep_different_index_types:
; ELF-NEXT: str x30, [sp, #-16]!
; ELF-NEXT: bl f
; ELF-NEXT: ldr x30, [sp], #16
; ELF-NEXT: ret
%tmp0 = call i32 ptrauth(ptr @f, i32 1, i64 0, ptr getelementptr ({ ptr }, ptr @f_struct.ref.ib.0.addr, i64 0, i32 0))() [ "ptrauth"(i32 1, i64 ptrtoint (ptr getelementptr ({ ptr }, ptr @f_struct.ref.ib.0.addr, i32 0, i32 0) to i64)) ]
ret i32 %tmp0
}
define i32 @test_direct_call_addr_blend_gep_different_index_types() #0 {
; DARWIN-LABEL: test_direct_call_addr_blend_gep_different_index_types:
; DARWIN-NEXT: stp x29, x30, [sp, #-16]!
; DARWIN-NEXT: bl _f
; DARWIN-NEXT: ldp x29, x30, [sp], #16
; DARWIN-NEXT: ret
;
; ELF-LABEL: test_direct_call_addr_blend_gep_different_index_types:
; ELF-NEXT: str x30, [sp, #-16]!
; ELF-NEXT: bl f
; ELF-NEXT: ldr x30, [sp], #16
; ELF-NEXT: ret
%tmp0 = call i64 @llvm.ptrauth.blend(i64 ptrtoint (ptr getelementptr ({ ptr }, ptr @f_struct.ref.ib.123.addr, i32 0, i32 0) to i64), i64 123)
%tmp1 = call i32 ptrauth(ptr @f, i32 1, i64 123, ptr getelementptr ({ ptr }, ptr @f_struct.ref.ib.123.addr, i64 0, i32 0))() [ "ptrauth"(i32 1, i64 %tmp0) ]
ret i32 %tmp1
}
@f.ref.ib.42.addr = external global ptr
@f.ref.ib.0.addr = external global ptr
@f_struct.ref.ib.0.addr = external global ptr
@f_struct.ref.ib.123.addr = external global ptr
declare void @f()
declare i64 @llvm.ptrauth.auth(i64, i32, i64)
declare i64 @llvm.ptrauth.blend(i64, i64)
attributes #0 = { nounwind }
Reference in New Issue View Git Blame Copy Permalink