From 8364901f2479effe999386b2abd04e7b5108782c Mon Sep 17 00:00:00 2001
From: Reid Kleckner <rnk@google.com>
Date: Wed, 14 Mar 2018 20:25:41 +0000
Subject: [PATCH] [COFF] Enable per-function and data sections in LTO

Summary: This allows post-LTO symbol reordering and ICF.

Reviewers: inglorion

Subscribers: mehdi_amini, llvm-commits

Differential Revision: https://reviews.llvm.org/D44492

llvm-svn: 327563
---
 lld/COFF/LTO.cpp         |  6 ++++++
 lld/test/COFF/lto-icf.ll | 26 ++++++++++++++++++++++++++
 2 files changed, 32 insertions(+)
 create mode 100644 lld/test/COFF/lto-icf.ll

diff --git a/lld/COFF/LTO.cpp b/lld/COFF/LTO.cpp
index 8a1c82c..56ca8c0 100644
--- a/lld/COFF/LTO.cpp
+++ b/lld/COFF/LTO.cpp
@@ -64,6 +64,12 @@ static void saveBuffer(StringRef Buffer, const Twine &Path) {
 static std::unique_ptr<lto::LTO> createLTO() {
   lto::Config Conf;
   Conf.Options = InitTargetOptionsFromCodeGenFlags();
+
+  // Always emit a section per function/datum with LTO. LLVM LTO should get most
+  // of the benefit of linker GC, but there are still opportunities for ICF.
+  Conf.Options.FunctionSections = true;
+  Conf.Options.DataSections = true;
+
   // Use static reloc model on 32-bit x86 because it usually results in more
   // compact code, and because there are also known code generation bugs when
   // using the PIC model (see PR34306).
diff --git a/lld/test/COFF/lto-icf.ll b/lld/test/COFF/lto-icf.ll
new file mode 100644
index 0000000..291188f
--- /dev/null
+++ b/lld/test/COFF/lto-icf.ll
@@ -0,0 +1,26 @@
+; Test that ICF works after LTO, i.e. both functions have the same address.
+; Previously, when we didn't enable function sections, ICF didn't work.
+
+; RUN: llvm-as %s -o %t.bc
+; RUN: lld-link -opt:icf -dll -noentry %t.bc -out:%t.dll
+; RUN: llvm-readobj -coff-exports %t.dll | FileCheck %s
+
+; CHECK: Export {
+; CHECK: Export {
+; CHECK:   RVA: 0x[[RVA:.*]]
+; CHECK: Export {
+; CHECK:   RVA: 0x[[RVA]]
+; CHECK-NOT: Export
+
+target datalayout = "e-m:w-i64:64-f80:128-n8:16:32:64-S128"
+target triple = "x86_64-pc-windows-msvc19.12.25835"
+
+define dllexport i8* @icf_ptr() {
+entry:
+  ret i8* null
+}
+
+define dllexport i64 @icf_int() {
+entry:
+  ret i64 0
+}
-- 
2.7.4