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