[elf2] Don't allocate VA space for TLS NOBITS sections.

[ruiu]

ELF/Writer.cpp

629–633

Factor out as a function.

[• rafael]

I am OK with the test update.
Just make sure Rui is OK with the refactoring he asked for.

test/elf2/tls.s

83

Make this check a tiny bit stronger. You know which PT_LOAD this is, so you can do

CHECK: PT_LOAD
CHECK: PT_LOAD
// CHECK: PT_LOAD
....

To make sure you skip the RO and RW segments.

[ruiu]

Are you still working on this?

ELF/Writer.cpp

if (Sec->getSize()) {
           Elf_Phdr *PH = &Phdrs.back();
           setPhdr(PH, PT_LOAD, Flags, FileOff, VA, Target->getPageSize());
         }
       }
     }
 
     uintX_t Align = Sec->getAlign();
     uintX_t Size = Sec->getSize();
-    if (Sec->getFlags() & SHF_ALLOC) {
+    if ((Sec->getFlags() & SHF_ALLOC) &&
+        // Don't allocate VA space for TLS NOBITS sections. The PT_TLS PHDR is
+        // responsible for allocating space for them, not the PT_LOAD that
+        // contains the TLS initialization image.
+        !((Sec->getFlags() & SHF_TLS) && Sec->getType() == SHT_NOBITS)) {
       VA = RoundUpToAlignment(VA, Align);
       Sec->setVA(VA);
       VA += Size;
     }
     FileOff = RoundUpToAlignment(FileOff, Align);
     Sec->setFileOffset(FileOff);
     if (Sec->getType() != SHT_NOBITS)
       FileOff += Size;

test/elf2/tls.s

+// REQUIRES: x86
+// RUN: llvm-mc -filetype=obj -triple=x86_64-unknown-linux %s -o %t
+// RUN: ld.lld2 %t -o %tout
+// RUN: llvm-readobj -sections -program-headers %tout | FileCheck %s
+
+.global _start
+_start:
+
+  .section  .tbss,"awT",@nobits
+  .long 0
+
+  .section  .tdata,"awT",@progbits
+  .long 1
+
+  .section  .thread_bss,"awT",@nobits
+  .long 0
+
+  .section  .thread_data,"awT",@progbits
+  .long 2
+
+// CHECK:          Name: .tdata
+// CHECK-NEXT:     Type: SHT_PROGBITS
+// CHECK-NEXT:     Flags [
+// CHECK-NEXT:       SHF_ALLOC
+// CHECK-NEXT:       SHF_TLS
+// CHECK-NEXT:       SHF_WRITE
+// CHECK-NEXT:     ]
+// CHECK-NEXT:     Address:
+// CHECK-NEXT:     Offset:
+// CHECK-NEXT:     Size: 4
+// CHECK-NEXT:     Link:
+// CHECK-NEXT:     Info:
+// CHECK-NEXT:     AddressAlignment:
+// CHECK-NEXT:     EntrySize:
+// CHECK-NEXT:   }
+// CHECK-NEXT:   Section {
+// CHECK-NEXT:     Index:
+// CHECK-NEXT:     Name: .thread_data
+// CHECK-NEXT:     Type: SHT_PROGBITS
+// CHECK-NEXT:     Flags [
+// CHECK-NEXT:       SHF_ALLOC
+// CHECK-NEXT:       SHF_TLS
+// CHECK-NEXT:       SHF_WRITE
+// CHECK-NEXT:     ]
+// CHECK-NEXT:     Address:
+// CHECK-NEXT:     Offset:
+// CHECK-NEXT:     Size: 4
+// CHECK-NEXT:     Link:
+// CHECK-NEXT:     Info:
+// CHECK-NEXT:     AddressAlignment:
+// CHECK-NEXT:     EntrySize:
+// CHECK-NEXT:   }
+// CHECK-NEXT:   Section {
+// CHECK-NEXT:     Index:
+// CHECK-NEXT:     Name: .tbss
+// CHECK-NEXT:     Type: SHT_NOBITS
+// CHECK-NEXT:     Flags [
+// CHECK-NEXT:       SHF_ALLOC
+// CHECK-NEXT:       SHF_TLS
+// CHECK-NEXT:       SHF_WRITE
+// CHECK-NEXT:     ]
+// CHECK-NEXT:     Address:
+// CHECK-NEXT:     Offset:
+// CHECK-NEXT:     Size: 4
+// CHECK-NEXT:     Link:
+// CHECK-NEXT:     Info:
+// CHECK-NEXT:     AddressAlignment:
+// CHECK-NEXT:     EntrySize:
+// CHECK-NEXT:   }
+// CHECK-NEXT:   Section {
+// CHECK-NEXT:     Index:
+// CHECK-NEXT:     Name: .thread_bss
+// CHECK-NEXT:     Type: SHT_NOBITS
+// CHECK-NEXT:     Flags [
+// CHECK-NEXT:       SHF_ALLOC
+// CHECK-NEXT:       SHF_TLS
+// CHECK-NEXT:       SHF_WRITE
+// CHECK-NEXT:     ]
+// CHECK-NEXT:     Address:
+// CHECK-NEXT:     Offset:
+// CHECK-NEXT:     Size: 4
+
+// Check that the TLS NOBITS sections weren't added to the R/W PT_LOAD's size.
+
+// CHECK:      ProgramHeaders [
+// CHECK:          FileSize: 8
+// CHECK-NEXT:     MemSize: 8
+// CHECK-NEXT:     Flags [
+// CHECK-NEXT:       PF_R
+// CHECK-NEXT:       PF_W
+// CHECK-NEXT:     ]