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 InitSections() {
  58     LogCall("InitSections");
  59     return Child->InitSections();
  60   }
  61
  62   virtual void EmitLabel(MCSymbol *Symbol) {
  63     LogCall("EmitLabel");
  64     return Child->EmitLabel(Symbol);
  65   }
  66
  67   virtual void EmitAssemblerFlag(MCAssemblerFlag Flag) {
  68     LogCall("EmitAssemblerFlag");
  69     return Child->EmitAssemblerFlag(Flag);
  70   }
  71
  72   virtual void EmitAssignment(MCSymbol *Symbol, const MCExpr *Value) {
  73     LogCall("EmitAssignment");
  74     return Child->EmitAssignment(Symbol, Value);
  75   }
  76
  77   virtual void EmitSymbolAttribute(MCSymbol *Symbol, MCSymbolAttr Attribute) {
  78     LogCall("EmitSymbolAttribute");
  79     return Child->EmitSymbolAttribute(Symbol, Attribute);
  80   }
  81
  82   virtual void EmitSymbolDesc(MCSymbol *Symbol, unsigned DescValue) {
  83     LogCall("EmitSymbolDesc");
  84     return Child->EmitSymbolDesc(Symbol, DescValue);
  85   }
  86
  87   virtual void BeginCOFFSymbolDef(const MCSymbol *Symbol) {
  88     LogCall("BeginCOFFSymbolDef");
  89     return Child->BeginCOFFSymbolDef(Symbol);
  90   }
  91
  92   virtual void EmitCOFFSymbolStorageClass(int StorageClass) {
  93     LogCall("EmitCOFFSymbolStorageClass");
  94     return Child->EmitCOFFSymbolStorageClass(StorageClass);
  95   }
  96
  97   virtual void EmitCOFFSymbolType(int Type) {
  98     LogCall("EmitCOFFSymbolType");
  99     return Child->EmitCOFFSymbolType(Type);
 100   }
 101
 102   virtual void EndCOFFSymbolDef() {
 103     LogCall("EndCOFFSymbolDef");
 104     return Child->EndCOFFSymbolDef();
 105   }
 106
 107   virtual void EmitELFSize(MCSymbol *Symbol, const MCExpr *Value) {
 108     LogCall("EmitELFSize");
 109     return Child->EmitELFSize(Symbol, Value);
 110   }
 111
 112   virtual void EmitCommonSymbol(MCSymbol *Symbol, uint64_t Size,
 113                                 unsigned ByteAlignment) {
 114     LogCall("EmitCommonSymbol");
 115     return Child->EmitCommonSymbol(Symbol, Size, ByteAlignment);
 116   }
 117
 118   virtual void EmitLocalCommonSymbol(MCSymbol *Symbol, uint64_t Size) {
 119     LogCall("EmitLocalCommonSymbol");
 120     return Child->EmitLocalCommonSymbol(Symbol, Size);
 121   }
 122
 123   virtual void EmitZerofill(const MCSection *Section, MCSymbol *Symbol = 0,
 124                             unsigned Size = 0, unsigned ByteAlignment = 0) {
 125     LogCall("EmitZerofill");
 126     return Child->EmitZerofill(Section, Symbol, Size, ByteAlignment);
 127   }
 128
 129   virtual void EmitTBSSSymbol (const MCSection *Section, MCSymbol *Symbol,
 130                                uint64_t Size, unsigned ByteAlignment = 0) {
 131     LogCall("EmitTBSSSymbol");
 132     return Child->EmitTBSSSymbol(Section, Symbol, Size, ByteAlignment);
 133   }
 134
 135   virtual void EmitBytes(StringRef Data, unsigned AddrSpace) {
 136     LogCall("EmitBytes");
 137     return Child->EmitBytes(Data, AddrSpace);
 138   }
 139
 140   virtual void EmitValue(const MCExpr *Value, unsigned Size,unsigned AddrSpace){
 141     LogCall("EmitValue");
 142     return Child->EmitValue(Value, Size, AddrSpace);
 143   }
 144
 145   virtual void EmitIntValue(uint64_t Value, unsigned Size, unsigned AddrSpace) {
 146     LogCall("EmitIntValue");
 147     return Child->EmitIntValue(Value, Size, AddrSpace);
 148   }
 149
 150   virtual void EmitGPRel32Value(const MCExpr *Value) {
 151     LogCall("EmitGPRel32Value");
 152     return Child->EmitGPRel32Value(Value);
 153   }
 154
 155   virtual void EmitFill(uint64_t NumBytes, uint8_t FillValue,
 156                         unsigned AddrSpace) {
 157     LogCall("EmitFill");
 158     return Child->EmitFill(NumBytes, FillValue, AddrSpace);
 159   }
 160
 161   virtual void EmitValueToAlignment(unsigned ByteAlignment, int64_t Value = 0,
 162                                     unsigned ValueSize = 1,
 163                                     unsigned MaxBytesToEmit = 0) {
 164     LogCall("EmitValueToAlignment");
 165     return Child->EmitValueToAlignment(ByteAlignment, Value,
 166                                        ValueSize, MaxBytesToEmit);
 167   }
 168
 169   virtual void EmitCodeAlignment(unsigned ByteAlignment,
 170                                  unsigned MaxBytesToEmit = 0) {
 171     LogCall("EmitCodeAlignment");
 172     return Child->EmitCodeAlignment(ByteAlignment, MaxBytesToEmit);
 173   }
 174
 175   virtual void EmitValueToOffset(const MCExpr *Offset,
 176                                  unsigned char Value = 0) {
 177     LogCall("EmitValueToOffset");
 178     return Child->EmitValueToOffset(Offset, Value);
 179   }
 180
 181   virtual void EmitFileDirective(StringRef Filename) {
 182     LogCall("EmitFileDirective", "FileName:" + Filename);
 183     return Child->EmitFileDirective(Filename);
 184   }
 185
 186   virtual void EmitDwarfFileDirective(unsigned FileNo, StringRef Filename) {
 187     LogCall("EmitDwarfFileDirective",
 188             "FileNo:" + Twine(FileNo) + " Filename:" + Filename);
 189     return Child->EmitDwarfFileDirective(FileNo, Filename);
 190   }
 191
 192   virtual void EmitInstruction(const MCInst &Inst) {
 193     LogCall("EmitInstruction");
 194     return Child->EmitInstruction(Inst);
 195   }
 196
 197   virtual void EmitRawText(StringRef String) {
 198     LogCall("EmitRawText", "\"" + String + "\"");
 199     return Child->EmitRawText(String);
 200   }
 201
 202   virtual void Finish() {
 203     LogCall("Finish");
 204     return Child->Finish();
 205   }
 206
 207 };
 208
 209 } // end anonymous namespace.
 210
 211 MCStreamer *llvm::createLoggingStreamer(MCStreamer *Child, raw_ostream &OS) {
 212   return new MCLoggingStreamer(Child, OS);
 213 }