lib/MC/MCLoggingStreamer.cpp

   1 //===- lib/MC/MCLoggingStreamer.cpp - API Logging Streamer ----------------===//
   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/OwningPtr.h"
  12 #include "llvm/ADT/Twine.h"
  13 #include "llvm/Support/raw_ostream.h"
  14 using namespace llvm;
  15
  16 namespace {
  17
  18 class MCLoggingStreamer : public MCStreamer {
  19   llvm::OwningPtr<MCStreamer> Child;
  20
  21   raw_ostream &OS;
  22
  23 public:
  24   MCLoggingStreamer(MCStreamer *_Child, raw_ostream &_OS)
  25     : MCStreamer(_Child->getContext()), Child(_Child), OS(_OS) {}
  26
  27   void LogCall(const char *Function) {
  28     OS << Function << "\n";
  29   }
  30
  31   void LogCall(const char *Function, const Twine &Message) {
  32     OS << Function << ": " << Message << "\n";
  33   }
  34
  35   virtual bool isVerboseAsm() const { return Child->isVerboseAsm(); }
  36
  37   virtual bool hasRawTextSupport() const { return Child->hasRawTextSupport(); }
  38
  39   virtual raw_ostream &GetCommentOS() { return Child->GetCommentOS(); }
  40
  41   virtual void AddComment(const Twine &T) {
  42     LogCall("AddComment", T);
  43     return Child->AddComment(T);
  44   }
  45
  46   virtual void AddBlankLine() {
  47     LogCall("AddBlankLine");
  48     return Child->AddBlankLine();
  49   }
  50
  51   virtual void SwitchSection(const MCSection *Section) {
  52     CurSection = Section;
  53     LogCall("SwitchSection");
  54     return Child->SwitchSection(Section);
  55   }
  56
  57   virtual void EmitLabel(MCSymbol *Symbol) {
  58     LogCall("EmitLabel");
  59     return Child->EmitLabel(Symbol);
  60   }
  61
  62   virtual void EmitAssemblerFlag(MCAssemblerFlag Flag) {
  63     LogCall("EmitAssemblerFlag");
  64     return Child->EmitAssemblerFlag(Flag);
  65   }
  66
  67   virtual void EmitAssignment(MCSymbol *Symbol, const MCExpr *Value) {
  68     LogCall("EmitAssignment");
  69     return Child->EmitAssignment(Symbol, Value);
  70   }
  71
  72   virtual void EmitSymbolAttribute(MCSymbol *Symbol, MCSymbolAttr Attribute) {
  73     LogCall("EmitSymbolAttribute");
  74     return Child->EmitSymbolAttribute(Symbol, Attribute);
  75   }
  76
  77   virtual void EmitSymbolDesc(MCSymbol *Symbol, unsigned DescValue) {
  78     LogCall("EmitSymbolDesc");
  79     return Child->EmitSymbolDesc(Symbol, DescValue);
  80   }
  81
  82   virtual void BeginCOFFSymbolDef(const MCSymbol *Symbol) {
  83     LogCall("BeginCOFFSymbolDef");
  84     return Child->BeginCOFFSymbolDef(Symbol);
  85   }
  86
  87   virtual void EmitCOFFSymbolStorageClass(int StorageClass) {
  88     LogCall("EmitCOFFSymbolStorageClass");
  89     return Child->EmitCOFFSymbolStorageClass(StorageClass);
  90   }
  91
  92   virtual void EmitCOFFSymbolType(int Type) {
  93     LogCall("EmitCOFFSymbolType");
  94     return Child->EmitCOFFSymbolType(Type);
  95   }
  96
  97   virtual void EndCOFFSymbolDef() {
  98     LogCall("EndCOFFSymbolDef");
  99     return Child->EndCOFFSymbolDef();
 100   }
 101
 102   virtual void EmitELFSize(MCSymbol *Symbol, const MCExpr *Value) {
 103     LogCall("EmitELFSize");
 104     return Child->EmitELFSize(Symbol, Value);
 105   }
 106
 107   virtual void EmitCommonSymbol(MCSymbol *Symbol, uint64_t Size,
 108                                 unsigned ByteAlignment) {
 109     LogCall("EmitCommonSymbol");
 110     return Child->EmitCommonSymbol(Symbol, Size, ByteAlignment);
 111   }
 112
 113   virtual void EmitLocalCommonSymbol(MCSymbol *Symbol, uint64_t Size) {
 114     LogCall("EmitLocalCommonSymbol");
 115     return Child->EmitLocalCommonSymbol(Symbol, Size);
 116   }
 117
 118   virtual void EmitZerofill(const MCSection *Section, MCSymbol *Symbol = 0,
 119                             unsigned Size = 0, unsigned ByteAlignment = 0) {
 120     LogCall("EmitZerofill");
 121     return Child->EmitZerofill(Section, Symbol, Size, ByteAlignment);
 122   }
 123
 124   virtual void EmitTBSSSymbol (const MCSection *Section, MCSymbol *Symbol,
 125                                uint64_t Size, unsigned ByteAlignment = 0) {
 126     LogCall("EmitTBSSSymbol");
 127     return Child->EmitTBSSSymbol(Section, Symbol, Size, ByteAlignment);
 128   }
 129
 130   virtual void EmitBytes(StringRef Data, unsigned AddrSpace) {
 131     LogCall("EmitBytes");
 132     return Child->EmitBytes(Data, AddrSpace);
 133   }
 134
 135   virtual void EmitValue(const MCExpr *Value, unsigned Size,unsigned AddrSpace){
 136     LogCall("EmitValue");
 137     return Child->EmitValue(Value, Size, AddrSpace);
 138   }
 139
 140   virtual void EmitIntValue(uint64_t Value, unsigned Size, unsigned AddrSpace) {
 141     LogCall("EmitIntValue");
 142     return Child->EmitIntValue(Value, Size, AddrSpace);
 143   }
 144
 145   virtual void EmitGPRel32Value(const MCExpr *Value) {
 146     LogCall("EmitGPRel32Value");
 147     return Child->EmitGPRel32Value(Value);
 148   }
 149
 150   virtual void EmitFill(uint64_t NumBytes, uint8_t FillValue,
 151                         unsigned AddrSpace) {
 152     LogCall("EmitFill");
 153     return Child->EmitFill(NumBytes, FillValue, AddrSpace);
 154   }
 155
 156   virtual void EmitValueToAlignment(unsigned ByteAlignment, int64_t Value = 0,
 157                                     unsigned ValueSize = 1,
 158                                     unsigned MaxBytesToEmit = 0) {
 159     LogCall("EmitValueToAlignment");
 160     return Child->EmitValueToAlignment(ByteAlignment, Value,
 161                                        ValueSize, MaxBytesToEmit);
 162   }
 163
 164   virtual void EmitCodeAlignment(unsigned ByteAlignment,
 165                                  unsigned MaxBytesToEmit = 0) {
 166     LogCall("EmitCodeAlignment");
 167     return Child->EmitCodeAlignment(ByteAlignment, MaxBytesToEmit);
 168   }
 169
 170   virtual void EmitValueToOffset(const MCExpr *Offset,
 171                                  unsigned char Value = 0) {
 172     LogCall("EmitValueToOffset");
 173     return Child->EmitValueToOffset(Offset, Value);
 174   }
 175
 176   virtual void EmitFileDirective(StringRef Filename) {
 177     LogCall("EmitFileDirective", "FileName:" + Filename);
 178     return Child->EmitFileDirective(Filename);
 179   }
 180
 181   virtual void EmitDwarfFileDirective(unsigned FileNo, StringRef Filename) {
 182     LogCall("EmitDwarfFileDirective",
 183             "FileNo:" + Twine(FileNo) + " Filename:" + Filename);
 184     return Child->EmitDwarfFileDirective(FileNo, Filename);
 185   }
 186
 187   virtual void EmitInstruction(const MCInst &Inst) {
 188     LogCall("EmitInstruction");
 189     return Child->EmitInstruction(Inst);
 190   }
 191
 192   virtual void EmitRawText(StringRef String) {
 193     LogCall("EmitRawText", "\"" + String + "\"");
 194     return Child->EmitRawText(String);
 195   }
 196
 197   virtual void Finish() {
 198     LogCall("Finish");
 199     return Child->Finish();
 200   }
 201
 202 };
 203
 204 } // end anonymous namespace.
 205
 206 MCStreamer *llvm::createLoggingStreamer(MCStreamer *Child, raw_ostream &OS) {
 207   return new MCLoggingStreamer(Child, OS);
 208 }