lib/MC/MCAsmStreamer.cpp

   1 //===- lib/MC/MCAsmStreamer.cpp - Text Assembly 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
  12 #include "llvm/MC/MCAtom.h"
  13 #include "llvm/MC/MCContext.h"
  14 #include "llvm/MC/MCSection.h"
  15 #include "llvm/MC/MCSymbol.h"
  16 #include "llvm/MC/MCValue.h"
  17 #include "llvm/Support/raw_ostream.h"
  18 using namespace llvm;
  19
  20 namespace {
  21
  22   class MCAsmStreamer : public MCStreamer {
  23     raw_ostream &OS;
  24
  25     MCSection *CurSection;
  26
  27   public:
  28     MCAsmStreamer(MCContext &Context, raw_ostream &_OS)
  29       : MCStreamer(Context), OS(_OS) {}
  30     ~MCAsmStreamer() {}
  31
  32     /// @name MCStreamer Interface
  33     /// @{
  34
  35     virtual void SwitchSection(MCSection *Section);
  36
  37     virtual void EmitLabel(MCSymbol *Symbol);
  38
  39     virtual void EmitAssignment(MCSymbol *Symbol, const MCValue &Value,
  40                                 bool MakeAbsolute = false);
  41
  42     virtual void EmitSymbolAttribute(MCSymbol *Symbol, SymbolAttr Attribute);
  43
  44     virtual void EmitBytes(const char *Data, unsigned Length);
  45
  46     virtual void EmitValue(const MCValue &Value, unsigned Size);
  47
  48     virtual void EmitInstruction(const MCInst &Inst);
  49
  50     virtual void Finish();
  51
  52     /// @}
  53   };
  54
  55 }
  56
  57 /// Allow printing values directly to a raw_ostream.
  58 inline raw_ostream &operator<<(raw_ostream &os, const MCValue &Value) {
  59   if (Value.getSymA()) {
  60     os << Value.getSymA()->getName();
  61     if (Value.getSymB())
  62       os << " - " << Value.getSymB()->getName();
  63     if (Value.getCst())
  64       os << " + " << Value.getCst();
  65   } else {
  66     assert(!Value.getSymB() && "Invalid machine code value!");
  67     os << Value.getCst();
  68   }
  69
  70   return os;
  71 }
  72
  73 void MCAsmStreamer::SwitchSection(MCSection *Section) {
  74   if (Section != CurSection) {
  75     CurSection = Section;
  76
  77     // FIXME: Really we would like the segment, flags, etc. to be separate
  78     // values instead of embedded in the name. Not all assemblers understand all
  79     // this stuff though.
  80     OS << ".section " << Section->getName() << "\n";
  81   }
  82 }
  83
  84 void MCAsmStreamer::EmitLabel(MCSymbol *Symbol) {
  85   // FIXME: We need to enforce that we aren't printing atoms which are more
  86   // complicated than the assembler understands.
  87   //assert(Symbol->getAtom()->getSection() == CurSection &&
  88   //       "The label for a symbol must match its section!");
  89   OS << Symbol->getName() << ":\n";
  90 }
  91
  92 void MCAsmStreamer::EmitAssignment(MCSymbol *Symbol, const MCValue &Value,
  93                                    bool MakeAbsolute) {
  94   if (MakeAbsolute) {
  95     OS << ".set " << Symbol->getName() << ", " << Value << '\n';
  96   } else {
  97     OS << Symbol->getName() << " = " << Value << '\n';
  98   }
  99 }
 100
 101 void MCAsmStreamer::EmitSymbolAttribute(MCSymbol *Symbol,
 102                                         SymbolAttr Attribute) {
 103   switch (Attribute) {
 104   case Global: OS << ".globl"; break;
 105   case Weak: OS << ".weak"; break;
 106   case PrivateExtern: OS << ".private_extern"; break;
 107   }
 108
 109   OS << ' ' << Symbol->getName() << '\n';
 110 }
 111
 112 void MCAsmStreamer::EmitBytes(const char *Data, unsigned Length) {
 113   for (unsigned i = 0; i != Length; ++i) {
 114     OS << ".byte " << (unsigned) Data[i] << '\n';
 115   }
 116 }
 117
 118 void MCAsmStreamer::EmitValue(const MCValue &Value, unsigned Size) {
 119   // Need target hooks to know how to print this.
 120   switch (Size) {
 121   default:
 122     assert(0 && "Invalid size for machine code value!");
 123   case 1: OS << ".byte"; break;
 124   case 2: OS << ".hword"; break;
 125   case 4: OS << ".long"; break;
 126   case 8: OS << ".quad"; break;
 127   }
 128
 129   OS << ' ' << Value << '\n';
 130 }
 131
 132 void MCAsmStreamer::EmitInstruction(const MCInst &Inst) {
 133   // FIXME: Implement.
 134   OS << "# FIXME: Implement instruction printing!\n";
 135 }
 136
 137 void MCAsmStreamer::Finish() {
 138   OS.flush();
 139 }
 140
 141 MCStreamer *llvm::createAsmStreamer(MCContext &Context, raw_ostream &OS) {
 142   return new MCAsmStreamer(Context, OS);
 143 }