llvm/test/Instrumentation/MemorySanitizer/AArch64/vararg-kmsan.ll

   1 ; RUN: opt < %s -S -passes=msan -msan-kernel=1 2>&1 | FileCheck %s
   2
   3 target datalayout = "e-m:e-i64:64-i128:128-n32:64-S128"
   4 target triple = "aarch64-unknown-linux-gnu"
   5
   6 %struct.__va_list = type { ptr, ptr, ptr, i32, i32 }
   7
   8 define i32 @foo(i32 %guard, ...) {
   9   %vl = alloca %struct.__va_list, align 8
  10   call void @llvm.lifetime.start.p0(i64 32, ptr %vl)
  11   call void @llvm.va_start(ptr %vl)
  12   call void @llvm.va_end(ptr %vl)
  13   call void @llvm.lifetime.end.p0(i64 32, ptr %vl)
  14   ret i32 0
  15 }
  16
  17 ; First check if the variadic shadow values are saved in stack with correct
  18 ; size (192 is total of general purpose registers size, 64, plus total of
  19 ; floating-point registers size, 128).
  20
  21 ; CHECK-LABEL: @foo
  22 ; CHECK: [[A:%.*]] = load {{.*}} ptr %va_arg_overflow_size
  23 ; CHECK: [[B:%.*]] = add i64 192, [[A]]
  24 ; CHECK: alloca {{.*}} [[B]]
  25
  26 ; We expect three memcpy operations: one for the general purpose registers,
  27 ; one for floating-point/SIMD ones, and one for thre remaining arguments.
  28
  29 ; Propagate the GR shadow values on for the va_list::__gp_top, adjust the
  30 ; offset in the __msan_va_arg_tls based on va_list:__gp_off, and finally
  31 ; issue the memcpy.
  32 ; CHECK: [[GRP:%.*]] = getelementptr inbounds i8, ptr {{%.*}}, i64 {{%.*}}
  33 ; CHECK: [[GRSIZE:%.*]] = sub i64 64, {{%.*}}
  34 ; CHECK: call void @llvm.memcpy.p0.p0.i64(ptr align 8 {{%.*}}, ptr align 8 [[GRP]], i64 [[GRSIZE]], i1 false)
  35
  36 ; Propagate the VR shadow values on for the va_list::__vr_top, adjust the
  37 ; offset in the __msan_va_arg_tls based on va_list:__vr_off, and finally
  38 ; issue the memcpy.
  39 ; CHECK: [[VRP:%.*]] = getelementptr inbounds i8, ptr {{%.*}}, i64 {{%.*}}
  40 ; CHECK: [[VRSIZE:%.*]] = sub i64 128, {{%.*}}
  41 ; CHECK: call void @llvm.memcpy.p0.p0.i64(ptr align 8 {{%.*}}, ptr align 8 [[VRP]], i64 [[VRSIZE]], i1 false)
  42
  43 ; Copy the remaining shadow values on the va_list::__stack position (it is
  44 ; on the constant offset of 192 from __msan_va_arg_tls).
  45 ; CHECK: [[STACK:%.*]] = getelementptr inbounds i8, ptr {{%.*}}, i32 192
  46 ; CHECK: call void @llvm.memcpy.p0.p0.i64(ptr align 16 {{%.*}}, ptr align 16 [[STACK]], i64 {{%.*}}, i1 false)
  47
  48 declare void @llvm.lifetime.start.p0(i64, ptr nocapture) #1
  49 declare void @llvm.va_start(ptr) #2
  50 declare void @llvm.va_end(ptr) #2
  51 declare void @llvm.lifetime.end.p0(i64, ptr nocapture) #1