mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-11-22 02:33:06 +01:00
[gold] Match lld WPD behavior for shared library symbols and add test
lld already marks shared library defs as ExportDynamic, which prevents potentially unsafe devirtualization of symbols defined in shared libraries. Match that behavior in the gold plugin, and add the same test. Depends on D96721. Differential Revision: https://reviews.llvm.org/D96722
This commit is contained in:
Teresa Johnson
parent
db301f643f
commit
8baf80ada5
19
test/tools/gold/X86/Inputs/devirt_vcall_vis_shared_def.ll
Normal file
19
test/tools/gold/X86/Inputs/devirt_vcall_vis_shared_def.ll
Normal file
@ -0,0 +1,19 @@
|
||||
target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
|
||||
target triple = "x86_64-grtev4-linux-gnu"
|
||||
|
||||
%struct.A = type { i32 (...)** }
|
||||
|
||||
@_ZTV1A = unnamed_addr constant { [4 x i8*] } { [4 x i8*] [i8* null, i8* undef, i8* bitcast (i32 (%struct.A*, i32)* @_ZN1A1fEi to i8*), i8* bitcast (i32 (%struct.A*, i32)* @_ZN1A1nEi to i8*)] }, !type !0, !vcall_visibility !1
|
||||
|
||||
define i32 @_ZN1A1fEi(%struct.A* %this, i32 %a) #0 {
|
||||
ret i32 0;
|
||||
}
|
||||
|
||||
define i32 @_ZN1A1nEi(%struct.A* %this, i32 %a) #0 {
|
||||
ret i32 0;
|
||||
}
|
||||
|
||||
attributes #0 = { noinline optnone }
|
||||
|
||||
!0 = !{i64 16, !"_ZTS1A"}
|
||||
!1 = !{i64 0}
|
||||
122
test/tools/gold/X86/devirt_vcall_vis_shared_def.ll
Normal file
122
test/tools/gold/X86/devirt_vcall_vis_shared_def.ll
Normal file
@ -0,0 +1,122 @@
|
||||
;; Test that symbols defined in shared libraries prevent devirtualization.
|
||||
|
||||
;; First check that we get devirtualization when the defs are in the
|
||||
;; LTO unit.
|
||||
|
||||
;; Index based WPD
|
||||
;; Generate unsplit module with summary for ThinLTO index-based WPD.
|
||||
; RUN: opt --thinlto-bc -o %t1a.o %s
|
||||
; RUN: opt --thinlto-bc -o %t2a.o %S/Inputs/devirt_vcall_vis_shared_def.ll
|
||||
; RUN: %gold -m elf_x86_64 -plugin %llvmshlibdir/LLVMgold%shlibext \
|
||||
; RUN: --plugin-opt=whole-program-visibility \
|
||||
; RUN: --plugin-opt=save-temps \
|
||||
; RUN: --plugin-opt=-pass-remarks=. \
|
||||
; RUN: %t1a.o %t2a.o -o %t3a 2>&1 | FileCheck %s --check-prefix=REMARK
|
||||
; RUN: llvm-dis %t1a.o.4.opt.bc -o - | FileCheck %s --check-prefix=CHECK-IR
|
||||
|
||||
;; Hybrid WPD
|
||||
;; Generate split module with summary for hybrid Thin/Regular LTO WPD.
|
||||
; RUN: opt --thinlto-bc --thinlto-split-lto-unit -o %t1b.o %s
|
||||
; RUN: opt --thinlto-bc --thinlto-split-lto-unit -o %t2b.o %S/Inputs/devirt_vcall_vis_shared_def.ll
|
||||
; RUN: %gold -m elf_x86_64 -plugin %llvmshlibdir/LLVMgold%shlibext \
|
||||
; RUN: --plugin-opt=whole-program-visibility \
|
||||
; RUN: --plugin-opt=save-temps \
|
||||
; RUN: --plugin-opt=-pass-remarks=. \
|
||||
; RUN: %t1b.o %t2b.o -o %t3b 2>&1 | FileCheck %s --check-prefix=REMARK
|
||||
; RUN: llvm-dis %t1b.o.4.opt.bc -o - | FileCheck %s --check-prefix=CHECK-IR
|
||||
|
||||
;; Regular LTO WPD
|
||||
; RUN: opt -o %t1c.o %s
|
||||
; RUN: opt -o %t2c.o %S/Inputs/devirt_vcall_vis_shared_def.ll
|
||||
; RUN: %gold -m elf_x86_64 -plugin %llvmshlibdir/LLVMgold%shlibext \
|
||||
; RUN: --plugin-opt=whole-program-visibility \
|
||||
; RUN: --plugin-opt=save-temps \
|
||||
; RUN: --plugin-opt=-pass-remarks=. \
|
||||
; RUN: %t1c.o %t2c.o -o %t3c 2>&1 | FileCheck %s --check-prefix=REMARK
|
||||
; RUN: llvm-dis %t3c.0.4.opt.bc -o - | FileCheck %s --check-prefix=CHECK-IR
|
||||
|
||||
; REMARK-DAG: single-impl: devirtualized a call to _ZN1A1nEi
|
||||
|
||||
;; Check that WPD fails with when linking against a shared library
|
||||
;; containing the strong defs of the vtables.
|
||||
; RUN: %gold -m elf_x86_64 -plugin %llvmshlibdir/LLVMgold%shlibext \
|
||||
; RUN: %t2c.o -o %t.so -shared
|
||||
|
||||
;; Index based WPD
|
||||
; RUN: %gold -m elf_x86_64 -plugin %llvmshlibdir/LLVMgold%shlibext \
|
||||
; RUN: --plugin-opt=whole-program-visibility \
|
||||
; RUN: --plugin-opt=-pass-remarks=. \
|
||||
; RUN: %t1a.o %t.so -o %t4a \
|
||||
; RUN: --export-dynamic 2>&1 | count 0
|
||||
|
||||
;; Hybrid WPD
|
||||
; RUN: opt --thinlto-bc --thinlto-split-lto-unit -o %t4.o %s
|
||||
; RUN: %gold -m elf_x86_64 -plugin %llvmshlibdir/LLVMgold%shlibext \
|
||||
; RUN: --plugin-opt=whole-program-visibility \
|
||||
; RUN: --plugin-opt=-pass-remarks=. \
|
||||
; RUN: %t1b.o %t.so -o %t4b \
|
||||
; RUN: --export-dynamic 2>&1 | count 0
|
||||
|
||||
;; Regular LTO WPD
|
||||
; RUN: opt -o %t4.o %s
|
||||
; RUN: %gold -m elf_x86_64 -plugin %llvmshlibdir/LLVMgold%shlibext \
|
||||
; RUN: --plugin-opt=whole-program-visibility \
|
||||
; RUN: --plugin-opt=-pass-remarks=. \
|
||||
; RUN: %t1c.o %t.so -o %t4c \
|
||||
; RUN: --export-dynamic 2>&1 | count 0
|
||||
|
||||
target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
|
||||
target triple = "x86_64-grtev4-linux-gnu"
|
||||
|
||||
%struct.A = type { i32 (...)** }
|
||||
%struct.B = type { %struct.A }
|
||||
|
||||
@_ZTV1A = available_externally unnamed_addr constant { [4 x i8*] } { [4 x i8*] [i8* null, i8* undef, i8* bitcast (i32 (%struct.A*, i32)* @_ZN1A1fEi to i8*), i8* bitcast (i32 (%struct.A*, i32)* @_ZN1A1nEi to i8*)] }, !type !0, !vcall_visibility !2
|
||||
@_ZTV1B = linkonce_odr unnamed_addr constant { [4 x i8*] } { [4 x i8*] [i8* null, i8* undef, i8* bitcast (i32 (%struct.B*, i32)* @_ZN1B1fEi to i8*), i8* bitcast (i32 (%struct.A*, i32)* @_ZN1A1nEi to i8*)] }, !type !0, !type !1, !vcall_visibility !2
|
||||
|
||||
;; Prevent the vtables from being dead code eliminated.
|
||||
@llvm.used = appending global [2 x i8*] [ i8* bitcast ( { [4 x i8*] }* @_ZTV1A to i8*), i8* bitcast ( { [4 x i8*] }* @_ZTV1B to i8*)]
|
||||
|
||||
; CHECK-IR-LABEL: define dso_local i32 @_start
|
||||
define i32 @_start(%struct.A* %obj, i32 %a) {
|
||||
entry:
|
||||
%0 = bitcast %struct.A* %obj to i8***
|
||||
%vtable = load i8**, i8*** %0
|
||||
%1 = bitcast i8** %vtable to i8*
|
||||
%p = call i1 @llvm.type.test(i8* %1, metadata !"_ZTS1A")
|
||||
call void @llvm.assume(i1 %p)
|
||||
%fptrptr = getelementptr i8*, i8** %vtable, i32 1
|
||||
%2 = bitcast i8** %fptrptr to i32 (%struct.A*, i32)**
|
||||
%fptr1 = load i32 (%struct.A*, i32)*, i32 (%struct.A*, i32)** %2, align 8
|
||||
|
||||
;; Check that the call was devirtualized.
|
||||
; CHECK-IR: %call = tail call i32 @_ZN1A1nEi
|
||||
; CHECK-NODEVIRT-IR: %call = tail call i32 %fptr1
|
||||
%call = tail call i32 %fptr1(%struct.A* nonnull %obj, i32 %a)
|
||||
|
||||
ret i32 %call
|
||||
}
|
||||
; CHECK-IR-LABEL: ret i32
|
||||
; CHECK-IR-LABEL: }
|
||||
|
||||
declare i1 @llvm.type.test(i8*, metadata)
|
||||
declare void @llvm.assume(i1)
|
||||
|
||||
define available_externally i32 @_ZN1A1fEi(%struct.A* %this, i32 %a) #0 {
|
||||
ret i32 0
|
||||
}
|
||||
|
||||
define available_externally i32 @_ZN1A1nEi(%struct.A* %this, i32 %a) #0 {
|
||||
ret i32 0
|
||||
}
|
||||
|
||||
define linkonce_odr i32 @_ZN1B1fEi(%struct.B* %this, i32 %a) #0 {
|
||||
ret i32 0
|
||||
}
|
||||
|
||||
;; Make sure we don't inline or otherwise optimize out the direct calls.
|
||||
attributes #0 = { noinline optnone }
|
||||
|
||||
!0 = !{i64 16, !"_ZTS1A"}
|
||||
!1 = !{i64 16, !"_ZTS1B"}
|
||||
!2 = !{i64 0}
|
||||
@ -748,12 +748,15 @@ static void addModule(LTO &Lto, claimed_file &F, const void *View,
|
||||
|
||||
case LDPR_RESOLVED_IR:
|
||||
case LDPR_RESOLVED_EXEC:
|
||||
case LDPR_RESOLVED_DYN:
|
||||
case LDPR_PREEMPTED_IR:
|
||||
case LDPR_PREEMPTED_REG:
|
||||
case LDPR_UNDEF:
|
||||
break;
|
||||
|
||||
case LDPR_RESOLVED_DYN:
|
||||
R.ExportDynamic = true;
|
||||
break;
|
||||
|
||||
case LDPR_PREVAILING_DEF_IRONLY:
|
||||
R.Prevailing = !isUndefined(Sym);
|
||||
break;
|
||||
|
||||
Loading…
Reference in New Issue
Block a user