a88c722e68
PAC/BTI-related codegen in the AArch64 backend is controlled by a set of LLVM IR function attributes, added to the function by Clang, based on command-line options and GCC-style function attributes. However, functions, generated in the LLVM middle end (for example, asan.module.ctor or __llvm_gcov_write_out) do not get any attributes and the backend incorrectly does not do any PAC/BTI code generation. This patch record the default state of PAC/BTI codegen in a set of LLVM IR module-level attributes, based on command-line options: * "sign-return-address", with non-zero value means generate code to sign return addresses (PAC-RET), zero value means disable PAC-RET. * "sign-return-address-all", with non-zero value means enable PAC-RET for all functions, zero value means enable PAC-RET only for functions, which spill LR. * "sign-return-address-with-bkey", with non-zero value means use B-key for signing, zero value mean use A-key. This set of attributes are always added for AArch64 targets (as opposed, for example, to interpreting a missing attribute as having a value 0) in order to be able to check for conflicts when combining module attributed during LTO. Module-level attributes are overridden by function level attributes. All the decision making about whether to not to generate PAC and/or BTI code is factored out into AArch64FunctionInfo, there shouldn't be any places left, other than AArch64FunctionInfo, which directly examine PAC/BTI attributes, except AArch64AsmPrinter.cpp, which is/will-be handled by a separate patch. Differential Revision: https://reviews.llvm.org/D85649
47 lines
2.5 KiB
C
47 lines
2.5 KiB
C
// RUN: %clang -target aarch64-arm-none-eabi -S -emit-llvm -o - -msign-return-address=none %s | FileCheck %s --check-prefix=CHECK --check-prefix=NONE
|
|
// RUN: %clang -target aarch64-arm-none-eabi -S -emit-llvm -o - -msign-return-address=all %s | FileCheck %s --check-prefix=CHECK --check-prefix=ALL
|
|
// RUN: %clang -target aarch64-arm-none-eabi -S -emit-llvm -o - -msign-return-address=non-leaf %s | FileCheck %s --check-prefix=CHECK --check-prefix=PART
|
|
|
|
// RUN: %clang -target aarch64-arm-none-eabi -S -emit-llvm -o - -mbranch-protection=none %s | FileCheck %s --check-prefix=CHECK --check-prefix=NONE
|
|
// RUN: %clang -target aarch64-arm-none-eabi -S -emit-llvm -o - -mbranch-protection=pac-ret+leaf %s | FileCheck %s --check-prefix=CHECK --check-prefix=ALL
|
|
// RUN: %clang -target aarch64-arm-none-eabi -S -emit-llvm -o - -mbranch-protection=pac-ret+b-key %s | FileCheck %s --check-prefix=CHECK --check-prefix=B-KEY
|
|
// RUN: %clang -target aarch64-arm-none-eabi -S -emit-llvm -o - -mbranch-protection=bti %s | FileCheck %s --check-prefix=CHECK --check-prefix=BTE
|
|
|
|
// REQUIRES: aarch64-registered-target
|
|
|
|
// Check there are no branch protection function attributes
|
|
|
|
// CHECK-LABEL: @foo() #[[#ATTR:]]
|
|
|
|
// CHECK-NOT: attributes #[[#ATTR]] = { {{.*}} "sign-return-address"
|
|
// CHECK-NOT: attributes #[[#ATTR]] = { {{.*}} "sign-return-address-key"
|
|
// CHECK-NOT: attributes #[[#ATTR]] = { {{.*}} "branch-target-enforcement"
|
|
|
|
// Check module attributes
|
|
|
|
// NONE: !{i32 1, !"branch-target-enforcement", i32 0}
|
|
// ALL: !{i32 1, !"branch-target-enforcement", i32 0}
|
|
// PART: !{i32 1, !"branch-target-enforcement", i32 0}
|
|
// BTE: !{i32 1, !"branch-target-enforcement", i32 1}
|
|
// B-KEY: !{i32 1, !"branch-target-enforcement", i32 0}
|
|
|
|
// NONE: !{i32 1, !"sign-return-address", i32 0}
|
|
// ALL: !{i32 1, !"sign-return-address", i32 1}
|
|
// PART: !{i32 1, !"sign-return-address", i32 1}
|
|
// BTE: !{i32 1, !"sign-return-address", i32 0}
|
|
// B-KEY: !{i32 1, !"sign-return-address", i32 1}
|
|
|
|
// NONE: !{i32 1, !"sign-return-address-all", i32 0}
|
|
// ALL: !{i32 1, !"sign-return-address-all", i32 1}
|
|
// PART: !{i32 1, !"sign-return-address-all", i32 0}
|
|
// BTE: !{i32 1, !"sign-return-address-all", i32 0}
|
|
// B-KEY: !{i32 1, !"sign-return-address-all", i32 0}
|
|
|
|
// NONE: !{i32 1, !"sign-return-address-with-bkey", i32 0}
|
|
// ALL: !{i32 1, !"sign-return-address-with-bkey", i32 0}
|
|
// PART: !{i32 1, !"sign-return-address-with-bkey", i32 0}
|
|
// BTE: !{i32 1, !"sign-return-address-with-bkey", i32 0}
|
|
// B-KEY: !{i32 1, !"sign-return-address-with-bkey", i32 1}
|
|
|
|
void foo() {}
|