contrib/llvm/lib/MC/MCStreamer.cpp

   1 //===- lib/MC/MCStreamer.cpp - Streaming Machine Code Output --------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #include "llvm/MC/MCStreamer.h"
  11 #include "llvm/ADT/SmallString.h"
  12 #include "llvm/ADT/StringRef.h"
  13 #include "llvm/ADT/Twine.h"
  14 #include "llvm/BinaryFormat/COFF.h"
  15 #include "llvm/MC/MCAsmBackend.h"
  16 #include "llvm/MC/MCAsmInfo.h"
  17 #include "llvm/MC/MCCodeView.h"
  18 #include "llvm/MC/MCContext.h"
  19 #include "llvm/MC/MCDwarf.h"
  20 #include "llvm/MC/MCExpr.h"
  21 #include "llvm/MC/MCInst.h"
  22 #include "llvm/MC/MCInstPrinter.h"
  23 #include "llvm/MC/MCObjectFileInfo.h"
  24 #include "llvm/MC/MCSection.h"
  25 #include "llvm/MC/MCSectionCOFF.h"
  26 #include "llvm/MC/MCSymbol.h"
  27 #include "llvm/MC/MCWin64EH.h"
  28 #include "llvm/MC/MCWinEH.h"
  29 #include "llvm/Support/Casting.h"
  30 #include "llvm/Support/ErrorHandling.h"
  31 #include "llvm/Support/LEB128.h"
  32 #include "llvm/Support/MathExtras.h"
  33 #include "llvm/Support/raw_ostream.h"
  34 #include <cassert>
  35 #include <cstdint>
  36 #include <cstdlib>
  37 #include <utility>
  38
  39 using namespace llvm;
  40
  41 MCTargetStreamer::MCTargetStreamer(MCStreamer &S) : Streamer(S) {
  42   S.setTargetStreamer(this);
  43 }
  44
  45 // Pin the vtables to this file.
  46 MCTargetStreamer::~MCTargetStreamer() = default;
  47
  48 void MCTargetStreamer::emitLabel(MCSymbol *Symbol) {}
  49
  50 void MCTargetStreamer::finish() {}
  51
  52 void MCTargetStreamer::changeSection(const MCSection *CurSection,
  53                                      MCSection *Section,
  54                                      const MCExpr *Subsection,
  55                                      raw_ostream &OS) {
  56   Section->PrintSwitchToSection(
  57       *Streamer.getContext().getAsmInfo(),
  58       Streamer.getContext().getObjectFileInfo()->getTargetTriple(), OS,
  59       Subsection);
  60 }
  61
  62 void MCTargetStreamer::emitDwarfFileDirective(StringRef Directive) {
  63   Streamer.EmitRawText(Directive);
  64 }
  65
  66 void MCTargetStreamer::emitValue(const MCExpr *Value) {
  67   SmallString<128> Str;
  68   raw_svector_ostream OS(Str);
  69
  70   Value->print(OS, Streamer.getContext().getAsmInfo());
  71   Streamer.EmitRawText(OS.str());
  72 }
  73
  74 void MCTargetStreamer::emitAssignment(MCSymbol *Symbol, const MCExpr *Value) {}
  75
  76 MCStreamer::MCStreamer(MCContext &Ctx)
  77     : Context(Ctx), CurrentWinFrameInfo(nullptr) {
  78   SectionStack.push_back(std::pair<MCSectionSubPair, MCSectionSubPair>());
  79 }
  80
  81 MCStreamer::~MCStreamer() {}
  82
  83 void MCStreamer::reset() {
  84   DwarfFrameInfos.clear();
  85   CurrentWinFrameInfo = nullptr;
  86   WinFrameInfos.clear();
  87   SymbolOrdering.clear();
  88   SectionStack.clear();
  89   SectionStack.push_back(std::pair<MCSectionSubPair, MCSectionSubPair>());
  90 }
  91
  92 raw_ostream &MCStreamer::GetCommentOS() {
  93   // By default, discard comments.
  94   return nulls();
  95 }
  96
  97 void MCStreamer::emitRawComment(const Twine &T, bool TabPrefix) {}
  98
  99 void MCStreamer::addExplicitComment(const Twine &T) {}
 100 void MCStreamer::emitExplicitComments() {}
 101
 102 void MCStreamer::generateCompactUnwindEncodings(MCAsmBackend *MAB) {
 103   for (auto &FI : DwarfFrameInfos)
 104     FI.CompactUnwindEncoding =
 105         (MAB ? MAB->generateCompactUnwindEncoding(FI.Instructions) : 0);
 106 }
 107
 108 /// EmitIntValue - Special case of EmitValue that avoids the client having to
 109 /// pass in a MCExpr for constant integers.
 110 void MCStreamer::EmitIntValue(uint64_t Value, unsigned Size) {
 111   assert(1 <= Size && Size <= 8 && "Invalid size");
 112   assert((isUIntN(8 * Size, Value) || isIntN(8 * Size, Value)) &&
 113          "Invalid size");
 114   char buf[8];
 115   const bool isLittleEndian = Context.getAsmInfo()->isLittleEndian();
 116   for (unsigned i = 0; i != Size; ++i) {
 117     unsigned index = isLittleEndian ? i : (Size - i - 1);
 118     buf[i] = uint8_t(Value >> (index * 8));
 119   }
 120   EmitBytes(StringRef(buf, Size));
 121 }
 122
 123 /// EmitULEB128Value - Special case of EmitULEB128Value that avoids the
 124 /// client having to pass in a MCExpr for constant integers.
 125 void MCStreamer::EmitPaddedULEB128IntValue(uint64_t Value, unsigned PadTo) {
 126   SmallString<128> Tmp;
 127   raw_svector_ostream OSE(Tmp);
 128   encodeULEB128(Value, OSE, PadTo);
 129   EmitBytes(OSE.str());
 130 }
 131
 132 void MCStreamer::EmitULEB128IntValue(uint64_t Value) {
 133   EmitPaddedULEB128IntValue(Value, 0);
 134 }
 135
 136 /// EmitSLEB128Value - Special case of EmitSLEB128Value that avoids the
 137 /// client having to pass in a MCExpr for constant integers.
 138 void MCStreamer::EmitSLEB128IntValue(int64_t Value) {
 139   SmallString<128> Tmp;
 140   raw_svector_ostream OSE(Tmp);
 141   encodeSLEB128(Value, OSE);
 142   EmitBytes(OSE.str());
 143 }
 144
 145 void MCStreamer::EmitValue(const MCExpr *Value, unsigned Size, SMLoc Loc) {
 146   EmitValueImpl(Value, Size, Loc);
 147 }
 148
 149 void MCStreamer::EmitSymbolValue(const MCSymbol *Sym, unsigned Size,
 150                                  bool IsSectionRelative) {
 151   assert((!IsSectionRelative || Size == 4) &&
 152          "SectionRelative value requires 4-bytes");
 153
 154   if (!IsSectionRelative)
 155     EmitValueImpl(MCSymbolRefExpr::create(Sym, getContext()), Size);
 156   else
 157     EmitCOFFSecRel32(Sym, /*Offset=*/0);
 158 }
 159
 160 void MCStreamer::EmitDTPRel64Value(const MCExpr *Value) {
 161   report_fatal_error("unsupported directive in streamer");
 162 }
 163
 164 void MCStreamer::EmitDTPRel32Value(const MCExpr *Value) {
 165   report_fatal_error("unsupported directive in streamer");
 166 }
 167
 168 void MCStreamer::EmitTPRel64Value(const MCExpr *Value) {
 169   report_fatal_error("unsupported directive in streamer");
 170 }
 171
 172 void MCStreamer::EmitTPRel32Value(const MCExpr *Value) {
 173   report_fatal_error("unsupported directive in streamer");
 174 }
 175
 176 void MCStreamer::EmitGPRel64Value(const MCExpr *Value) {
 177   report_fatal_error("unsupported directive in streamer");
 178 }
 179
 180 void MCStreamer::EmitGPRel32Value(const MCExpr *Value) {
 181   report_fatal_error("unsupported directive in streamer");
 182 }
 183
 184 /// Emit NumBytes bytes worth of the value specified by FillValue.
 185 /// This implements directives such as '.space'.
 186 void MCStreamer::emitFill(uint64_t NumBytes, uint8_t FillValue) {
 187   for (uint64_t i = 0, e = NumBytes; i != e; ++i)
 188     EmitIntValue(FillValue, 1);
 189 }
 190
 191 void MCStreamer::emitFill(uint64_t NumValues, int64_t Size, int64_t Expr) {
 192   int64_t NonZeroSize = Size > 4 ? 4 : Size;
 193   Expr &= ~0ULL >> (64 - NonZeroSize * 8);
 194   for (uint64_t i = 0, e = NumValues; i != e; ++i) {
 195     EmitIntValue(Expr, NonZeroSize);
 196     if (NonZeroSize < Size)
 197       EmitIntValue(0, Size - NonZeroSize);
 198   }
 199 }
 200
 201 /// The implementation in this class just redirects to emitFill.
 202 void MCStreamer::EmitZeros(uint64_t NumBytes) {
 203   emitFill(NumBytes, 0);
 204 }
 205
 206 unsigned MCStreamer::EmitDwarfFileDirective(unsigned FileNo,
 207                                             StringRef Directory,
 208                                             StringRef Filename, unsigned CUID) {
 209   return getContext().getDwarfFile(Directory, Filename, FileNo, CUID);
 210 }
 211
 212 void MCStreamer::EmitDwarfLocDirective(unsigned FileNo, unsigned Line,
 213                                        unsigned Column, unsigned Flags,
 214                                        unsigned Isa,
 215                                        unsigned Discriminator,
 216                                        StringRef FileName) {
 217   getContext().setCurrentDwarfLoc(FileNo, Line, Column, Flags, Isa,
 218                                   Discriminator);
 219 }
 220
 221 MCSymbol *MCStreamer::getDwarfLineTableSymbol(unsigned CUID) {
 222   MCDwarfLineTable &Table = getContext().getMCDwarfLineTable(CUID);
 223   if (!Table.getLabel()) {
 224     StringRef Prefix = Context.getAsmInfo()->getPrivateGlobalPrefix();
 225     Table.setLabel(
 226         Context.getOrCreateSymbol(Prefix + "line_table_start" + Twine(CUID)));
 227   }
 228   return Table.getLabel();
 229 }
 230
 231 bool MCStreamer::hasUnfinishedDwarfFrameInfo() {
 232   return !DwarfFrameInfos.empty() && !DwarfFrameInfos.back().End;
 233 }
 234
 235 MCDwarfFrameInfo *MCStreamer::getCurrentDwarfFrameInfo() {
 236   if (!hasUnfinishedDwarfFrameInfo()) {
 237     getContext().reportError(SMLoc(), "this directive must appear between "
 238                                       ".cfi_startproc and .cfi_endproc "
 239                                       "directives");
 240     return nullptr;
 241   }
 242   return &DwarfFrameInfos.back();
 243 }
 244
 245 bool MCStreamer::EmitCVFileDirective(unsigned FileNo, StringRef Filename,
 246                                      ArrayRef<uint8_t> Checksum,
 247                                      unsigned ChecksumKind) {
 248   return getContext().getCVContext().addFile(*this, FileNo, Filename, Checksum,
 249                                              ChecksumKind);
 250 }
 251
 252 bool MCStreamer::EmitCVFuncIdDirective(unsigned FunctionId) {
 253   return getContext().getCVContext().recordFunctionId(FunctionId);
 254 }
 255
 256 bool MCStreamer::EmitCVInlineSiteIdDirective(unsigned FunctionId,
 257                                              unsigned IAFunc, unsigned IAFile,
 258                                              unsigned IALine, unsigned IACol,
 259                                              SMLoc Loc) {
 260   if (getContext().getCVContext().getCVFunctionInfo(IAFunc) == nullptr) {
 261     getContext().reportError(Loc, "parent function id not introduced by "
 262                                   ".cv_func_id or .cv_inline_site_id");
 263     return true;
 264   }
 265
 266   return getContext().getCVContext().recordInlinedCallSiteId(
 267       FunctionId, IAFunc, IAFile, IALine, IACol);
 268 }
 269
 270 void MCStreamer::EmitCVLocDirective(unsigned FunctionId, unsigned FileNo,
 271                                     unsigned Line, unsigned Column,
 272                                     bool PrologueEnd, bool IsStmt,
 273                                     StringRef FileName, SMLoc Loc) {
 274   CodeViewContext &CVC = getContext().getCVContext();
 275   MCCVFunctionInfo *FI = CVC.getCVFunctionInfo(FunctionId);
 276   if (!FI)
 277     return getContext().reportError(
 278         Loc, "function id not introduced by .cv_func_id or .cv_inline_site_id");
 279
 280   // Track the section
 281   if (FI->Section == nullptr)
 282     FI->Section = getCurrentSectionOnly();
 283   else if (FI->Section != getCurrentSectionOnly())
 284     return getContext().reportError(
 285         Loc,
 286         "all .cv_loc directives for a function must be in the same section");
 287
 288   CVC.setCurrentCVLoc(FunctionId, FileNo, Line, Column, PrologueEnd, IsStmt);
 289 }
 290
 291 void MCStreamer::EmitCVLinetableDirective(unsigned FunctionId,
 292                                           const MCSymbol *Begin,
 293                                           const MCSymbol *End) {}
 294
 295 void MCStreamer::EmitCVInlineLinetableDirective(unsigned PrimaryFunctionId,
 296                                                 unsigned SourceFileId,
 297                                                 unsigned SourceLineNum,
 298                                                 const MCSymbol *FnStartSym,
 299                                                 const MCSymbol *FnEndSym) {}
 300
 301 void MCStreamer::EmitCVDefRangeDirective(
 302     ArrayRef<std::pair<const MCSymbol *, const MCSymbol *>> Ranges,
 303     StringRef FixedSizePortion) {}
 304
 305 void MCStreamer::EmitEHSymAttributes(const MCSymbol *Symbol,
 306                                      MCSymbol *EHSymbol) {
 307 }
 308
 309 void MCStreamer::InitSections(bool NoExecStack) {
 310   SwitchSection(getContext().getObjectFileInfo()->getTextSection());
 311 }
 312
 313 void MCStreamer::AssignFragment(MCSymbol *Symbol, MCFragment *Fragment) {
 314   assert(Fragment);
 315   Symbol->setFragment(Fragment);
 316
 317   // As we emit symbols into a section, track the order so that they can
 318   // be sorted upon later. Zero is reserved to mean 'unemitted'.
 319   SymbolOrdering[Symbol] = 1 + SymbolOrdering.size();
 320 }
 321
 322 void MCStreamer::EmitLabel(MCSymbol *Symbol, SMLoc Loc) {
 323   Symbol->redefineIfPossible();
 324
 325   if (!Symbol->isUndefined() || Symbol->isVariable())
 326     return getContext().reportError(Loc, "invalid symbol redefinition");
 327
 328   assert(!Symbol->isVariable() && "Cannot emit a variable symbol!");
 329   assert(getCurrentSectionOnly() && "Cannot emit before setting section!");
 330   assert(!Symbol->getFragment() && "Unexpected fragment on symbol data!");
 331   assert(Symbol->isUndefined() && "Cannot define a symbol twice!");
 332
 333   Symbol->setFragment(&getCurrentSectionOnly()->getDummyFragment());
 334
 335   MCTargetStreamer *TS = getTargetStreamer();
 336   if (TS)
 337     TS->emitLabel(Symbol);
 338 }
 339
 340 void MCStreamer::EmitCFISections(bool EH, bool Debug) {
 341   assert(EH || Debug);
 342 }
 343
 344 void MCStreamer::EmitCFIStartProc(bool IsSimple) {
 345   if (hasUnfinishedDwarfFrameInfo())
 346     getContext().reportError(
 347         SMLoc(), "starting new .cfi frame before finishing the previous one");
 348
 349   MCDwarfFrameInfo Frame;
 350   Frame.IsSimple = IsSimple;
 351   EmitCFIStartProcImpl(Frame);
 352
 353   const MCAsmInfo* MAI = Context.getAsmInfo();
 354   if (MAI) {
 355     for (const MCCFIInstruction& Inst : MAI->getInitialFrameState()) {
 356       if (Inst.getOperation() == MCCFIInstruction::OpDefCfa ||
 357           Inst.getOperation() == MCCFIInstruction::OpDefCfaRegister) {
 358         Frame.CurrentCfaRegister = Inst.getRegister();
 359       }
 360     }
 361   }
 362
 363   DwarfFrameInfos.push_back(Frame);
 364 }
 365
 366 void MCStreamer::EmitCFIStartProcImpl(MCDwarfFrameInfo &Frame) {
 367 }
 368
 369 void MCStreamer::EmitCFIEndProc() {
 370   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 371   if (!CurFrame)
 372     return;
 373   EmitCFIEndProcImpl(*CurFrame);
 374 }
 375
 376 void MCStreamer::EmitCFIEndProcImpl(MCDwarfFrameInfo &Frame) {
 377   // Put a dummy non-null value in Frame.End to mark that this frame has been
 378   // closed.
 379   Frame.End = (MCSymbol *)1;
 380 }
 381
 382 MCSymbol *MCStreamer::EmitCFILabel() {
 383   // Return a dummy non-null value so that label fields appear filled in when
 384   // generating textual assembly.
 385   return (MCSymbol *)1;
 386 }
 387
 388 void MCStreamer::EmitCFIDefCfa(int64_t Register, int64_t Offset) {
 389   MCSymbol *Label = EmitCFILabel();
 390   MCCFIInstruction Instruction =
 391     MCCFIInstruction::createDefCfa(Label, Register, Offset);
 392   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 393   if (!CurFrame)
 394     return;
 395   CurFrame->Instructions.push_back(Instruction);
 396   CurFrame->CurrentCfaRegister = static_cast<unsigned>(Register);
 397 }
 398
 399 void MCStreamer::EmitCFIDefCfaOffset(int64_t Offset) {
 400   MCSymbol *Label = EmitCFILabel();
 401   MCCFIInstruction Instruction =
 402     MCCFIInstruction::createDefCfaOffset(Label, Offset);
 403   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 404   if (!CurFrame)
 405     return;
 406   CurFrame->Instructions.push_back(Instruction);
 407 }
 408
 409 void MCStreamer::EmitCFIAdjustCfaOffset(int64_t Adjustment) {
 410   MCSymbol *Label = EmitCFILabel();
 411   MCCFIInstruction Instruction =
 412     MCCFIInstruction::createAdjustCfaOffset(Label, Adjustment);
 413   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 414   if (!CurFrame)
 415     return;
 416   CurFrame->Instructions.push_back(Instruction);
 417 }
 418
 419 void MCStreamer::EmitCFIDefCfaRegister(int64_t Register) {
 420   MCSymbol *Label = EmitCFILabel();
 421   MCCFIInstruction Instruction =
 422     MCCFIInstruction::createDefCfaRegister(Label, Register);
 423   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 424   if (!CurFrame)
 425     return;
 426   CurFrame->Instructions.push_back(Instruction);
 427   CurFrame->CurrentCfaRegister = static_cast<unsigned>(Register);
 428 }
 429
 430 void MCStreamer::EmitCFIOffset(int64_t Register, int64_t Offset) {
 431   MCSymbol *Label = EmitCFILabel();
 432   MCCFIInstruction Instruction =
 433     MCCFIInstruction::createOffset(Label, Register, Offset);
 434   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 435   if (!CurFrame)
 436     return;
 437   CurFrame->Instructions.push_back(Instruction);
 438 }
 439
 440 void MCStreamer::EmitCFIRelOffset(int64_t Register, int64_t Offset) {
 441   MCSymbol *Label = EmitCFILabel();
 442   MCCFIInstruction Instruction =
 443     MCCFIInstruction::createRelOffset(Label, Register, Offset);
 444   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 445   if (!CurFrame)
 446     return;
 447   CurFrame->Instructions.push_back(Instruction);
 448 }
 449
 450 void MCStreamer::EmitCFIPersonality(const MCSymbol *Sym,
 451                                     unsigned Encoding) {
 452   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 453   if (!CurFrame)
 454     return;
 455   CurFrame->Personality = Sym;
 456   CurFrame->PersonalityEncoding = Encoding;
 457 }
 458
 459 void MCStreamer::EmitCFILsda(const MCSymbol *Sym, unsigned Encoding) {
 460   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 461   if (!CurFrame)
 462     return;
 463   CurFrame->Lsda = Sym;
 464   CurFrame->LsdaEncoding = Encoding;
 465 }
 466
 467 void MCStreamer::EmitCFIRememberState() {
 468   MCSymbol *Label = EmitCFILabel();
 469   MCCFIInstruction Instruction = MCCFIInstruction::createRememberState(Label);
 470   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 471   if (!CurFrame)
 472     return;
 473   CurFrame->Instructions.push_back(Instruction);
 474 }
 475
 476 void MCStreamer::EmitCFIRestoreState() {
 477   // FIXME: Error if there is no matching cfi_remember_state.
 478   MCSymbol *Label = EmitCFILabel();
 479   MCCFIInstruction Instruction = MCCFIInstruction::createRestoreState(Label);
 480   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 481   if (!CurFrame)
 482     return;
 483   CurFrame->Instructions.push_back(Instruction);
 484 }
 485
 486 void MCStreamer::EmitCFISameValue(int64_t Register) {
 487   MCSymbol *Label = EmitCFILabel();
 488   MCCFIInstruction Instruction =
 489     MCCFIInstruction::createSameValue(Label, Register);
 490   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 491   if (!CurFrame)
 492     return;
 493   CurFrame->Instructions.push_back(Instruction);
 494 }
 495
 496 void MCStreamer::EmitCFIRestore(int64_t Register) {
 497   MCSymbol *Label = EmitCFILabel();
 498   MCCFIInstruction Instruction =
 499     MCCFIInstruction::createRestore(Label, Register);
 500   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 501   if (!CurFrame)
 502     return;
 503   CurFrame->Instructions.push_back(Instruction);
 504 }
 505
 506 void MCStreamer::EmitCFIEscape(StringRef Values) {
 507   MCSymbol *Label = EmitCFILabel();
 508   MCCFIInstruction Instruction = MCCFIInstruction::createEscape(Label, Values);
 509   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 510   if (!CurFrame)
 511     return;
 512   CurFrame->Instructions.push_back(Instruction);
 513 }
 514
 515 void MCStreamer::EmitCFIGnuArgsSize(int64_t Size) {
 516   MCSymbol *Label = EmitCFILabel();
 517   MCCFIInstruction Instruction =
 518     MCCFIInstruction::createGnuArgsSize(Label, Size);
 519   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 520   if (!CurFrame)
 521     return;
 522   CurFrame->Instructions.push_back(Instruction);
 523 }
 524
 525 void MCStreamer::EmitCFISignalFrame() {
 526   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 527   if (!CurFrame)
 528     return;
 529   CurFrame->IsSignalFrame = true;
 530 }
 531
 532 void MCStreamer::EmitCFIUndefined(int64_t Register) {
 533   MCSymbol *Label = EmitCFILabel();
 534   MCCFIInstruction Instruction =
 535     MCCFIInstruction::createUndefined(Label, Register);
 536   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 537   if (!CurFrame)
 538     return;
 539   CurFrame->Instructions.push_back(Instruction);
 540 }
 541
 542 void MCStreamer::EmitCFIRegister(int64_t Register1, int64_t Register2) {
 543   MCSymbol *Label = EmitCFILabel();
 544   MCCFIInstruction Instruction =
 545     MCCFIInstruction::createRegister(Label, Register1, Register2);
 546   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 547   if (!CurFrame)
 548     return;
 549   CurFrame->Instructions.push_back(Instruction);
 550 }
 551
 552 void MCStreamer::EmitCFIWindowSave() {
 553   MCSymbol *Label = EmitCFILabel();
 554   MCCFIInstruction Instruction =
 555     MCCFIInstruction::createWindowSave(Label);
 556   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 557   if (!CurFrame)
 558     return;
 559   CurFrame->Instructions.push_back(Instruction);
 560 }
 561
 562 void MCStreamer::EmitCFIReturnColumn(int64_t Register) {
 563   MCDwarfFrameInfo *CurFrame = getCurrentDwarfFrameInfo();
 564   if (!CurFrame)
 565     return;
 566   CurFrame->RAReg = Register;
 567 }
 568
 569 WinEH::FrameInfo *MCStreamer::EnsureValidWinFrameInfo(SMLoc Loc) {
 570   const MCAsmInfo *MAI = Context.getAsmInfo();
 571   if (!MAI->usesWindowsCFI()) {
 572     getContext().reportError(
 573         Loc, ".seh_* directives are not supported on this target");
 574     return nullptr;
 575   }
 576   if (!CurrentWinFrameInfo || CurrentWinFrameInfo->End) {
 577     getContext().reportError(
 578         Loc, ".seh_ directive must appear within an active frame");
 579     return nullptr;
 580   }
 581   return CurrentWinFrameInfo;
 582 }
 583
 584 void MCStreamer::EmitWinCFIStartProc(const MCSymbol *Symbol, SMLoc Loc) {
 585   const MCAsmInfo *MAI = Context.getAsmInfo();
 586   if (!MAI->usesWindowsCFI())
 587     return getContext().reportError(
 588         Loc, ".seh_* directives are not supported on this target");
 589   if (CurrentWinFrameInfo && !CurrentWinFrameInfo->End)
 590     getContext().reportError(
 591         Loc, "Starting a function before ending the previous one!");
 592
 593   MCSymbol *StartProc = EmitCFILabel();
 594
 595   WinFrameInfos.emplace_back(
 596       llvm::make_unique<WinEH::FrameInfo>(Symbol, StartProc));
 597   CurrentWinFrameInfo = WinFrameInfos.back().get();
 598   CurrentWinFrameInfo->TextSection = getCurrentSectionOnly();
 599 }
 600
 601 void MCStreamer::EmitWinCFIEndProc(SMLoc Loc) {
 602   WinEH::FrameInfo *CurFrame = EnsureValidWinFrameInfo(Loc);
 603   if (!CurFrame)
 604     return;
 605   if (CurFrame->ChainedParent)
 606     getContext().reportError(Loc, "Not all chained regions terminated!");
 607
 608   MCSymbol *Label = EmitCFILabel();
 609   CurFrame->End = Label;
 610 }
 611
 612 void MCStreamer::EmitWinCFIStartChained(SMLoc Loc) {
 613   WinEH::FrameInfo *CurFrame = EnsureValidWinFrameInfo(Loc);
 614   if (!CurFrame)
 615     return;
 616
 617   MCSymbol *StartProc = EmitCFILabel();
 618
 619   WinFrameInfos.emplace_back(llvm::make_unique<WinEH::FrameInfo>(
 620       CurFrame->Function, StartProc, CurFrame));
 621   CurrentWinFrameInfo = WinFrameInfos.back().get();
 622   CurrentWinFrameInfo->TextSection = getCurrentSectionOnly();
 623 }
 624
 625 void MCStreamer::EmitWinCFIEndChained(SMLoc Loc) {
 626   WinEH::FrameInfo *CurFrame = EnsureValidWinFrameInfo(Loc);
 627   if (!CurFrame)
 628     return;
 629   if (!CurFrame->ChainedParent)
 630     return getContext().reportError(
 631         Loc, "End of a chained region outside a chained region!");
 632
 633   MCSymbol *Label = EmitCFILabel();
 634
 635   CurFrame->End = Label;
 636   CurrentWinFrameInfo = const_cast<WinEH::FrameInfo *>(CurFrame->ChainedParent);
 637 }
 638
 639 void MCStreamer::EmitWinEHHandler(const MCSymbol *Sym, bool Unwind, bool Except,
 640                                   SMLoc Loc) {
 641   WinEH::FrameInfo *CurFrame = EnsureValidWinFrameInfo(Loc);
 642   if (!CurFrame)
 643     return;
 644   if (CurFrame->ChainedParent)
 645     return getContext().reportError(
 646         Loc, "Chained unwind areas can't have handlers!");
 647   CurFrame->ExceptionHandler = Sym;
 648   if (!Except && !Unwind)
 649     getContext().reportError(Loc, "Don't know what kind of handler this is!");
 650   if (Unwind)
 651     CurFrame->HandlesUnwind = true;
 652   if (Except)
 653     CurFrame->HandlesExceptions = true;
 654 }
 655
 656 void MCStreamer::EmitWinEHHandlerData(SMLoc Loc) {
 657   WinEH::FrameInfo *CurFrame = EnsureValidWinFrameInfo(Loc);
 658   if (!CurFrame)
 659     return;
 660   if (CurFrame->ChainedParent)
 661     getContext().reportError(Loc, "Chained unwind areas can't have handlers!");
 662 }
 663
 664 static MCSection *getWinCFISection(MCContext &Context, unsigned *NextWinCFIID,
 665                                    MCSection *MainCFISec,
 666                                    const MCSection *TextSec) {
 667   // If this is the main .text section, use the main unwind info section.
 668   if (TextSec == Context.getObjectFileInfo()->getTextSection())
 669     return MainCFISec;
 670
 671   const auto *TextSecCOFF = cast<MCSectionCOFF>(TextSec);
 672   unsigned UniqueID = TextSecCOFF->getOrAssignWinCFISectionID(NextWinCFIID);
 673
 674   // If this section is COMDAT, this unwind section should be COMDAT associative
 675   // with its group.
 676   const MCSymbol *KeySym = nullptr;
 677   if (TextSecCOFF->getCharacteristics() & COFF::IMAGE_SCN_LNK_COMDAT)
 678     KeySym = TextSecCOFF->getCOMDATSymbol();
 679
 680   return Context.getAssociativeCOFFSection(cast<MCSectionCOFF>(MainCFISec),
 681                                            KeySym, UniqueID);
 682 }
 683
 684 MCSection *MCStreamer::getAssociatedPDataSection(const MCSection *TextSec) {
 685   return getWinCFISection(getContext(), &NextWinCFIID,
 686                           getContext().getObjectFileInfo()->getPDataSection(),
 687                           TextSec);
 688 }
 689
 690 MCSection *MCStreamer::getAssociatedXDataSection(const MCSection *TextSec) {
 691   return getWinCFISection(getContext(), &NextWinCFIID,
 692                           getContext().getObjectFileInfo()->getXDataSection(),
 693                           TextSec);
 694 }
 695
 696 void MCStreamer::EmitSyntaxDirective() {}
 697
 698 void MCStreamer::EmitWinCFIPushReg(unsigned Register, SMLoc Loc) {
 699   WinEH::FrameInfo *CurFrame = EnsureValidWinFrameInfo(Loc);
 700   if (!CurFrame)
 701     return;
 702
 703   MCSymbol *Label = EmitCFILabel();
 704
 705   WinEH::Instruction Inst = Win64EH::Instruction::PushNonVol(Label, Register);
 706   CurFrame->Instructions.push_back(Inst);
 707 }
 708
 709 void MCStreamer::EmitWinCFISetFrame(unsigned Register, unsigned Offset,
 710                                     SMLoc Loc) {
 711   WinEH::FrameInfo *CurFrame = EnsureValidWinFrameInfo(Loc);
 712   if (!CurFrame)
 713     return;
 714   if (CurFrame->LastFrameInst >= 0)
 715     return getContext().reportError(
 716         Loc, "frame register and offset can be set at most once");
 717   if (Offset & 0x0F)
 718     return getContext().reportError(Loc, "offset is not a multiple of 16");
 719   if (Offset > 240)
 720     return getContext().reportError(
 721         Loc, "frame offset must be less than or equal to 240");
 722
 723   MCSymbol *Label = EmitCFILabel();
 724
 725   WinEH::Instruction Inst =
 726       Win64EH::Instruction::SetFPReg(Label, Register, Offset);
 727   CurFrame->LastFrameInst = CurFrame->Instructions.size();
 728   CurFrame->Instructions.push_back(Inst);
 729 }
 730
 731 void MCStreamer::EmitWinCFIAllocStack(unsigned Size, SMLoc Loc) {
 732   WinEH::FrameInfo *CurFrame = EnsureValidWinFrameInfo(Loc);
 733   if (!CurFrame)
 734     return;
 735   if (Size == 0)
 736     return getContext().reportError(Loc,
 737                                     "stack allocation size must be non-zero");
 738   if (Size & 7)
 739     return getContext().reportError(
 740         Loc, "stack allocation size is not a multiple of 8");
 741
 742   MCSymbol *Label = EmitCFILabel();
 743
 744   WinEH::Instruction Inst = Win64EH::Instruction::Alloc(Label, Size);
 745   CurFrame->Instructions.push_back(Inst);
 746 }
 747
 748 void MCStreamer::EmitWinCFISaveReg(unsigned Register, unsigned Offset,
 749                                    SMLoc Loc) {
 750   WinEH::FrameInfo *CurFrame = EnsureValidWinFrameInfo(Loc);
 751   if (!CurFrame)
 752     return;
 753
 754   if (Offset & 7)
 755     return getContext().reportError(
 756         Loc, "register save offset is not 8 byte aligned");
 757
 758   MCSymbol *Label = EmitCFILabel();
 759
 760   WinEH::Instruction Inst =
 761       Win64EH::Instruction::SaveNonVol(Label, Register, Offset);
 762   CurFrame->Instructions.push_back(Inst);
 763 }
 764
 765 void MCStreamer::EmitWinCFISaveXMM(unsigned Register, unsigned Offset,
 766                                    SMLoc Loc) {
 767   WinEH::FrameInfo *CurFrame = EnsureValidWinFrameInfo(Loc);
 768   if (!CurFrame)
 769     return;
 770   if (Offset & 0x0F)
 771     return getContext().reportError(Loc, "offset is not a multiple of 16");
 772
 773   MCSymbol *Label = EmitCFILabel();
 774
 775   WinEH::Instruction Inst =
 776       Win64EH::Instruction::SaveXMM(Label, Register, Offset);
 777   CurFrame->Instructions.push_back(Inst);
 778 }
 779
 780 void MCStreamer::EmitWinCFIPushFrame(bool Code, SMLoc Loc) {
 781   WinEH::FrameInfo *CurFrame = EnsureValidWinFrameInfo(Loc);
 782   if (!CurFrame)
 783     return;
 784   if (!CurFrame->Instructions.empty())
 785     return getContext().reportError(
 786         Loc, "If present, PushMachFrame must be the first UOP");
 787
 788   MCSymbol *Label = EmitCFILabel();
 789
 790   WinEH::Instruction Inst = Win64EH::Instruction::PushMachFrame(Label, Code);
 791   CurFrame->Instructions.push_back(Inst);
 792 }
 793
 794 void MCStreamer::EmitWinCFIEndProlog(SMLoc Loc) {
 795   WinEH::FrameInfo *CurFrame = EnsureValidWinFrameInfo(Loc);
 796   if (!CurFrame)
 797     return;
 798
 799   MCSymbol *Label = EmitCFILabel();
 800
 801   CurFrame->PrologEnd = Label;
 802 }
 803
 804 void MCStreamer::EmitCOFFSafeSEH(MCSymbol const *Symbol) {
 805 }
 806
 807 void MCStreamer::EmitCOFFSectionIndex(MCSymbol const *Symbol) {
 808 }
 809
 810 void MCStreamer::EmitCOFFSecRel32(MCSymbol const *Symbol, uint64_t Offset) {}
 811
 812 /// EmitRawText - If this file is backed by an assembly streamer, this dumps
 813 /// the specified string in the output .s file.  This capability is
 814 /// indicated by the hasRawTextSupport() predicate.
 815 void MCStreamer::EmitRawTextImpl(StringRef String) {
 816   errs() << "EmitRawText called on an MCStreamer that doesn't support it, "
 817   " something must not be fully mc'ized\n";
 818   abort();
 819 }
 820
 821 void MCStreamer::EmitRawText(const Twine &T) {
 822   SmallString<128> Str;
 823   EmitRawTextImpl(T.toStringRef(Str));
 824 }
 825
 826 void MCStreamer::EmitWindowsUnwindTables() {
 827 }
 828
 829 void MCStreamer::Finish() {
 830   if (!DwarfFrameInfos.empty() && !DwarfFrameInfos.back().End)
 831     getContext().reportError(SMLoc(), "Unfinished frame!");
 832   if (!WinFrameInfos.empty() && !WinFrameInfos.back()->End)
 833     getContext().reportError(SMLoc(), "Unfinished frame!");
 834
 835   MCTargetStreamer *TS = getTargetStreamer();
 836   if (TS)
 837     TS->finish();
 838
 839   FinishImpl();
 840 }
 841
 842 void MCStreamer::EmitAssignment(MCSymbol *Symbol, const MCExpr *Value) {
 843   visitUsedExpr(*Value);
 844   Symbol->setVariableValue(Value);
 845
 846   MCTargetStreamer *TS = getTargetStreamer();
 847   if (TS)
 848     TS->emitAssignment(Symbol, Value);
 849 }
 850
 851 void MCTargetStreamer::prettyPrintAsm(MCInstPrinter &InstPrinter, raw_ostream &OS,
 852                               const MCInst &Inst, const MCSubtargetInfo &STI) {
 853   InstPrinter.printInst(&Inst, OS, "", STI);
 854 }
 855
 856 void MCStreamer::visitUsedSymbol(const MCSymbol &Sym) {
 857 }
 858
 859 void MCStreamer::visitUsedExpr(const MCExpr &Expr) {
 860   switch (Expr.getKind()) {
 861   case MCExpr::Target:
 862     cast<MCTargetExpr>(Expr).visitUsedExpr(*this);
 863     break;
 864
 865   case MCExpr::Constant:
 866     break;
 867
 868   case MCExpr::Binary: {
 869     const MCBinaryExpr &BE = cast<MCBinaryExpr>(Expr);
 870     visitUsedExpr(*BE.getLHS());
 871     visitUsedExpr(*BE.getRHS());
 872     break;
 873   }
 874
 875   case MCExpr::SymbolRef:
 876     visitUsedSymbol(cast<MCSymbolRefExpr>(Expr).getSymbol());
 877     break;
 878
 879   case MCExpr::Unary:
 880     visitUsedExpr(*cast<MCUnaryExpr>(Expr).getSubExpr());
 881     break;
 882   }
 883 }
 884
 885 void MCStreamer::EmitInstruction(const MCInst &Inst, const MCSubtargetInfo &STI,
 886                                  bool) {
 887   // Scan for values.
 888   for (unsigned i = Inst.getNumOperands(); i--;)
 889     if (Inst.getOperand(i).isExpr())
 890       visitUsedExpr(*Inst.getOperand(i).getExpr());
 891 }
 892
 893 void MCStreamer::emitAbsoluteSymbolDiff(const MCSymbol *Hi, const MCSymbol *Lo,
 894                                         unsigned Size) {
 895   // Get the Hi-Lo expression.
 896   const MCExpr *Diff =
 897       MCBinaryExpr::createSub(MCSymbolRefExpr::create(Hi, Context),
 898                               MCSymbolRefExpr::create(Lo, Context), Context);
 899
 900   const MCAsmInfo *MAI = Context.getAsmInfo();
 901   if (!MAI->doesSetDirectiveSuppressReloc()) {
 902     EmitValue(Diff, Size);
 903     return;
 904   }
 905
 906   // Otherwise, emit with .set (aka assignment).
 907   MCSymbol *SetLabel = Context.createTempSymbol("set", true);
 908   EmitAssignment(SetLabel, Diff);
 909   EmitSymbolValue(SetLabel, Size);
 910 }
 911
 912 void MCStreamer::EmitAssemblerFlag(MCAssemblerFlag Flag) {}
 913 void MCStreamer::EmitThumbFunc(MCSymbol *Func) {}
 914 void MCStreamer::EmitSymbolDesc(MCSymbol *Symbol, unsigned DescValue) {}
 915 void MCStreamer::BeginCOFFSymbolDef(const MCSymbol *Symbol) {
 916   llvm_unreachable("this directive only supported on COFF targets");
 917 }
 918 void MCStreamer::EndCOFFSymbolDef() {
 919   llvm_unreachable("this directive only supported on COFF targets");
 920 }
 921 void MCStreamer::EmitFileDirective(StringRef Filename) {}
 922 void MCStreamer::EmitCOFFSymbolStorageClass(int StorageClass) {
 923   llvm_unreachable("this directive only supported on COFF targets");
 924 }
 925 void MCStreamer::EmitCOFFSymbolType(int Type) {
 926   llvm_unreachable("this directive only supported on COFF targets");
 927 }
 928 void MCStreamer::emitELFSize(MCSymbol *Symbol, const MCExpr *Value) {}
 929 void MCStreamer::emitELFSymverDirective(MCSymbol *Alias,
 930                                         const MCSymbol *Aliasee) {}
 931 void MCStreamer::EmitLocalCommonSymbol(MCSymbol *Symbol, uint64_t Size,
 932                                        unsigned ByteAlignment) {}
 933 void MCStreamer::EmitTBSSSymbol(MCSection *Section, MCSymbol *Symbol,
 934                                 uint64_t Size, unsigned ByteAlignment) {}
 935 void MCStreamer::ChangeSection(MCSection *, const MCExpr *) {}
 936 void MCStreamer::EmitWeakReference(MCSymbol *Alias, const MCSymbol *Symbol) {}
 937 void MCStreamer::EmitBytes(StringRef Data) {}
 938 void MCStreamer::EmitBinaryData(StringRef Data) { EmitBytes(Data); }
 939 void MCStreamer::EmitValueImpl(const MCExpr *Value, unsigned Size, SMLoc Loc) {
 940   visitUsedExpr(*Value);
 941 }
 942 void MCStreamer::EmitULEB128Value(const MCExpr *Value) {}
 943 void MCStreamer::EmitSLEB128Value(const MCExpr *Value) {}
 944 void MCStreamer::emitFill(const MCExpr &NumBytes, uint64_t Value, SMLoc Loc) {}
 945 void MCStreamer::emitFill(const MCExpr &NumValues, int64_t Size, int64_t Expr,
 946                           SMLoc Loc) {}
 947 void MCStreamer::EmitValueToAlignment(unsigned ByteAlignment, int64_t Value,
 948                                       unsigned ValueSize,
 949                                       unsigned MaxBytesToEmit) {}
 950 void MCStreamer::EmitCodeAlignment(unsigned ByteAlignment,
 951                                    unsigned MaxBytesToEmit) {}
 952 void MCStreamer::emitValueToOffset(const MCExpr *Offset, unsigned char Value,
 953                                    SMLoc Loc) {}
 954 void MCStreamer::EmitBundleAlignMode(unsigned AlignPow2) {}
 955 void MCStreamer::EmitBundleLock(bool AlignToEnd) {}
 956 void MCStreamer::FinishImpl() {}
 957 void MCStreamer::EmitBundleUnlock() {}
 958
 959 void MCStreamer::SwitchSection(MCSection *Section, const MCExpr *Subsection) {
 960   assert(Section && "Cannot switch to a null section!");
 961   MCSectionSubPair curSection = SectionStack.back().first;
 962   SectionStack.back().second = curSection;
 963   if (MCSectionSubPair(Section, Subsection) != curSection) {
 964     ChangeSection(Section, Subsection);
 965     SectionStack.back().first = MCSectionSubPair(Section, Subsection);
 966     assert(!Section->hasEnded() && "Section already ended");
 967     MCSymbol *Sym = Section->getBeginSymbol();
 968     if (Sym && !Sym->isInSection())
 969       EmitLabel(Sym);
 970   }
 971 }
 972
 973 MCSymbol *MCStreamer::endSection(MCSection *Section) {
 974   // TODO: keep track of the last subsection so that this symbol appears in the
 975   // correct place.
 976   MCSymbol *Sym = Section->getEndSymbol(Context);
 977   if (Sym->isInSection())
 978     return Sym;
 979
 980   SwitchSection(Section);
 981   EmitLabel(Sym);
 982   return Sym;
 983 }
 984
 985 void MCStreamer::EmitVersionForTarget(const Triple &Target) {
 986   if (!Target.isOSBinFormatMachO() || !Target.isOSDarwin())
 987     return;
 988   // Do we even know the version?
 989   if (Target.getOSMajorVersion() == 0)
 990     return;
 991
 992   unsigned Major;
 993   unsigned Minor;
 994   unsigned Update;
 995   MCVersionMinType VersionType;
 996   if (Target.isWatchOS()) {
 997     VersionType = MCVM_WatchOSVersionMin;
 998     Target.getWatchOSVersion(Major, Minor, Update);
 999   } else if (Target.isTvOS()) {
1000     VersionType = MCVM_TvOSVersionMin;
1001     Target.getiOSVersion(Major, Minor, Update);
1002   } else if (Target.isMacOSX()) {
1003     VersionType = MCVM_OSXVersionMin;
1004     if (!Target.getMacOSXVersion(Major, Minor, Update))
1005       Major = 0;
1006   } else {
1007     VersionType = MCVM_IOSVersionMin;
1008     Target.getiOSVersion(Major, Minor, Update);
1009   }
1010   if (Major != 0)
1011     EmitVersionMin(VersionType, Major, Minor, Update);
1012 }