[llvm] r350050 - Fix build - follow-up to r350048 which broke headerless (v4) address pool

[David Blaikie via llvm-commits]

Author: dblaikie
Date: Sun Dec 23 23:56:40 2018
New Revision: 350050

URL: http://llvm.org/viewvc/llvm-project?rev=350050&view=rev
Log:
Fix build - follow-up to r350048 which broke headerless (v4) address pool

Modified:
    llvm/trunk/lib/CodeGen/AsmPrinter/AddressPool.cpp
    llvm/trunk/lib/CodeGen/AsmPrinter/AddressPool.h

Modified: llvm/trunk/lib/CodeGen/AsmPrinter/AddressPool.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/CodeGen/AsmPrinter/AddressPool.cpp?rev=350050&r1=350049&r2=350050&view=diff
==============================================================================
--- llvm/trunk/lib/CodeGen/AsmPrinter/AddressPool.cpp (original)
+++ llvm/trunk/lib/CodeGen/AsmPrinter/AddressPool.cpp Sun Dec 23 23:56:40 2018
@@ -24,12 +24,11 @@ unsigned AddressPool::getIndex(const MCS
   return IterBool.first->second.Number;
 }
 
-
-void AddressPool::emitHeader(AsmPrinter &Asm, MCSection *Section) {
+MCSymbol *AddressPool::emitHeader(AsmPrinter &Asm, MCSection *Section) {
   static const uint8_t AddrSize = Asm.getDataLayout().getPointerSize();
   StringRef Prefix = "debug_addr_";
   MCSymbol *BeginLabel = Asm.createTempSymbol(Prefix + "start");
-  EndLabel = Asm.createTempSymbol(Prefix + "end");
+  MCSymbol *EndLabel = Asm.createTempSymbol(Prefix + "end");
   Asm.OutStreamer->AddComment("Length of contribution");
   Asm.EmitLabelDifference(EndLabel, BeginLabel,
                           4); // TODO: Support DWARF64 format.
@@ -40,6 +39,8 @@ void AddressPool::emitHeader(AsmPrinter
   Asm.emitInt8(AddrSize);
   Asm.OutStreamer->AddComment("Segment selector size");
   Asm.emitInt8(0); // TODO: Support non-zero segment_selector_size.
+
+  return EndLabel;
 }
 
 // Emit addresses into the section given.
@@ -50,8 +51,10 @@ void AddressPool::emit(AsmPrinter &Asm,
   // Start the dwarf addr section.
   Asm.OutStreamer->SwitchSection(AddrSection);
 
+  MCSymbol *EndLabel = nullptr;
+
   if (Asm.getDwarfVersion() >= 5)
-    emitHeader(Asm, AddrSection);
+    EndLabel = emitHeader(Asm, AddrSection);
 
   // Define the symbol that marks the start of the contribution.
   // It is referenced via DW_AT_addr_base.
@@ -69,5 +72,6 @@ void AddressPool::emit(AsmPrinter &Asm,
   for (const MCExpr *Entry : Entries)
     Asm.OutStreamer->EmitValue(Entry, Asm.getDataLayout().getPointerSize());
 
-  Asm.OutStreamer->EmitLabel(EndLabel);
+  if (EndLabel)
+    Asm.OutStreamer->EmitLabel(EndLabel);
 }

Modified: llvm/trunk/lib/CodeGen/AsmPrinter/AddressPool.h
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/CodeGen/AsmPrinter/AddressPool.h?rev=350050&r1=350049&r2=350050&view=diff
==============================================================================
--- llvm/trunk/lib/CodeGen/AsmPrinter/AddressPool.h (original)
+++ llvm/trunk/lib/CodeGen/AsmPrinter/AddressPool.h Sun Dec 23 23:56:40 2018
@@ -30,8 +30,6 @@ class AddressPool {
   };
   DenseMap<const MCSymbol *, AddressPoolEntry> Pool;
 
-  MCSymbol *EndLabel;
-
   /// Record whether the AddressPool has been queried for an address index since
   /// the last "resetUsedFlag" call. Used to implement type unit fallback - a
   /// type that references addresses cannot be placed in a type unit when using
@@ -57,7 +55,7 @@ public:
   void setLabel(MCSymbol *Sym) { AddressTableBaseSym = Sym; }
 
 private:
-  void emitHeader(AsmPrinter &Asm, MCSection *Section);
+  MCSymbol *emitHeader(AsmPrinter &Asm, MCSection *Section);
 
   /// Symbol designates the start of the contribution to the address table.
   MCSymbol *AddressTableBaseSym = nullptr;