110 changes: 110 additions & 0 deletions clang/test/CodeGen/arm64-mte.c
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,110 @@
// Test memory tagging extension intrinsics
// RUN: %clang_cc1 -triple aarch64-none-linux-eabi -target-feature +mte -O3 -S -emit-llvm -o - %s | FileCheck %s
#include <stddef.h>
#include <arm_acle.h>

// CHECK-LABEL: define i32* @create_tag1
int *create_tag1(int *a, unsigned b) {
// CHECK: [[T0:%[0-9]+]] = bitcast i32* %a to i8*
// CHECK: [[T1:%[0-9]+]] = zext i32 %b to i64
// CHECK: [[T2:%[0-9]+]] = tail call i8* @llvm.aarch64.irg(i8* [[T0]], i64 [[T1]])
// CHECK: bitcast i8* [[T2]] to i32*
return __arm_mte_create_random_tag(a,b);
}

// CHECK-LABEL: define i16* @create_tag2
short *create_tag2(short *a, unsigned b) {
// CHECK: [[T0:%[0-9]+]] = bitcast i16* %a to i8*
// CHECK: [[T1:%[0-9]+]] = zext i32 %b to i64
// CHECK: [[T2:%[0-9]+]] = tail call i8* @llvm.aarch64.irg(i8* [[T0]], i64 [[T1]])
// CHECK: bitcast i8* [[T2]] to i16*
return __arm_mte_create_random_tag(a,b);
}

// CHECK-LABEL: define i8* @create_tag3
char *create_tag3(char *a, unsigned b) {
// CHECK: [[T1:%[0-9]+]] = zext i32 %b to i64
// CHECK: [[T2:%[0-9]+]] = tail call i8* @llvm.aarch64.irg(i8* %a, i64 [[T1]])
// CHECK: ret i8* [[T2:%[0-9]+]]
return __arm_mte_create_random_tag(a,b);
}

// CHECK-LABEL: define i8* @increment_tag1
char *increment_tag1(char *a) {
// CHECK: call i8* @llvm.aarch64.addg(i8* %a, i64 3)
return __arm_mte_increment_tag(a,3);
}

// CHECK-LABEL: define i16* @increment_tag2
short *increment_tag2(short *a) {
// CHECK: [[T0:%[0-9]+]] = bitcast i16* %a to i8*
// CHECK: [[T1:%[0-9]+]] = tail call i8* @llvm.aarch64.addg(i8* [[T0]], i64 3)
// CHECK: [[T2:%[0-9]+]] = bitcast i8* [[T1]] to i16*
return __arm_mte_increment_tag(a,3);
}

// CHECK-LABEL: define i32 @exclude_tag
unsigned exclude_tag(int *a, unsigned m) {
// CHECK: [[T0:%[0-9]+]] = zext i32 %m to i64
// CHECK: [[T1:%[0-9]+]] = bitcast i32* %a to i8*
// CHECK: [[T2:%[0-9]+]] = tail call i64 @llvm.aarch64.gmi(i8* [[T1]], i64 [[T0]])
// CHECK: trunc i64 [[T2]] to i32
return __arm_mte_exclude_tag(a, m);
}

// CHECK-LABEL: define i32* @get_tag1
int *get_tag1(int *a) {
// CHECK: [[T0:%[0-9]+]] = bitcast i32* %a to i8*
// CHECK: [[T1:%[0-9]+]] = tail call i8* @llvm.aarch64.ldg(i8* [[T0]], i8* [[T0]])
// CHECK: [[T2:%[0-9]+]] = bitcast i8* [[T1]] to i32*
return __arm_mte_get_tag(a);
}

// CHECK-LABEL: define i16* @get_tag2
short *get_tag2(short *a) {
// CHECK: [[T0:%[0-9]+]] = bitcast i16* %a to i8*
// CHECK: [[T1:%[0-9]+]] = tail call i8* @llvm.aarch64.ldg(i8* [[T0]], i8* [[T0]])
// CHECK: [[T2:%[0-9]+]] = bitcast i8* [[T1]] to i16*
return __arm_mte_get_tag(a);
}

// CHECK-LABEL: define void @set_tag1
void set_tag1(int *a) {
// CHECK: [[T0:%[0-9]+]] = bitcast i32* %a to i8*
// CHECK: tail call void @llvm.aarch64.stg(i8* [[T0]], i8* [[T0]])
__arm_mte_set_tag(a);
}

