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) { |
Benjamin Kramer | 1674b0b | 2010-09-02 18:53:37 +0000 | [diff] [blame^] | 220 | PrevSection = CurSection; |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 221 | CurSection = Section; |
Chris Lattner | 33adcfb | 2009-08-22 21:43:10 +0000 | [diff] [blame] | 222 | Section->PrintSwitchToSection(MAI, OS); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 223 | } |
| 224 | } |
| 225 | |
| 226 | void MCAsmStreamer::EmitLabel(MCSymbol *Symbol) { |
Daniel Dunbar | 8906ff1 | 2009-08-22 07:22:36 +0000 | [diff] [blame] | 227 | assert(Symbol->isUndefined() && "Cannot define a symbol twice!"); |
Daniel Dunbar | c304718 | 2010-05-05 19:01:00 +0000 | [diff] [blame] | 228 | assert(!Symbol->isVariable() && "Cannot emit a variable symbol!"); |
Daniel Dunbar | 71d259b | 2009-06-24 17:00:42 +0000 | [diff] [blame] | 229 | assert(CurSection && "Cannot emit before setting section!"); |
Daniel Dunbar | 71d259b | 2009-06-24 17:00:42 +0000 | [diff] [blame] | 230 | |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 231 | OS << *Symbol << ":"; |
| 232 | EmitEOL(); |
Daniel Dunbar | 8906ff1 | 2009-08-22 07:22:36 +0000 | [diff] [blame] | 233 | Symbol->setSection(*CurSection); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 234 | } |
| 235 | |
Chris Lattner | a5ad93a | 2010-01-23 06:39:22 +0000 | [diff] [blame] | 236 | void MCAsmStreamer::EmitAssemblerFlag(MCAssemblerFlag Flag) { |
Kevin Enderby | f96db46 | 2009-07-16 17:56:39 +0000 | [diff] [blame] | 237 | switch (Flag) { |
Daniel Dunbar | 011e4db | 2009-08-13 23:36:34 +0000 | [diff] [blame] | 238 | default: assert(0 && "Invalid flag!"); |
Chris Lattner | a5ad93a | 2010-01-23 06:39:22 +0000 | [diff] [blame] | 239 | case MCAF_SubsectionsViaSymbols: OS << ".subsections_via_symbols"; break; |
Kevin Enderby | f96db46 | 2009-07-16 17:56:39 +0000 | [diff] [blame] | 240 | } |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 241 | EmitEOL(); |
Kevin Enderby | a5c7832 | 2009-07-13 21:03:15 +0000 | [diff] [blame] | 242 | } |
| 243 | |
Daniel Dunbar | 821e333 | 2009-08-31 08:09:28 +0000 | [diff] [blame] | 244 | void MCAsmStreamer::EmitAssignment(MCSymbol *Symbol, const MCExpr *Value) { |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 245 | OS << *Symbol << " = " << *Value; |
| 246 | EmitEOL(); |
Daniel Dunbar | fffff91 | 2009-10-16 01:34:54 +0000 | [diff] [blame] | 247 | |
| 248 | // FIXME: Lift context changes into super class. |
Daniel Dunbar | 08a408a | 2010-05-05 17:41:00 +0000 | [diff] [blame] | 249 | Symbol->setVariableValue(Value); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 250 | } |
| 251 | |
Daniel Dunbar | fffff91 | 2009-10-16 01:34:54 +0000 | [diff] [blame] | 252 | void MCAsmStreamer::EmitSymbolAttribute(MCSymbol *Symbol, |
Chris Lattner | a5ad93a | 2010-01-23 06:39:22 +0000 | [diff] [blame] | 253 | MCSymbolAttr Attribute) { |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 254 | switch (Attribute) { |
Chris Lattner | a5ad93a | 2010-01-23 06:39:22 +0000 | [diff] [blame] | 255 | case MCSA_Invalid: assert(0 && "Invalid symbol attribute"); |
Chris Lattner | ed0ab15 | 2010-01-25 18:30:45 +0000 | [diff] [blame] | 256 | case MCSA_ELF_TypeFunction: /// .type _foo, STT_FUNC # aka @function |
| 257 | case MCSA_ELF_TypeIndFunction: /// .type _foo, STT_GNU_IFUNC |
| 258 | case MCSA_ELF_TypeObject: /// .type _foo, STT_OBJECT # aka @object |
| 259 | case MCSA_ELF_TypeTLS: /// .type _foo, STT_TLS # aka @tls_object |
| 260 | case MCSA_ELF_TypeCommon: /// .type _foo, STT_COMMON # aka @common |
| 261 | case MCSA_ELF_TypeNoType: /// .type _foo, STT_NOTYPE # aka @notype |
| 262 | assert(MAI.hasDotTypeDotSizeDirective() && "Symbol Attr not supported"); |
Dan Gohman | 523d70e | 2010-02-04 01:42:13 +0000 | [diff] [blame] | 263 | OS << "\t.type\t" << *Symbol << ',' |
Chris Lattner | ed0ab15 | 2010-01-25 18:30:45 +0000 | [diff] [blame] | 264 | << ((MAI.getCommentString()[0] != '@') ? '@' : '%'); |
| 265 | switch (Attribute) { |
| 266 | default: assert(0 && "Unknown ELF .type"); |
| 267 | case MCSA_ELF_TypeFunction: OS << "function"; break; |
| 268 | case MCSA_ELF_TypeIndFunction: OS << "gnu_indirect_function"; break; |
| 269 | case MCSA_ELF_TypeObject: OS << "object"; break; |
| 270 | case MCSA_ELF_TypeTLS: OS << "tls_object"; break; |
| 271 | case MCSA_ELF_TypeCommon: OS << "common"; break; |
| 272 | case MCSA_ELF_TypeNoType: OS << "no_type"; break; |
| 273 | } |
| 274 | EmitEOL(); |
| 275 | return; |
| 276 | case MCSA_Global: // .globl/.global |
| 277 | OS << MAI.getGlobalDirective(); |
| 278 | break; |
Chris Lattner | 61abd7b | 2010-06-21 20:35:01 +0000 | [diff] [blame] | 279 | case MCSA_Hidden: OS << "\t.hidden\t"; break; |
| 280 | case MCSA_IndirectSymbol: OS << "\t.indirect_symbol\t"; break; |
| 281 | case MCSA_Internal: OS << "\t.internal\t"; break; |
| 282 | case MCSA_LazyReference: OS << "\t.lazy_reference\t"; break; |
| 283 | case MCSA_Local: OS << "\t.local\t"; break; |
| 284 | case MCSA_NoDeadStrip: OS << "\t.no_dead_strip\t"; break; |
| 285 | case MCSA_PrivateExtern: OS << "\t.private_extern\t"; break; |
| 286 | case MCSA_Protected: OS << "\t.protected\t"; break; |
| 287 | case MCSA_Reference: OS << "\t.reference\t"; break; |
| 288 | case MCSA_Weak: OS << "\t.weak\t"; break; |
| 289 | case MCSA_WeakDefinition: OS << "\t.weak_definition\t"; break; |
Chris Lattner | a5ad93a | 2010-01-23 06:39:22 +0000 | [diff] [blame] | 290 | // .weak_reference |
| 291 | case MCSA_WeakReference: OS << MAI.getWeakRefDirective(); break; |
Kevin Enderby | f59cac5 | 2010-07-08 17:22:42 +0000 | [diff] [blame] | 292 | case MCSA_WeakDefAutoPrivate: OS << "\t.weak_def_can_be_hidden\t"; break; |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 293 | } |
| 294 | |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 295 | OS << *Symbol; |
| 296 | EmitEOL(); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 297 | } |
| 298 | |
Kevin Enderby | 95cf30c | 2009-07-14 18:17:10 +0000 | [diff] [blame] | 299 | void MCAsmStreamer::EmitSymbolDesc(MCSymbol *Symbol, unsigned DescValue) { |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 300 | OS << ".desc" << ' ' << *Symbol << ',' << DescValue; |
| 301 | EmitEOL(); |
Kevin Enderby | 95cf30c | 2009-07-14 18:17:10 +0000 | [diff] [blame] | 302 | } |
| 303 | |
Chris Lattner | b54b9dd | 2010-05-08 19:54:22 +0000 | [diff] [blame] | 304 | void MCAsmStreamer::BeginCOFFSymbolDef(const MCSymbol *Symbol) { |
| 305 | OS << "\t.def\t " << *Symbol << ';'; |
| 306 | EmitEOL(); |
| 307 | } |
| 308 | |
| 309 | void MCAsmStreamer::EmitCOFFSymbolStorageClass (int StorageClass) { |
| 310 | OS << "\t.scl\t" << StorageClass << ';'; |
| 311 | EmitEOL(); |
| 312 | } |
| 313 | |
| 314 | void MCAsmStreamer::EmitCOFFSymbolType (int Type) { |
| 315 | OS << "\t.type\t" << Type << ';'; |
| 316 | EmitEOL(); |
| 317 | } |
| 318 | |
| 319 | void MCAsmStreamer::EndCOFFSymbolDef() { |
| 320 | OS << "\t.endef"; |
| 321 | EmitEOL(); |
| 322 | } |
| 323 | |
Chris Lattner | 99328ad | 2010-01-25 07:52:13 +0000 | [diff] [blame] | 324 | void MCAsmStreamer::EmitELFSize(MCSymbol *Symbol, const MCExpr *Value) { |
| 325 | assert(MAI.hasDotTypeDotSizeDirective()); |
| 326 | OS << "\t.size\t" << *Symbol << ", " << *Value << '\n'; |
| 327 | } |
| 328 | |
Chris Lattner | 9eb158d | 2010-01-23 07:47:02 +0000 | [diff] [blame] | 329 | void MCAsmStreamer::EmitCommonSymbol(MCSymbol *Symbol, uint64_t Size, |
Daniel Dunbar | 7092c7e | 2009-08-30 06:17:16 +0000 | [diff] [blame] | 330 | unsigned ByteAlignment) { |
Chris Lattner | 9eb158d | 2010-01-23 07:47:02 +0000 | [diff] [blame] | 331 | OS << "\t.comm\t" << *Symbol << ',' << Size; |
Chris Lattner | 6559d76 | 2010-01-25 07:29:13 +0000 | [diff] [blame] | 332 | if (ByteAlignment != 0) { |
Rafael Espindola | 2e2563b | 2010-01-26 20:21:43 +0000 | [diff] [blame] | 333 | if (MAI.getCOMMDirectiveAlignmentIsInBytes()) |
Chris Lattner | 4ed5438 | 2010-01-19 06:01:04 +0000 | [diff] [blame] | 334 | OS << ',' << ByteAlignment; |
| 335 | else |
| 336 | OS << ',' << Log2_32(ByteAlignment); |
| 337 | } |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 338 | EmitEOL(); |
Chris Lattner | 4e4db7a | 2009-07-07 20:30:46 +0000 | [diff] [blame] | 339 | } |
| 340 | |
Chris Lattner | 9eb158d | 2010-01-23 07:47:02 +0000 | [diff] [blame] | 341 | /// EmitLocalCommonSymbol - Emit a local common (.lcomm) symbol. |
| 342 | /// |
| 343 | /// @param Symbol - The common symbol to emit. |
| 344 | /// @param Size - The size of the common symbol. |
| 345 | void MCAsmStreamer::EmitLocalCommonSymbol(MCSymbol *Symbol, uint64_t Size) { |
| 346 | assert(MAI.hasLCOMMDirective() && "Doesn't have .lcomm, can't emit it!"); |
| 347 | OS << "\t.lcomm\t" << *Symbol << ',' << Size; |
| 348 | EmitEOL(); |
| 349 | } |
| 350 | |
Daniel Dunbar | 8751b94 | 2009-08-28 05:48:22 +0000 | [diff] [blame] | 351 | void MCAsmStreamer::EmitZerofill(const MCSection *Section, MCSymbol *Symbol, |
Daniel Dunbar | 7092c7e | 2009-08-30 06:17:16 +0000 | [diff] [blame] | 352 | unsigned Size, unsigned ByteAlignment) { |
Daniel Dunbar | 011e4db | 2009-08-13 23:36:34 +0000 | [diff] [blame] | 353 | // Note: a .zerofill directive does not switch sections. |
Chris Lattner | 93b6db3 | 2009-08-08 23:39:42 +0000 | [diff] [blame] | 354 | OS << ".zerofill "; |
| 355 | |
| 356 | // This is a mach-o specific directive. |
Chris Lattner | ff4bc46 | 2009-08-10 01:39:42 +0000 | [diff] [blame] | 357 | const MCSectionMachO *MOSection = ((const MCSectionMachO*)Section); |
Daniel Dunbar | 12de0df | 2009-08-14 18:51:45 +0000 | [diff] [blame] | 358 | OS << MOSection->getSegmentName() << "," << MOSection->getSectionName(); |
Chris Lattner | 93b6db3 | 2009-08-08 23:39:42 +0000 | [diff] [blame] | 359 | |
Chris Lattner | 9be3fee | 2009-07-10 22:20:30 +0000 | [diff] [blame] | 360 | if (Symbol != NULL) { |
Chris Lattner | 10b318b | 2010-01-17 21:43:43 +0000 | [diff] [blame] | 361 | OS << ',' << *Symbol << ',' << Size; |
Daniel Dunbar | 7092c7e | 2009-08-30 06:17:16 +0000 | [diff] [blame] | 362 | if (ByteAlignment != 0) |
| 363 | OS << ',' << Log2_32(ByteAlignment); |
Chris Lattner | 9be3fee | 2009-07-10 22:20:30 +0000 | [diff] [blame] | 364 | } |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 365 | EmitEOL(); |
Chris Lattner | 9be3fee | 2009-07-10 22:20:30 +0000 | [diff] [blame] | 366 | } |
| 367 | |
Eric Christopher | d04d98d | 2010-05-17 02:13:02 +0000 | [diff] [blame] | 368 | // .tbss sym, size, align |
| 369 | // This depends that the symbol has already been mangled from the original, |
| 370 | // e.g. _a. |
Eric Christopher | 4d01cbe | 2010-05-18 21:16:04 +0000 | [diff] [blame] | 371 | void MCAsmStreamer::EmitTBSSSymbol(const MCSection *Section, MCSymbol *Symbol, |
| 372 | uint64_t Size, unsigned ByteAlignment) { |
Eric Christopher | 482eba0 | 2010-05-14 01:50:28 +0000 | [diff] [blame] | 373 | assert(Symbol != NULL && "Symbol shouldn't be NULL!"); |
Eric Christopher | 4d01cbe | 2010-05-18 21:16:04 +0000 | [diff] [blame] | 374 | // Instead of using the Section we'll just use the shortcut. |
| 375 | // This is a mach-o specific directive and section. |
Eric Christopher | d04d98d | 2010-05-17 02:13:02 +0000 | [diff] [blame] | 376 | OS << ".tbss " << *Symbol << ", " << Size; |
Eric Christopher | 482eba0 | 2010-05-14 01:50:28 +0000 | [diff] [blame] | 377 | |
Eric Christopher | 4d01cbe | 2010-05-18 21:16:04 +0000 | [diff] [blame] | 378 | // Output align if we have it. We default to 1 so don't bother printing |
| 379 | // that. |
| 380 | if (ByteAlignment > 1) OS << ", " << Log2_32(ByteAlignment); |
Eric Christopher | 482eba0 | 2010-05-14 01:50:28 +0000 | [diff] [blame] | 381 | |
| 382 | EmitEOL(); |
| 383 | } |
| 384 | |
Chris Lattner | 12e555c | 2010-01-23 00:15:00 +0000 | [diff] [blame] | 385 | static inline char toOctal(int X) { return (X&7)+'0'; } |
| 386 | |
Chris Lattner | a6594fc | 2010-01-25 18:58:59 +0000 | [diff] [blame] | 387 | static void PrintQuotedString(StringRef Data, raw_ostream &OS) { |
| 388 | OS << '"'; |
| 389 | |
| 390 | for (unsigned i = 0, e = Data.size(); i != e; ++i) { |
| 391 | unsigned char C = Data[i]; |
| 392 | if (C == '"' || C == '\\') { |
| 393 | OS << '\\' << (char)C; |
| 394 | continue; |
| 395 | } |
| 396 | |
| 397 | if (isprint((unsigned char)C)) { |
| 398 | OS << (char)C; |
| 399 | continue; |
| 400 | } |
| 401 | |
| 402 | switch (C) { |
| 403 | case '\b': OS << "\\b"; break; |
| 404 | case '\f': OS << "\\f"; break; |
| 405 | case '\n': OS << "\\n"; break; |
| 406 | case '\r': OS << "\\r"; break; |
| 407 | case '\t': OS << "\\t"; break; |
| 408 | default: |
| 409 | OS << '\\'; |
| 410 | OS << toOctal(C >> 6); |
| 411 | OS << toOctal(C >> 3); |
| 412 | OS << toOctal(C >> 0); |
| 413 | break; |
| 414 | } |
| 415 | } |
| 416 | |
| 417 | OS << '"'; |
| 418 | } |
| 419 | |
| 420 | |
Chris Lattner | aaec205 | 2010-01-19 19:46:13 +0000 | [diff] [blame] | 421 | void MCAsmStreamer::EmitBytes(StringRef Data, unsigned AddrSpace) { |
Daniel Dunbar | 71d259b | 2009-06-24 17:00:42 +0000 | [diff] [blame] | 422 | assert(CurSection && "Cannot emit contents before setting section!"); |
Chris Lattner | 12e555c | 2010-01-23 00:15:00 +0000 | [diff] [blame] | 423 | if (Data.empty()) return; |
| 424 | |
| 425 | if (Data.size() == 1) { |
| 426 | OS << MAI.getData8bitsDirective(AddrSpace); |
| 427 | OS << (unsigned)(unsigned char)Data[0]; |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 428 | EmitEOL(); |
Chris Lattner | 12e555c | 2010-01-23 00:15:00 +0000 | [diff] [blame] | 429 | return; |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 430 | } |
Chris Lattner | 12e555c | 2010-01-23 00:15:00 +0000 | [diff] [blame] | 431 | |
| 432 | // If the data ends with 0 and the target supports .asciz, use it, otherwise |
| 433 | // use .ascii |
| 434 | if (MAI.getAscizDirective() && Data.back() == 0) { |
| 435 | OS << MAI.getAscizDirective(); |
| 436 | Data = Data.substr(0, Data.size()-1); |
| 437 | } else { |
| 438 | OS << MAI.getAsciiDirective(); |
| 439 | } |
| 440 | |
Chris Lattner | a6594fc | 2010-01-25 18:58:59 +0000 | [diff] [blame] | 441 | OS << ' '; |
| 442 | PrintQuotedString(Data, OS); |
Chris Lattner | 12e555c | 2010-01-23 00:15:00 +0000 | [diff] [blame] | 443 | EmitEOL(); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 444 | } |
| 445 | |
Chris Lattner | 32ae3fe | 2010-01-19 22:03:38 +0000 | [diff] [blame] | 446 | /// EmitIntValue - Special case of EmitValue that avoids the client having |
| 447 | /// to pass in a MCExpr for constant integers. |
| 448 | void MCAsmStreamer::EmitIntValue(uint64_t Value, unsigned Size, |
| 449 | unsigned AddrSpace) { |
| 450 | assert(CurSection && "Cannot emit contents before setting section!"); |
Chris Lattner | 32ae3fe | 2010-01-19 22:03:38 +0000 | [diff] [blame] | 451 | const char *Directive = 0; |
| 452 | switch (Size) { |
| 453 | default: break; |
| 454 | case 1: Directive = MAI.getData8bitsDirective(AddrSpace); break; |
| 455 | case 2: Directive = MAI.getData16bitsDirective(AddrSpace); break; |
| 456 | case 4: Directive = MAI.getData32bitsDirective(AddrSpace); break; |
Chris Lattner | 5eaa54e | 2010-01-20 06:45:39 +0000 | [diff] [blame] | 457 | case 8: |
| 458 | Directive = MAI.getData64bitsDirective(AddrSpace); |
| 459 | // If the target doesn't support 64-bit data, emit as two 32-bit halves. |
| 460 | if (Directive) break; |
| 461 | if (isLittleEndian()) { |
| 462 | EmitIntValue((uint32_t)(Value >> 0 ), 4, AddrSpace); |
| 463 | EmitIntValue((uint32_t)(Value >> 32), 4, AddrSpace); |
| 464 | } else { |
| 465 | EmitIntValue((uint32_t)(Value >> 32), 4, AddrSpace); |
| 466 | EmitIntValue((uint32_t)(Value >> 0 ), 4, AddrSpace); |
| 467 | } |
| 468 | return; |
Chris Lattner | 32ae3fe | 2010-01-19 22:03:38 +0000 | [diff] [blame] | 469 | } |
| 470 | |
| 471 | assert(Directive && "Invalid size for machine code value!"); |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 472 | OS << Directive << truncateToSize(Value, Size); |
| 473 | EmitEOL(); |
Chris Lattner | 32ae3fe | 2010-01-19 22:03:38 +0000 | [diff] [blame] | 474 | } |
| 475 | |
Chris Lattner | aaec205 | 2010-01-19 19:46:13 +0000 | [diff] [blame] | 476 | void MCAsmStreamer::EmitValue(const MCExpr *Value, unsigned Size, |
| 477 | unsigned AddrSpace) { |
Daniel Dunbar | 71d259b | 2009-06-24 17:00:42 +0000 | [diff] [blame] | 478 | assert(CurSection && "Cannot emit contents before setting section!"); |
Chris Lattner | 32ae3fe | 2010-01-19 22:03:38 +0000 | [diff] [blame] | 479 | const char *Directive = 0; |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 480 | switch (Size) { |
Chris Lattner | 32ae3fe | 2010-01-19 22:03:38 +0000 | [diff] [blame] | 481 | default: break; |
| 482 | case 1: Directive = MAI.getData8bitsDirective(AddrSpace); break; |
| 483 | case 2: Directive = MAI.getData16bitsDirective(AddrSpace); break; |
| 484 | case 4: Directive = MAI.getData32bitsDirective(AddrSpace); break; |
| 485 | case 8: Directive = MAI.getData64bitsDirective(AddrSpace); break; |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 486 | } |
Chris Lattner | aaec205 | 2010-01-19 19:46:13 +0000 | [diff] [blame] | 487 | |
Chris Lattner | 32ae3fe | 2010-01-19 22:03:38 +0000 | [diff] [blame] | 488 | assert(Directive && "Invalid size for machine code value!"); |
Chris Lattner | 718fb59 | 2010-01-25 21:28:50 +0000 | [diff] [blame] | 489 | OS << Directive << *Value; |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 490 | EmitEOL(); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 491 | } |
| 492 | |
Chris Lattner | 718fb59 | 2010-01-25 21:28:50 +0000 | [diff] [blame] | 493 | void MCAsmStreamer::EmitGPRel32Value(const MCExpr *Value) { |
| 494 | assert(MAI.getGPRel32Directive() != 0); |
| 495 | OS << MAI.getGPRel32Directive() << *Value; |
| 496 | EmitEOL(); |
| 497 | } |
| 498 | |
| 499 | |
Chris Lattner | 6113b3d | 2010-01-19 18:52:28 +0000 | [diff] [blame] | 500 | /// EmitFill - Emit NumBytes bytes worth of the value specified by |
| 501 | /// FillValue. This implements directives such as '.space'. |
Chris Lattner | aaec205 | 2010-01-19 19:46:13 +0000 | [diff] [blame] | 502 | void MCAsmStreamer::EmitFill(uint64_t NumBytes, uint8_t FillValue, |
| 503 | unsigned AddrSpace) { |
Chris Lattner | 8a6d7ac | 2010-01-19 18:58:52 +0000 | [diff] [blame] | 504 | if (NumBytes == 0) return; |
| 505 | |
Chris Lattner | aaec205 | 2010-01-19 19:46:13 +0000 | [diff] [blame] | 506 | if (AddrSpace == 0) |
| 507 | if (const char *ZeroDirective = MAI.getZeroDirective()) { |
| 508 | OS << ZeroDirective << NumBytes; |
| 509 | if (FillValue != 0) |
| 510 | OS << ',' << (int)FillValue; |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 511 | EmitEOL(); |
Chris Lattner | aaec205 | 2010-01-19 19:46:13 +0000 | [diff] [blame] | 512 | return; |
| 513 | } |
| 514 | |
| 515 | // Emit a byte at a time. |
| 516 | MCStreamer::EmitFill(NumBytes, FillValue, AddrSpace); |
Chris Lattner | 6113b3d | 2010-01-19 18:52:28 +0000 | [diff] [blame] | 517 | } |
| 518 | |
Daniel Dunbar | 84a2926 | 2009-06-24 19:25:34 +0000 | [diff] [blame] | 519 | void MCAsmStreamer::EmitValueToAlignment(unsigned ByteAlignment, int64_t Value, |
| 520 | unsigned ValueSize, |
| 521 | unsigned MaxBytesToEmit) { |
Chris Lattner | 663c2d2 | 2009-08-19 06:12:02 +0000 | [diff] [blame] | 522 | // Some assemblers don't support non-power of two alignments, so we always |
| 523 | // emit alignments as a power of two if possible. |
| 524 | if (isPowerOf2_32(ByteAlignment)) { |
Chris Lattner | 6e579c6 | 2009-08-19 06:35:36 +0000 | [diff] [blame] | 525 | switch (ValueSize) { |
| 526 | default: llvm_unreachable("Invalid size for machine code value!"); |
Chris Lattner | 33adcfb | 2009-08-22 21:43:10 +0000 | [diff] [blame] | 527 | case 1: OS << MAI.getAlignDirective(); break; |
| 528 | // FIXME: use MAI for this! |
Chris Lattner | 6e579c6 | 2009-08-19 06:35:36 +0000 | [diff] [blame] | 529 | case 2: OS << ".p2alignw "; break; |
| 530 | case 4: OS << ".p2alignl "; break; |
| 531 | case 8: llvm_unreachable("Unsupported alignment size!"); |
| 532 | } |
Chris Lattner | 663c2d2 | 2009-08-19 06:12:02 +0000 | [diff] [blame] | 533 | |
Chris Lattner | 33adcfb | 2009-08-22 21:43:10 +0000 | [diff] [blame] | 534 | if (MAI.getAlignmentIsInBytes()) |
Chris Lattner | 663c2d2 | 2009-08-19 06:12:02 +0000 | [diff] [blame] | 535 | OS << ByteAlignment; |
| 536 | else |
| 537 | OS << Log2_32(ByteAlignment); |
Daniel Dunbar | 84a2926 | 2009-06-24 19:25:34 +0000 | [diff] [blame] | 538 | |
Chris Lattner | 663c2d2 | 2009-08-19 06:12:02 +0000 | [diff] [blame] | 539 | if (Value || MaxBytesToEmit) { |
Chris Lattner | 579531c | 2009-09-03 04:01:10 +0000 | [diff] [blame] | 540 | OS << ", 0x"; |
| 541 | OS.write_hex(truncateToSize(Value, ValueSize)); |
Chris Lattner | 663c2d2 | 2009-08-19 06:12:02 +0000 | [diff] [blame] | 542 | |
| 543 | if (MaxBytesToEmit) |
| 544 | OS << ", " << MaxBytesToEmit; |
| 545 | } |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 546 | EmitEOL(); |
Chris Lattner | 663c2d2 | 2009-08-19 06:12:02 +0000 | [diff] [blame] | 547 | return; |
| 548 | } |
| 549 | |
| 550 | // Non-power of two alignment. This is not widely supported by assemblers. |
Chris Lattner | 33adcfb | 2009-08-22 21:43:10 +0000 | [diff] [blame] | 551 | // FIXME: Parameterize this based on MAI. |
Daniel Dunbar | 84a2926 | 2009-06-24 19:25:34 +0000 | [diff] [blame] | 552 | switch (ValueSize) { |
Chris Lattner | 663c2d2 | 2009-08-19 06:12:02 +0000 | [diff] [blame] | 553 | default: llvm_unreachable("Invalid size for machine code value!"); |
| 554 | case 1: OS << ".balign"; break; |
| 555 | case 2: OS << ".balignw"; break; |
| 556 | case 4: OS << ".balignl"; break; |
| 557 | case 8: llvm_unreachable("Unsupported alignment size!"); |
Daniel Dunbar | 84a2926 | 2009-06-24 19:25:34 +0000 | [diff] [blame] | 558 | } |
| 559 | |
Chris Lattner | 663c2d2 | 2009-08-19 06:12:02 +0000 | [diff] [blame] | 560 | OS << ' ' << ByteAlignment; |
Daniel Dunbar | 304f6a4 | 2009-06-25 21:03:18 +0000 | [diff] [blame] | 561 | OS << ", " << truncateToSize(Value, ValueSize); |
Daniel Dunbar | 84a2926 | 2009-06-24 19:25:34 +0000 | [diff] [blame] | 562 | if (MaxBytesToEmit) |
| 563 | OS << ", " << MaxBytesToEmit; |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 564 | EmitEOL(); |
Daniel Dunbar | 84a2926 | 2009-06-24 19:25:34 +0000 | [diff] [blame] | 565 | } |
| 566 | |
Kevin Enderby | 6e72048 | 2010-02-23 18:26:34 +0000 | [diff] [blame] | 567 | void MCAsmStreamer::EmitCodeAlignment(unsigned ByteAlignment, |
| 568 | unsigned MaxBytesToEmit) { |
Chris Lattner | ec167fd | 2010-02-23 18:44:31 +0000 | [diff] [blame] | 569 | // Emit with a text fill value. |
| 570 | EmitValueToAlignment(ByteAlignment, MAI.getTextAlignFillValue(), |
| 571 | 1, MaxBytesToEmit); |
Kevin Enderby | 6e72048 | 2010-02-23 18:26:34 +0000 | [diff] [blame] | 572 | } |
| 573 | |
Daniel Dunbar | 821e333 | 2009-08-31 08:09:28 +0000 | [diff] [blame] | 574 | void MCAsmStreamer::EmitValueToOffset(const MCExpr *Offset, |
Daniel Dunbar | 84a2926 | 2009-06-24 19:25:34 +0000 | [diff] [blame] | 575 | unsigned char Value) { |
| 576 | // FIXME: Verify that Offset is associated with the current section. |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 577 | OS << ".org " << *Offset << ", " << (unsigned) Value; |
| 578 | EmitEOL(); |
Daniel Dunbar | 84a2926 | 2009-06-24 19:25:34 +0000 | [diff] [blame] | 579 | } |
| 580 | |
Chris Lattner | a6594fc | 2010-01-25 18:58:59 +0000 | [diff] [blame] | 581 | |
| 582 | void MCAsmStreamer::EmitFileDirective(StringRef Filename) { |
| 583 | assert(MAI.hasSingleParameterDotFile()); |
| 584 | OS << "\t.file\t"; |
| 585 | PrintQuotedString(Filename, OS); |
| 586 | EmitEOL(); |
| 587 | } |
| 588 | |
| 589 | void MCAsmStreamer::EmitDwarfFileDirective(unsigned FileNo, StringRef Filename){ |
| 590 | OS << "\t.file\t" << FileNo << ' '; |
| 591 | PrintQuotedString(Filename, OS); |
| 592 | EmitEOL(); |
| 593 | } |
| 594 | |
Daniel Dunbar | 6b71653 | 2010-02-09 23:00:14 +0000 | [diff] [blame] | 595 | void MCAsmStreamer::AddEncodingComment(const MCInst &Inst) { |
| 596 | raw_ostream &OS = GetCommentOS(); |
| 597 | SmallString<256> Code; |
| 598 | SmallVector<MCFixup, 4> Fixups; |
| 599 | raw_svector_ostream VecOS(Code); |
| 600 | Emitter->EncodeInstruction(Inst, VecOS, Fixups); |
| 601 | VecOS.flush(); |
Chris Lattner | a6594fc | 2010-01-25 18:58:59 +0000 | [diff] [blame] | 602 | |
Daniel Dunbar | 6b71653 | 2010-02-09 23:00:14 +0000 | [diff] [blame] | 603 | // If we are showing fixups, create symbolic markers in the encoded |
| 604 | // representation. We do this by making a per-bit map to the fixup item index, |
| 605 | // then trying to display it as nicely as possible. |
Daniel Dunbar | 5532cf4 | 2010-02-10 01:41:14 +0000 | [diff] [blame] | 606 | SmallVector<uint8_t, 64> FixupMap; |
| 607 | FixupMap.resize(Code.size() * 8); |
Daniel Dunbar | 6b71653 | 2010-02-09 23:00:14 +0000 | [diff] [blame] | 608 | for (unsigned i = 0, e = Code.size() * 8; i != e; ++i) |
| 609 | FixupMap[i] = 0; |
| 610 | |
| 611 | for (unsigned i = 0, e = Fixups.size(); i != e; ++i) { |
| 612 | MCFixup &F = Fixups[i]; |
Chris Lattner | 8d31de6 | 2010-02-11 21:27:18 +0000 | [diff] [blame] | 613 | const MCFixupKindInfo &Info = Emitter->getFixupKindInfo(F.getKind()); |
Daniel Dunbar | 6b71653 | 2010-02-09 23:00:14 +0000 | [diff] [blame] | 614 | for (unsigned j = 0; j != Info.TargetSize; ++j) { |
| 615 | unsigned Index = F.getOffset() * 8 + Info.TargetOffset + j; |
| 616 | assert(Index < Code.size() * 8 && "Invalid offset in fixup!"); |
| 617 | FixupMap[Index] = 1 + i; |
| 618 | } |
| 619 | } |
| 620 | |
| 621 | OS << "encoding: ["; |
| 622 | for (unsigned i = 0, e = Code.size(); i != e; ++i) { |
| 623 | if (i) |
| 624 | OS << ','; |
| 625 | |
| 626 | // See if all bits are the same map entry. |
| 627 | uint8_t MapEntry = FixupMap[i * 8 + 0]; |
| 628 | for (unsigned j = 1; j != 8; ++j) { |
| 629 | if (FixupMap[i * 8 + j] == MapEntry) |
| 630 | continue; |
| 631 | |
| 632 | MapEntry = uint8_t(~0U); |
| 633 | break; |
| 634 | } |
| 635 | |
| 636 | if (MapEntry != uint8_t(~0U)) { |
| 637 | if (MapEntry == 0) { |
| 638 | OS << format("0x%02x", uint8_t(Code[i])); |
| 639 | } else { |
| 640 | assert(Code[i] == 0 && "Encoder wrote into fixed up bit!"); |
| 641 | OS << char('A' + MapEntry - 1); |
| 642 | } |
| 643 | } else { |
| 644 | // Otherwise, write out in binary. |
| 645 | OS << "0b"; |
| 646 | for (unsigned j = 8; j--;) { |
| 647 | unsigned Bit = (Code[i] >> j) & 1; |
| 648 | if (uint8_t MapEntry = FixupMap[i * 8 + j]) { |
| 649 | assert(Bit == 0 && "Encoder wrote into fixed up bit!"); |
| 650 | OS << char('A' + MapEntry - 1); |
| 651 | } else |
| 652 | OS << Bit; |
| 653 | } |
| 654 | } |
| 655 | } |
| 656 | OS << "]\n"; |
| 657 | |
| 658 | for (unsigned i = 0, e = Fixups.size(); i != e; ++i) { |
| 659 | MCFixup &F = Fixups[i]; |
Chris Lattner | 8d31de6 | 2010-02-11 21:27:18 +0000 | [diff] [blame] | 660 | const MCFixupKindInfo &Info = Emitter->getFixupKindInfo(F.getKind()); |
Daniel Dunbar | 6b71653 | 2010-02-09 23:00:14 +0000 | [diff] [blame] | 661 | OS << " fixup " << char('A' + i) << " - " << "offset: " << F.getOffset() |
Daniel Dunbar | 5d5a1e1 | 2010-02-10 04:47:08 +0000 | [diff] [blame] | 662 | << ", value: " << *F.getValue() << ", kind: " << Info.Name << "\n"; |
Daniel Dunbar | 6b71653 | 2010-02-09 23:00:14 +0000 | [diff] [blame] | 663 | } |
| 664 | } |
| 665 | |
| 666 | void MCAsmStreamer::EmitInstruction(const MCInst &Inst) { |
| 667 | assert(CurSection && "Cannot emit contents before setting section!"); |
| 668 | |
| 669 | // Show the encoding in a comment if we have a code emitter. |
| 670 | if (Emitter) |
| 671 | AddEncodingComment(Inst); |
| 672 | |
Chris Lattner | 30d9a64 | 2010-02-09 00:54:51 +0000 | [diff] [blame] | 673 | // Show the MCInst if enabled. |
Daniel Dunbar | c9adb8c | 2010-05-26 15:18:13 +0000 | [diff] [blame] | 674 | if (ShowInst) { |
Daniel Dunbar | 67c076c | 2010-03-22 21:49:34 +0000 | [diff] [blame] | 675 | Inst.dump_pretty(GetCommentOS(), &MAI, InstPrinter.get(), "\n "); |
Daniel Dunbar | c9adb8c | 2010-05-26 15:18:13 +0000 | [diff] [blame] | 676 | GetCommentOS() << "\n"; |
| 677 | } |
| 678 | |
Daniel Dunbar | 67c076c | 2010-03-22 21:49:34 +0000 | [diff] [blame] | 679 | // 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] | 680 | if (InstPrinter) |
Chris Lattner | d374087 | 2010-04-04 05:04:31 +0000 | [diff] [blame] | 681 | InstPrinter->printInst(&Inst, OS); |
Daniel Dunbar | 9dee8e3 | 2010-02-03 18:18:30 +0000 | [diff] [blame] | 682 | else |
| 683 | Inst.print(OS, &MAI); |
Chris Lattner | 7d1e49c | 2010-01-22 07:36:39 +0000 | [diff] [blame] | 684 | EmitEOL(); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 685 | } |
| 686 | |
Chris Lattner | 91bead7 | 2010-04-03 21:35:55 +0000 | [diff] [blame] | 687 | /// EmitRawText - If this file is backed by a assembly streamer, this dumps |
| 688 | /// the specified string in the output .s file. This capability is |
| 689 | /// indicated by the hasRawTextSupport() predicate. |
| 690 | void MCAsmStreamer::EmitRawText(StringRef String) { |
Chris Lattner | d5928dc | 2010-04-03 22:06:56 +0000 | [diff] [blame] | 691 | if (!String.empty() && String.back() == '\n') |
| 692 | String = String.substr(0, String.size()-1); |
Chris Lattner | 91bead7 | 2010-04-03 21:35:55 +0000 | [diff] [blame] | 693 | OS << String; |
Chris Lattner | d5928dc | 2010-04-03 22:06:56 +0000 | [diff] [blame] | 694 | EmitEOL(); |
Chris Lattner | 91bead7 | 2010-04-03 21:35:55 +0000 | [diff] [blame] | 695 | } |
| 696 | |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 697 | void MCAsmStreamer::Finish() { |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 698 | } |
Daniel Dunbar | 9dee8e3 | 2010-02-03 18:18:30 +0000 | [diff] [blame] | 699 | |
Chris Lattner | 86e2211 | 2010-01-22 07:29:22 +0000 | [diff] [blame] | 700 | MCStreamer *llvm::createAsmStreamer(MCContext &Context, |
| 701 | formatted_raw_ostream &OS, |
Chris Lattner | fdab14b | 2010-03-12 18:28:53 +0000 | [diff] [blame] | 702 | bool isLittleEndian, |
Chris Lattner | 0740441 | 2010-01-22 07:06:15 +0000 | [diff] [blame] | 703 | bool isVerboseAsm, MCInstPrinter *IP, |
Daniel Dunbar | 5532cf4 | 2010-02-10 01:41:14 +0000 | [diff] [blame] | 704 | MCCodeEmitter *CE, bool ShowInst) { |
Chris Lattner | fdab14b | 2010-03-12 18:28:53 +0000 | [diff] [blame] | 705 | return new MCAsmStreamer(Context, OS, isLittleEndian, isVerboseAsm, |
Daniel Dunbar | 5532cf4 | 2010-02-10 01:41:14 +0000 | [diff] [blame] | 706 | IP, CE, ShowInst); |
Daniel Dunbar | a11af53 | 2009-06-24 01:03:06 +0000 | [diff] [blame] | 707 | } |