80e326a8c4
This is a child diff of D92261. It extended TLS arg/ret to work with aggregate types. For a function t foo(t1 a1, t2 a2, ... tn an) Its arguments shadow are saved in TLS args like a1_s, a2_s, ..., an_s TLS ret simply includes r_s. By calculating the type size of each shadow value, we can get their offset. This is similar to what MSan does. See __msan_retval_tls and __msan_param_tls from llvm/lib/Transforms/Instrumentation/MemorySanitizer.cpp. Note that this change does not add test cases for overflowed TLS arg/ret because this is hard to test w/o supporting aggregate shdow types. We will be adding them after supporting that. Reviewed-by: morehouse Differential Revision: https://reviews.llvm.org/D92440
75 lines
2.4 KiB
LLVM
75 lines
2.4 KiB
LLVM
; RUN: opt < %s -dfsan -S | FileCheck %s
|
|
target datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128-n8:16:32:64-S128"
|
|
target triple = "x86_64-unknown-linux-gnu"
|
|
|
|
define i8 @add(i8 %a, i8 %b) {
|
|
; CHECK: @"dfs$add"
|
|
; CHECK-DAG: %[[ALABEL:.*]] = load [[SHADOWTYPE:i16]], [[SHADOWTYPE]]* bitcast ([[ARGTLSTYPE:\[100 x i64\]]]* @__dfsan_arg_tls to [[SHADOWTYPE]]*), align [[ALIGN:2]]
|
|
; CHECK-DAG: %[[BLABEL:.*]] = load [[SHADOWTYPE]], [[SHADOWTYPE]]* inttoptr (i64 add (i64 ptrtoint ([[ARGTLSTYPE]]* @__dfsan_arg_tls to i64), i64 2) to [[SHADOWTYPE]]*), align [[ALIGN]]
|
|
; CHECK: %[[UNION:.*]] = call zeroext [[SHADOWTYPE]] @__dfsan_union([[SHADOWTYPE]] zeroext %[[ALABEL]], [[SHADOWTYPE]] zeroext %[[BLABEL]])
|
|
; CHECK: %[[ADDLABEL:.*]] = phi [[SHADOWTYPE]] [ %[[UNION]], {{.*}} ], [ %[[ALABEL]], {{.*}} ]
|
|
; CHECK: add i8
|
|
; CHECK: store [[SHADOWTYPE]] %[[ADDLABEL]], [[SHADOWTYPE]]* bitcast ([100 x i64]* @__dfsan_retval_tls to [[SHADOWTYPE]]*), align [[ALIGN]]
|
|
; CHECK: ret i8
|
|
%c = add i8 %a, %b
|
|
ret i8 %c
|
|
}
|
|
|
|
define i8 @sub(i8 %a, i8 %b) {
|
|
; CHECK: @"dfs$sub"
|
|
; CHECK: load{{.*}}__dfsan_arg_tls
|
|
; CHECK: load{{.*}}__dfsan_arg_tls
|
|
; CHECK: call{{.*}}__dfsan_union
|
|
; CHECK: sub i8
|
|
; CHECK: store{{.*}}__dfsan_retval_tls
|
|
; CHECK: ret i8
|
|
%c = sub i8 %a, %b
|
|
ret i8 %c
|
|
}
|
|
|
|
define i8 @mul(i8 %a, i8 %b) {
|
|
; CHECK: @"dfs$mul"
|
|
; CHECK: load{{.*}}__dfsan_arg_tls
|
|
; CHECK: load{{.*}}__dfsan_arg_tls
|
|
; CHECK: call{{.*}}__dfsan_union
|
|
; CHECK: mul i8
|
|
; CHECK: store{{.*}}__dfsan_retval_tls
|
|
; CHECK: ret i8
|
|
%c = mul i8 %a, %b
|
|
ret i8 %c
|
|
}
|
|
|
|
define i8 @sdiv(i8 %a, i8 %b) {
|
|
; CHECK: @"dfs$sdiv"
|
|
; CHECK: load{{.*}}__dfsan_arg_tls
|
|
; CHECK: load{{.*}}__dfsan_arg_tls
|
|
; CHECK: call{{.*}}__dfsan_union
|
|
; CHECK: sdiv i8
|
|
; CHECK: store{{.*}}__dfsan_retval_tls
|
|
; CHECK: ret i8
|
|
%c = sdiv i8 %a, %b
|
|
ret i8 %c
|
|
}
|
|
|
|
define i8 @udiv(i8 %a, i8 %b) {
|
|
; CHECK: @"dfs$udiv"
|
|
; CHECK: load{{.*}}__dfsan_arg_tls
|
|
; CHECK: load{{.*}}__dfsan_arg_tls
|
|
; CHECK: call{{.*}}__dfsan_union
|
|
; CHECK: udiv i8
|
|
; CHECK: store{{.*}}__dfsan_retval_tls
|
|
; CHECK: ret i8
|
|
%c = udiv i8 %a, %b
|
|
ret i8 %c
|
|
}
|
|
|
|
define double @fneg(double %a) {
|
|
; CHECK: @"dfs$fneg"
|
|
; CHECK: load{{.*}}__dfsan_arg_tls
|
|
; CHECK: fneg double
|
|
; CHECK: store{{.*}}__dfsan_retval_tls
|
|
; CHECK: ret double
|
|
%c = fneg double %a
|
|
ret double %c
|
|
}
|