llvm/lib/MC/MCGOFFStreamer.cpp

//===- lib/MC/MCGOFFStreamer.cpp - GOFF Object Output ---------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// This file assembles .s files and emits GOFF .o object files.
//
//===----------------------------------------------------------------------===//

#include "llvm/MC/MCGOFFStreamer.h"
#include "llvm/BinaryFormat/GOFF.h"
#include "llvm/MC/MCAsmBackend.h"
#include "llvm/MC/MCAssembler.h"
#include "llvm/MC/MCCodeEmitter.h"
#include "llvm/MC/MCContext.h"
#include "llvm/MC/MCDirectives.h"
#include "llvm/MC/MCGOFFObjectWriter.h"
#include "llvm/MC/MCObjectStreamer.h"
#include "llvm/MC/MCSymbolGOFF.h"
#include "llvm/MC/TargetRegistry.h"

using namespace llvm;

MCGOFFStreamer::MCGOFFStreamer(MCContext &Context,
                               std::unique_ptr<MCAsmBackend> MAB,
                               std::unique_ptr<MCObjectWriter> OW,
                               std::unique_ptr<MCCodeEmitter> Emitter)
    : MCObjectStreamer(Context, std::move(MAB), std::move(OW),
                       std::move(Emitter)) {}

MCGOFFStreamer::~MCGOFFStreamer() = default;

void MCGOFFStreamer::finishImpl() {
  getWriter().setRootSD(static_cast<MCSectionGOFF *>(
                            getContext().getObjectFileInfo()->getTextSection())
                            ->getParent());
  MCObjectStreamer::finishImpl();
}

GOFFObjectWriter &MCGOFFStreamer::getWriter() {
  return static_cast<GOFFObjectWriter &>(getAssembler().getWriter());
}

void MCGOFFStreamer::changeSection(MCSection *Section, uint32_t Subsection) {
  // Make sure that all section are registered in the correct order.
  SmallVector<MCSectionGOFF *> Sections;
  for (auto *S = static_cast<MCSectionGOFF *>(Section); S; S = S->getParent())
    Sections.push_back(S);
  while (!Sections.empty()) {
    auto *S = Sections.pop_back_val();
    MCObjectStreamer::changeSection(S, Sections.empty() ? Subsection : 0);
  }
}

void MCGOFFStreamer::emitLabel(MCSymbol *Symbol, SMLoc Loc) {
  MCSectionGOFF *Section =
      static_cast<MCSectionGOFF *>(getCurrentSectionOnly());
  if (Section->isPR()) {
    if (Section->getBeginSymbol() == nullptr)
      Section->setBeginSymbol(Symbol);
    else
      getContext().reportError(
          Loc, "only one symbol can be defined in a PR section.");
  }
  MCObjectStreamer::emitLabel(Symbol, Loc);
}

bool MCGOFFStreamer::emitSymbolAttribute(MCSymbol *Sym,
                                         MCSymbolAttr Attribute) {
  return static_cast<MCSymbolGOFF *>(Sym)->setSymbolAttribute(Attribute);
}

void MCGOFFStreamer::emitCommonSymbol(MCSymbol *S, uint64_t Size,
                                      Align ByteAlignment) {
  auto *Symbol = static_cast<MCSymbolGOFF *>(S);

  MCSectionGOFF *SD = getContext().getGOFFSection(
      SectionKind::getMetadata(), Symbol->getName(),
      GOFF::SDAttr{GOFF::ESD_TA_Unspecified, GOFF::ESD_BSC_Unspecified});

  MCSectionGOFF *ED = getContext().getGOFFSection(
      SectionKind::getMetadata(), GOFF::CLASS_WSA,
      GOFF::EDAttr{false, GOFF::ESD_RMODE_64, GOFF::ESD_NS_Parts,
                   GOFF::ESD_TS_ByteOriented, GOFF::ESD_BA_Merge,
                   GOFF::ESD_LB_Deferred, GOFF::ESD_RQ_0, 0},
      SD);
  ED->setAlignment(ByteAlignment);

  MCSectionGOFF *Section = getContext().getGOFFSection(
      SectionKind::getBSS(), Symbol->getName(),
      GOFF::PRAttr{false, GOFF::ESD_EXE_DATA, GOFF::ESD_LT_XPLink,
                   Symbol->getBindingScope(), 0},
      ED);

  pushSection();
  switchSection(Section);
  emitLabel(Symbol);
  emitZeros(Size);
  popSection();
}

MCStreamer *llvm::createGOFFStreamer(MCContext &Context,
                                     std::unique_ptr<MCAsmBackend> &&MAB,
                                     std::unique_ptr<MCObjectWriter> &&OW,
                                     std::unique_ptr<MCCodeEmitter> &&CE) {
  MCGOFFStreamer *S =
      new MCGOFFStreamer(Context, std::move(MAB), std::move(OW), std::move(CE));
  return S;
}