Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 1 | //===- lib/MC/MCAsmStreamer.cpp - Text Assembly Output --------------------===// |
| 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" |
Daniel Dunbar | 8c2eebe | 2009-08-31 08:08:38 +0000 | [diff] [blame] | 11 | #include "llvm/MC/MCAsmInfo.h" |
Daniel Dunbar | 4a0abd8 | 2009-08-27 00:51:57 +0000 | [diff] [blame] | 12 | #include "llvm/MC/MCCodeEmitter.h" |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 13 | #include "llvm/MC/MCContext.h" |
Daniel Dunbar | 8c2eebe | 2009-08-31 08:08:38 +0000 | [diff] [blame] | 14 | #include "llvm/MC/MCExpr.h" |
Daniel Dunbar | abde298 | 2009-07-01 06:35:03 +0000 | [diff] [blame] | 15 | #include "llvm/MC/MCInst.h" |
Chris Lattner | 90edac0 | 2009-09-14 03:02:37 +0000 | [diff] [blame] | 16 | #include "llvm/MC/MCInstPrinter.h" |
Chris Lattner | f9bdedd | 2009-08-10 18:15:01 +0000 | [diff] [blame] | 17 | #include "llvm/MC/MCSectionMachO.h" |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 18 | #include "llvm/MC/MCSymbol.h" |
Chris Lattner | 4c42a6d | 2010-03-19 05:48:53 +0000 | [diff] [blame] | 19 | #include "llvm/ADT/OwningPtr.h" |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 20 | #include "llvm/ADT/SmallString.h" |
| 21 | #include "llvm/ADT/Twine.h" |
Torok Edwin | c25e758 | 2009-07-11 20:10:48 +0000 | [diff] [blame] | 22 | #include "llvm/Support/ErrorHandling.h" |
Chris Lattner | 663c2d2 | 2009-08-19 06:12:02 +0000 | [diff] [blame] | 23 | #include "llvm/Support/MathExtras.h" |
Daniel Dunbar | 4a0abd8 | 2009-08-27 00:51:57 +0000 | [diff] [blame] | 24 | #include "llvm/Support/Format.h" |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 25 | #include "llvm/Support/FormattedStream.h" |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 26 | using namespace llvm; |
| 27 | |
| 28 | namespace { |
| 29 | |
Chris Lattner | 46a947d | 2009-08-17 04:17:34 +0000 | [diff] [blame] | 30 | class MCAsmStreamer : public MCStreamer { |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 31 | formatted_raw_ostream &OS; |
Chris Lattner | 33adcfb | 2009-08-22 21:43:10 +0000 | [diff] [blame] | 32 | const MCAsmInfo &MAI; |
Chris Lattner | 4c42a6d | 2010-03-19 05:48:53 +0000 | [diff] [blame] | 33 | OwningPtr<MCInstPrinter> InstPrinter; |
Benjamin Kramer | 1abcd06 | 2010-07-29 17:48:06 +0000 | [diff] [blame] | 34 | OwningPtr<MCCodeEmitter> Emitter; |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 35 | |
| 36 | SmallString<128> CommentToEmit; |
Chris Lattner | d79d9dc | 2010-01-22 19:17:48 +0000 | [diff] [blame] | 37 | raw_svector_ostream CommentStream; |
Daniel Dunbar | 9dee8e3 | 2010-02-03 18:18:30 +0000 | [diff] [blame] | 38 | |
| 39 | unsigned IsLittleEndian : 1; |
| 40 | unsigned IsVerboseAsm : 1; |
| 41 | unsigned ShowInst : 1; |
| 42 | |
Chris Lattner | 46a947d | 2009-08-17 04:17:34 +0000 | [diff] [blame] | 43 | public: |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 44 | MCAsmStreamer(MCContext &Context, formatted_raw_ostream &os, |
Chris Lattner | 0740441 | 2010-01-22 07:06:15 +0000 | [diff] [blame] | 45 | bool isLittleEndian, bool isVerboseAsm, MCInstPrinter *printer, |
Daniel Dunbar | 5532cf4 | 2010-02-10 01:41:14 +0000 | [diff] [blame] | 46 | MCCodeEmitter *emitter, bool showInst) |
Chris Lattner | fdab14b | 2010-03-12 18:28:53 +0000 | [diff] [blame] | 47 | : MCStreamer(Context), OS(os), MAI(Context.getAsmInfo()), |
| 48 | InstPrinter(printer), Emitter(emitter), CommentStream(CommentToEmit), |
Daniel Dunbar | 9dee8e3 | 2010-02-03 18:18:30 +0000 | [diff] [blame] | 49 | IsLittleEndian(isLittleEndian), IsVerboseAsm(isVerboseAsm), |
Daniel Dunbar | 5532cf4 | 2010-02-10 01:41:14 +0000 | [diff] [blame] | 50 | ShowInst(showInst) { |
Chris Lattner | 5d672cf | 2010-02-10 00:10:18 +0000 | [diff] [blame] | 51 | if (InstPrinter && IsVerboseAsm) |
| 52 | InstPrinter->setCommentStream(CommentStream); |
| 53 | } |
Chris Lattner | 46a947d | 2009-08-17 04:17:34 +0000 | [diff] [blame] | 54 | ~MCAsmStreamer() {} |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 55 | |
Chris Lattner | 1658202 | 2010-01-20 06:39:07 +0000 | [diff] [blame] | 56 | bool isLittleEndian() const { return IsLittleEndian; } |
Daniel Dunbar | 6b71653 | 2010-02-09 23:00:14 +0000 | [diff] [blame] | 57 | |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 58 | inline void EmitEOL() { |
Chris Lattner | d79d9dc | 2010-01-22 19:17:48 +0000 | [diff] [blame] | 59 | // If we don't have any comments, just emit a \n. |
| 60 | if (!IsVerboseAsm) { |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 61 | OS << '\n'; |
| 62 | return; |
| 63 | } |
| 64 | EmitCommentsAndEOL(); |
| 65 | } |
| 66 | void EmitCommentsAndEOL(); |
Chris Lattner | 56591ab | 2010-02-02 23:37:42 +0000 | [diff] [blame] | 67 | |
| 68 | /// isVerboseAsm - Return true if this streamer supports verbose assembly at |
| 69 | /// all. |
| 70 | virtual bool isVerboseAsm() const { return IsVerboseAsm; } |
Chris Lattner | 91bead7 | 2010-04-03 21:35:55 +0000 | [diff] [blame] | 71 | |
| 72 | /// hasRawTextSupport - We support EmitRawText. |
| 73 | virtual bool hasRawTextSupport() const { return true; } |
Daniel Dunbar | 6b71653 | 2010-02-09 23:00:14 +0000 | [diff] [blame] | 74 | |
Chris Lattner | d32c7cf | 2010-01-22 18:21:35 +0000 | [diff] [blame] | 75 | /// AddComment - Add a comment that can be emitted to the generated .s |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 76 | /// file if applicable as a QoI issue to make the output of the compiler |
| 77 | /// more readable. This only affects the MCAsmStreamer, and only when |
| 78 | /// verbose assembly output is enabled. |
Chris Lattner | d32c7cf | 2010-01-22 18:21:35 +0000 | [diff] [blame] | 79 | virtual void AddComment(const Twine &T); |
Daniel Dunbar | 6b71653 | 2010-02-09 23:00:14 +0000 | [diff] [blame] | 80 | |
| 81 | /// AddEncodingComment - Add a comment showing the encoding of an instruction. |
| 82 | virtual void AddEncodingComment(const MCInst &Inst); |
| 83 | |
Chris Lattner | d79d9dc | 2010-01-22 19:17:48 +0000 | [diff] [blame] | 84 | /// GetCommentOS - Return a raw_ostream that comments can be written to. |
| 85 | /// Unlike AddComment, you are required to terminate comments with \n if you |
| 86 | /// use this method. |
| 87 | virtual raw_ostream &GetCommentOS() { |
| 88 | if (!IsVerboseAsm) |
| 89 | return nulls(); // Discard comments unless in verbose asm mode. |
| 90 | return CommentStream; |
| 91 | } |
Daniel Dunbar | 6b71653 | 2010-02-09 23:00:14 +0000 | [diff] [blame] | 92 | |
Chris Lattner | 0fd90fd | 2010-01-22 19:52:01 +0000 | [diff] [blame] | 93 | /// AddBlankLine - Emit a blank line to a .s file to pretty it up. |
| 94 | virtual void AddBlankLine() { |
| 95 | EmitEOL(); |
| 96 | } |
Daniel Dunbar | 6b71653 | 2010-02-09 23:00:14 +0000 | [diff] [blame] | 97 | |
Chris Lattner | 46a947d | 2009-08-17 04:17:34 +0000 | [diff] [blame] | 98 | /// @name MCStreamer Interface |
| 99 | /// @{ |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 100 | |
Chris Lattner | 975780b | 2009-08-17 05:49:08 +0000 | [diff] [blame] | 101 | virtual void SwitchSection(const MCSection *Section); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 102 | |
Chris Lattner | 46a947d | 2009-08-17 04:17:34 +0000 | [diff] [blame] | 103 | virtual void EmitLabel(MCSymbol *Symbol); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 104 | |
Chris Lattner | a5ad93a | 2010-01-23 06:39:22 +0000 | [diff] [blame] | 105 | virtual void EmitAssemblerFlag(MCAssemblerFlag Flag); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 106 | |
Daniel Dunbar | 821e333 | 2009-08-31 08:09:28 +0000 | [diff] [blame] | 107 | virtual void EmitAssignment(MCSymbol *Symbol, const MCExpr *Value); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 108 | |
Chris Lattner | a5ad93a | 2010-01-23 06:39:22 +0000 | [diff] [blame] | 109 | virtual void EmitSymbolAttribute(MCSymbol *Symbol, MCSymbolAttr Attribute); |
Kevin Enderby | a5c7832 | 2009-07-13 21:03:15 +0000 | [diff] [blame] | 110 | |
Chris Lattner | 46a947d | 2009-08-17 04:17:34 +0000 | [diff] [blame] | 111 | virtual void EmitSymbolDesc(MCSymbol *Symbol, unsigned DescValue); |
Chris Lattner | b54b9dd | 2010-05-08 19:54:22 +0000 | [diff] [blame] | 112 | virtual void BeginCOFFSymbolDef(const MCSymbol *Symbol); |
| 113 | virtual void EmitCOFFSymbolStorageClass(int StorageClass); |
| 114 | virtual void EmitCOFFSymbolType(int Type); |
| 115 | virtual void EndCOFFSymbolDef(); |
Chris Lattner | 99328ad | 2010-01-25 07:52:13 +0000 | [diff] [blame] | 116 | virtual void EmitELFSize(MCSymbol *Symbol, const MCExpr *Value); |
Chris Lattner | 9eb158d | 2010-01-23 07:47:02 +0000 | [diff] [blame] | 117 | virtual void EmitCommonSymbol(MCSymbol *Symbol, uint64_t Size, |
Daniel Dunbar | 7092c7e | 2009-08-30 06:17:16 +0000 | [diff] [blame] | 118 | unsigned ByteAlignment); |
Kevin Enderby | 95cf30c | 2009-07-14 18:17:10 +0000 | [diff] [blame] | 119 | |
Chris Lattner | 9eb158d | 2010-01-23 07:47:02 +0000 | [diff] [blame] | 120 | /// EmitLocalCommonSymbol - Emit a local common (.lcomm) symbol. |
| 121 | /// |
| 122 | /// @param Symbol - The common symbol to emit. |
| 123 | /// @param Size - The size of the common symbol. |
| 124 | virtual void EmitLocalCommonSymbol(MCSymbol *Symbol, uint64_t Size); |
| 125 | |
Daniel Dunbar | 8751b94 | 2009-08-28 05:48:22 +0000 | [diff] [blame] | 126 | virtual void EmitZerofill(const MCSection *Section, MCSymbol *Symbol = 0, |
Daniel Dunbar | 7092c7e | 2009-08-30 06:17:16 +0000 | [diff] [blame] | 127 | unsigned Size = 0, unsigned ByteAlignment = 0); |
Kevin Enderby | 7114824 | 2009-07-14 21:35:03 +0000 | [diff] [blame] | 128 | |
Eric Christopher | 4d01cbe | 2010-05-18 21:16:04 +0000 | [diff] [blame] | 129 | virtual void EmitTBSSSymbol (const MCSection *Section, MCSymbol *Symbol, |
| 130 | uint64_t Size, unsigned ByteAlignment = 0); |
Eric Christopher | 482eba0 | 2010-05-14 01:50:28 +0000 | [diff] [blame] | 131 | |
Chris Lattner | aaec205 | 2010-01-19 19:46:13 +0000 | [diff] [blame] | 132 | virtual void EmitBytes(StringRef Data, unsigned AddrSpace); |
Chris Lattner | 4e4db7a | 2009-07-07 20:30:46 +0000 | [diff] [blame] | 133 | |
Chris Lattner | aaec205 | 2010-01-19 19:46:13 +0000 | [diff] [blame] | 134 | virtual void EmitValue(const MCExpr *Value, unsigned Size,unsigned AddrSpace); |
Chris Lattner | 32ae3fe | 2010-01-19 22:03:38 +0000 | [diff] [blame] | 135 | virtual void EmitIntValue(uint64_t Value, unsigned Size, unsigned AddrSpace); |
Chris Lattner | 718fb59 | 2010-01-25 21:28:50 +0000 | [diff] [blame] | 136 | virtual void EmitGPRel32Value(const MCExpr *Value); |
| 137 | |
Chris Lattner | 32ae3fe | 2010-01-19 22:03:38 +0000 | [diff] [blame] | 138 | |
Chris Lattner | aaec205 | 2010-01-19 19:46:13 +0000 | [diff] [blame] | 139 | virtual void EmitFill(uint64_t NumBytes, uint8_t FillValue, |
| 140 | unsigned AddrSpace); |
Chris Lattner | 9be3fee | 2009-07-10 22:20:30 +0000 | [diff] [blame] | 141 | |
Chris Lattner | 46a947d | 2009-08-17 04:17:34 +0000 | [diff] [blame] | 142 | virtual void EmitValueToAlignment(unsigned ByteAlignment, int64_t Value = 0, |
| 143 | unsigned ValueSize = 1, |
| 144 | unsigned MaxBytesToEmit = 0); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 145 | |
Kevin Enderby | 6e72048 | 2010-02-23 18:26:34 +0000 | [diff] [blame] | 146 | virtual void EmitCodeAlignment(unsigned ByteAlignment, |
| 147 | unsigned MaxBytesToEmit = 0); |
| 148 | |
Daniel Dunbar | 821e333 | 2009-08-31 08:09:28 +0000 | [diff] [blame] | 149 | virtual void EmitValueToOffset(const MCExpr *Offset, |
Chris Lattner | 46a947d | 2009-08-17 04:17:34 +0000 | [diff] [blame] | 150 | unsigned char Value = 0); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 151 | |
Chris Lattner | a6594fc | 2010-01-25 18:58:59 +0000 | [diff] [blame] | 152 | virtual void EmitFileDirective(StringRef Filename); |
| 153 | virtual void EmitDwarfFileDirective(unsigned FileNo, StringRef Filename); |
| 154 | |
| 155 | virtual void EmitInstruction(const MCInst &Inst); |
| 156 | |
Chris Lattner | 91bead7 | 2010-04-03 21:35:55 +0000 | [diff] [blame] | 157 | /// EmitRawText - If this file is backed by a assembly streamer, this dumps |
| 158 | /// the specified string in the output .s file. This capability is |
| 159 | /// indicated by the hasRawTextSupport() predicate. |
| 160 | virtual void EmitRawText(StringRef String); |
| 161 | |
Chris Lattner | 46a947d | 2009-08-17 04:17:34 +0000 | [diff] [blame] | 162 | virtual void Finish(); |
| 163 | |
| 164 | /// @} |
| 165 | }; |
Daniel Dunbar | 84a2926 | 2009-06-24 19:25:34 +0000 | [diff] [blame] | 166 | |
Chris Lattner | 46a947d | 2009-08-17 04:17:34 +0000 | [diff] [blame] | 167 | } // end anonymous namespace. |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 168 | |
Chris Lattner | d32c7cf | 2010-01-22 18:21:35 +0000 | [diff] [blame] | 169 | /// AddComment - Add a comment that can be emitted to the generated .s |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 170 | /// file if applicable as a QoI issue to make the output of the compiler |
| 171 | /// more readable. This only affects the MCAsmStreamer, and only when |
| 172 | /// verbose assembly output is enabled. |
Chris Lattner | d32c7cf | 2010-01-22 18:21:35 +0000 | [diff] [blame] | 173 | void MCAsmStreamer::AddComment(const Twine &T) { |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 174 | if (!IsVerboseAsm) return; |
Chris Lattner | d79d9dc | 2010-01-22 19:17:48 +0000 | [diff] [blame] | 175 | |
| 176 | // Make sure that CommentStream is flushed. |
| 177 | CommentStream.flush(); |
| 178 | |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 179 | T.toVector(CommentToEmit); |
Chris Lattner | d79d9dc | 2010-01-22 19:17:48 +0000 | [diff] [blame] | 180 | // Each comment goes on its own line. |
| 181 | CommentToEmit.push_back('\n'); |
Chris Lattner | 14ca177 | 2010-01-22 21:16:10 +0000 | [diff] [blame] | 182 | |
| 183 | // Tell the comment stream that the vector changed underneath it. |
| 184 | CommentStream.resync(); |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 185 | } |
| 186 | |
| 187 | void MCAsmStreamer::EmitCommentsAndEOL() { |
Chris Lattner | d79d9dc | 2010-01-22 19:17:48 +0000 | [diff] [blame] | 188 | if (CommentToEmit.empty() && CommentStream.GetNumBytesInBuffer() == 0) { |
| 189 | OS << '\n'; |
| 190 | return; |
| 191 | } |
| 192 | |
| 193 | CommentStream.flush(); |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 194 | StringRef Comments = CommentToEmit.str(); |
Chris Lattner | d79d9dc | 2010-01-22 19:17:48 +0000 | [diff] [blame] | 195 | |
| 196 | assert(Comments.back() == '\n' && |
| 197 | "Comment array not newline terminated"); |
| 198 | do { |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 199 | // Emit a line of comments. |
| 200 | OS.PadToColumn(MAI.getCommentColumn()); |
| 201 | size_t Position = Comments.find('\n'); |
| 202 | OS << MAI.getCommentString() << ' ' << Comments.substr(0, Position) << '\n'; |
| 203 | |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 204 | Comments = Comments.substr(Position+1); |
Chris Lattner | d79d9dc | 2010-01-22 19:17:48 +0000 | [diff] [blame] | 205 | } while (!Comments.empty()); |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 206 | |
Chris Lattner | 14ca177 | 2010-01-22 21:16:10 +0000 | [diff] [blame] | 207 | CommentToEmit.clear(); |
| 208 | // Tell the comment stream that the vector changed underneath it. |
| 209 | CommentStream.resync(); |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 210 | } |
| 211 | |
Daniel Dunbar | 304f6a4 | 2009-06-25 21:03:18 +0000 | [diff] [blame] | 212 | static inline int64_t truncateToSize(int64_t Value, unsigned Bytes) { |
| 213 | assert(Bytes && "Invalid size!"); |
| 214 | return Value & ((uint64_t) (int64_t) -1 >> (64 - Bytes * 8)); |
| 215 | } |
| 216 | |
Chris Lattner | 975780b | 2009-08-17 05:49:08 +0000 | [diff] [blame] | 217 | void MCAsmStreamer::SwitchSection(const MCSection *Section) { |
Chris Lattner | 6c2f9e1 | 2009-08-19 05:49:37 +0000 | [diff] [blame] | 218 | assert(Section && "Cannot switch to a null section!"); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 219 | if (Section != CurSection) { |
| 220 | CurSection = Section; |
Chris Lattner | 33adcfb | 2009-08-22 21:43:10 +0000 | [diff] [blame] | 221 | Section->PrintSwitchToSection(MAI, OS); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 222 | } |
| 223 | } |
| 224 | |
| 225 | void MCAsmStreamer::EmitLabel(MCSymbol *Symbol) { |
Daniel Dunbar | 8906ff1 | 2009-08-22 07:22:36 +0000 | [diff] [blame] | 226 | assert(Symbol->isUndefined() && "Cannot define a symbol twice!"); |
Daniel Dunbar | c304718 | 2010-05-05 19:01:00 +0000 | [diff] [blame] | 227 | assert(!Symbol->isVariable() && "Cannot emit a variable symbol!"); |
Daniel Dunbar | 71d259b | 2009-06-24 17:00:42 +0000 | [diff] [blame] | 228 | assert(CurSection && "Cannot emit before setting section!"); |
Daniel Dunbar | 71d259b | 2009-06-24 17:00:42 +0000 | [diff] [blame] | 229 | |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 230 | OS << *Symbol << ":"; |
| 231 | EmitEOL(); |
Daniel Dunbar | 8906ff1 | 2009-08-22 07:22:36 +0000 | [diff] [blame] | 232 | Symbol->setSection(*CurSection); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 233 | } |
| 234 | |
Chris Lattner | a5ad93a | 2010-01-23 06:39:22 +0000 | [diff] [blame] | 235 | void MCAsmStreamer::EmitAssemblerFlag(MCAssemblerFlag Flag) { |
Kevin Enderby | f96db46 | 2009-07-16 17:56:39 +0000 | [diff] [blame] | 236 | switch (Flag) { |
Daniel Dunbar | 011e4db | 2009-08-13 23:36:34 +0000 | [diff] [blame] | 237 | default: assert(0 && "Invalid flag!"); |
Chris Lattner | a5ad93a | 2010-01-23 06:39:22 +0000 | [diff] [blame] | 238 | case MCAF_SubsectionsViaSymbols: OS << ".subsections_via_symbols"; break; |
Kevin Enderby | f96db46 | 2009-07-16 17:56:39 +0000 | [diff] [blame] | 239 | } |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 240 | EmitEOL(); |
Kevin Enderby | a5c7832 | 2009-07-13 21:03:15 +0000 | [diff] [blame] | 241 | } |
| 242 | |
Daniel Dunbar | 821e333 | 2009-08-31 08:09:28 +0000 | [diff] [blame] | 243 | void MCAsmStreamer::EmitAssignment(MCSymbol *Symbol, const MCExpr *Value) { |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 244 | OS << *Symbol << " = " << *Value; |
| 245 | EmitEOL(); |
Daniel Dunbar | fffff91 | 2009-10-16 01:34:54 +0000 | [diff] [blame] | 246 | |
| 247 | // FIXME: Lift context changes into super class. |
Daniel Dunbar | 08a408a | 2010-05-05 17:41:00 +0000 | [diff] [blame] | 248 | Symbol->setVariableValue(Value); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 249 | } |
| 250 | |
Daniel Dunbar | fffff91 | 2009-10-16 01:34:54 +0000 | [diff] [blame] | 251 | void MCAsmStreamer::EmitSymbolAttribute(MCSymbol *Symbol, |
Chris Lattner | a5ad93a | 2010-01-23 06:39:22 +0000 | [diff] [blame] | 252 | MCSymbolAttr Attribute) { |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 253 | switch (Attribute) { |
Chris Lattner | a5ad93a | 2010-01-23 06:39:22 +0000 | [diff] [blame] | 254 | case MCSA_Invalid: assert(0 && "Invalid symbol attribute"); |
Chris Lattner | ed0ab15 | 2010-01-25 18:30:45 +0000 | [diff] [blame] | 255 | case MCSA_ELF_TypeFunction: /// .type _foo, STT_FUNC # aka @function |
| 256 | case MCSA_ELF_TypeIndFunction: /// .type _foo, STT_GNU_IFUNC |
| 257 | case MCSA_ELF_TypeObject: /// .type _foo, STT_OBJECT # aka @object |
| 258 | case MCSA_ELF_TypeTLS: /// .type _foo, STT_TLS # aka @tls_object |
| 259 | case MCSA_ELF_TypeCommon: /// .type _foo, STT_COMMON # aka @common |
| 260 | case MCSA_ELF_TypeNoType: /// .type _foo, STT_NOTYPE # aka @notype |
| 261 | assert(MAI.hasDotTypeDotSizeDirective() && "Symbol Attr not supported"); |
Dan Gohman | 523d70e | 2010-02-04 01:42:13 +0000 | [diff] [blame] | 262 | OS << "\t.type\t" << *Symbol << ',' |
Chris Lattner | ed0ab15 | 2010-01-25 18:30:45 +0000 | [diff] [blame] | 263 | << ((MAI.getCommentString()[0] != '@') ? '@' : '%'); |
| 264 | switch (Attribute) { |
| 265 | default: assert(0 && "Unknown ELF .type"); |
| 266 | case MCSA_ELF_TypeFunction: OS << "function"; break; |
| 267 | case MCSA_ELF_TypeIndFunction: OS << "gnu_indirect_function"; break; |
| 268 | case MCSA_ELF_TypeObject: OS << "object"; break; |
| 269 | case MCSA_ELF_TypeTLS: OS << "tls_object"; break; |
| 270 | case MCSA_ELF_TypeCommon: OS << "common"; break; |
| 271 | case MCSA_ELF_TypeNoType: OS << "no_type"; break; |
| 272 | } |
| 273 | EmitEOL(); |
| 274 | return; |
| 275 | case MCSA_Global: // .globl/.global |
| 276 | OS << MAI.getGlobalDirective(); |
| 277 | break; |
Chris Lattner | 61abd7b | 2010-06-21 20:35:01 +0000 | [diff] [blame] | 278 | case MCSA_Hidden: OS << "\t.hidden\t"; break; |
| 279 | case MCSA_IndirectSymbol: OS << "\t.indirect_symbol\t"; break; |
| 280 | case MCSA_Internal: OS << "\t.internal\t"; break; |
| 281 | case MCSA_LazyReference: OS << "\t.lazy_reference\t"; break; |
| 282 | case MCSA_Local: OS << "\t.local\t"; break; |
| 283 | case MCSA_NoDeadStrip: OS << "\t.no_dead_strip\t"; break; |
| 284 | case MCSA_PrivateExtern: OS << "\t.private_extern\t"; break; |
| 285 | case MCSA_Protected: OS << "\t.protected\t"; break; |
| 286 | case MCSA_Reference: OS << "\t.reference\t"; break; |
| 287 | case MCSA_Weak: OS << "\t.weak\t"; break; |
| 288 | case MCSA_WeakDefinition: OS << "\t.weak_definition\t"; break; |
Chris Lattner | a5ad93a | 2010-01-23 06:39:22 +0000 | [diff] [blame] | 289 | // .weak_reference |
| 290 | case MCSA_WeakReference: OS << MAI.getWeakRefDirective(); break; |
Kevin Enderby | f59cac5 | 2010-07-08 17:22:42 +0000 | [diff] [blame] | 291 | case MCSA_WeakDefAutoPrivate: OS << "\t.weak_def_can_be_hidden\t"; break; |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 292 | } |
| 293 | |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 294 | OS << *Symbol; |
| 295 | EmitEOL(); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 296 | } |
| 297 | |
Kevin Enderby | 95cf30c | 2009-07-14 18:17:10 +0000 | [diff] [blame] | 298 | void MCAsmStreamer::EmitSymbolDesc(MCSymbol *Symbol, unsigned DescValue) { |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 299 | OS << ".desc" << ' ' << *Symbol << ',' << DescValue; |
| 300 | EmitEOL(); |
Kevin Enderby | 95cf30c | 2009-07-14 18:17:10 +0000 | [diff] [blame] | 301 | } |
| 302 | |
Chris Lattner | b54b9dd | 2010-05-08 19:54:22 +0000 | [diff] [blame] | 303 | void MCAsmStreamer::BeginCOFFSymbolDef(const MCSymbol *Symbol) { |
| 304 | OS << "\t.def\t " << *Symbol << ';'; |
| 305 | EmitEOL(); |
| 306 | } |
| 307 | |
| 308 | void MCAsmStreamer::EmitCOFFSymbolStorageClass (int StorageClass) { |
| 309 | OS << "\t.scl\t" << StorageClass << ';'; |
| 310 | EmitEOL(); |
| 311 | } |
| 312 | |
| 313 | void MCAsmStreamer::EmitCOFFSymbolType (int Type) { |
| 314 | OS << "\t.type\t" << Type << ';'; |
| 315 | EmitEOL(); |
| 316 | } |
| 317 | |
| 318 | void MCAsmStreamer::EndCOFFSymbolDef() { |
| 319 | OS << "\t.endef"; |
| 320 | EmitEOL(); |
| 321 | } |
| 322 | |
Chris Lattner | 99328ad | 2010-01-25 07:52:13 +0000 | [diff] [blame] | 323 | void MCAsmStreamer::EmitELFSize(MCSymbol *Symbol, const MCExpr *Value) { |
| 324 | assert(MAI.hasDotTypeDotSizeDirective()); |
| 325 | OS << "\t.size\t" << *Symbol << ", " << *Value << '\n'; |
| 326 | } |
| 327 | |
Chris Lattner | 9eb158d | 2010-01-23 07:47:02 +0000 | [diff] [blame] | 328 | void MCAsmStreamer::EmitCommonSymbol(MCSymbol *Symbol, uint64_t Size, |
Daniel Dunbar | 7092c7e | 2009-08-30 06:17:16 +0000 | [diff] [blame] | 329 | unsigned ByteAlignment) { |
Chris Lattner | 9eb158d | 2010-01-23 07:47:02 +0000 | [diff] [blame] | 330 | OS << "\t.comm\t" << *Symbol << ',' << Size; |
Chris Lattner | 6559d76 | 2010-01-25 07:29:13 +0000 | [diff] [blame] | 331 | if (ByteAlignment != 0) { |
Rafael Espindola | 2e2563b | 2010-01-26 20:21:43 +0000 | [diff] [blame] | 332 | if (MAI.getCOMMDirectiveAlignmentIsInBytes()) |
Chris Lattner | 4ed5438 | 2010-01-19 06:01:04 +0000 | [diff] [blame] | 333 | OS << ',' << ByteAlignment; |
| 334 | else |
| 335 | OS << ',' << Log2_32(ByteAlignment); |
| 336 | } |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 337 | EmitEOL(); |
Chris Lattner | 4e4db7a | 2009-07-07 20:30:46 +0000 | [diff] [blame] | 338 | } |
| 339 | |
Chris Lattner | 9eb158d | 2010-01-23 07:47:02 +0000 | [diff] [blame] | 340 | /// EmitLocalCommonSymbol - Emit a local common (.lcomm) symbol. |
| 341 | /// |
| 342 | /// @param Symbol - The common symbol to emit. |
| 343 | /// @param Size - The size of the common symbol. |
| 344 | void MCAsmStreamer::EmitLocalCommonSymbol(MCSymbol *Symbol, uint64_t Size) { |
| 345 | assert(MAI.hasLCOMMDirective() && "Doesn't have .lcomm, can't emit it!"); |
| 346 | OS << "\t.lcomm\t" << *Symbol << ',' << Size; |
| 347 | EmitEOL(); |
| 348 | } |
| 349 | |
Daniel Dunbar | 8751b94 | 2009-08-28 05:48:22 +0000 | [diff] [blame] | 350 | void MCAsmStreamer::EmitZerofill(const MCSection *Section, MCSymbol *Symbol, |
Daniel Dunbar | 7092c7e | 2009-08-30 06:17:16 +0000 | [diff] [blame] | 351 | unsigned Size, unsigned ByteAlignment) { |
Daniel Dunbar | 011e4db | 2009-08-13 23:36:34 +0000 | [diff] [blame] | 352 | // Note: a .zerofill directive does not switch sections. |
Chris Lattner | 93b6db3 | 2009-08-08 23:39:42 +0000 | [diff] [blame] | 353 | OS << ".zerofill "; |
| 354 | |
| 355 | // This is a mach-o specific directive. |
Chris Lattner | ff4bc46 | 2009-08-10 01:39:42 +0000 | [diff] [blame] | 356 | const MCSectionMachO *MOSection = ((const MCSectionMachO*)Section); |
Daniel Dunbar | 12de0df | 2009-08-14 18:51:45 +0000 | [diff] [blame] | 357 | OS << MOSection->getSegmentName() << "," << MOSection->getSectionName(); |
Chris Lattner | 93b6db3 | 2009-08-08 23:39:42 +0000 | [diff] [blame] | 358 | |
Chris Lattner | 9be3fee | 2009-07-10 22:20:30 +0000 | [diff] [blame] | 359 | if (Symbol != NULL) { |
Chris Lattner | 10b318b | 2010-01-17 21:43:43 +0000 | [diff] [blame] | 360 | OS << ',' << *Symbol << ',' << Size; |
Daniel Dunbar | 7092c7e | 2009-08-30 06:17:16 +0000 | [diff] [blame] | 361 | if (ByteAlignment != 0) |
| 362 | OS << ',' << Log2_32(ByteAlignment); |
Chris Lattner | 9be3fee | 2009-07-10 22:20:30 +0000 | [diff] [blame] | 363 | } |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 364 | EmitEOL(); |
Chris Lattner | 9be3fee | 2009-07-10 22:20:30 +0000 | [diff] [blame] | 365 | } |
| 366 | |
Eric Christopher | d04d98d | 2010-05-17 02:13:02 +0000 | [diff] [blame] | 367 | // .tbss sym, size, align |
| 368 | // This depends that the symbol has already been mangled from the original, |
| 369 | // e.g. _a. |
Eric Christopher | 4d01cbe | 2010-05-18 21:16:04 +0000 | [diff] [blame] | 370 | void MCAsmStreamer::EmitTBSSSymbol(const MCSection *Section, MCSymbol *Symbol, |
| 371 | uint64_t Size, unsigned ByteAlignment) { |
Eric Christopher | 482eba0 | 2010-05-14 01:50:28 +0000 | [diff] [blame] | 372 | assert(Symbol != NULL && "Symbol shouldn't be NULL!"); |
Eric Christopher | 4d01cbe | 2010-05-18 21:16:04 +0000 | [diff] [blame] | 373 | // Instead of using the Section we'll just use the shortcut. |
| 374 | // This is a mach-o specific directive and section. |
Eric Christopher | d04d98d | 2010-05-17 02:13:02 +0000 | [diff] [blame] | 375 | OS << ".tbss " << *Symbol << ", " << Size; |
Eric Christopher | 482eba0 | 2010-05-14 01:50:28 +0000 | [diff] [blame] | 376 | |
Eric Christopher | 4d01cbe | 2010-05-18 21:16:04 +0000 | [diff] [blame] | 377 | // Output align if we have it. We default to 1 so don't bother printing |
| 378 | // that. |
| 379 | if (ByteAlignment > 1) OS << ", " << Log2_32(ByteAlignment); |
Eric Christopher | 482eba0 | 2010-05-14 01:50:28 +0000 | [diff] [blame] | 380 | |
| 381 | EmitEOL(); |
| 382 | } |
| 383 | |
Chris Lattner | 12e555c | 2010-01-23 00:15:00 +0000 | [diff] [blame] | 384 | static inline char toOctal(int X) { return (X&7)+'0'; } |
| 385 | |
Chris Lattner | a6594fc | 2010-01-25 18:58:59 +0000 | [diff] [blame] | 386 | static void PrintQuotedString(StringRef Data, raw_ostream &OS) { |
| 387 | OS << '"'; |
| 388 | |
| 389 | for (unsigned i = 0, e = Data.size(); i != e; ++i) { |
| 390 | unsigned char C = Data[i]; |
| 391 | if (C == '"' || C == '\\') { |
| 392 | OS << '\\' << (char)C; |
| 393 | continue; |
| 394 | } |
| 395 | |
| 396 | if (isprint((unsigned char)C)) { |
| 397 | OS << (char)C; |
| 398 | continue; |
| 399 | } |
| 400 | |
| 401 | switch (C) { |
| 402 | case '\b': OS << "\\b"; break; |
| 403 | case '\f': OS << "\\f"; break; |
| 404 | case '\n': OS << "\\n"; break; |
| 405 | case '\r': OS << "\\r"; break; |
| 406 | case '\t': OS << "\\t"; break; |
| 407 | default: |
| 408 | OS << '\\'; |
| 409 | OS << toOctal(C >> 6); |
| 410 | OS << toOctal(C >> 3); |
| 411 | OS << toOctal(C >> 0); |
| 412 | break; |
| 413 | } |
| 414 | } |
| 415 | |
| 416 | OS << '"'; |
| 417 | } |
| 418 | |
| 419 | |
Chris Lattner | aaec205 | 2010-01-19 19:46:13 +0000 | [diff] [blame] | 420 | void MCAsmStreamer::EmitBytes(StringRef Data, unsigned AddrSpace) { |
Daniel Dunbar | 71d259b | 2009-06-24 17:00:42 +0000 | [diff] [blame] | 421 | assert(CurSection && "Cannot emit contents before setting section!"); |
Chris Lattner | 12e555c | 2010-01-23 00:15:00 +0000 | [diff] [blame] | 422 | if (Data.empty()) return; |
| 423 | |
| 424 | if (Data.size() == 1) { |
| 425 | OS << MAI.getData8bitsDirective(AddrSpace); |
| 426 | OS << (unsigned)(unsigned char)Data[0]; |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 427 | EmitEOL(); |
Chris Lattner | 12e555c | 2010-01-23 00:15:00 +0000 | [diff] [blame] | 428 | return; |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 429 | } |
Chris Lattner | 12e555c | 2010-01-23 00:15:00 +0000 | [diff] [blame] | 430 | |
| 431 | // If the data ends with 0 and the target supports .asciz, use it, otherwise |
| 432 | // use .ascii |
| 433 | if (MAI.getAscizDirective() && Data.back() == 0) { |
| 434 | OS << MAI.getAscizDirective(); |
| 435 | Data = Data.substr(0, Data.size()-1); |
| 436 | } else { |
| 437 | OS << MAI.getAsciiDirective(); |
| 438 | } |
| 439 | |
Chris Lattner | a6594fc | 2010-01-25 18:58:59 +0000 | [diff] [blame] | 440 | OS << ' '; |
| 441 | PrintQuotedString(Data, OS); |
Chris Lattner | 12e555c | 2010-01-23 00:15:00 +0000 | [diff] [blame] | 442 | EmitEOL(); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 443 | } |
| 444 | |
Chris Lattner | 32ae3fe | 2010-01-19 22:03:38 +0000 | [diff] [blame] | 445 | /// EmitIntValue - Special case of EmitValue that avoids the client having |
| 446 | /// to pass in a MCExpr for constant integers. |
| 447 | void MCAsmStreamer::EmitIntValue(uint64_t Value, unsigned Size, |
| 448 | unsigned AddrSpace) { |
| 449 | assert(CurSection && "Cannot emit contents before setting section!"); |
Chris Lattner | 32ae3fe | 2010-01-19 22:03:38 +0000 | [diff] [blame] | 450 | const char *Directive = 0; |
| 451 | switch (Size) { |
| 452 | default: break; |
| 453 | case 1: Directive = MAI.getData8bitsDirective(AddrSpace); break; |
| 454 | case 2: Directive = MAI.getData16bitsDirective(AddrSpace); break; |
| 455 | case 4: Directive = MAI.getData32bitsDirective(AddrSpace); break; |
Chris Lattner | 5eaa54e | 2010-01-20 06:45:39 +0000 | [diff] [blame] | 456 | case 8: |
| 457 | Directive = MAI.getData64bitsDirective(AddrSpace); |
| 458 | // If the target doesn't support 64-bit data, emit as two 32-bit halves. |
| 459 | if (Directive) break; |
| 460 | if (isLittleEndian()) { |
| 461 | EmitIntValue((uint32_t)(Value >> 0 ), 4, AddrSpace); |
| 462 | EmitIntValue((uint32_t)(Value >> 32), 4, AddrSpace); |
| 463 | } else { |
| 464 | EmitIntValue((uint32_t)(Value >> 32), 4, AddrSpace); |
| 465 | EmitIntValue((uint32_t)(Value >> 0 ), 4, AddrSpace); |
| 466 | } |
| 467 | return; |
Chris Lattner | 32ae3fe | 2010-01-19 22:03:38 +0000 | [diff] [blame] | 468 | } |
| 469 | |
| 470 | assert(Directive && "Invalid size for machine code value!"); |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 471 | OS << Directive << truncateToSize(Value, Size); |
| 472 | EmitEOL(); |
Chris Lattner | 32ae3fe | 2010-01-19 22:03:38 +0000 | [diff] [blame] | 473 | } |
| 474 | |
Chris Lattner | aaec205 | 2010-01-19 19:46:13 +0000 | [diff] [blame] | 475 | void MCAsmStreamer::EmitValue(const MCExpr *Value, unsigned Size, |
| 476 | unsigned AddrSpace) { |
Daniel Dunbar | 71d259b | 2009-06-24 17:00:42 +0000 | [diff] [blame] | 477 | assert(CurSection && "Cannot emit contents before setting section!"); |
Chris Lattner | 32ae3fe | 2010-01-19 22:03:38 +0000 | [diff] [blame] | 478 | const char *Directive = 0; |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 479 | switch (Size) { |
Chris Lattner | 32ae3fe | 2010-01-19 22:03:38 +0000 | [diff] [blame] | 480 | default: break; |
| 481 | case 1: Directive = MAI.getData8bitsDirective(AddrSpace); break; |
| 482 | case 2: Directive = MAI.getData16bitsDirective(AddrSpace); break; |
| 483 | case 4: Directive = MAI.getData32bitsDirective(AddrSpace); break; |
| 484 | case 8: Directive = MAI.getData64bitsDirective(AddrSpace); break; |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 485 | } |
Chris Lattner | aaec205 | 2010-01-19 19:46:13 +0000 | [diff] [blame] | 486 | |
Chris Lattner | 32ae3fe | 2010-01-19 22:03:38 +0000 | [diff] [blame] | 487 | assert(Directive && "Invalid size for machine code value!"); |
Chris Lattner | 718fb59 | 2010-01-25 21:28:50 +0000 | [diff] [blame] | 488 | OS << Directive << *Value; |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 489 | EmitEOL(); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 490 | } |
| 491 | |
Chris Lattner | 718fb59 | 2010-01-25 21:28:50 +0000 | [diff] [blame] | 492 | void MCAsmStreamer::EmitGPRel32Value(const MCExpr *Value) { |
| 493 | assert(MAI.getGPRel32Directive() != 0); |
| 494 | OS << MAI.getGPRel32Directive() << *Value; |
| 495 | EmitEOL(); |
| 496 | } |
| 497 | |
| 498 | |
Chris Lattner | 6113b3d | 2010-01-19 18:52:28 +0000 | [diff] [blame] | 499 | /// EmitFill - Emit NumBytes bytes worth of the value specified by |
| 500 | /// FillValue. This implements directives such as '.space'. |
Chris Lattner | aaec205 | 2010-01-19 19:46:13 +0000 | [diff] [blame] | 501 | void MCAsmStreamer::EmitFill(uint64_t NumBytes, uint8_t FillValue, |
| 502 | unsigned AddrSpace) { |
Chris Lattner | 8a6d7ac | 2010-01-19 18:58:52 +0000 | [diff] [blame] | 503 | if (NumBytes == 0) return; |
| 504 | |
Chris Lattner | aaec205 | 2010-01-19 19:46:13 +0000 | [diff] [blame] | 505 | if (AddrSpace == 0) |
| 506 | if (const char *ZeroDirective = MAI.getZeroDirective()) { |
| 507 | OS << ZeroDirective << NumBytes; |
| 508 | if (FillValue != 0) |
| 509 | OS << ',' << (int)FillValue; |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 510 | EmitEOL(); |
Chris Lattner | aaec205 | 2010-01-19 19:46:13 +0000 | [diff] [blame] | 511 | return; |
| 512 | } |
| 513 | |
| 514 | // Emit a byte at a time. |
| 515 | MCStreamer::EmitFill(NumBytes, FillValue, AddrSpace); |
Chris Lattner | 6113b3d | 2010-01-19 18:52:28 +0000 | [diff] [blame] | 516 | } |
| 517 | |
Daniel Dunbar | 84a2926 | 2009-06-24 19:25:34 +0000 | [diff] [blame] | 518 | void MCAsmStreamer::EmitValueToAlignment(unsigned ByteAlignment, int64_t Value, |
| 519 | unsigned ValueSize, |
| 520 | unsigned MaxBytesToEmit) { |
Chris Lattner | 663c2d2 | 2009-08-19 06:12:02 +0000 | [diff] [blame] | 521 | // Some assemblers don't support non-power of two alignments, so we always |
| 522 | // emit alignments as a power of two if possible. |
| 523 | if (isPowerOf2_32(ByteAlignment)) { |
Chris Lattner | 6e579c6 | 2009-08-19 06:35:36 +0000 | [diff] [blame] | 524 | switch (ValueSize) { |
| 525 | default: llvm_unreachable("Invalid size for machine code value!"); |
Chris Lattner | 33adcfb | 2009-08-22 21:43:10 +0000 | [diff] [blame] | 526 | case 1: OS << MAI.getAlignDirective(); break; |
| 527 | // FIXME: use MAI for this! |
Chris Lattner | 6e579c6 | 2009-08-19 06:35:36 +0000 | [diff] [blame] | 528 | case 2: OS << ".p2alignw "; break; |
| 529 | case 4: OS << ".p2alignl "; break; |
| 530 | case 8: llvm_unreachable("Unsupported alignment size!"); |
| 531 | } |
Chris Lattner | 663c2d2 | 2009-08-19 06:12:02 +0000 | [diff] [blame] | 532 | |
Chris Lattner | 33adcfb | 2009-08-22 21:43:10 +0000 | [diff] [blame] | 533 | if (MAI.getAlignmentIsInBytes()) |
Chris Lattner | 663c2d2 | 2009-08-19 06:12:02 +0000 | [diff] [blame] | 534 | OS << ByteAlignment; |
| 535 | else |
| 536 | OS << Log2_32(ByteAlignment); |
Daniel Dunbar | 84a2926 | 2009-06-24 19:25:34 +0000 | [diff] [blame] | 537 | |
Chris Lattner | 663c2d2 | 2009-08-19 06:12:02 +0000 | [diff] [blame] | 538 | if (Value || MaxBytesToEmit) { |
Chris Lattner | 579531c | 2009-09-03 04:01:10 +0000 | [diff] [blame] | 539 | OS << ", 0x"; |
| 540 | OS.write_hex(truncateToSize(Value, ValueSize)); |
Chris Lattner | 663c2d2 | 2009-08-19 06:12:02 +0000 | [diff] [blame] | 541 | |
| 542 | if (MaxBytesToEmit) |
| 543 | OS << ", " << MaxBytesToEmit; |
| 544 | } |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 545 | EmitEOL(); |
Chris Lattner | 663c2d2 | 2009-08-19 06:12:02 +0000 | [diff] [blame] | 546 | return; |
| 547 | } |
| 548 | |
| 549 | // Non-power of two alignment. This is not widely supported by assemblers. |
Chris Lattner | 33adcfb | 2009-08-22 21:43:10 +0000 | [diff] [blame] | 550 | // FIXME: Parameterize this based on MAI. |
Daniel Dunbar | 84a2926 | 2009-06-24 19:25:34 +0000 | [diff] [blame] | 551 | switch (ValueSize) { |
Chris Lattner | 663c2d2 | 2009-08-19 06:12:02 +0000 | [diff] [blame] | 552 | default: llvm_unreachable("Invalid size for machine code value!"); |
| 553 | case 1: OS << ".balign"; break; |
| 554 | case 2: OS << ".balignw"; break; |
| 555 | case 4: OS << ".balignl"; break; |
| 556 | case 8: llvm_unreachable("Unsupported alignment size!"); |
Daniel Dunbar | 84a2926 | 2009-06-24 19:25:34 +0000 | [diff] [blame] | 557 | } |
| 558 | |
Chris Lattner | 663c2d2 | 2009-08-19 06:12:02 +0000 | [diff] [blame] | 559 | OS << ' ' << ByteAlignment; |
Daniel Dunbar | 304f6a4 | 2009-06-25 21:03:18 +0000 | [diff] [blame] | 560 | OS << ", " << truncateToSize(Value, ValueSize); |
Daniel Dunbar | 84a2926 | 2009-06-24 19:25:34 +0000 | [diff] [blame] | 561 | if (MaxBytesToEmit) |
| 562 | OS << ", " << MaxBytesToEmit; |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 563 | EmitEOL(); |
Daniel Dunbar | 84a2926 | 2009-06-24 19:25:34 +0000 | [diff] [blame] | 564 | } |
| 565 | |
Kevin Enderby | 6e72048 | 2010-02-23 18:26:34 +0000 | [diff] [blame] | 566 | void MCAsmStreamer::EmitCodeAlignment(unsigned ByteAlignment, |
| 567 | unsigned MaxBytesToEmit) { |
Chris Lattner | ec167fd | 2010-02-23 18:44:31 +0000 | [diff] [blame] | 568 | // Emit with a text fill value. |
| 569 | EmitValueToAlignment(ByteAlignment, MAI.getTextAlignFillValue(), |
| 570 | 1, MaxBytesToEmit); |
Kevin Enderby | 6e72048 | 2010-02-23 18:26:34 +0000 | [diff] [blame] | 571 | } |
| 572 | |
Daniel Dunbar | 821e333 | 2009-08-31 08:09:28 +0000 | [diff] [blame] | 573 | void MCAsmStreamer::EmitValueToOffset(const MCExpr *Offset, |
Daniel Dunbar | 84a2926 | 2009-06-24 19:25:34 +0000 | [diff] [blame] | 574 | unsigned char Value) { |
| 575 | // FIXME: Verify that Offset is associated with the current section. |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 576 | OS << ".org " << *Offset << ", " << (unsigned) Value; |
| 577 | EmitEOL(); |
Daniel Dunbar | 84a2926 | 2009-06-24 19:25:34 +0000 | [diff] [blame] | 578 | } |
| 579 | |
Chris Lattner | a6594fc | 2010-01-25 18:58:59 +0000 | [diff] [blame] | 580 | |
| 581 | void MCAsmStreamer::EmitFileDirective(StringRef Filename) { |
| 582 | assert(MAI.hasSingleParameterDotFile()); |
| 583 | OS << "\t.file\t"; |
| 584 | PrintQuotedString(Filename, OS); |
| 585 | EmitEOL(); |
| 586 | } |
| 587 | |
| 588 | void MCAsmStreamer::EmitDwarfFileDirective(unsigned FileNo, StringRef Filename){ |
| 589 | OS << "\t.file\t" << FileNo << ' '; |
| 590 | PrintQuotedString(Filename, OS); |
| 591 | EmitEOL(); |
| 592 | } |
| 593 | |
Daniel Dunbar | 6b71653 | 2010-02-09 23:00:14 +0000 | [diff] [blame] | 594 | void MCAsmStreamer::AddEncodingComment(const MCInst &Inst) { |
| 595 | raw_ostream &OS = GetCommentOS(); |
| 596 | SmallString<256> Code; |
| 597 | SmallVector<MCFixup, 4> Fixups; |
| 598 | raw_svector_ostream VecOS(Code); |
| 599 | Emitter->EncodeInstruction(Inst, VecOS, Fixups); |
| 600 | VecOS.flush(); |
Chris Lattner | a6594fc | 2010-01-25 18:58:59 +0000 | [diff] [blame] | 601 | |
Daniel Dunbar | 6b71653 | 2010-02-09 23:00:14 +0000 | [diff] [blame] | 602 | // If we are showing fixups, create symbolic markers in the encoded |
| 603 | // representation. We do this by making a per-bit map to the fixup item index, |
| 604 | // then trying to display it as nicely as possible. |
Daniel Dunbar | 5532cf4 | 2010-02-10 01:41:14 +0000 | [diff] [blame] | 605 | SmallVector<uint8_t, 64> FixupMap; |
| 606 | FixupMap.resize(Code.size() * 8); |
Daniel Dunbar | 6b71653 | 2010-02-09 23:00:14 +0000 | [diff] [blame] | 607 | for (unsigned i = 0, e = Code.size() * 8; i != e; ++i) |
| 608 | FixupMap[i] = 0; |
| 609 | |
| 610 | for (unsigned i = 0, e = Fixups.size(); i != e; ++i) { |
| 611 | MCFixup &F = Fixups[i]; |
Chris Lattner | 8d31de6 | 2010-02-11 21:27:18 +0000 | [diff] [blame] | 612 | const MCFixupKindInfo &Info = Emitter->getFixupKindInfo(F.getKind()); |
Daniel Dunbar | 6b71653 | 2010-02-09 23:00:14 +0000 | [diff] [blame] | 613 | for (unsigned j = 0; j != Info.TargetSize; ++j) { |
| 614 | unsigned Index = F.getOffset() * 8 + Info.TargetOffset + j; |
| 615 | assert(Index < Code.size() * 8 && "Invalid offset in fixup!"); |
| 616 | FixupMap[Index] = 1 + i; |
| 617 | } |
| 618 | } |
| 619 | |
| 620 | OS << "encoding: ["; |
| 621 | for (unsigned i = 0, e = Code.size(); i != e; ++i) { |
| 622 | if (i) |
| 623 | OS << ','; |
| 624 | |
| 625 | // See if all bits are the same map entry. |
| 626 | uint8_t MapEntry = FixupMap[i * 8 + 0]; |
| 627 | for (unsigned j = 1; j != 8; ++j) { |
| 628 | if (FixupMap[i * 8 + j] == MapEntry) |
| 629 | continue; |
| 630 | |
| 631 | MapEntry = uint8_t(~0U); |
| 632 | break; |
| 633 | } |
| 634 | |
| 635 | if (MapEntry != uint8_t(~0U)) { |
| 636 | if (MapEntry == 0) { |
| 637 | OS << format("0x%02x", uint8_t(Code[i])); |
| 638 | } else { |
| 639 | assert(Code[i] == 0 && "Encoder wrote into fixed up bit!"); |
| 640 | OS << char('A' + MapEntry - 1); |
| 641 | } |
| 642 | } else { |
| 643 | // Otherwise, write out in binary. |
| 644 | OS << "0b"; |
| 645 | for (unsigned j = 8; j--;) { |
| 646 | unsigned Bit = (Code[i] >> j) & 1; |
| 647 | if (uint8_t MapEntry = FixupMap[i * 8 + j]) { |
| 648 | assert(Bit == 0 && "Encoder wrote into fixed up bit!"); |
| 649 | OS << char('A' + MapEntry - 1); |
| 650 | } else |
| 651 | OS << Bit; |
| 652 | } |
| 653 | } |
| 654 | } |
| 655 | OS << "]\n"; |
| 656 | |
| 657 | for (unsigned i = 0, e = Fixups.size(); i != e; ++i) { |
| 658 | MCFixup &F = Fixups[i]; |
Chris Lattner | 8d31de6 | 2010-02-11 21:27:18 +0000 | [diff] [blame] | 659 | const MCFixupKindInfo &Info = Emitter->getFixupKindInfo(F.getKind()); |
Daniel Dunbar | 6b71653 | 2010-02-09 23:00:14 +0000 | [diff] [blame] | 660 | OS << " fixup " << char('A' + i) << " - " << "offset: " << F.getOffset() |
Daniel Dunbar | 5d5a1e1 | 2010-02-10 04:47:08 +0000 | [diff] [blame] | 661 | << ", value: " << *F.getValue() << ", kind: " << Info.Name << "\n"; |
Daniel Dunbar | 6b71653 | 2010-02-09 23:00:14 +0000 | [diff] [blame] | 662 | } |
| 663 | } |
| 664 | |
| 665 | void MCAsmStreamer::EmitInstruction(const MCInst &Inst) { |
| 666 | assert(CurSection && "Cannot emit contents before setting section!"); |
| 667 | |
| 668 | // Show the encoding in a comment if we have a code emitter. |
| 669 | if (Emitter) |
| 670 | AddEncodingComment(Inst); |
| 671 | |
Chris Lattner | 30d9a64 | 2010-02-09 00:54:51 +0000 | [diff] [blame] | 672 | // Show the MCInst if enabled. |
Daniel Dunbar | c9adb8c | 2010-05-26 15:18:13 +0000 | [diff] [blame] | 673 | if (ShowInst) { |
Daniel Dunbar | 67c076c | 2010-03-22 21:49:34 +0000 | [diff] [blame] | 674 | Inst.dump_pretty(GetCommentOS(), &MAI, InstPrinter.get(), "\n "); |
Daniel Dunbar | c9adb8c | 2010-05-26 15:18:13 +0000 | [diff] [blame] | 675 | GetCommentOS() << "\n"; |
| 676 | } |
| 677 | |
Daniel Dunbar | 67c076c | 2010-03-22 21:49:34 +0000 | [diff] [blame] | 678 | // If we have an AsmPrinter, use that to print, otherwise print the MCInst. |
Daniel Dunbar | 9dee8e3 | 2010-02-03 18:18:30 +0000 | [diff] [blame] | 679 | if (InstPrinter) |
Chris Lattner | d374087 | 2010-04-04 05:04:31 +0000 | [diff] [blame] | 680 | InstPrinter->printInst(&Inst, OS); |
Daniel Dunbar | 9dee8e3 | 2010-02-03 18:18:30 +0000 | [diff] [blame] | 681 | else |
| 682 | Inst.print(OS, &MAI); |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 683 | EmitEOL(); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 684 | } |
| 685 | |
Chris Lattner | 91bead7 | 2010-04-03 21:35:55 +0000 | [diff] [blame] | 686 | /// EmitRawText - If this file is backed by a assembly streamer, this dumps |
| 687 | /// the specified string in the output .s file. This capability is |
| 688 | /// indicated by the hasRawTextSupport() predicate. |
| 689 | void MCAsmStreamer::EmitRawText(StringRef String) { |
Chris Lattner | d5928dc | 2010-04-03 22:06:56 +0000 | [diff] [blame] | 690 | if (!String.empty() && String.back() == '\n') |
| 691 | String = String.substr(0, String.size()-1); |
Chris Lattner | 91bead7 | 2010-04-03 21:35:55 +0000 | [diff] [blame] | 692 | OS << String; |
Chris Lattner | d5928dc | 2010-04-03 22:06:56 +0000 | [diff] [blame] | 693 | EmitEOL(); |
Chris Lattner | 91bead7 | 2010-04-03 21:35:55 +0000 | [diff] [blame] | 694 | } |
| 695 | |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 696 | void MCAsmStreamer::Finish() { |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 697 | } |
Daniel Dunbar | 9dee8e3 | 2010-02-03 18:18:30 +0000 | [diff] [blame] | 698 | |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 699 | MCStreamer *llvm::createAsmStreamer(MCContext &Context, |
| 700 | formatted_raw_ostream &OS, |
Chris Lattner | fdab14b | 2010-03-12 18:28:53 +0000 | [diff] [blame] | 701 | bool isLittleEndian, |
Chris Lattner | 0740441 | 2010-01-22 07:06:15 +0000 | [diff] [blame] | 702 | bool isVerboseAsm, MCInstPrinter *IP, |
Daniel Dunbar | 5532cf4 | 2010-02-10 01:41:14 +0000 | [diff] [blame] | 703 | MCCodeEmitter *CE, bool ShowInst) { |
Chris Lattner | fdab14b | 2010-03-12 18:28:53 +0000 | [diff] [blame] | 704 | return new MCAsmStreamer(Context, OS, isLittleEndian, isVerboseAsm, |
Daniel Dunbar | 5532cf4 | 2010-02-10 01:41:14 +0000 | [diff] [blame] | 705 | IP, CE, ShowInst); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 706 | } |