clang/test/CodeGen/ms_abi_aarch64.c

   1 // RUN: %clang_cc1 -triple aarch64-linux-gnu -emit-llvm < %s | FileCheck -check-prefixes=LINUX,COMMON %s
   2 // RUN: %clang_cc1 -triple aarch64-pc-win32 -emit-llvm < %s | FileCheck -check-prefixes=WIN64,COMMON %s
   3
   4 struct small_odd {
   5   char a, b, c;
   6 };
   7
   8 struct larger {
   9   int a, b, c, d, e;
  10 };
  11
  12 void __attribute__((ms_abi)) f1(void);
  13 void f2(void);
  14 void f3(void) {
  15   // LINUX-LABEL: define{{.*}} void @f3()
  16   // WIN64-LABEL: define dso_local void @f3()
  17   f1();
  18   // LINUX: call win64cc void @f1()
  19   // WIN64: call void @f1()
  20   f2();
  21   // COMMON: call void @f2()
  22 }
  23 // LINUX: declare win64cc void @f1()
  24 // LINUX: declare void @f2()
  25 // WIN64: declare dso_local void @f1()
  26 // WIN64: declare dso_local void @f2()
  27
  28 // Win64 ABI varargs
  29 void __attribute__((ms_abi)) f4(int a, ...) {
  30   // LINUX-LABEL: define{{.*}} win64cc void @f4
  31   // WIN64-LABEL: define dso_local void @f4
  32   __builtin_ms_va_list ap;
  33   __builtin_ms_va_start(ap, a);
  34   // COMMON: %[[AP:.*]] = alloca ptr
  35   // COMMON: call void @llvm.va_start
  36   int b = __builtin_va_arg(ap, int);
  37   // COMMON: %[[AP_CUR:.*]] = load ptr, ptr %[[AP]]
  38   // COMMON-NEXT: %[[AP_NEXT:.*]] = getelementptr inbounds i8, ptr %[[AP_CUR]], i64 8
  39   // COMMON-NEXT: store ptr %[[AP_NEXT]], ptr %[[AP]]
  40   __builtin_ms_va_list ap2;
  41   __builtin_ms_va_copy(ap2, ap);
  42   // COMMON: %[[AP_VAL:.*]] = load ptr, ptr %[[AP]]
  43   // COMMON-NEXT: store ptr %[[AP_VAL]], ptr %[[AP2:.*]]
  44   __builtin_ms_va_end(ap);
  45   // COMMON: call void @llvm.va_end
  46 }
  47
  48 void __attribute__((ms_abi)) f4_2(int a, ...) {
  49   // LINUX-LABEL: define{{.*}} win64cc void @f4_2
  50   // WIN64-LABEL: define dso_local void @f4_2
  51   __builtin_ms_va_list ap;
  52   __builtin_ms_va_start(ap, a);
  53   // COMMON: %[[AP:.*]] = alloca ptr
  54   // COMMON: call void @llvm.va_start
  55   struct small_odd s1 = __builtin_va_arg(ap, struct small_odd);
  56   // COMMON: %[[AP_CUR:.*]] = load ptr, ptr %[[AP]]
  57   // COMMON-NEXT: %[[AP_NEXT:.*]] = getelementptr inbounds i8, ptr %[[AP_CUR]], i64 8
  58   // COMMON-NEXT: store ptr %[[AP_NEXT]], ptr %[[AP]]
  59   struct larger s2 = __builtin_va_arg(ap, struct larger);
  60   // COMMON: %[[AP_CUR2:.*]] = load ptr, ptr %[[AP]]
  61   // COMMON-NEXT: %[[AP_NEXT3:.*]] = getelementptr inbounds i8, ptr %[[AP_CUR2]], i64 8
  62   // COMMON-NEXT: store ptr %[[AP_NEXT3]], ptr %[[AP]]
  63   __builtin_ms_va_end(ap);
  64 }
  65
  66 // Let's verify that normal va_lists work right on Win64, too.
  67 void f5(int a, ...) {
  68   // WIN64-LABEL: define dso_local void @f5
  69   __builtin_va_list ap;
  70   __builtin_va_start(ap, a);
  71   // WIN64: %[[AP:.*]] = alloca ptr
  72   // WIN64: call void @llvm.va_start
  73   int b = __builtin_va_arg(ap, int);
  74   // WIN64: %[[AP_CUR:.*]] = load ptr, ptr %[[AP]]
  75   // WIN64-NEXT: %[[AP_NEXT:.*]] = getelementptr inbounds i8, ptr %[[AP_CUR]], i64 8
  76   // WIN64-NEXT: store ptr %[[AP_NEXT]], ptr %[[AP]]
  77   __builtin_va_list ap2;
  78   __builtin_va_copy(ap2, ap);
  79   // WIN64: call void @llvm.va_copy
  80   __builtin_va_end(ap);
  81   // WIN64: call void @llvm.va_end
  82 }
  83
  84 struct HFA {
  85   float a, b, c;
  86 };
  87
  88 __attribute__((ms_abi)) void msabi_hfa(struct HFA a);
  89 __attribute__((ms_abi)) void msabi_hfa_vararg(struct HFA a, int b, ...);
  90
  91 void call_msabi_hfa(void) {
  92   // COMMON-LABEL: define{{.*}} void @call_msabi_hfa()
  93   // WIN64: call void @msabi_hfa([3 x float] {{.*}})
  94   // LINUX: call win64cc void @msabi_hfa([3 x float] {{.*}})
  95   msabi_hfa((struct HFA){1.0f, 2.0f, 3.0f});
  96 }
  97
  98 void call_msabi_hfa_vararg(void) {
  99   // COMMON-LABEL: define{{.*}} void @call_msabi_hfa_vararg()
 100   // WIN64: call void ([2 x i64], i32, ...) @msabi_hfa_vararg([2 x i64] {{.*}}, i32 noundef 4, [2 x i64] {{.*}})
 101   // LINUX: call win64cc void ([2 x i64], i32, ...) @msabi_hfa_vararg([2 x i64] {{.*}}, i32 noundef 4, [2 x i64] {{.*}})
 102   msabi_hfa_vararg((struct HFA){1.0f, 2.0f, 3.0f}, 4,
 103                    (struct HFA){5.0f, 6.0f, 7.0f});
 104 }
 105
 106 __attribute__((ms_abi)) void get_msabi_hfa_vararg(int a, ...) {
 107   // COMMON-LABEL: define{{.*}} void @get_msabi_hfa_vararg
 108   __builtin_ms_va_list ap;
 109   __builtin_ms_va_start(ap, a);
 110   // COMMON: %[[AP:.*]] = alloca ptr
 111   // COMMON: call void @llvm.va_start
 112   struct HFA b = __builtin_va_arg(ap, struct HFA);
 113   // COMMON: %[[AP_CUR:.*]] = load ptr, ptr %[[AP]]
 114   // COMMON-NEXT: %[[AP_NEXT:.*]] = getelementptr inbounds i8, ptr %[[AP_CUR]], i64 16
 115   // COMMON-NEXT: store ptr %[[AP_NEXT]], ptr %[[AP]]
 116   __builtin_ms_va_end(ap);
 117 }