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