Index: llvm/trunk/docs/LangRef.rst =================================================================== --- llvm/trunk/docs/LangRef.rst +++ llvm/trunk/docs/LangRef.rst @@ -1450,6 +1450,10 @@ ``noredzone`` This attribute indicates that the code generator should not use a red zone, even if the target-specific ABI normally permits it. +``indirect-tls-seg-refs`` + This attribute indicates that the code generator should not use + direct TLS access through segment registers, even if the + target-specific ABI normally permits it. ``noreturn`` This function attribute indicates that the function never returns normally. This produces undefined behavior at runtime if the Index: llvm/trunk/lib/Target/X86/X86ISelDAGToDAG.cpp =================================================================== --- llvm/trunk/lib/Target/X86/X86ISelDAGToDAG.cpp +++ llvm/trunk/lib/Target/X86/X86ISelDAGToDAG.cpp @@ -165,6 +165,9 @@ /// If true, selector should try to optimize for minimum code size. bool OptForMinSize; + /// Disable direct TLS access through segment registers. + bool IndirectTlsSegRefs; + public: explicit X86DAGToDAGISel(X86TargetMachine &tm, CodeGenOpt::Level OptLevel) : SelectionDAGISel(tm, OptLevel), OptForSize(false), @@ -177,6 +180,8 @@ bool runOnMachineFunction(MachineFunction &MF) override { // Reset the subtarget each time through. Subtarget = &MF.getSubtarget(); + IndirectTlsSegRefs = MF.getFunction().hasFnAttribute( + "indirect-tls-seg-refs"); SelectionDAGISel::runOnMachineFunction(MF); return true; } @@ -981,6 +986,7 @@ // For more information see http://people.redhat.com/drepper/tls.pdf if (ConstantSDNode *C = dyn_cast(Address)) if (C->getSExtValue() == 0 && AM.Segment.getNode() == nullptr && + !IndirectTlsSegRefs && (Subtarget->isTargetGlibc() || Subtarget->isTargetAndroid() || Subtarget->isTargetFuchsia())) switch (N->getPointerInfo().getAddrSpace()) { Index: llvm/trunk/test/CodeGen/X86/tls.ll =================================================================== --- llvm/trunk/test/CodeGen/X86/tls.ll +++ llvm/trunk/test/CodeGen/X86/tls.ll @@ -1,5 +1,7 @@ ; RUN: llc < %s -mtriple=i386-linux-gnu | FileCheck -check-prefix=X86_LINUX %s ; RUN: llc < %s -mtriple=x86_64-linux-gnu | FileCheck -check-prefix=X64_LINUX %s +; RUN: llc < %s -mtriple=i386-linux-gnu -fast-isel | FileCheck -check-prefix=X86_ISEL_LINUX %s +; RUN: llc < %s -mtriple=x86_64-linux-gnu -fast-isel | FileCheck -check-prefix=X64_ISEL_LINUX %s ; RUN: llc < %s -mtriple=i686-pc-win32 | FileCheck -check-prefix=X86_WIN %s ; RUN: llc < %s -mtriple=x86_64-pc-win32 | FileCheck -check-prefix=X64_WIN %s ; RUN: llc < %s -mtriple=i686-pc-windows-gnu | FileCheck -check-prefix=MINGW32 %s @@ -453,3 +455,59 @@ ret i32* @i6 } + +; NOTE: Similar to f1() but with direct TLS segment access disabled +define i32 @f17() #0 { +; X86_LINUX-LABEL: f17: +; X86_LINUX: movl %gs:0, %eax +; X86_LINUX-NEXT: movl i1@NTPOFF(%eax), %eax +; X86_LINUX-NEXT: ret +; X64_LINUX-LABEL: f17: +; X64_LINUX: movq %fs:0, %rax +; X64_LINUX-NEXT: movl i1@TPOFF(%rax), %eax +; X64_LINUX-NEXT: ret +; X86_ISEL_LINUX-LABEL: f17: +; X86_ISEL_LINUX: movl %gs:0, %eax +; X86_ISEL_LINUX-NEXT: movl i1@NTPOFF(%eax), %eax +; X86_ISEL_LINUX-NEXT: ret +; X64_ISEL_LINUX-LABEL: f17: +; X64_ISEL_LINUX: movq %fs:0, %rax +; X64_ISEL_LINUX-NEXT: movl i1@TPOFF(%rax), %eax +; X64_ISEL_LINUX-NEXT: ret + +entry: + %tmp1 = load i32, i32* @i1 + ret i32 %tmp1 +} + +; NOTE: Similar to f3() but with direct TLS segment access disabled +define i32 @f18() #1 { +; X86_LINUX-LABEL: f18: +; X86_LINUX: movl i2@INDNTPOFF, %eax +; X86_LINUX-NEXT: movl %gs:0, %ecx +; X86_LINUX-NEXT: movl (%ecx,%eax), %eax +; X86_LINUX-NEXT: ret +; X64_LINUX-LABEL: f18: +; X64_LINUX: movq i2@GOTTPOFF(%rip), %rax +; X64_LINUX-NEXT: movq %fs:0, %rcx +; X64_LINUX-NEXT: movl (%rcx,%rax), %eax +; X64_LINUX-NEXT: ret +; X86_ISEL_LINUX-LABEL: f18: +; X86_ISEL_LINUX: movl i2@INDNTPOFF, %eax +; X86_ISEL_LINUX-NEXT: movl %gs:0, %ecx +; X86_ISEL_LINUX-NEXT: movl (%ecx,%eax), %eax +; X86_ISEL_LINUX-NEXT: ret +; X64_ISEL_LINUX-LABEL: f18: +; X64_ISEL_LINUX: movq i2@GOTTPOFF(%rip), %rax +; X64_ISEL_LINUX-NEXT: movq %fs:0, %rcx +; X64_ISEL_LINUX-NEXT: movl (%rcx,%rax), %eax +; X64_ISEL_LINUX-NEXT: ret + + +entry: + %tmp1 = load i32, i32* @i2 + ret i32 %tmp1 +} + +attributes #0 = { "indirect-tls-seg-refs" } +attributes #1 = { nounwind "indirect-tls-seg-refs" }