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