// CHECK-LABEL: define i64 @subtract_pointers
ptrdiff_t subtract_pointers(int *a, int *b) {
// CHECK: [[T0:%[0-9]+]] = bitcast i32* %a to i8*
// CHECK: [[T1:%[0-9]+]] = bitcast i32* %b to i8*
// CHECK: [[T2:%[0-9]+]] = tail call i64 @llvm.aarch64.subp(i8* [[T0]], i8* [[T1]])
// CHECK: ret i64 [[T2]]
return __arm_mte_ptrdiff(a, b);
}

// CHECK-LABEL: define i64 @subtract_pointers_null_1
ptrdiff_t subtract_pointers_null_1(int *a) {
// CHECK: [[T0:%[0-9]+]] = bitcast i32* %a to i8*
// CHECK: [[T1:%[0-9]+]] = tail call i64 @llvm.aarch64.subp(i8* [[T0]], i8* null)
// CHECK: ret i64 [[T1]]
return __arm_mte_ptrdiff(a, NULL);
}

// CHECK-LABEL: define i64 @subtract_pointers_null_2
ptrdiff_t subtract_pointers_null_2(int *a) {
// CHECK: [[T0:%[0-9]+]] = bitcast i32* %a to i8*
// CHECK: [[T1:%[0-9]+]] = tail call i64 @llvm.aarch64.subp(i8* null, i8* [[T0]])
// CHECK: ret i64 [[T1]]
return __arm_mte_ptrdiff(NULL, a);
}

// Check arithmetic promotion on return type
// CHECK-LABEL: define i32 @subtract_pointers4
int subtract_pointers4(void* a, void *b) {
// CHECK: [[T0:%[0-9]+]] = tail call i64 @llvm.aarch64.subp(i8* %a, i8* %b)
// CHECK-NEXT: %cmp = icmp slt i64 [[T0]], 1
// CHECK-NEXT: = zext i1 %cmp to i32
return __arm_mte_ptrdiff(a,b) <= 0;
}
3 changes: 3 additions & 0 deletions clang/test/Preprocessor/aarch64-target-features.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -316,3 +316,6 @@
// CHECK-V81A-FEATURE-2: "-cc1"{{.*}} "-triple" "aarch64{{.*}}" "-target-feature" "+neon" "-target-feature" "+v8.1a" "-target-feature" "-crypto"
// CHECK-V81A-FEATURE-3: "-cc1"{{.*}} "-triple" "aarch64{{.*}}" "-target-feature" "+v8.1a" "-target-feature" "-neon"

// ================== Check Memory Tagging Extensions (MTE).
// RUN: %clang -target arm64-none-linux-gnu -march=armv8.5-a+memtag -x c -E -dM %s -o - 2>&1 | FileCheck -check-prefix=CHECK-MEMTAG %s
// CHECK-MEMTAG: __ARM_FEATURE_MEMORY_TAGGING 1
136 changes: 136 additions & 0 deletions clang/test/Sema/builtins-arm64-mte.c
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,136 @@
// RUN: %clang_cc1 -triple arm64-arm-eabi %s -target-feature +mte -fsyntax-only -verify
// RUN: %clang_cc1 -triple arm64-arm-eabi %s -target-feature +mte -x c++ -fsyntax-only -verify
#include <stddef.h>
#include <arm_acle.h>

int *create_tag1(int a, unsigned b) {
// expected-error@+1 {{first argument of MTE builtin function must be a pointer ('int' invalid)}}
return __arm_mte_create_random_tag(a,b);
}

int *create_tag2(int *a, unsigned *b) {
// expected-error@+1 {{second argument of MTE builtin function must be an integer type ('unsigned int *' invalid)}}
return __arm_mte_create_random_tag(a,b);
}

int *create_tag3(const int *a, unsigned b) {
#ifdef __cplusplus
// expected-error@+1 {{cannot initialize return object of type 'int *' with an rvalue of type 'const int *'}}
return __arm_mte_create_random_tag(a,b);
#else
// expected-warning@+1 {{returning 'const int *' from a function with result type 'int *' discards qualifiers}}
return __arm_mte_create_random_tag(a,b);
#endif
}

