Home | History | Annotate | Download | only in MC
      1 //===-- llvm/MC/MCObjectWriter.h - Object File Writer Interface -*- C++ -*-===//
      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 #ifndef LLVM_MC_MCOBJECTWRITER_H
     11 #define LLVM_MC_MCOBJECTWRITER_H
     12 
     13 #include "llvm/ADT/Triple.h"
     14 #include "llvm/Support/raw_ostream.h"
     15 #include "llvm/Support/DataTypes.h"
     16 #include <cassert>
     17 
     18 namespace llvm {
     19 class MCAsmLayout;
     20 class MCAssembler;
     21 class MCFixup;
     22 class MCFragment;
     23 class MCSymbol;
     24 class MCSymbolData;
     25 class MCSymbolRefExpr;
     26 class MCValue;
     27 class raw_ostream;
     28 
     29 /// MCObjectWriter - Defines the object file and target independent interfaces
     30 /// used by the assembler backend to write native file format object files.
     31 ///
     32 /// The object writer contains a few callbacks used by the assembler to allow
     33 /// the object writer to modify the assembler data structures at appropriate
     34 /// points. Once assembly is complete, the object writer is given the
     35 /// MCAssembler instance, which contains all the symbol and section data which
     36 /// should be emitted as part of WriteObject().
     37 ///
     38 /// The object writer also contains a number of helper methods for writing
     39 /// binary data to the output stream.
     40 class MCObjectWriter {
     41   MCObjectWriter(const MCObjectWriter &); // DO NOT IMPLEMENT
     42   void operator=(const MCObjectWriter &); // DO NOT IMPLEMENT
     43 
     44 protected:
     45   raw_ostream &OS;
     46 
     47   unsigned IsLittleEndian : 1;
     48 
     49 protected: // Can only create subclasses.
     50   MCObjectWriter(raw_ostream &_OS, bool _IsLittleEndian)
     51     : OS(_OS), IsLittleEndian(_IsLittleEndian) {}
     52 
     53 public:
     54   virtual ~MCObjectWriter();
     55 
     56   bool isLittleEndian() const { return IsLittleEndian; }
     57 
     58   raw_ostream &getStream() { return OS; }
     59 
     60   /// @name High-Level API
     61   /// @{
     62 
     63   /// Perform any late binding of symbols (for example, to assign symbol indices
     64   /// for use when generating relocations).
     65   ///
     66   /// This routine is called by the assembler after layout and relaxation is
     67   /// complete.
     68   virtual void ExecutePostLayoutBinding(MCAssembler &Asm,
     69                                         const MCAsmLayout &Layout) = 0;
     70 
     71   /// Record a relocation entry.
     72   ///
     73   /// This routine is called by the assembler after layout and relaxation, and
     74   /// post layout binding. The implementation is responsible for storing
     75   /// information about the relocation so that it can be emitted during
     76   /// WriteObject().
     77   virtual void RecordRelocation(const MCAssembler &Asm,
     78                                 const MCAsmLayout &Layout,
     79                                 const MCFragment *Fragment,
     80                                 const MCFixup &Fixup, MCValue Target,
     81                                 uint64_t &FixedValue) = 0;
     82 
     83   /// \brief Check whether the difference (A - B) between two symbol
     84   /// references is fully resolved.
     85   ///
     86   /// Clients are not required to answer precisely and may conservatively return
     87   /// false, even when a difference is fully resolved.
     88   bool
     89   IsSymbolRefDifferenceFullyResolved(const MCAssembler &Asm,
     90                                      const MCSymbolRefExpr *A,
     91                                      const MCSymbolRefExpr *B,
     92                                      bool InSet) const;
     93 
     94   virtual bool
     95   IsSymbolRefDifferenceFullyResolvedImpl(const MCAssembler &Asm,
     96                                          const MCSymbolData &DataA,
     97                                          const MCFragment &FB,
     98                                          bool InSet,
     99                                          bool IsPCRel) const;
    100 
    101 
    102   /// Write the object file.
    103   ///
    104   /// This routine is called by the assembler after layout and relaxation is
    105   /// complete, fixups have been evaluated and applied, and relocations
    106   /// generated.
    107   virtual void WriteObject(MCAssembler &Asm,
    108                            const MCAsmLayout &Layout) = 0;
    109 
    110   /// @}
    111   /// @name Binary Output
    112   /// @{
    113 
    114   void Write8(uint8_t Value) {
    115     OS << char(Value);
    116   }
    117 
    118   void WriteLE16(uint16_t Value) {
    119     Write8(uint8_t(Value >> 0));
    120     Write8(uint8_t(Value >> 8));
    121   }
    122 
    123   void WriteLE32(uint32_t Value) {
    124     WriteLE16(uint16_t(Value >> 0));
    125     WriteLE16(uint16_t(Value >> 16));
    126   }
    127 
    128   void WriteLE64(uint64_t Value) {
    129     WriteLE32(uint32_t(Value >> 0));
    130     WriteLE32(uint32_t(Value >> 32));
    131   }
    132 
    133   void WriteBE16(uint16_t Value) {
    134     Write8(uint8_t(Value >> 8));
    135     Write8(uint8_t(Value >> 0));
    136   }
    137 
    138   void WriteBE32(uint32_t Value) {
    139     WriteBE16(uint16_t(Value >> 16));
    140     WriteBE16(uint16_t(Value >> 0));
    141   }
    142 
    143   void WriteBE64(uint64_t Value) {
    144     WriteBE32(uint32_t(Value >> 32));
    145     WriteBE32(uint32_t(Value >> 0));
    146   }
    147 
    148   void Write16(uint16_t Value) {
    149     if (IsLittleEndian)
    150       WriteLE16(Value);
    151     else
    152       WriteBE16(Value);
    153   }
    154 
    155   void Write32(uint32_t Value) {
    156     if (IsLittleEndian)
    157       WriteLE32(Value);
    158     else
    159       WriteBE32(Value);
    160   }
    161 
    162   void Write64(uint64_t Value) {
    163     if (IsLittleEndian)
    164       WriteLE64(Value);
    165     else
    166       WriteBE64(Value);
    167   }
    168 
    169   void WriteZeros(unsigned N) {
    170     const char Zeros[16] = { 0 };
    171 
    172     for (unsigned i = 0, e = N / 16; i != e; ++i)
    173       OS << StringRef(Zeros, 16);
    174 
    175     OS << StringRef(Zeros, N % 16);
    176   }
    177 
    178   void WriteBytes(StringRef Str, unsigned ZeroFillSize = 0) {
    179     assert((ZeroFillSize == 0 || Str.size () <= ZeroFillSize) &&
    180       "data size greater than fill size, unexpected large write will occur");
    181     OS << Str;
    182     if (ZeroFillSize)
    183       WriteZeros(ZeroFillSize - Str.size());
    184   }
    185 
    186   /// @}
    187 
    188   /// Utility function to encode a SLEB128 value.
    189   static void EncodeSLEB128(int64_t Value, raw_ostream &OS);
    190   /// Utility function to encode a ULEB128 value.
    191   static void EncodeULEB128(uint64_t Value, raw_ostream &OS);
    192 };
    193 
    194 MCObjectWriter *createWinCOFFObjectWriter(raw_ostream &OS, bool is64Bit);
    195 
    196 } // End llvm namespace
    197 
    198 #endif
    199