int *create_tag4(volatile int *a, unsigned b) {
#ifdef __cplusplus
// expected-error@+1 {{cannot initialize return object of type 'int *' with an rvalue of type 'volatile int *'}}
return __arm_mte_create_random_tag(a,b);
#else
// expected-warning@+1 {{returning 'volatile int *' from a function with result type 'int *' discards qualifiers}}
return __arm_mte_create_random_tag(a,b);
#endif
}

int *increment_tag1(int *a, unsigned b) {
// expected-error@+1 {{argument to '__builtin_arm_addg' must be a constant integer}}
return __arm_mte_increment_tag(a,b);
}

int *increment_tag2(int *a) {
// expected-error@+1 {{argument value 16 is outside the valid range [0, 15]}}
return __arm_mte_increment_tag(a,16);
}

int *increment_tag3(int *a) {
// expected-error@+1 {{argument value -1 is outside the valid range [0, 15]}}
return __arm_mte_increment_tag(a,-1);
}

int *increment_tag4(const int *a) {
#ifdef __cplusplus
// expected-error@+1 {{cannot initialize return object of type 'int *' with an rvalue of type 'const int *'}}
return __arm_mte_increment_tag(a,5);
#else
// expected-warning@+1 {{returning 'const int *' from a function with result type 'int *' discards qualifiers}}
return __arm_mte_increment_tag(a,5);
#endif
}

int *increment_tag5(const volatile int *a) {
#ifdef __cplusplus
// expected-error@+1 {{cannot initialize return object of type 'int *' with an rvalue of type 'const volatile int *'}}
return __arm_mte_increment_tag(a,5);
#else
// expected-warning@+1 {{returning 'const volatile int *' from a function with result type 'int *' discards qualifiers}}
return __arm_mte_increment_tag(a,5);
#endif
}

unsigned exclude_tag1(int *ptr, unsigned m) {
// expected-error@+1 {{first argument of MTE builtin function must be a pointer ('int' invalid)}}
return __arm_mte_exclude_tag(*ptr, m);
}

unsigned exclude_tag2(int *ptr, int *m) {
// expected-error@+1 {{second argument of MTE builtin function must be an integer type ('int *' invalid)}}
return __arm_mte_exclude_tag(ptr, m);
}

void get_tag1() {
// expected-error@+1 {{too few arguments to function call, expected 1, have 0}}
__arm_mte_get_tag();
}

int *get_tag2(int ptr) {
// expected-error@+1 {{first argument of MTE builtin function must be a pointer ('int' invalid)}}
return __arm_mte_get_tag(ptr);
}

int *get_tag3(const volatile int *ptr) {
#ifdef __cplusplus
// expected-error@+1 {{cannot initialize return object of type 'int *' with an rvalue of type 'const volatile int *'}}
return __arm_mte_get_tag(ptr);
#else
// expected-warning@+1 {{returning 'const volatile int *' from a function with result type 'int *' discards qualifiers}}
return __arm_mte_get_tag(ptr);
#endif
}

void set_tag1() {
// expected-error@+1 {{too few arguments to function call, expected 1, have 0}}
__arm_mte_set_tag();
}

void set_tag2(int ptr) {
// expected-error@+1 {{first argument of MTE builtin function must be a pointer ('int' invalid)}}
__arm_mte_set_tag(ptr);
}

ptrdiff_t subtract_pointers1(int a, int *b) {
// expected-error@+1 {{first argument of MTE builtin function must be a null or a pointer ('int' invalid)}}
return __arm_mte_ptrdiff(a, b);
}

ptrdiff_t subtract_pointers2(int *a, int b) {
// expected-error@+1 {{second argument of MTE builtin function must be a null or a pointer ('int' invalid)}}
return __arm_mte_ptrdiff(a, b);
}

ptrdiff_t subtract_pointers3(char *a, int *b) {
// expected-error@+1 {{'char *' and 'int *' are not pointers to compatible types}}
return __arm_mte_ptrdiff(a, b);
}

ptrdiff_t subtract_pointers4(int *a, char *b) {
// expected-error@+1 {{'int *' and 'char *' are not pointers to compatible types}}
return __arm_mte_ptrdiff(a, b);
}

#ifdef __cplusplus
ptrdiff_t subtract_pointers5() {
// expected-error@+1 {{at least one argument of MTE builtin function must be a pointer ('nullptr_t', 'nullptr_t' invalid)}}
return __arm_mte_ptrdiff(nullptr, nullptr);
}
#endif